PWGDQ/dielectron/macrosJPSI/ConfigJpsi_mf_pp.C

   1 void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   2 void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   3 void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   4 void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   5
   6 AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition);
   7
   8 TString namesDieleData=("EMCal+QA;EMCal+ITSfirstclass;EMCal+SPDAny+cut");
   9 TObjArray *arrNamesDieleData=namesDieleData.Tokenize("; ");
  10
  11 const Int_t nDie=arrNamesDieleData->GetEntries();
  12
  13 AliDielectron* ConfigJpsi_mf_pp(Int_t cutDefinition, Bool_t isAOD=kFALSE)
  14 {
  15   //
  16   // Setup the instance of AliDielectron
  17   //
  18
  19   // create the actual framework object
  20   TString name=Form("%02d",cutDefinition);
  21   if (cutDefinition<arrNamesDieleData->GetEntriesFast()){
  22     name=arrNamesDieleData->At(cutDefinition)->GetName();
  23   }
  24   AliDielectron *diele = new AliDielectron(Form("%s",name.Data()),
  25                              Form("Track cuts: %s",name.Data()));
  26
  27   // cut setup
  28   SetupEventCutsDieleFilter(diele, cutDefinition, isAOD);
  29   SetupTrackCutsDieleData(diele, cutDefinition, isAOD);
  30   SetupPairCutsDieleData(diele, cutDefinition, isAOD);
  31
  32   //
  33   // histogram setup
  34   // only if an AliDielectronHistos object is attached to the
  35   // dielelectron framework histograms will be filled
  36   //
  37   InitHistogramsDieleData(diele, cutDefinition, isAOD);
  38
  39   // the last definition uses no cuts and only the QA histograms should be filled!
  40   if(cutDefinition>=1)  InitCFDieleData(diele, cutDefinition, isAOD);
  41
  42   AliDielectronTrackRotator *rot=new AliDielectronTrackRotator;
  43   rot->SetConeAnglePhi(TMath::Pi());
  44   rot->SetIterations(10);
  45   diele->SetTrackRotator(rot);
  46   return diele;
  47 }
  48
  49 //______________________________________________________________________________________
  50 void SetupEventCutsDieleFilter(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD=kFALSE)
  51 {
  52   //
  53   // Setup the event cuts
  54   //
  55   AliDielectronVarCuts *vtxZ = new AliDielectronVarCuts("vtxZ","Vertex z cut");
  56   vtxZ->AddCut(AliDielectronVarManager::kZvPrim,-10.,10.);
  57
  58   AliDielectronVarCuts *Centrality = new AliDielectronVarCuts("Centrality","Centrality Percentile");
  59   Centrality->AddCut(AliDielectronVarManager::kCentrality,0.,90.);
  60
  61   diele->GetEventFilter().AddCuts(vtxZ);
  62   diele->GetEventFilter().AddCuts(Centrality);
  63
  64 }
  65
  66
  67 //______________________________________________________________________________________
  68 void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
  69 {
  70   //
  71   // Setup the track cuts
  72   //
  73
  74   //ESD quality cuts DielectronTrackCuts
  75   if (!isAOD) {
  76     diele->GetTrackFilter().AddCuts(SetupESDtrackCutsDieleData(cutDefinition));
  77   } else {
  78     AliDielectronTrackCuts *trackCuts=new AliDielectronTrackCuts("trackCuts","trackCuts");
  79     if(cutDefinition>=2)        trackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
  80     trackCuts->SetRequireTPCRefit(kTRUE);
  81     trackCuts->SetRequireITSRefit(kTRUE);
  82     diele->GetTrackFilter().AddCuts(trackCuts);
  83   }
  84
  85   //Pt cut ----------------------------------------------------------
  86   AliDielectronVarCuts *pt = new AliDielectronVarCuts("ptCut","pt cut");
  87   pt->AddCut(AliDielectronVarManager::kPt,1.,1e30);
  88   pt->AddCut(AliDielectronVarManager::kKinkIndex0,0.);
  89   //AOD additions since there are no AliESDtrackCuts -----------------
  90   pt->AddCut(AliDielectronVarManager::kEta,-0.9,0.9);
  91   pt->AddCut(AliDielectronVarManager::kImpactParXY, -1.0,   1.0);
  92   pt->AddCut(AliDielectronVarManager::kImpactParZ,  -3.0,   3.0);
  93   //pt->AddCut(AliDielectronVarManager::kTPCchi2Cl,    0.0,   4.0);
  94   pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-2.5,4.1);
  95   pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCsignal,65.,100.);
  96
  97   if(cutDefinition==0)    pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-4.5,6.);
  98   if(cutDefinition==0)    pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCsignal,45.,130.);
  99
 100
 101   pt->AddCut(AliDielectronVarManager::kNclsTPC,90.,160.);
 102   if(cutDefinition==2)    pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-1.5,3.);
 103   if(cutDefinition==2)    pt->AddCut(AliDielectronVarManager::kPt,1.1,1e30);
 104   diele->GetTrackFilter().AddCuts(pt);
 105 }
 106
 107 //______________________________________________________________________________________
 108 void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
 109 {
 110   //
 111   // Setup the pair cuts
 112   //
 113   //Invariant mass and rapidity selection
 114 AliDielectronVarCuts *pairCut=new AliDielectronVarCuts("0<M<5+|Y|<.9+PtEMCalleg","0<M<5 + |Y|<.9+PtEMCalleg");
 115 pairCut->AddCut(AliDielectronVarManager::kM,1.,5.);
 116 pairCut->AddCut(AliDielectronVarManager::kY,-0.9,0.9);
 117 pairCut->AddCut(AliDielectronVarManager::kPt,4.,100.);
 118 diele->GetPairFilter().AddCuts(pairCut);
 119
 120 AliDielectronVarCuts *mycut = new AliDielectronVarCuts("ptCutEMCAL","cut for EMCal");
 121 mycut->AddCut(AliDielectronVarManager::kEMCALE,4.,1e30);
 122 mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.7,2.);
 123 if(cutDefinition==2)  mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.9,1.4);
 124 AliDielectronPairLegCuts *varpair=new AliDielectronPairLegCuts();
 125 varpair->GetLeg1Filter().AddCuts(mycut);
 126 varpair->GetLeg2Filter().AddCuts(mycut);
 127 varpair->SetCutType(AliDielectronPairLegCuts::kAnyLeg);
 128 diele->GetPairFilter().AddCuts(varpair);
 129
 130 }
 131
 132 //______________________________________________________________________________________
 133 AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition)
 134 {
 135   //
 136   // Setup default AliESDtrackCuts
 137   //
 138   AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
 139
 140   // basic track quality cuts  (basicQ)
 141   esdTrackCuts->SetMaxDCAToVertexZ(3.0);
 142   esdTrackCuts->SetMaxDCAToVertexXY(1.0);
 143   esdTrackCuts->SetEtaRange( -0.9 , 0.9 );
 144   esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
 145   esdTrackCuts->SetRequireITSRefit(kTRUE);
 146   esdTrackCuts->SetRequireTPCRefit(kTRUE);
 147   esdTrackCuts->SetPtRange(1.,1e30);
 148   esdTrackCuts->SetMinNClustersTPC(70);
 149   esdTrackCuts->SetMaxChi2PerClusterTPC(4);
 150   esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
 151
 152   return esdTrackCuts;
 153 }
 154
 155
 156 //______________________________________________________________________________________
 157 void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
 158 {
 159   //
 160   // Initialise the histograms
 161   //
 162
 163   //Setup histogram Manager
 164   AliDielectronHistos *histos=new AliDielectronHistos(diele->GetName(),diele->GetTitle());
 165
 166   //Initialise histogram classes
 167   histos->SetReservedWords("Track;Pair");
 168
 169   //Track classes
 170   //to fill also track info from 2nd event loop until 2
 171   for (Int_t i=0; i<2; ++i){
 172     histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
 173   }
 174
 175   //Pair classes
 176   // to fill also mixed event histograms loop until 10
 177   for (Int_t i=0; i<3; ++i){
 178     histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 179   }
 180
 181   //legs from pair
 182   for (Int_t i=0; i<3; ++i){
 183     histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(i)));
 184   }
 185
 186   //track rotation
 187   if (cutDefinition==1) {
 188       histos->AddClass(Form("Pair_%s",PairClassName(AliDielectron::kEv1PMRot)));
 189       histos->AddClass(Form("Track_Legs_%s",PairClassName(AliDielectron::kEv1PMRot)));
 190   }
 191
 192   //add histograms to event class
 193   if (cutDefinition==0) {
 194     histos->AddClass("Event");
 195     histos->UserHistogram("Event","VtxZ","Vertex Z;Z[cm]",300,-15.,15.,AliDielectronVarManager::kZvPrim);
 196     histos->UserHistogram("Event","Centrality","Centrality;Cent(%)",100,0.,100.,AliDielectronVarManager::kCentrality);
 197     histos->UserHistogram("Event","Multiplicity","Multiplicity V0;Multiplicity V0",500,0.,25000.,AliDielectronVarManager::kMultV0);
 198     histos->UserHistogram("Event","Multiplicity_cent","Multiplicity V0 x Cent;Cent(%);Multiplicity V0",10,0,100.,500,0.,25000.,AliDielectronVarManager::kCentrality,AliDielectronVarManager::kMultV0);
 199   }
 200
 201   //add histograms to Track classes
 202   histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",400,0,20.,AliDielectronVarManager::kPt);
 203   histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",160,-0.5,159.5,AliDielectronVarManager::kNclsTPC);
 204
 205   histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",500,-1.,1.,AliDielectronVarManager::kImpactParXY);
 206   histos->UserHistogram("Track","dZ","dZ;dZ [cm];#tracks",600,-3.,3.,AliDielectronVarManager::kImpactParZ);
 207   histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 208                         100,-1,1,144,0,6.285,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 209
 210   histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 211                         200,0.2,20.,400,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 212
 213   histos->UserHistogram("Track","dEdx_Phi","dEdx;#phi [rad];TPC signal (arb units);#tracks",
 214                         200,0.,2*TMath::Pi(),400,0.,200.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCsignal,kTRUE);
 215
 216   histos->UserHistogram("Track","dEdx_nSigmaEMCal","dEdx;NsigmaEmcal;TPC signal (arb units);NSigmaEMCAL",
 217                         200,-5.,5.,400,0.,200.,AliDielectronVarManager::kEMCALnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
 218
 219   histos->UserHistogram("Track","dEdx_TPCnSigmaEle","dEdx;TPC signal (arbunits);TPC number of sigmas Electrons;TPC signal (a.u.);#tracks",
 220                         100,-10.,10.,400,0.,200.,AliDielectronVarManager::kTPCnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
 221
 222   histos->UserHistogram("Track","dEdx_EoverP","dEdx;EoverP;TPC signal (arbunits);E/P",200,0.,5.,400,0.,200.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kTPCsignal,kTRUE);
 223
 224   histos->UserHistogram("Track","nSigmaEMCal_EoverP","NsigmaEmcal;EoverP;NSigmaEMCAL;E/P",200,0.,5.,200,-5.,5.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kEMCALnSigmaEle,kTRUE);
 225
 226   histos->UserHistogram("Track","EMCal_E","EmcalE;Cluster Energy [GeV];#Clusters",200,0.,40.,AliDielectronVarManager::kEMCALE,kTRUE);
 227
 228   //add histograms to Pair classes
 229   histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
 230                         251,-.01,5.01,AliDielectronVarManager::kM);
 231   histos->UserHistogram("Pair","InvMass2D","Inv.Mass;Pt [GeV]; Inv. Mass [GeV]",
 232                         20,0.,20.,251,-.01,5.01,AliDielectronVarManager::kPt,AliDielectronVarManager::kM);
 233   histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
 234                         100,-1.,1.,AliDielectronVarManager::kY);
 235   histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
 236                         100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
 237
 238   diele->SetHistogramManager(histos);
 239 }
 240
 241 void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
 242 {
 243   //
 244   // Setupd the CF Manager if needed
 245   //
 246
 247   AliDielectronCF *cf=new AliDielectronCF(diele->GetName(),diele->GetTitle());
 248
 249   //pair variables
 250   cf->AddVariable(AliDielectronVarManager::kPt,"3.5, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 16.0, 20.0, 100.0");
 251   cf->AddVariable(AliDielectronVarManager::kM,125,0.,125*.04); //40Mev Steps
 252   cf->AddVariable(AliDielectronVarManager::kPairType,12,0,12);
 253 //   cf->AddVariable(AliDielectronVarManager::kOpeningAngle,"0.,0.1,0.2,0.4,0.6,0.8,1.0,2.0,3.14");
 254   //leg variables
 255   cf->AddVariable(AliDielectronVarManager::kPt,"1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 2.0, 3.0, 10.0, 100.0",kTRUE);
 256 //   cf->AddVariable(AliDielectronVarManager::kP,"0.0, 1.0, 1.2, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 5.0, 10.0, 100.0",kTRUE);
 257   cf->AddVariable(AliDielectronVarManager::kNclsTPC,"85, 90, 95, 100, 120, 160",kTRUE);
 258 //   cf->AddVariable(AliDielectronVarManager::kEta,"-5.0,-1.0,-0.9,-0.7,0.7,0.9,1.0,5.0",kTRUE);
 259 //   cf->AddVariable(AliDielectronVarManager::kPhi,"0.0,1.0,2.0,3.0,4.0,5.0,6.0",kTRUE);
 260   cf->AddVariable(AliDielectronVarManager::kEMCALE,"1.0, 1.5, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 100.0",kTRUE);
 261   cf->AddVariable(AliDielectronVarManager::kEMCALnSigmaEle,"-3.5,-3.0,-2.0,-1.0,1.0,2.0,3.0,4.0,5.0",kTRUE);
 262
 263 //   cf->AddVariable(AliDielectronVarManager::kEMCALM02,"0.,0.02,0.2,0.3,0.4,0.5,0.8,1.0,2.0,4.0",kTRUE);
 264   cf->AddVariable(AliDielectronVarManager::kEMCALNCells,25,0,25,kTRUE);
 265   cf->AddVariable(AliDielectronVarManager::kEMCALEoverP,"0.6,0.7,0.8,0.9,1.1,1.2,1.3,1.4,1.8,2.0,4.0",kTRUE);
 266   cf->AddVariable(AliDielectronVarManager::kTPCnSigmaEle,"-3.0,-2.5,-2.0,-1.5,-1.0,0.0,1.0,2.0,3.0,3.5,4.0",kTRUE);
 267 //   cf->AddVariable(AliDielectronVarManager::kTPCsignal,"66.0,67.0,68.0,69.0,70.0,72.0,75.0,80.0,90.0,100.0,110.0",kTRUE);
 268   //event variables
 269   cf->AddVariable(AliDielectronVarManager::kCentrality,10,0.,100.);
 270   cf->AddVariable(AliDielectronVarManager::kTPCnSigmaPio,"2.5,3.0,3.5,4.0,4.5,100",kTRUE);
 271   cf->AddVariable(AliDielectronVarManager::kITSLayerFirstCls,4,0.,4.,kTRUE);
 272
 273   if (!isAOD){
 274     Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
 275     if (hasMC){
 276       cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
 277       cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
 278       cf->AddVariable(AliDielectronVarManager::kPdgCodeGrandMother,10000,-5000.5,4999.5,kTRUE);
 279     }
 280   }
 281     //only in this case write MC truth info
 282   if (cutDefinition==0){
 283     cf->SetStepForMCtruth();
 284   }
 285
 286   diele->SetCFManagerPair(cf);
 287
 288 }
 289