major updates to plotting macros

[u/mrichter/AliRoot.git] / PWG4 / macros / electrons / ConfigAnalysisElectron.C

//------------------------------------\r
// Configuration macro example:\r
//\r
// Configure EMCal electron analysis\r
//\r
// Modified by: K. Read\r
//\r
//------------------------------------\r
\r
AliAnaPartCorrMaker*  ConfigAnalysis()\r
{\r
  //\r
  // Configuration goes here\r
  // \r
  printf("======================== \n");\r
  printf("ConfigAnalysisElectron() \n");\r
  printf("======================== \n");\r
  Bool_t kInputIsESD = kTRUE;     //uncomment for input ESD\r
//Bool_t kInputIsESD = kFALSE;    //uncomment for input AODs\r
  Bool_t kFollowsFilter = kTRUE;  //uncomment if follows ESD filter task\r
//Bool_t kFollowsFilter = kFALSE; //uncomment if no ESD filter task\r
\r
  //enum for the different electron cut sets\r
  //defined for dR and p/E\r
  //kTight2 is the default standard cuts\r
  enum kCutTypes {kTight2, kLooseTight, kTightLoose, kLoose2};\r
  Int_t kCutSet = kTight2;\r
  Double_t pOverEmin = 0.8;  //tight\r
  Double_t pOverEmax = 1.2;  //tight\r
  Double_t dRmax     = 0.02; //tight\r
  if (gSystem->Getenv("ELECUTSET")){\r
    kCutSet = atoi(gSystem->Getenv("ELECUTSET"));\r
  }\r
  if(kCutSet == kLooseTight) {\r
    pOverEmin = 0.6;  //loose\r
    pOverEmax = 1.4;  //loose\r
    dRmax     = 0.02; //tight\r
  }\r
  if(kCutSet == kTightLoose) {\r
    pOverEmin = 0.8;  //tight\r
    pOverEmax = 1.2;  //tight\r
    dRmax     = 0.05; //loose\r
  }\r
  if(kCutSet == kLoose2) {\r
    pOverEmin = 0.6;  //loose\r
    pOverEmax = 1.4;  //loose\r
    dRmax     = 0.05; //loose\r
  }    \r
\r
  //Alternatively, select input via anaInputData environment variable.\r
  if (gSystem->Getenv("anaInputData")){\r
    TString kInputData = gSystem->Getenv("anaInputData");\r
    if( kInputData == "AOD" ){\r
      kInputIsESD = kFALSE;\r
      kFollowsFilter = kFALSE;\r
    }\r
  }\r
\r
  //Detector Fidutial Cuts\r
  AliFidutialCut * fidCut = new AliFidutialCut();\r
  fidCut->DoCTSFidutialCut(kFALSE) ;\r
  fidCut->DoEMCALFidutialCut(kFALSE) ;\r
  fidCut->DoPHOSFidutialCut(kFALSE) ;\r
\r
  //fidCut->SetSimpleCTSFidutialCut(0.9,0.,360.);\r
  //fidCut->SetSimpleEMCALFidutialCut(0.7,80.,190.);\r
  //fidCut->SetSimplePHOSFidutialCut(0.13,220.,320.);\r
\r
  fidCut->Print("");\r
\r
  //-----------------------------------------------------------  \r
  // Reader\r
  //-----------------------------------------------------------\r
  if(kInputIsESD && !kFollowsFilter)AliCaloTrackESDReader *reader = new AliCaloTrackESDReader();\r
  else           AliCaloTrackAODReader *reader = new AliCaloTrackAODReader();\r
  reader->SetDebug(-1);//10 for lots of messages\r
\r
  //Switch on or off the detectors information that you want\r
  reader->SwitchOnEMCAL();\r
  reader->SwitchOnCTS();\r
  //reader->SwitchOffEMCALCells();	\r
  reader->SwitchOffPHOS();\r
  //reader->SwitchOffPHOSCells();	\r
\r
  //Kine\r
  if(!kInputIsESD){\r
    reader->SwitchOffStack();          // On  by default, remember to SwitchOnMCData() in analysis classes\r
    reader->SwitchOnAODMCParticles();  // Off by default, remember to SwitchOnMCData() in analysis classes\r
  }\r
\r
  //Min particle pT\r
  reader->SetCTSPtMin(0.0);   //new\r
  reader->SetEMCALPtMin(0.0); //new\r
  if(kFollowsFilter)reader->SetTrackStatus(0);  //to prevent automatic TPC and ITS refit\r
\r
  //In case of generating jet events (with PYTHIA), if pt hard bin is small\r
  //reject events with large difference between ptHard and triggered jet\r
  //reader->SetPtHardAndJetPtComparison(kTRUE);\r
\r
  reader->SetFidutialCut(fidCut);\r
\r
  if(!kInputIsESD){\r
    // Analysis with tracks, select only tracks with\r
    // following bits\r
\r
    //     //We want tracks fitted in the detectors:\r
    //     ULong_t status=AliAODTrack::kTPCrefit;\r
    //     status|=AliAODTrack::kITSrefit;\r
   \r
    //     We want tracks whose PID bit is set:\r
    //     ULong_t status =AliAODTrack::kITSpid;\r
    //     status|=AliAODTrack::kTPCpid;	\r
\r
    //	reader->SetTrackStatus(status);\r
  }\r
\r
  reader->Print("");\r
\r
\r
  //Detector Fidutial Cuts\r
  AliFidutialCut * fidCut2 = new AliFidutialCut();\r
  fidCut2->DoEMCALFidutialCut(kTRUE) ;\r
  fidCut2->SetSimpleEMCALFidutialCut(0.7,80.,190.);\r
\r
  fidCut2->DoCTSFidutialCut(kTRUE) ;\r
  fidCut2->SetSimpleCTSFidutialCut(0.9,0.,360.); \r
\r
  fidCut2->Print("");\r
\r
  //---------------------------------------------------------------------\r
  // Analysis algorithm\r
  //---------------------------------------------------------------------\r
\r
  AliAnaElectron *anaelectron = new AliAnaElectron();\r
  anaelectron->SetDebug(-1); //10 for lots of messages\r
  anaelectron->SetCalorimeter("EMCAL");\r
  anaelectron->SwitchOnDataMC();\r
  anaelectron->SetMinPt(1.);\r
  anaelectron->SetOutputAODName("Electrons");\r
  anaelectron->SetOutputAODClassName("AliAODPWG4Particle");\r
  anaelectron->SetWriteNtuple(kFALSE);\r
  //Determine which cuts to use based on enum\r
  anaelectron->SetpOverEmin(pOverEmin);\r
  anaelectron->SetpOverEmax(pOverEmax);\r
  anaelectron->SetResidualCut(dRmax);\r
  //Set Histrograms bins and ranges\r
  anaelectron->SetHistoPtRangeAndNBins(0, 100, 100) ;\r
  anaelectron->SetHistoPhiRangeAndNBins(0, TMath::TwoPi(), 100) ;\r
  anaelectron->SetHistoEtaRangeAndNBins(-0.7, 0.7, 100) ;\r
  anaelectron->Print("");\r
\r
  //---------------------------------------------------------------------\r
  // Set  analysis algorithm and reader\r
  //---------------------------------------------------------------------\r
  maker = new AliAnaPartCorrMaker();\r
  maker->SetReader(reader);//pointer to reader\r
  maker->AddAnalysis(anaelectron,0);\r
  maker->SetAnaDebug(-1)  ;\r
  maker->SwitchOnHistogramsMaker()  ;\r
  maker->SwitchOnAODsMaker()  ;\r
\r
  maker->Print("");\r
  //\r
  printf("============================ \n");\r
  printf("END ConfigAnalysisElectron() \n");\r
  printf("============================ \n");\r
  return maker ;\r
}\r