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;TPCTOFCentPhiV");
   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)
  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         selectedCentrality = LMEECutLib::kPbPb2011Central;
  74         rejectionStep = kFALSE;
  75   }
  76   else Semi{
  77         cout << " =============================== " << endl;
  78         cout << " ==== INVALID CONFIGURATION ==== " << endl;
  79         cout << " =============================== " << endl;
  80   }
  81
  82
  83   //Now configure task
  84
  85   //Apply correct Pre-Filter Scheme, if necessary
  86   die->SetPreFilterAllSigns();
  87
  88         if (rejectionStep) {
  89                 die->GetTrackFilter().AddCuts(LMCL->GetPIDCutsPre(selectedPID) );
  90                 die->GetPairPreFilterLegs().AddCuts(LMCL->GetPIDCutsAna(selectedPID) );
  91                 die->GetPairPreFilter().AddCuts(LMCL->GetPairCuts(selectedPID) );
  92         }
  93         else { //No Prefilter, no Pairfilter
  94                 die->GetTrackFilter().AddCuts( LMCL->GetPIDCutsAna(selectedPID) );
  95                 if ((cutDefinition >=6) &&  (cutDefinition <=7)) {
  96                   die->GetPairFilter().AddCuts(LMCL->GetPairCuts2(selectedPID,kFALSE));
  97                 }
  98
  99         }
 100         //Introduce NULL-check for pp?
 101         die->GetEventFilter().AddCuts(LMCL->GetCentralityCuts(selectedCentrality));
 102
 103
 104
 105
 106   AliDielectronTrackRotator *rot= 0x0;
 107   /*AliDielectronTrackRotator *rot= LMCL->GetTrackRotator(selectedPID);
 108   die->SetTrackRotator(rot);
 109    */
 110   AliDielectronMixingHandler *mix=LMCL->GetMixingHandler(selectedPID);
 111   die->SetMixingHandler(mix);
 112
 113   // histogram setup
 114   // only if an AliDielectronHistos object is attached to the
 115   // dielectron framework histograms will be filled
 116   //
 117   InitHistograms(die,cutDefinition);
 118
 119   // the last definition uses no cuts and only the QA histograms should be filled!
 120   InitCF(die,cutDefinition);
 121
 122   return die;
 123 }
 124
 125 //______________________________________________________________________________________
 126
 127 void InitHistograms(AliDielectron *die, Int_t cutDefinition)
 128 {
 129   //
 130   // Initialise the histograms
 131   //
 132
 133   //Setup histogram Manager
 134   AliDielectronHistos *histos=
 135         new AliDielectronHistos(die->GetName(),
 136                 die->GetTitle());
 137   //Initialise histogram classes
 138   histos->SetReservedWords("Track;Pair;Pre;RejTrack;RejPair");
 139
 140   //Event class
 141 //  if (cutDefinition==nDie-1)
 142                 histos->AddClass("Event");
 143
 144   //Track classes
 145   //to fill also track info from 2nd event loop until 2
 146   for (Int_t i=0; i<2; ++i){
 147         histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
 148   }
 149
 150   //Pair classes
 151   // to fill also mixed event histograms loop until 10
 152   for (Int_t i=0; i<3; ++i){
 153         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 154   }
 155
 156   //ME and track rot
 157   if (die->GetMixingHandler()) {
 158         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(3)));
 159         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(4)));
 160         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(6)));
 161         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(7)));
 162   }
 163   if (die->GetTrackRotator()) {
 164         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(10)));
 165   }
 166
 167   //PreFilter Classes
 168   //to fill also track info from 2nd event loop until 2
 169   for (Int_t i=0; i<2; ++i){
 170         histos->AddClass(Form("Pre_%s",AliDielectron::TrackClassName(i)));
 171   }
 172
 173
 174   //Create Classes for Rejected Tracks/Pairs:
 175   for (Int_t i=0; i<2; ++i){
 176         histos->AddClass(Form("RejTrack_%s",AliDielectron::TrackClassName(i)));
 177   }
 178   for (Int_t i=0; i<3; ++i){
 179         histos->AddClass(Form("RejPair_%s",AliDielectron::PairClassName(i)));
 180   }
 181
 182   /*
 183   //track rotation
 184
 185   histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(AliDielectron::kEv1PMRot)));
 186   histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(AliDielectron::kEv1PMRot)));
 187   */
 188         //add histograms to event class
 189         histos->UserHistogram("Event","nEvents","Number of processed events after cuts;Number events",
 190                 1,0.,1.,AliDielectronVarManager::kNevents);
 191         histos->UserHistogram("Event","Centrality","Centrality;Centrality [%]","0,10,20,40,80,100,101",
 192                 AliDielectronVarManager::kCentrality);
 193
 194
 195   //add histograms to Track classes
 196   histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPt);
 197   histos->UserHistogram("Track","NclsSFracTPC","NclsSFracTPC; NclsSFracTPC;#tracks",200,0,10.,AliDielectronVarManager::kNclsSFracTPC);
 198   histos->UserHistogram("Track","TPCclsDiff","TPCclsDiff; TPCclsDiff;#tracks",200,0,10.,AliDielectronVarManager::kTPCclsDiff);
 199
 200   histos->UserHistogram("Track","ITS_dEdx_P","ITS_dEdx;P [GeV];ITS signal (arb units);#tracks",
 201           400,0.0,20.,1000,0.,1000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kITSsignal,kTRUE);
 202
 203   histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 204           400,0.0,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 205
 206   histos->UserHistogram("Track","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 207           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
 208   histos->UserHistogram("Track","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 209           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao,kTRUE);
 210   histos->UserHistogram("Track","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 211           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
 212
 213   histos->UserHistogram("Track","TRDpidPobEle_P","TRD PID probability Electrons;P [GeV];TRD prob Electrons;#tracks",
 214           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobEle,kTRUE);
 215   histos->UserHistogram("Track","TRDpidPobPio_P","TRD PID probability Pions;P [GeV];TRD prob Pions;#tracks",
 216           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobPio,kTRUE);
 217
 218   histos->UserHistogram("Track","TOFnSigmaKao_P","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 219           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao,kTRUE);
 220   histos->UserHistogram("Track","TOFnSigmaPro_P","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 221           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro,kTRUE);
 222
 223   histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 224           200,-2,2,200,0,3.15,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 225
 226   histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParXY);
 227
 228   histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",159,0.,159.,AliDielectronVarManager::kNclsTPC);
 229
 230   histos->UserHistogram("Track","TPCnCls_kNFclsTPCr","nTPC vs nTPCr;nTPC vs nTPCr;#tracks",159,0.,159.,159,0.,159.,AliDielectronVarManager::kNclsTPC,AliDielectronVarManager::kNFclsTPCr);
 231
 232   histos->UserHistogram("Track","kNFclsTPCr_pT","nTPCr vs pt;nTPCr vs pt;#tracks",159,0.,159.,200,0.,20.,AliDielectronVarManager::kNFclsTPCr,AliDielectronVarManager::kPt);
 233
 234   //add histograms to Pair classes
 235   histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
 236           500,0.0,5.00,AliDielectronVarManager::kM);
 237   histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
 238           100,-2.,2.,AliDielectronVarManager::kY);
 239   histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
 240           100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
 241   //2D Histo Plot
 242   histos->UserHistogram("Pair","InvMassPairPt","Inv.Mass vs PairPt;Inv. Mass [GeV], pT [GeV];#pairs",
 243           500,0.0,5.0,500,0.,50.,AliDielectronVarManager::kM,AliDielectronVarManager::kPt);
 244
 245   histos->UserHistogram("Pair","InvMassOpeningAngle","Opening Angle vs Inv.Mass;Inv. Mass [GeV];#pairs",
 246           500,0.0,5.0,200,0.,6.3,AliDielectronVarManager::kM,AliDielectronVarManager::kOpeningAngle);
 247
 248   //add histograms to Track classes
 249   histos->UserHistogram("Pre","Pt","Pt;Pt [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPt);
 250
 251   histos->UserHistogram("Pre","ITS_dEdx_P","ITS_dEdx;P [GeV];ITS signal (arb units);#tracks",
 252           400,0.0,20.,1000,0.,1000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kITSsignal,kTRUE);
 253
 254   histos->UserHistogram("Pre","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 255           400,0.0,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 256
 257
 258   histos->UserHistogram("Pre","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 259           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
 260   histos->UserHistogram("Pre","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 261           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao,kTRUE);
 262   histos->UserHistogram("Pre","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 263           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
 264
 265   histos->UserHistogram("Pre","TRDpidPobEle_P","TRD PID probability Electrons;P [GeV];TRD prob Electrons;#tracks",
 266           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobEle,kTRUE);
 267   histos->UserHistogram("Pre","TRDpidPobPio_P","TRD PID probability Pions;P [GeV];TRD prob Pions;#tracks",
 268           400,0.0,20.,100,0.,1.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTRDprobPio,kTRUE);
 269
 270   histos->UserHistogram("Pre","TOFnSigmaKao_P","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 271           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao,kTRUE);
 272   histos->UserHistogram("Pre","TOFnSigmaPro_P","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 273           400,0.0,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro,kTRUE);
 274
 275   histos->UserHistogram("Pre","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 276           200,-2,2,200,0,3.15,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 277
 278   histos->UserHistogram("Pre","dXY","dXY;dXY [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParXY);
 279
 280   histos->UserHistogram("Pre","ZVertex ","ZVertex ;ZVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kZv);
 281   histos->UserHistogram("Pre","XVertex ","XVertex ;XVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kXv);
 282   histos->UserHistogram("Pre","YVertex ","YVertex ;YVertex[cm];#tracks",20,-20,20,AliDielectronVarManager::kYv);
 283
 284   histos->UserHistogram("Pre","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",159,0.,159.,AliDielectronVarManager::kNclsTPC);
 285
 286   //add histograms to Pair classes For Rejected Pairs:
 287   die->SetHistogramManager(histos);
 288 }
 289
 290
 291 void InitCF(AliDielectron* die, Int_t cutDefinition)
 292 {
 293   //
 294   // Setupd the CF Manager if needed
 295   //
 296   AliDielectronCF *cf=new AliDielectronCF(die->GetName(),die->GetTitle());
 297
 298   //pair variables
 299   cf->AddVariable(AliDielectronVarManager::kP,200,0,20);
 300   cf->AddVariable(AliDielectronVarManager::kM,201,-0.01,4.01); //20Mev Steps
 301   cf->AddVariable(AliDielectronVarManager::kPairType,10,0,10);
 302
 303   cf->AddVariable(AliDielectronVarManager::kCentrality,"0.,10.0,30.0,40.0,60.,80.,100.");
 304
 305   //leg variables
 306   cf->AddVariable(AliDielectronVarManager::kP,200,0.,20.,kTRUE);
 307     cf->AddVariable(AliDielectronVarManager::kITSsignal,1000,0.0.,1000.,kTRUE);
 308   cf->AddVariable(AliDielectronVarManager::kTPCsignal,500,0.0.,500.,kTRUE);
 309   cf->AddVariable(AliDielectronVarManager::kHaveSameMother,21,-10,10,kTRUE);
 310
 311   //only in this case write MC truth info
 312   if (MCenabled) {
 313         cf->SetStepForMCtruth();
 314         cf->SetStepsForMCtruthOnly();
 315         cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
 316         cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
 317   }
 318
 319   cf->SetStepsForSignal();
 320   die->SetCFManagerPair(cf);
 321 }
 322
 323 //--------------------------------------
 324 void EnableMC() {
 325   MCenabled=kTRUE;
 326 }
 327