[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosJPSI / ConfigJpsi_mf_PbPb.C

void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
void AddMCSignals(AliDielectron *diele);
AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition);

TString namesDieleData=("EMCal+QA;TPC+QA;EMCal+ITSfirstclass");
TObjArray *arrNamesDieleData=namesDieleData.Tokenize("; ");
Bool_t hasMC=kFALSE;
TString list  = gSystem->Getenv("LIST");
const Int_t nDie=arrNamesDieleData->GetEntries();

AliDielectron* ConfigJpsi_mf_PbPb(Int_t cutDefinition, Bool_t isAOD=kFALSE)
{
  //
  // Setup the instance of AliDielectron
  //
    // find mc or not?
//   if( list.IsNull()) list=prod;
  if( list.Contains("LHC10h")   || list.Contains("LHC11h")   ) hasMC=kFALSE;
  if( list.Contains("LHC11a10") || list.Contains("LHC12a17") ) hasMC=kTRUE;
  
  // create the actual framework object
  TString name=Form("%02d",cutDefinition);
  if (cutDefinition<arrNamesDieleData->GetEntriesFast()){
    name=arrNamesDieleData->At(cutDefinition)->GetName();
  }
  
   
  AliDielectron *diele = new AliDielectron(Form("%s",name.Data()),
                             Form("Track cuts: %s",name.Data()));

  diele->SetHasMC(hasMC);

//   printf(" Add %s %s config %s for %s \n",(isESD?"ESD":"AOD"),(hasMC?"MC":""),name.Data(),list.Data());

  // Monte Carlo Signals and TRD efficiency tables
  if(hasMC) {
    AddMCSignals(diele);
    printf(" Add %d MC signals \n",diele->GetMCSignals()->GetEntriesFast());
  }
  
  
  // cut setup
//   SetupEventCutsDieleFilter(diele, cutDefinition, isAOD);
  SetupTrackCutsDieleData(diele, cutDefinition, isAOD);
  SetupPairCutsDieleData(diele, cutDefinition, isAOD);
  
  //
  // histogram setup
  // only if an AliDielectronHistos object is attached to the
  // dielelectron framework histograms will be filled
  //
  InitHistogramsDieleData(diele, cutDefinition, isAOD);

  // the last definition uses no cuts and only the QA histograms should be filled!
  if(cutDefinition==2||cutDefinition==0)	InitCFDieleData(diele, cutDefinition, isAOD);

  if(!hasMC){
      AliDielectronTrackRotator *rot=new AliDielectronTrackRotator;
      rot->SetIterations(10);
      rot->SetConeAnglePhi(TMath::Pi());
      rot->SetStartAnglePhi(TMath::Pi());
      if(cutDefinition>=2)  diele->SetTrackRotator(rot);
      
      AliDielectronMixingHandler *mix=new AliDielectronMixingHandler;
      mix->AddVariable(AliDielectronVarManager::kZvPrim,     20,-10.,10.);
      mix->AddVariable(AliDielectronVarManager::kCentrality,  9,  0.,90.);
      mix->SetMixType(AliDielectronMixingHandler::kAll);
      mix->SetDepth(120);
      if(cutDefinition>=2) diele->SetMixingHandler(mix);
  }
  
  return diele;
}

//______________________________________________________________________________________
void SetupEventCutsDieleFilter(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD=kFALSE)
{
  //
  // Setup the event cuts
  //
  AliDielectronVarCuts *vtxZ = new AliDielectronVarCuts("vtxZ","Vertex z cut");
  vtxZ->AddCut(AliDielectronVarManager::kZvPrim,-10.,10.);
  
  AliDielectronVarCuts *Centrality = new AliDielectronVarCuts("Centrality","Centrality Percentile");
  Centrality->AddCut(AliDielectronVarManager::kCentrality,0.,90.);
  
  diele->GetEventFilter().AddCuts(vtxZ);
  diele->GetEventFilter().AddCuts(Centrality);
  
}


//______________________________________________________________________________________
void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
{
  //
  // Setup the track cuts
  //
  
  //ESD quality cuts DielectronTrackCuts
  if (!isAOD) {
    diele->GetTrackFilter().AddCuts(SetupESDtrackCutsDieleData(cutDefinition));
  } else {
    AliDielectronTrackCuts *trackCuts=new AliDielectronTrackCuts("trackCuts","trackCuts");
    trackCuts->SetRequireTPCRefit(kTRUE);
    trackCuts->SetRequireITSRefit(kTRUE);
    diele->GetTrackFilter().AddCuts(trackCuts);
  }

  //Pt cut ----------------------------------------------------------
  AliDielectronVarCuts *pt = new AliDielectronVarCuts("ptCut","pt cut");
  pt->AddCut(AliDielectronVarManager::kPt,1.,100.);
  pt->AddCut(AliDielectronVarManager::kKinkIndex0,0.);
  //AOD additions since there are no AliESDtrackCuts -----------------
  pt->AddCut(AliDielectronVarManager::kEta,-0.9,0.9);
  pt->AddCut(AliDielectronVarManager::kImpactParXY, -1.0,   1.0);
  pt->AddCut(AliDielectronVarManager::kImpactParZ,  -3.0,   3.0);
  //pt->AddCut(AliDielectronVarManager::kTPCchi2Cl,    0.0,   4.0);
  pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-6,6.);
  
  if(cutDefinition==0)    pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-2,3.);
  if(cutDefinition==1)    pt->AddCut(AliDielectronVarManager::kEMCALEoverP,0.9,1.2);
  if(cutDefinition<=1)	  pt->AddCut(AliDielectronVarManager::kNclsTPC,90.,160.);
  
  if(cutDefinition>=2){
    pt->AddCut(AliDielectronVarManager::kNclsTPC,80.,160.);
    pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-3.5,4.1);
    pt->AddCut(AliDielectronVarManager::kPt,1.1,100.);
  }
    
  diele->GetTrackFilter().AddCuts(pt);
}

//______________________________________________________________________________________
void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
{
  //
  // Setup the pair cuts
  //
  //Invariant mass and rapidity selection
AliDielectronVarCuts *pairCut=new AliDielectronVarCuts("0<M<5+|Y|<.9+PtEMCalleg","0<M<5 + |Y|<.9+PtEMCalleg");
pairCut->AddCut(AliDielectronVarManager::kM,1.,5.);
pairCut->AddCut(AliDielectronVarManager::kY,-0.9,0.9);
pairCut->AddCut(AliDielectronVarManager::kPt,4.5,100.);
diele->GetPairFilter().AddCuts(pairCut);
  
AliDielectronVarCuts *mycut = new AliDielectronVarCuts("ptCutEMCAL","cut for EMCal");
mycut->AddCut(AliDielectronVarManager::kEMCALE,4.,100.);
mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.65,2.);
AliDielectronPairLegCuts *varpair=new AliDielectronPairLegCuts();
varpair->GetLeg1Filter().AddCuts(mycut);
varpair->GetLeg2Filter().AddCuts(mycut);
varpair->SetCutType(AliDielectronPairLegCuts::kAnyLeg);
if(cutDefinition>=2) diele->GetPairFilter().AddCuts(varpair);
  
}

//______________________________________________________________________________________
AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition)
{
  //
  // Setup default AliESDtrackCuts
  //
  AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;

  // basic track quality cuts  (basicQ)
  esdTrackCuts->SetMaxDCAToVertexZ(3.0);
  esdTrackCuts->SetMaxDCAToVertexXY(1.0);
  esdTrackCuts->SetEtaRange( -0.9 , 0.9 );
  esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
  esdTrackCuts->SetRequireITSRefit(kTRUE);
  esdTrackCuts->SetRequireTPCRefit(kTRUE);
  esdTrackCuts->SetPtRange(1.,1e30);
  esdTrackCuts->SetMinNClustersTPC(70);
  esdTrackCuts->SetMaxChi2PerClusterTPC(4);
  esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);  

  return esdTrackCuts;
}


//______________________________________________________________________________________
void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
{
  //
  // Initialise the histograms
  //
  
  //Setup histogram Manager
  AliDielectronHistos *histos=new AliDielectronHistos(diele->GetName(),diele->GetTitle());
  
  //Initialise histogram classes
  histos->SetReservedWords("Track;Pair");
  
  //Track classes
  //to fill also track info from 2nd event loop until 2
  for (Int_t i=0; i<2; ++i){
    histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
  }
  
  //Pair classes
  // to fill also mixed event histograms loop until 10
   if (cutDefinition==2) {
     for (Int_t i=0; i<10; ++i){
	histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
     }
   }
  
  //legs from pair
  for (Int_t i=0; i<3; ++i){
    histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(i)));
  }
  
  //track rotation
  if (cutDefinition==2) {     
      histos->AddClass(Form("Pair_%s",PairClassName(AliDielectron::kEv1PMRot)));
      histos->AddClass(Form("Track_Legs_%s",PairClassName(AliDielectron::kEv1PMRot)));
  }
  
  //add histograms to event class
  if (cutDefinition==0) {
    histos->AddClass("Event");
    histos->UserHistogram("Event","VtxZ","Vertex Z;Z[cm]",300,-15.,15.,AliDielectronVarManager::kZvPrim);
    histos->UserHistogram("Event","Centrality","Centrality;Cent(%)",100,0.,100.,AliDielectronVarManager::kCentrality);
    histos->UserHistogram("Event","Multiplicity","Multiplicity V0;Multiplicity V0",500,0.,25000.,AliDielectronVarManager::kMultV0);
    histos->UserHistogram("Event","Multiplicity_cent","Multiplicity V0 x Cent;Cent(%);Multiplicity V0",10,0,100.,500,0.,25000.,AliDielectronVarManager::kCentrality,AliDielectronVarManager::kMultV0);
  }
  
   //add histograms to Track classes
  histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",100,0,20.,AliDielectronVarManager::kPt);
  histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",160,-0.5,159.5,AliDielectronVarManager::kNclsTPC);
  
  histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-1.,1.,AliDielectronVarManager::kImpactParXY);
  histos->UserHistogram("Track","dZ","dZ;dZ [cm];#tracks",200,-3.,3.,AliDielectronVarManager::kImpactParZ);
  histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
                        100,-1,1,144,0,6.285,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);

  histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
                        200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);

  histos->UserHistogram("Track","dEdx_Pt","dEdx;Pt [GeV];TPC signal (arb units);#tracks",
                        200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCsignal,kTRUE);

  histos->UserHistogram("Track","TPCnSigmaPion_P","TPCnSigmaPion;P [GeV];TPCnSigmaEle;#tracks",
                        200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
  
  histos->UserHistogram("Track","TPCnSigmaEle_P","TPCnSigmaEle;P [GeV];TPCnSigmaEle;#tracks",
                        200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
  
  histos->UserHistogram("Track","TPCnSigmaEle_Pt","TPCnSigmaEle;Pt [GeV];TPCnSigmaEle;#tracks",
                        200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);

  histos->UserHistogram("Track","TPCnSigmaEle_Phi","TPCnSigmaEle;#phi [rad];TPCnSigmaEle;#tracks",
                        200,0.,2*TMath::Pi(),800,-12.,12.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
  
  histos->UserHistogram("Track","TPCnSigmaEle_Eta","TPCnSigmaEle;#eta;TPCnSigmaEle;#tracks",
                        200,-1.,1.,800,-12.,12.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
    

  histos->UserHistogram("Track","dEdx_Phi","dEdx;#phi [rad];TPC signal (arb units);#tracks",
                        200,0.,2*TMath::Pi(),800,20.,200.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCsignal,kTRUE);

  histos->UserHistogram("Track","dEdx_Eta","dEdx;#eta;TPC signal (arb units);#tracks",
                        200,-1.,1.,800,20.,200.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCsignal,kTRUE);


  histos->UserHistogram("Track","dEdx_nSigmaEMCal","dEdx;NsigmaEmcal;TPC signal (arb units);NSigmaEMCAL",
                        200,-5.,5.,800,20.,200.,AliDielectronVarManager::kEMCALnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);

  histos->UserHistogram("Track","dEdx_TPCnSigmaEle","dEdx;TPC signal (arbunits);TPC number of sigmas Electrons;TPC signal (a.u.);#tracks",
                        100,-10.,10.,800,20.,200.,AliDielectronVarManager::kTPCnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);

  histos->UserHistogram("Track","dEdx_EoverP","dEdx;EoverP;TPC signal (arbunits);E/P",100,0.,5.,800,20.,200.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kTPCsignal,kTRUE);
  
  histos->UserHistogram("Track","nSigmaEMCal_EoverP","NsigmaEmcal;EoverP;NSigmaEMCAL;E/P",100,0.,5.,200,-5.,5.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kEMCALnSigmaEle,kTRUE);
 
  histos->UserHistogram("Track","EMCal_E","EmcalE;Cluster Energy [GeV];#Clusters",200,0.,40.,AliDielectronVarManager::kEMCALE,kTRUE);

  histos->UserHistogram("Track","EMCal_E_nCells","EmcalE_nCells;# cells;Cluster Energy [GeV]",200,0.,40.,AliDielectronVarManager::kEMCALNCells,AliDielectronVarManager::kEMCALE,kTRUE);
  
  histos->UserHistogram("Track","ITS_FirstCls","ITS First Layer;ITS First Layer;#Entries",6,0.,6.,AliDielectronVarManager::kITSLayerFirstCls,kTRUE);

  //add histograms to Pair classes
  histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
                        251,-.01,5.01,AliDielectronVarManager::kM);
  histos->UserHistogram("Pair","InvMass2D","Inv.Mass;Pt [GeV]; Inv. Mass [GeV]",
                        20,0.,20.,251,-.01,5.01,AliDielectronVarManager::kPt,AliDielectronVarManager::kM);
  histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
                        100,-1.,1.,AliDielectronVarManager::kY);
  histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
                        100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
  
  diele->SetHistogramManager(histos);
}

void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
{
  //
  // Setupd the CF Manager if needed
  //
  
  AliDielectronCF *cf=new AliDielectronCF(diele->GetName(),diele->GetTitle());
  
  //pair variables
  cf->AddVariable(AliDielectronVarManager::kPt,"4.0, 4.5, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 20.0, 100.0");
  cf->AddVariable(AliDielectronVarManager::kM,125,0.,125*.04); //40Mev Steps
  cf->AddVariable(AliDielectronVarManager::kPairType,12,0,12);
  cf->AddVariable(AliDielectronVarManager::kOpeningAngle,"0.,0.1,0.2,0.4,0.6,0.8,1.0,2.0,3.14");
  cf->AddVariable(AliDielectronVarManager::kY,20,-1.,1.); 
  //leg variables
  cf->AddVariable(AliDielectronVarManager::kPt,"1.0, 1.1, 1.2, 1.3, 1.4, 1.5,1.6,1.7,1.8,1.9, 2.0, 3.0, 10.0, 100.0",kTRUE);
//   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);
  cf->AddVariable(AliDielectronVarManager::kNclsTPC,"80, 85, 90, 95, 100, 110, 120, 160",kTRUE);
//   cf->AddVariable(AliDielectronVarManager::kEta,"-5.0,-1.0,-0.9,-0.7,0.7,0.9,1.0,5.0",kTRUE);
//   cf->AddVariable(AliDielectronVarManager::kPhi,"0.0,1.0,2.0,3.0,4.0,5.0,6.0",kTRUE);
  cf->AddVariable(AliDielectronVarManager::kEMCALE,"1.0,2.0, 3.0, 4.0, 4.5, 5.0, 5.5, 6.0, 7.0, 8.0, 9.0, 10.0,11.0,12.0,16.0,100.0",kTRUE); 
  cf->AddVariable(AliDielectronVarManager::kEMCALnSigmaEle,"-3.5,-3.0,-2.0,-1.0,1.0,2.0,3.0,4.0,5.0",kTRUE);
  cf->AddVariable(AliDielectronVarManager::kEMCALNCells,25,0,25,kTRUE);
  cf->AddVariable(AliDielectronVarManager::kEMCALEoverP,"0.6, 0.65,0.7,0.8,0.9,1.1,1.2,1.3,1.4,1.8,2.0,2.1",kTRUE);
  cf->AddVariable(AliDielectronVarManager::kTPCnSigmaEle,"-3.5,-3.0,-2.5,-2.0,-1.5,-1.0,0.0,1.0,2.0,3.0,3.5,4.0",kTRUE);
  cf->AddVariable(AliDielectronVarManager::kTPCnSigmaPio,"2.,2.5,3.0,3.5,4.0,4.5,100",kTRUE);
  //event variables
  cf->AddVariable(AliDielectronVarManager::kCentrality,20,0.,100.);
  cf->AddVariable(AliDielectronVarManager::kMultV0,26,0.,26000.);
  cf->AddVariable(AliDielectronVarManager::kZvPrim,20,-10.,10.);

  cf->AddVariable(AliDielectronVarManager::kITSLayerFirstCls,6,0.,6.,kTRUE);

  if (hasMC){
      cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
      cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
      cf->AddVariable(AliDielectronVarManager::kPdgCodeGrandMother,10000,-5000.5,4999.5,kTRUE);
  }
    //only in this case write MC truth info
  if (cutDefinition==0){
    cf->SetStepForMCtruth();
//     cf->SetStepsForMCtruthOnly();
 }

  diele->SetCFManagerPair(cf);
  
}

void AddMCSignals(AliDielectron *diele){
  //Do we have an MC handler?
  //Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
  if (!hasMC) return;
  
  AliDielectronSignalMC* inclusiveJpsi = new AliDielectronSignalMC("inclusiveJpsi","Inclusive J/psi");
  inclusiveJpsi->SetLegPDGs(11,-11);
  inclusiveJpsi->SetMotherPDGs(443,443);
  inclusiveJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
  inclusiveJpsi->SetFillPureMCStep(kTRUE);
  inclusiveJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
  inclusiveJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
  diele->AddSignalMC(inclusiveJpsi);
  
  AliDielectronSignalMC* promptJpsi = new AliDielectronSignalMC("promptJpsi","Prompt J/psi");   // prompt J/psi (not from beauty decays)
  promptJpsi->SetLegPDGs(11,-11);
  promptJpsi->SetMotherPDGs(443,443);
  promptJpsi->SetGrandMotherPDGs(503,503,kTRUE,kTRUE);   // not from beauty hadrons
  promptJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
  promptJpsi->SetFillPureMCStep(kTRUE);
  promptJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
  promptJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
  promptJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
  promptJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
  diele->AddSignalMC(promptJpsi);
  
  AliDielectronSignalMC* beautyJpsi = new AliDielectronSignalMC("beautyJpsi","Beauty J/psi");
  beautyJpsi->SetLegPDGs(11,-11);
  beautyJpsi->SetMotherPDGs(443,443);
  beautyJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
  beautyJpsi->SetGrandMotherPDGs(500,500);
  beautyJpsi->SetFillPureMCStep(kTRUE);
  beautyJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
  beautyJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
  beautyJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
  diele->AddSignalMC(beautyJpsi);
  
  AliDielectronSignalMC* directJpsi = new AliDielectronSignalMC("directJpsi","Direct J/psi");   // embedded J/psi
  directJpsi->SetLegPDGs(11,-11);
  directJpsi->SetMotherPDGs(443,443);
  directJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
  directJpsi->SetFillPureMCStep(kTRUE);
  directJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
  directJpsi->SetMotherSources(AliDielectronSignalMC::kDirect, AliDielectronSignalMC::kDirect);
  directJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
  directJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
  diele->AddSignalMC(directJpsi);
}
Commit	Line	Data
bffbeda1	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);
6fdf1dbb	5	void AddMCSignals(AliDielectron *diele);
bffbeda1	6	AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition);
bffbeda1	7
6fdf1dbb	8	TString namesDieleData=("EMCal+QA;TPC+QA;EMCal+ITSfirstclass");
bffbeda1	9	TObjArray *arrNamesDieleData=namesDieleData.Tokenize("; ");
6fdf1dbb	10	Bool_t hasMC=kFALSE;
6fdf1dbb	11	TString list = gSystem->Getenv("LIST");
bffbeda1	12	const Int_t nDie=arrNamesDieleData->GetEntries();
bffbeda1	13
6fdf1dbb	14	AliDielectron* ConfigJpsi_mf_PbPb(Int_t cutDefinition, Bool_t isAOD=kFALSE)
bffbeda1	15	{
	16	//
	17	// Setup the instance of AliDielectron
	18	//
6fdf1dbb	19	// find mc or not?
	20	// if( list.IsNull()) list=prod;
	21	if( list.Contains("LHC10h") \|\| list.Contains("LHC11h") ) hasMC=kFALSE;
	22	if( list.Contains("LHC11a10") \|\| list.Contains("LHC12a17") ) hasMC=kTRUE;
bffbeda1	23
	24	// create the actual framework object
	25	TString name=Form("%02d",cutDefinition);
	26	if (cutDefinition<arrNamesDieleData->GetEntriesFast()){
	27	name=arrNamesDieleData->At(cutDefinition)->GetName();
	28	}
6fdf1dbb	29
6fdf1dbb	30
bffbeda1	31	AliDielectron *diele = new AliDielectron(Form("%s",name.Data()),
	32	Form("Track cuts: %s",name.Data()));
	33
6fdf1dbb	34	diele->SetHasMC(hasMC);
	35
	36	// printf(" Add %s %s config %s for %s \n",(isESD?"ESD":"AOD"),(hasMC?"MC":""),name.Data(),list.Data());
	37
	38	// Monte Carlo Signals and TRD efficiency tables
	39	if(hasMC) {
	40	AddMCSignals(diele);
	41	printf(" Add %d MC signals \n",diele->GetMCSignals()->GetEntriesFast());
	42	}
	43
	44
bffbeda1	45	// cut setup
6fdf1dbb	46	// SetupEventCutsDieleFilter(diele, cutDefinition, isAOD);
bffbeda1	47	SetupTrackCutsDieleData(diele, cutDefinition, isAOD);
	48	SetupPairCutsDieleData(diele, cutDefinition, isAOD);
	49
	50	//
	51	// histogram setup
	52	// only if an AliDielectronHistos object is attached to the
	53	// dielelectron framework histograms will be filled
	54	//
	55	InitHistogramsDieleData(diele, cutDefinition, isAOD);
	56
	57	// the last definition uses no cuts and only the QA histograms should be filled!
6fdf1dbb	58	if(cutDefinition==2\|\|cutDefinition==0) InitCFDieleData(diele, cutDefinition, isAOD);
	59
	60	if(!hasMC){
	61	AliDielectronTrackRotator *rot=new AliDielectronTrackRotator;
	62	rot->SetIterations(10);
	63	rot->SetConeAnglePhi(TMath::Pi());
	64	rot->SetStartAnglePhi(TMath::Pi());
	65	if(cutDefinition>=2) diele->SetTrackRotator(rot);
	66
	67	AliDielectronMixingHandler *mix=new AliDielectronMixingHandler;
	68	mix->AddVariable(AliDielectronVarManager::kZvPrim, 20,-10.,10.);
	69	mix->AddVariable(AliDielectronVarManager::kCentrality, 9, 0.,90.);
	70	mix->SetMixType(AliDielectronMixingHandler::kAll);
	71	mix->SetDepth(120);
	72	if(cutDefinition>=2) diele->SetMixingHandler(mix);
	73	}
	74
bffbeda1	75	return diele;
	76	}
	77
	78	//______________________________________________________________________________________
	79	void SetupEventCutsDieleFilter(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD=kFALSE)
	80	{
	81	//
	82	// Setup the event cuts
	83	//
	84	AliDielectronVarCuts *vtxZ = new AliDielectronVarCuts("vtxZ","Vertex z cut");
	85	vtxZ->AddCut(AliDielectronVarManager::kZvPrim,-10.,10.);
	86
	87	AliDielectronVarCuts *Centrality = new AliDielectronVarCuts("Centrality","Centrality Percentile");
	88	Centrality->AddCut(AliDielectronVarManager::kCentrality,0.,90.);
	89
	90	diele->GetEventFilter().AddCuts(vtxZ);
	91	diele->GetEventFilter().AddCuts(Centrality);
	92
	93	}
	94
	95
	96	//______________________________________________________________________________________
	97	void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
	98	{
	99	//
	100	// Setup the track cuts
	101	//
	102
	103	//ESD quality cuts DielectronTrackCuts
	104	if (!isAOD) {
	105	diele->GetTrackFilter().AddCuts(SetupESDtrackCutsDieleData(cutDefinition));
	106	} else {
	107	AliDielectronTrackCuts *trackCuts=new AliDielectronTrackCuts("trackCuts","trackCuts");
bffbeda1	108	trackCuts->SetRequireTPCRefit(kTRUE);
	109	trackCuts->SetRequireITSRefit(kTRUE);
	110	diele->GetTrackFilter().AddCuts(trackCuts);
	111	}
	112
	113	//Pt cut ----------------------------------------------------------
	114	AliDielectronVarCuts *pt = new AliDielectronVarCuts("ptCut","pt cut");
6fdf1dbb	115	pt->AddCut(AliDielectronVarManager::kPt,1.,100.);
bffbeda1	116	pt->AddCut(AliDielectronVarManager::kKinkIndex0,0.);
	117	//AOD additions since there are no AliESDtrackCuts -----------------
	118	pt->AddCut(AliDielectronVarManager::kEta,-0.9,0.9);
	119	pt->AddCut(AliDielectronVarManager::kImpactParXY, -1.0, 1.0);
	120	pt->AddCut(AliDielectronVarManager::kImpactParZ, -3.0, 3.0);
	121	//pt->AddCut(AliDielectronVarManager::kTPCchi2Cl, 0.0, 4.0);
6fdf1dbb	122	pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-6,6.);
	123
	124	if(cutDefinition==0) pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-2,3.);
	125	if(cutDefinition==1) pt->AddCut(AliDielectronVarManager::kEMCALEoverP,0.9,1.2);
	126	if(cutDefinition<=1) pt->AddCut(AliDielectronVarManager::kNclsTPC,90.,160.);
	127
	128	if(cutDefinition>=2){
	129	pt->AddCut(AliDielectronVarManager::kNclsTPC,80.,160.);
	130	pt->AddCut(AliDielectronVarManager::AliDielectronVarManager::kTPCnSigmaEle,-3.5,4.1);
	131	pt->AddCut(AliDielectronVarManager::kPt,1.1,100.);
	132	}
	133
bffbeda1	134	diele->GetTrackFilter().AddCuts(pt);
	135	}
	136
	137	//______________________________________________________________________________________
	138	void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
	139	{
	140	//
	141	// Setup the pair cuts
	142	//
	143	//Invariant mass and rapidity selection
	144	AliDielectronVarCuts *pairCut=new AliDielectronVarCuts("0<M<5+\|Y\|<.9+PtEMCalleg","0<M<5 + \|Y\|<.9+PtEMCalleg");
	145	pairCut->AddCut(AliDielectronVarManager::kM,1.,5.);
	146	pairCut->AddCut(AliDielectronVarManager::kY,-0.9,0.9);
6fdf1dbb	147	pairCut->AddCut(AliDielectronVarManager::kPt,4.5,100.);
bffbeda1	148	diele->GetPairFilter().AddCuts(pairCut);
	149
	150	AliDielectronVarCuts *mycut = new AliDielectronVarCuts("ptCutEMCAL","cut for EMCal");
6fdf1dbb	151	mycut->AddCut(AliDielectronVarManager::kEMCALE,4.,100.);
6fdf1dbb	152	mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.65,2.);
bffbeda1	153	AliDielectronPairLegCuts *varpair=new AliDielectronPairLegCuts();
	154	varpair->GetLeg1Filter().AddCuts(mycut);
	155	varpair->GetLeg2Filter().AddCuts(mycut);
	156	varpair->SetCutType(AliDielectronPairLegCuts::kAnyLeg);
6fdf1dbb	157	if(cutDefinition>=2) diele->GetPairFilter().AddCuts(varpair);
bffbeda1	158
	159	}
	160
	161	//______________________________________________________________________________________
	162	AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition)
	163	{
	164	//
	165	// Setup default AliESDtrackCuts
	166	//
	167	AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
	168
	169	// basic track quality cuts (basicQ)
	170	esdTrackCuts->SetMaxDCAToVertexZ(3.0);
	171	esdTrackCuts->SetMaxDCAToVertexXY(1.0);
	172	esdTrackCuts->SetEtaRange( -0.9 , 0.9 );
	173	esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
	174	esdTrackCuts->SetRequireITSRefit(kTRUE);
	175	esdTrackCuts->SetRequireTPCRefit(kTRUE);
	176	esdTrackCuts->SetPtRange(1.,1e30);
	177	esdTrackCuts->SetMinNClustersTPC(70);
	178	esdTrackCuts->SetMaxChi2PerClusterTPC(4);
	179	esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
	180
	181	return esdTrackCuts;
	182	}
	183
	184
	185	//______________________________________________________________________________________
	186	void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
	187	{
	188	//
	189	// Initialise the histograms
	190	//
	191
	192	//Setup histogram Manager
	193	AliDielectronHistos *histos=new AliDielectronHistos(diele->GetName(),diele->GetTitle());
	194
	195	//Initialise histogram classes
	196	histos->SetReservedWords("Track;Pair");
	197
	198	//Track classes
	199	//to fill also track info from 2nd event loop until 2
	200	for (Int_t i=0; i<2; ++i){
	201	histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
	202	}
	203
	204	//Pair classes
	205	// to fill also mixed event histograms loop until 10
6fdf1dbb	206	if (cutDefinition==2) {
	207	for (Int_t i=0; i<10; ++i){
	208	histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
	209	}
	210	}
bffbeda1	211
	212	//legs from pair
	213	for (Int_t i=0; i<3; ++i){
	214	histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(i)));
	215	}
	216
	217	//track rotation
6fdf1dbb	218	if (cutDefinition==2) {
bffbeda1	219	histos->AddClass(Form("Pair_%s",PairClassName(AliDielectron::kEv1PMRot)));
	220	histos->AddClass(Form("Track_Legs_%s",PairClassName(AliDielectron::kEv1PMRot)));
	221	}
	222
	223	//add histograms to event class
	224	if (cutDefinition==0) {
	225	histos->AddClass("Event");
	226	histos->UserHistogram("Event","VtxZ","Vertex Z;Z[cm]",300,-15.,15.,AliDielectronVarManager::kZvPrim);
	227	histos->UserHistogram("Event","Centrality","Centrality;Cent(%)",100,0.,100.,AliDielectronVarManager::kCentrality);
	228	histos->UserHistogram("Event","Multiplicity","Multiplicity V0;Multiplicity V0",500,0.,25000.,AliDielectronVarManager::kMultV0);
	229	histos->UserHistogram("Event","Multiplicity_cent","Multiplicity V0 x Cent;Cent(%);Multiplicity V0",10,0,100.,500,0.,25000.,AliDielectronVarManager::kCentrality,AliDielectronVarManager::kMultV0);
	230	}
	231
6fdf1dbb	232	//add histograms to Track classes
6fdf1dbb	233	histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",100,0,20.,AliDielectronVarManager::kPt);
bffbeda1	234	histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",160,-0.5,159.5,AliDielectronVarManager::kNclsTPC);
bffbeda1	235
6fdf1dbb	236	histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-1.,1.,AliDielectronVarManager::kImpactParXY);
6fdf1dbb	237	histos->UserHistogram("Track","dZ","dZ;dZ [cm];#tracks",200,-3.,3.,AliDielectronVarManager::kImpactParZ);
bffbeda1	238	histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
	239	100,-1,1,144,0,6.285,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
	240
	241	histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
6fdf1dbb	242	200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal,kTRUE);
	243
	244	histos->UserHistogram("Track","dEdx_Pt","dEdx;Pt [GeV];TPC signal (arb units);#tracks",
	245	200,0.2,20.,800,20.,200.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCsignal,kTRUE);
	246
	247	histos->UserHistogram("Track","TPCnSigmaPion_P","TPCnSigmaPion;P [GeV];TPCnSigmaEle;#tracks",
	248	200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio,kTRUE);
	249
	250	histos->UserHistogram("Track","TPCnSigmaEle_P","TPCnSigmaEle;P [GeV];TPCnSigmaEle;#tracks",
	251	200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
	252
	253	histos->UserHistogram("Track","TPCnSigmaEle_Pt","TPCnSigmaEle;Pt [GeV];TPCnSigmaEle;#tracks",
	254	200,0.2,20.,800,-12.,12.,AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
	255
	256	histos->UserHistogram("Track","TPCnSigmaEle_Phi","TPCnSigmaEle;#phi [rad];TPCnSigmaEle;#tracks",
	257	200,0.,2*TMath::Pi(),800,-12.,12.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
	258
	259	histos->UserHistogram("Track","TPCnSigmaEle_Eta","TPCnSigmaEle;#eta;TPCnSigmaEle;#tracks",
	260	200,-1.,1.,800,-12.,12.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaEle,kTRUE);
	261
	262
bffbeda1	263
bffbeda1	264	histos->UserHistogram("Track","dEdx_Phi","dEdx;#phi [rad];TPC signal (arb units);#tracks",
6fdf1dbb	265	200,0.,2*TMath::Pi(),800,20.,200.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kTPCsignal,kTRUE);
	266
	267	histos->UserHistogram("Track","dEdx_Eta","dEdx;#eta;TPC signal (arb units);#tracks",
	268	200,-1.,1.,800,20.,200.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCsignal,kTRUE);
	269
	270
bffbeda1	271
bffbeda1	272	histos->UserHistogram("Track","dEdx_nSigmaEMCal","dEdx;NsigmaEmcal;TPC signal (arb units);NSigmaEMCAL",
6fdf1dbb	273	200,-5.,5.,800,20.,200.,AliDielectronVarManager::kEMCALnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
bffbeda1	274
bffbeda1	275	histos->UserHistogram("Track","dEdx_TPCnSigmaEle","dEdx;TPC signal (arbunits);TPC number of sigmas Electrons;TPC signal (a.u.);#tracks",
6fdf1dbb	276	100,-10.,10.,800,20.,200.,AliDielectronVarManager::kTPCnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
bffbeda1	277
6fdf1dbb	278	histos->UserHistogram("Track","dEdx_EoverP","dEdx;EoverP;TPC signal (arbunits);E/P",100,0.,5.,800,20.,200.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kTPCsignal,kTRUE);
bffbeda1	279
6fdf1dbb	280	histos->UserHistogram("Track","nSigmaEMCal_EoverP","NsigmaEmcal;EoverP;NSigmaEMCAL;E/P",100,0.,5.,200,-5.,5.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kEMCALnSigmaEle,kTRUE);
6fdf1dbb	281
bffbeda1	282	histos->UserHistogram("Track","EMCal_E","EmcalE;Cluster Energy [GeV];#Clusters",200,0.,40.,AliDielectronVarManager::kEMCALE,kTRUE);
bffbeda1	283
6fdf1dbb	284	histos->UserHistogram("Track","EMCal_E_nCells","EmcalE_nCells;# cells;Cluster Energy [GeV]",200,0.,40.,AliDielectronVarManager::kEMCALNCells,AliDielectronVarManager::kEMCALE,kTRUE);
	285
	286	histos->UserHistogram("Track","ITS_FirstCls","ITS First Layer;ITS First Layer;#Entries",6,0.,6.,AliDielectronVarManager::kITSLayerFirstCls,kTRUE);
	287
bffbeda1	288	//add histograms to Pair classes
	289	histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
	290	251,-.01,5.01,AliDielectronVarManager::kM);
	291	histos->UserHistogram("Pair","InvMass2D","Inv.Mass;Pt [GeV]; Inv. Mass [GeV]",
	292	20,0.,20.,251,-.01,5.01,AliDielectronVarManager::kPt,AliDielectronVarManager::kM);
	293	histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
	294	100,-1.,1.,AliDielectronVarManager::kY);
	295	histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
	296	100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
	297
	298	diele->SetHistogramManager(histos);
	299	}
	300
	301	void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD)
	302	{
	303	//
	304	// Setupd the CF Manager if needed
	305	//
	306
	307	AliDielectronCF *cf=new AliDielectronCF(diele->GetName(),diele->GetTitle());
	308
	309	//pair variables
6fdf1dbb	310	cf->AddVariable(AliDielectronVarManager::kPt,"4.0, 4.5, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 20.0, 100.0");
bffbeda1	311	cf->AddVariable(AliDielectronVarManager::kM,125,0.,125*.04); //40Mev Steps
bffbeda1	312	cf->AddVariable(AliDielectronVarManager::kPairType,12,0,12);
6fdf1dbb	313	cf->AddVariable(AliDielectronVarManager::kOpeningAngle,"0.,0.1,0.2,0.4,0.6,0.8,1.0,2.0,3.14");
6fdf1dbb	314	cf->AddVariable(AliDielectronVarManager::kY,20,-1.,1.);
bffbeda1	315	//leg variables
6fdf1dbb	316	cf->AddVariable(AliDielectronVarManager::kPt,"1.0, 1.1, 1.2, 1.3, 1.4, 1.5,1.6,1.7,1.8,1.9, 2.0, 3.0, 10.0, 100.0",kTRUE);
bffbeda1	317	// 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);
6fdf1dbb	318	cf->AddVariable(AliDielectronVarManager::kNclsTPC,"80, 85, 90, 95, 100, 110, 120, 160",kTRUE);
bffbeda1	319	// cf->AddVariable(AliDielectronVarManager::kEta,"-5.0,-1.0,-0.9,-0.7,0.7,0.9,1.0,5.0",kTRUE);
bffbeda1	320	// cf->AddVariable(AliDielectronVarManager::kPhi,"0.0,1.0,2.0,3.0,4.0,5.0,6.0",kTRUE);
6fdf1dbb	321	cf->AddVariable(AliDielectronVarManager::kEMCALE,"1.0,2.0, 3.0, 4.0, 4.5, 5.0, 5.5, 6.0, 7.0, 8.0, 9.0, 10.0,11.0,12.0,16.0,100.0",kTRUE);
bffbeda1	322	cf->AddVariable(AliDielectronVarManager::kEMCALnSigmaEle,"-3.5,-3.0,-2.0,-1.0,1.0,2.0,3.0,4.0,5.0",kTRUE);
bffbeda1	323	cf->AddVariable(AliDielectronVarManager::kEMCALNCells,25,0,25,kTRUE);
6fdf1dbb	324	cf->AddVariable(AliDielectronVarManager::kEMCALEoverP,"0.6, 0.65,0.7,0.8,0.9,1.1,1.2,1.3,1.4,1.8,2.0,2.1",kTRUE);
	325	cf->AddVariable(AliDielectronVarManager::kTPCnSigmaEle,"-3.5,-3.0,-2.5,-2.0,-1.5,-1.0,0.0,1.0,2.0,3.0,3.5,4.0",kTRUE);
	326	cf->AddVariable(AliDielectronVarManager::kTPCnSigmaPio,"2.,2.5,3.0,3.5,4.0,4.5,100",kTRUE);
bffbeda1	327	//event variables
	328	cf->AddVariable(AliDielectronVarManager::kCentrality,20,0.,100.);
	329	cf->AddVariable(AliDielectronVarManager::kMultV0,26,0.,26000.);
6fdf1dbb	330	cf->AddVariable(AliDielectronVarManager::kZvPrim,20,-10.,10.);
bffbeda1	331
bffbeda1	332	cf->AddVariable(AliDielectronVarManager::kITSLayerFirstCls,6,0.,6.,kTRUE);
bffbeda1	333
6fdf1dbb	334	if (hasMC){
bffbeda1	335	cf->AddVariable(AliDielectronVarManager::kPdgCode,10000,-5000.5,4999.5,kTRUE);
	336	cf->AddVariable(AliDielectronVarManager::kPdgCodeMother,10000,-5000.5,4999.5,kTRUE);
	337	cf->AddVariable(AliDielectronVarManager::kPdgCodeGrandMother,10000,-5000.5,4999.5,kTRUE);
bffbeda1	338	}
	339	//only in this case write MC truth info
	340	if (cutDefinition==0){
	341	cf->SetStepForMCtruth();
6fdf1dbb	342	// cf->SetStepsForMCtruthOnly();
6fdf1dbb	343	}
bffbeda1	344
	345	diele->SetCFManagerPair(cf);
	346
	347	}
	348
6fdf1dbb	349	void AddMCSignals(AliDielectron *diele){
	350	//Do we have an MC handler?
	351	//Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
	352	if (!hasMC) return;
	353
	354	AliDielectronSignalMC* inclusiveJpsi = new AliDielectronSignalMC("inclusiveJpsi","Inclusive J/psi");
	355	inclusiveJpsi->SetLegPDGs(11,-11);
	356	inclusiveJpsi->SetMotherPDGs(443,443);
	357	inclusiveJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
	358	inclusiveJpsi->SetFillPureMCStep(kTRUE);
	359	inclusiveJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
	360	inclusiveJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
	361	diele->AddSignalMC(inclusiveJpsi);
	362
	363	AliDielectronSignalMC* promptJpsi = new AliDielectronSignalMC("promptJpsi","Prompt J/psi"); // prompt J/psi (not from beauty decays)
	364	promptJpsi->SetLegPDGs(11,-11);
	365	promptJpsi->SetMotherPDGs(443,443);
	366	promptJpsi->SetGrandMotherPDGs(503,503,kTRUE,kTRUE); // not from beauty hadrons
	367	promptJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
	368	promptJpsi->SetFillPureMCStep(kTRUE);
	369	promptJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
	370	promptJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
	371	promptJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
	372	promptJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
	373	diele->AddSignalMC(promptJpsi);
	374
	375	AliDielectronSignalMC* beautyJpsi = new AliDielectronSignalMC("beautyJpsi","Beauty J/psi");
	376	beautyJpsi->SetLegPDGs(11,-11);
	377	beautyJpsi->SetMotherPDGs(443,443);
	378	beautyJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
	379	beautyJpsi->SetGrandMotherPDGs(500,500);
	380	beautyJpsi->SetFillPureMCStep(kTRUE);
	381	beautyJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
	382	beautyJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
	383	beautyJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
	384	diele->AddSignalMC(beautyJpsi);
	385
	386	AliDielectronSignalMC* directJpsi = new AliDielectronSignalMC("directJpsi","Direct J/psi"); // embedded J/psi
	387	directJpsi->SetLegPDGs(11,-11);
	388	directJpsi->SetMotherPDGs(443,443);
	389	directJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
	390	directJpsi->SetFillPureMCStep(kTRUE);
	391	directJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
	392	directJpsi->SetMotherSources(AliDielectronSignalMC::kDirect, AliDielectronSignalMC::kDirect);
	393	directJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
	394	directJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
	395	diele->AddSignalMC(directJpsi);
	396	}