PWGDQ/dielectron/macros/ConfigJpsi2eeDataAOD.C

   1
   2 void InitHistograms(AliDielectron *die, Int_t cutDefinition);
   3 void InitCF(AliDielectron* die, Int_t cutDefinition);
   4
   5 void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition);
   6 void SetupPairCuts(AliDielectron *die, Int_t cutDefinition);
   7
   8 TString names=("trackQ+Pt>1GeV+loose_PID;trackQ+Pt>1GeV+tight_PID;trackQ+Pt>.5GeV");
   9 TObjArray *arrNames=names.Tokenize(";");
  10
  11 const Int_t nDie=arrNames->GetEntries();
  12
  13 AliDielectron* ConfigJpsi2ee(Int_t cutDefinition)
  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<arrNames->GetEntriesFast()){
  22     name=arrNames->At(cutDefinition)->GetName();
  23   }
  24   AliDielectron *die =
  25     new AliDielectron(Form("%s",name.Data()),
  26                       Form("Track cuts: %s",name.Data()));
  27
  28   // cut setup
  29   SetupTrackCuts(die,cutDefinition);
  30   SetupPairCuts(die,cutDefinition);
  31
  32   //
  33   // histogram setup
  34   // only if an AliDielectronHistos object is attached to the
  35   //  dielectron framework the QA histograms will be filled
  36   //
  37   InitHistograms(die,cutDefinition);
  38
  39   // the last definition uses no cuts and only the QA histograms should be filled!
  40   if (cutDefinition<nDie-1) InitCF(die,cutDefinition);
  41
  42   return die;
  43 }
  44
  45 //______________________________________________________________________________________
  46 void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition)
  47 {
  48   //
  49   // Setup the track cuts
  50   //
  51
  52
  53
  54   //QA no CF
  55   if (cutDefinition==nDie-1) {
  56     //Pt cut
  57     AliDielectronVarCuts *pt = new AliDielectronVarCuts("Pt>.5","Pt>.5");
  58     pt->AddCut(AliDielectronVarManager::kPt,.5,1e30);
  59     die->GetTrackFilter().AddCuts(pt);
  60
  61     return;
  62   }
  63
  64
  65   AliDielectronVarCuts *pt = new AliDielectronVarCuts("Pt>1","Pt>1");
  66   pt->AddCut(AliDielectronVarManager::kPt,1.,1e30);
  67   pt->AddCut(AliDielectronVarManager::kImpactParZ,-3.,3.);
  68   pt->AddCut(AliDielectronVarManager::kImpactParXY,-.07,.07);
  69   die->GetTrackFilter().AddCuts(pt);
  70
  71   //loose PID
  72   if (cutDefinition==0){
  73     //ESD pid cuts (TPC nSigma Electrons)
  74     AliDielectronVarCuts *pid = new AliDielectronVarCuts("TPCpid","TPC nSigma cut |e|<3 pi>2 |P|>2 |K|>2");
  75     //include
  76     pid->AddCut(AliDielectronVarManager::kTPCnSigmaEle, -3., 3.);
  77     //exclude
  78     pid->AddCut(AliDielectronVarManager::kTPCnSigmaPio, -20., 2., kTRUE);
  79     pid->AddCut(AliDielectronVarManager::kTPCnSigmaPro, -2., 2., kTRUE);
  80     pid->AddCut(AliDielectronVarManager::kTPCnSigmaKao, -2., 2., kTRUE);
  81     die->GetTrackFilter().AddCuts(pid);
  82   }
  83
  84   //tight PID
  85   if (cutDefinition==1){
  86     //ESD pid cuts (TPC nSigma Electrons)
  87     AliDielectronVarCuts *pid = new AliDielectronVarCuts("TPCpid","TPC nSigma cut |e|<2 pi>2 |P|>2 |K|>2; |dXY|<200um");
  88     //include
  89     pid->AddCut(AliDielectronVarManager::kTPCnSigmaEle, -2., 2.);
  90     //exclude
  91     pid->AddCut(AliDielectronVarManager::kTPCnSigmaPio, -2., 2., kTRUE);
  92     pid->AddCut(AliDielectronVarManager::kTPCnSigmaPro, -2., 2., kTRUE);
  93     pid->AddCut(AliDielectronVarManager::kTPCnSigmaKao, -2., 2., kTRUE);
  94     //tighten Impact parameter to 200um
  95     pid->AddCut(AliDielectronVarManager::kImpactParXY, -0.02, 0.02);
  96     die->GetTrackFilter().AddCuts(pid);
  97   }
  98
  99 }
 100
 101 //______________________________________________________________________________________
 102 void SetupPairCuts(AliDielectron *die, Int_t cutDefinition)
 103 {
 104   //
 105   // Setup the pair cuts
 106   //
 107
 108   //Invariant mass and rapidity selection
 109   AliDielectronVarCuts *pairCut=new AliDielectronVarCuts("2<M<4+|Y|<.8","2<M<4 + |Y|<.8");
 110   pairCut->AddCut(AliDielectronVarManager::kM,2.,4.);
 111   pairCut->AddCut(AliDielectronVarManager::kY,-0.8,0.8);
 112 }
 113
 114 //______________________________________________________________________________________
 115 void InitHistograms(AliDielectron *die, Int_t cutDefinition)
 116 {
 117   //
 118   // Initialise the QA histograms
 119   //
 120
 121   //Setup QA histograms
 122   AliDielectronHistos *histos=
 123     new AliDielectronHistos(die->GetName(),
 124                             die->GetTitle());
 125
 126   //Initialise histogram classes
 127   histos->SetReservedWords("Track;Pair");
 128
 129   //Event class (only for last QA)
 130   if (cutDefinition==nDie-1) histos->AddClass("Event");
 131
 132   //Track classes, only first event
 133   for (Int_t i=0; i<2; ++i){
 134     histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
 135   }
 136
 137   //Pair classes, only first event
 138   for (Int_t i=0; i<3; ++i){
 139     histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 140   }
 141
 142   //Event histograms
 143   if (cutDefinition==nDie-1){
 144     //add histograms to event class
 145     histos->UserHistogram("Event","nEvents","Number of processed events after cuts;Number events",
 146                           1,0.,1.,AliDielectronVarManager::kNevents);
 147   }
 148
 149   //add histograms to Track classes
 150   histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",200,0,20.,AliDielectronVarManager::kPt);
 151   histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 152                         400,0.2,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
 153   histos->UserHistogram("Track","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 154                         400,0.2,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
 155   histos->UserHistogram("Track","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 156                         400,0.2,20.,100,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
 157   histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-2.,2.,AliDielectronVarManager::kImpactParXY);
 158
 159   //add histograms to Pair classes
 160   histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
 161                         500,0.,4.,AliDielectronVarManager::kM);
 162   histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
 163                         100,-2.,2.,AliDielectronVarManager::kY);
 164   histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
 165                         100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
 166   histos->UserHistogram("Pair","Chi2/NDF","#Chi^{2}/NDF;#Chi^{2}/NDF",
 167                         100, 0., 20., AliDielectronVarManager::kChi2NDF);
 168
 169   die->SetHistogramManager(histos);
 170 }
 171
 172
 173 void InitCF(AliDielectron* die, Int_t cutDefinition)
 174 {
 175   //
 176   // Setupd the CF Manager if needed
 177   //
 178   AliDielectronCF *cf=new AliDielectronCF(die->GetName(),die->GetTitle());
 179
 180   //pair variables
 181   cf->AddVariable(AliDielectronVarManager::kPt,20,0,10);
 182   cf->AddVariable(AliDielectronVarManager::kY,40,-2,2);
 183   cf->AddVariable(AliDielectronVarManager::kM,200,-0.01,3.99);
 184   cf->AddVariable(AliDielectronVarManager::kPairType,10,0,10);
 185   //leg variables
 186   cf->AddVariable(AliDielectronVarManager::kPt,20,0,10,kTRUE);
 187
 188   //only in this case write MC truth info
 189   cf->SetStepsForCutsIncreasing();
 190   if (cutDefinition==0){
 191     cf->SetStepForMCtruth();
 192   }
 193
 194   die->SetCFManagerPair(cf);
 195 }