PWGDQ/dielectron/macrosLMEE/ConfigLMEEPbPb2011AOD.C

   1 //#include "PWGDQ/dielectron/macrosLMEE/LMEECutLib.C"
   2
   3 void InitHistograms(AliDielectron *die, Int_t cutDefinition);
   4 void InitCF(AliDielectron* die, Int_t cutDefinition);
   5 void EnableMC();
   6
   7 TString names=("noPairing;TPCTOFCentnoRej;TPCTOFSemiCentnoRej;TPCTOFPerinoRej;TPCTOFCent;TPCTOFSemiCent;TPCTOFCentnoTOF");
   8 TObjArray *arrNames=names.Tokenize(";");
   9 const Int_t nDie=arrNames->GetEntries();
  10
  11 Bool_t MCenabled=kFALSE;
  12
  13
  14 AliDielectron* ConfigLMEEPbPb2011AOD(Int_t cutDefinition, Bool_t hasMC=kFALSE, Bool_t ESDanalysis=kFALSE)
  15 {
  16
  17   Int_t selectedPID=-1;
  18   Int_t selectedCentrality=-1;
  19   Bool_t rejectionStep=kFALSE;
  20   LMEECutLib*  LMCL = new LMEECutLib();
  21
  22   //
  23   // Setup the instance of AliDielectron
  24   //
  25
  26   MCenabled=hasMC;
  27
  28   // create the actual framework object
  29
  30   TString name=Form("%02d",cutDefinition);
  31   if ((cutDefinition)<arrNames->GetEntriesFast()){
  32         name=arrNames->At((cutDefinition))->GetName();
  33   }
  34
  35   //thisCut only relevant for MC:
  36   AliDielectron *die =
  37         new AliDielectron(Form
  38                 ("%s",name.Data()),
  39                 Form("Track cuts: %s",name.Data()));
  40
  41
  42   //Setup AnalysisSelection:
  43   if (cutDefinition==0) {
  44         //not yet implemented
  45   }
  46   else if (cutDefinition==1) {
  47         selectedPID = LMEECutLib::kPbPb2011TPCandTOF;
  48         selectedCentrality = LMEECutLib::kPbPb2011Central;
  49         rejectionStep = kFALSE;
  50   }
  51   else if (cutDefinition==2) {
  52         selectedPID = LMEECutLib::kPbPb2011TPCandTOF;
  53         selectedCentrality = LMEECutLib::kPbPb2011SemiCentral;
  54         rejectionStep = kFALSE;
  55   }
  56   else if (cutDefinition==3) {
  57         selectedPID = LMEECutLib::kPbPb2011TPCandTOF;
  58         selectedCentrality = LMEECutLib::kPbPb2011Peripheral;
  59         rejectionStep = kFALSE;
  60   }
  61   else if (cutDefinition==4) {
  62         selectedPID = LMEECutLib::kPbPb2011TPCandTOFwide;
  63         selectedCentrality = LMEECutLib::kPbPb2011Central;
  64         rejectionStep = kFALSE;
  65   }
  66   else if (cutDefinition==5) {
  67         selectedPID = LMEECutLib::kPbPb2011TPCandTOF;
  68         selectedCentrality = LMEECutLib::kPbPb2011SemiCentral;
  69         rejectionStep = kTRUE;
  70   }
  71   else if (cutDefinition==6) {
  72         //selectedPID = LMEECutLib::kPbPb2011TPCandTOFwide;
  73         selectedPID = LMEECutLib::kPbPb2011TPC;
  74         selectedCentrality = LMEECutLib::kPbPb2011Central;
  75         rejectionStep = kFALSE;
  76   }
  77   else Semi{
  78         cout << " =============================== " << endl;
  79         cout << " ==== INVALID CONFIGURATION ==== " << endl;
  80         cout << " =============================== " << endl;
  81   }
  82
  83
  84   //Now configure task
  85
  86   //Apply correct Pre-Filter Scheme, if necessary
  87   die->SetPreFilterAllSigns();
  88
  89   //switch off KF PArticle:
  90   die->SetUseKF(kFALSE);
  91
  92         if (rejectionStep) {
  93                 die->GetTrackFilter().AddCuts(LMCL->GetPIDCutsPre(selectedPID) );
  94                 die->GetPairPreFilterLegs().AddCuts(LMCL->GetPIDCutsAna(selectedPID) );
  95                 die->GetPairPreFilter().AddCuts(LMCL->GetPairCuts(selectedPID) );
  96         }
  97         else { //No Prefilter, no Pairfilter
  98
  99           if (ESDanalysis) {
 100                 die->GetTrackFilter().AddCuts( LMCL->GetESDTrackCutsAna(selectedPID) );
 101           }
 102
 103           die->GetTrackFilter().AddCuts( LMCL->GetTrackCutsAna(selectedPID) );
 104           die->GetTrackFilter().AddCuts( LMCL->GetPIDCutsAna(selectedPID) );
 105           /*
 106                 if ((cutDefinition >=6) &&  (cutDefinition <=7)) {
 107                   die->GetPairFilter().AddCuts(LMCL->GetPairCuts2(selectedPID,kFALSE));
 108                 }
 109 */
 110
 111         }
 112         //Introduce NULL-check for pp?
 113         die->GetEventFilter().AddCuts(LMCL->GetCentralityCuts(selectedCentrality));
 114
 115
 116
 117
 118   AliDielectronTrackRotator *rot= 0x0;
 119   /*AliDielectronTrackRotator *rot= LMCL->GetTrackRotator(selectedPID);
 120   die->SetTrackRotator(rot);
 121    */
 122   AliDielectronMixingHandler *mix=LMCL->GetMixingHandler(selectedPID);
 123   die->SetMixingHandler(mix);
 124
 125   // histogram setup
 126   // only if an AliDielectronHistos object is attached to the
 127   // dielectron framework histograms will be filled
 128   //
 129   InitHistograms(die,cutDefinition);
 130
 131   // the last definition uses no cuts and only the QA histograms should be filled!
 132 //  InitCF(die,cutDefinition);
 133
 134   return die;
 135 }
 136
 137 //______________________________________________________________________________________
 138
 139 void InitHistograms(AliDielectron *die, Int_t cutDefinition)
 140 {
 141   //
 142   // Initialise the histograms
 143   //
 144
 145   //Setup histogram Manager
 146   AliDielectronHistos *histos=
 147         new AliDielectronHistos(die->GetName(),
 148                 die->GetTitle());
 149   //Initialise histogram classes
 150   histos->SetReservedWords("Track;Pair;Pre;RejTrack;RejPair");
 151
 152   //Event class
 153 //  if (cutDefinition==nDie-1)
 154                 histos->AddClass("Event");
 155
 156   //Track classes
 157   //to fill also track info from 2nd event loop until 2
 158   for (Int_t i=0; i<2; ++i){
 159         histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
 160   }
 161
 162   //Pair classes
 163   // to fill also mixed event histograms loop until 10
 164   for (Int_t i=0; i<3; ++i){
 165         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 166   }
 167
 168   //ME and track rot
 169   if (die->GetMixingHandler()) {
 170         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(3)));
 171         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(4)));
 172         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(6)));
 173         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(7)));
 174   }
 175   if (die->GetTrackRotator()) {
 176         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(10)));
 177   }
 178
 179   //PreFilter Classes
 180   //to fill also track info from 2nd event loop until 2
 181   for (Int_t i=0; i<2; ++i){
 182         histos->AddClass(Form("Pre_%s",AliDielectron::TrackClassName(i)));
 183   }
 184
 185
 186   //Create Classes for Rejected Tracks/Pairs:
 187   for (Int_t i=0; i<2; ++i){
 188         histos->AddClass(Form("RejTrack_%s",AliDielectron::TrackClassName(i)));
 189   }
 190   for (Int_t i=0; i<3; ++i){
 191         histos->AddClass(Form("RejPair_%s",AliDielectron::PairClassName(i)));
 192   }
 193
 194   /*
 195   //track rotation
 196
 197   histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(AliDielectron::kEv1PMRot)));
 198   histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(AliDielectron::kEv1PMRot)));
 199   */
 200         //add histograms to event class
 201         histos->UserHistogram("Event","nEvents","Number of processed events after cuts;Number events",
 202                 1,0.,1.,AliDielectronVarManager::kNevents);
 203         histos->UserHistogram("Event","Centrality","Centrality;Centrality [%]","0,10,20,40,80,100,101",
 204                 AliDielectronVarManager::kCentrality);
 205
 206
 207   //add histograms to Track classes
 208   histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPt);
 209   histos->UserHistogram("Track","Px","Px;Px [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPx);
 210   histos->UserHistogram("Track","Py","Py;Py [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPy);
 211   histos->UserHistogram("Track","Pz","Pz;Pz [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPz);
 212
 213   histos->UserHistogram("Track","NclsSFracTPC","NclsSFracTPC; NclsSFracTPC;#tracks",200,0,10.,AliDielectronVarManager::kNclsSFracTPC);
 214   histos->UserHistogram("Track","TPCclsDiff","TPCclsDiff; TPCclsDiff;#tracks",200,0,10.,AliDielectronVarManager::kTPCclsDiff);
 215
 216   histos->UserHistogram("Track","ITS_dEdx_P","ITS_dEdx;P [GeV];ITS signal (arb units);#tracks",
 217           400,0.0,20.,1000,0.,1000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kITSsignal,kTRUE);
 218
 219   histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 220           400,0.0,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 221
 222   histos->UserHistogram("Track","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 223           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
 224   histos->UserHistogram("Track","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 225           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao,kTRUE);
 226   histos->UserHistogram("Track","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 227           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
 228
 229   histos->UserHistogram("Track","TRDpidPobEle_P","TRD PID probability Electrons;P [GeV];TRD prob Electrons;#tracks",
 230           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobEle,kTRUE);
 231   histos->UserHistogram("Track","TRDpidPobPio_P","TRD PID probability Pions;P [GeV];TRD prob Pions;#tracks",
 232           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobPio,kTRUE);
 233
 234   histos->UserHistogram("Track","TOFnSigmaKao_P","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 235           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao,kTRUE);
 236   histos->UserHistogram("Track","TOFnSigmaPro_P","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 237           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro,kTRUE);
 238
 239   histos->UserHistogram("Track","TOFbeta","TOF beta;P [GeV];TOF beta;#tracks",
 240           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFbeta,kTRUE);
 241
 242
 243   histos->UserHistogram("Track","Eta","Eta; Eta;#tracks",
 244           200,-2,2,AliDielectronVarManager::kEta);
 245   histos->UserHistogram("Track","Phi","Phi; Phi;#tracks",
 246           200,0.,3.15,AliDielectronVarManager::kPhi);
 247
 248   histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 249           200,-2,2,200,0,3.15,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 250
 251   histos->UserHistogram("Track","dXY_dZ","dXY dZ Map; dXY; dZ;#tracks",
 252           200,-2,2,200,-2,2.,AliDielectronVarManager::kImpactParXY,AliDielectronVarManager::kImpactParZ);
 253
 254
 255   histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParXY);
 256   histos->UserHistogram("Track","dZ","dZ;dZ [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParZ);
 257
 258           histos->UserHistogram("Track","TPCcrossedRowsOverFindable","Number of Crossed Rows TPC over Findable;TPC crossed rows over findable;#tracks",100,0.,1.,AliDielectronVarManager::kNFclsTPCfCross);
 259           histos->UserHistogram("Track","TPCcrossedRows","Number of Crossed Rows TPC;TPC crossed rows;#tracks",159,0.,159.,AliDielectronVarManager::kNFclsTPCr);
 260           histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",159,0.,159.,AliDielectronVarManager::kNclsTPC);
 261           histos->UserHistogram("Track","ITSnCls","Number of Clusters ITS;ITS number clusteres;#tracks",159,0.,159.,AliDielectronVarManager::kNclsITS);
 262
 263           histos->UserHistogram("Track","TPCchi2","TPC Chi2 value;TPC chi2;#tracks",100,0.,10.,AliDielectronVarManager::kTPCchi2Cl);
 264           histos->UserHistogram("Track","ITSchi2","ITS Chi2 value;ITS chi2;#tracks",100,0.,10.,AliDielectronVarManager::kITSchi2Cl);
 265
 266           histos->UserHistogram("Track","TPCnCls_kNFclsTPCr","nTPC vs nTPCr;nTPC vs nTPCr;#tracks",159,0.,159.,159,0.,159.,AliDielectronVarManager::kNclsTPC,AliDielectronVarManager::kNFclsTPCr);
 267
 268           histos->UserHistogram("Track","kNFclsTPCr_pT","nTPCr vs pt;nTPCr vs pt;#tracks",159,0.,159.,200,0.,20.,AliDielectronVarManager::kNFclsTPCr,AliDielectronVarManager::kPt);
 269
 270           //add histograms to Pair classes
 271           histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
 272                   500,0.0,5.00,AliDielectronVarManager::kM);
 273           histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
 274                   100,-2.,2.,AliDielectronVarManager::kY);
 275           histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
 276                   100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
 277           //2D Histo Plot
 278           histos->UserHistogram("Pair","InvMassPairPt","Inv.Mass vs PairPt;Inv. Mass [GeV], pT [GeV];#pairs",
 279                   500,0.0,5.0,500,0.,50.,AliDielectronVarManager::kM,AliDielectronVarManager::kPt);
 280
 281           histos->UserHistogram("Pair","InvMassOpeningAngle","Opening Angle vs Inv.Mass;Inv. Mass [GeV];#pairs",
 282                   500,0.0,5.0,200,0.,6.3,AliDielectronVarManager::kM,AliDielectronVarManager::kOpeningAngle);
 283
 284           //add histograms to Track classes
 285           histos->UserHistogram("Pre","Pt","Pt;Pt [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPt);
 286
 287           histos->UserHistogram("Pre","ITS_dEdx_P","ITS_dEdx;P [GeV];ITS signal (arb units);#tracks",
 288                   400,0.0,20.,1000,0.,1000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kITSsignal,kTRUE);
 289
 290           histos->UserHistogram("Pre","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 291                   400,0.0,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 292
 293
 294           histos->UserHistogram("Pre","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 295                   400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
 296           histos->UserHistogram("Pre","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 297                   400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao,kTRUE);
 298           histos->UserHistogram("Pre","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 299                   400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
 300
 301           histos->UserHistogram("Pre","TRDpidPobEle_P","TRD PID probability Electrons;P [GeV];TRD prob Electrons;#tracks",
 302                   400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobEle,kTRUE);
 303           histos->UserHistogram("Pre","TRDpidPobPio_P","TRD PID probability Pions;P [GeV];TRD prob Pions;#tracks",
 304                   400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobPio,kTRUE);
 305
 306           histos->UserHistogram("Pre","TOFnSigmaKao_P","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 307                   400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao,kTRUE);
 308           histos->UserHistogram("Pre","TOFnSigmaPro_P","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 309                   400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro,kTRUE);
 310
 311           histos->UserHistogram("Pre","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 312                   200,-2,2,200,0,3.15,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 313
 314           histos->UserHistogram("Pre","dXY","dXY;dXY [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParXY);
 315
 316   histos->UserHistogram("Pre","ZVertex ","ZVertex ;ZVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kZv);
 317   histos->UserHistogram("Pre","XVertex ","XVertex ;XVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kXv);
 318   histos->UserHistogram("Pre","YVertex ","YVertex ;YVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kYv);
 319
 320   histos->UserHistogram("Pre","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",159,0.,159.,AliDielectronVarManager::kNclsTPC);
 321
 322   //add histograms to Pair classes For Rejected Pairs:
 323   die->SetHistogramManager(histos);
 324 }
 325
 326
 327 void InitCF(AliDielectron* die, Int_t cutDefinition)
 328 {
 329   //
 330   // Setupd the CF Manager if needed
 331   //
 332   AliDielectronCF *cf=new AliDielectronCF(die->GetName(),die->GetTitle());
 333
 334   //pair variables
 335   cf->AddVariable(AliDielectronVarManager::kP,200,0,20);
 336   cf->AddVariable(AliDielectronVarManager::kM,201,-0.01,4.01); //20Mev Steps
 337   cf->AddVariable(AliDielectronVarManager::kPairType,10,0,10);
 338
 339   cf->AddVariable(AliDielectronVarManager::kCentrality,"0.,10.0,30.0,40.0,60.,80.,100.");
 340
 341   //leg variables
 342   cf->AddVariable(AliDielectronVarManager::kP,200,0.,20.,kTRUE);
 343     cf->AddVariable(AliDielectronVarManager::kITSsignal,1000,0.0.,1000.,kTRUE);
 344   cf->AddVariable(AliDielectronVarManager::kTPCsignal,500,0.0.,500.,kTRUE);
 345   cf->AddVariable(AliDielectronVarManager::kHaveSameMother,21,-10,10,kTRUE);
 346
 347   //only in this case write MC truth info
 348   if (MCenabled) {
 349         cf->SetStepForMCtruth();
 350         cf->SetStepsForMCtruthOnly();
 351         cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
 352         cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
 353   }
 354
 355   cf->SetStepsForSignal();
 356   die->SetCFManagerPair(cf);
 357 }
 358
 359 //--------------------------------------
 360 void EnableMC() {
 361   MCenabled=kTRUE;
 362 }
 363