PWGDQ/dielectron/macrosJPSI/AddTask_ReducedTree.C

   1
   2 //__________________________________________________________________________________________
   3 AliAnalysisTask *AddTask_ReducedTree(){
   4   //get the current analysis manager
   5   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
   6   if (!mgr) {
   7     Error("AddTask_ReducedTree", "No analysis manager found.");
   8     return 0;
   9   }
  10
  11   //Do we have an MC handler?
  12   Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
  13
  14   //create task and add it to the manager
  15   AliAnalysisTaskReducedTree *task=new AliAnalysisTaskReducedTree("DSTTreeMaker");
  16   task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
  17   //if(trainConfig=="pp") task->SetRejectPileup();
  18   if(!hasMC) task->UsePhysicsSelection(kFALSE);
  19   mgr->AddTask(task);
  20
  21   task->SetFillV0Info(kFALSE);
  22   task->SetFillCaloClusterInfo(kFALSE);
  23   task->SetFillFriendInfo(kFALSE);
  24
  25   task->SetEventFilter(CreateEventFilter());
  26   task->SetTrackFilter(CreateGlobalTrackFilter());
  27   task->SetFlowTrackFilter(CreateFlowTrackFilter());
  28   task->SetK0sPionCuts(CreateK0sPionCuts());
  29   task->SetLambdaProtonCuts(CreateLambdaProtonCuts());
  30   task->SetLambdaPionCuts(CreateLambdaPionCuts());
  31   task->SetK0sCuts(CreateK0sCuts());
  32   task->SetK0sMassRange(0.44,0.55);
  33   task->SetLambdaMassRange(1.090,1.14);
  34   task->SetLambdaCuts(CreateLambdaCuts());
  35   //task->SetV0Histograms(CreateV0Histograms());
  36
  37   task->AddDielectron(ConfigDielectron(0));   // J/psi -> e+e-
  38   //task->AddDielectron(ConfigDielectron(1));   // phi -> K+K-
  39
  40   //create output container
  41   AliAnalysisDataContainer *coutput1 =
  42     mgr->CreateContainer("diele_defaultTree",
  43                          TTree::Class(),
  44                          AliAnalysisManager::kExchangeContainer,
  45                          "diele_default");
  46
  47   AliAnalysisDataContainer *cOutputHist1 =
  48     mgr->CreateContainer("qaHistos",
  49                          TList::Class(),
  50                          AliAnalysisManager::kOutputContainer,
  51                          "dstTree.root");
  52
  53   AliAnalysisDataContainer *cOutputHist2 =
  54     mgr->CreateContainer("dstTree",
  55                          TTree::Class(),
  56                          AliAnalysisManager::kOutputContainer,
  57                          "dstTree.root");
  58
  59   /*AliAnalysisDataContainer *cOutputHist3 =
  60     mgr->CreateContainer("friendTree",
  61                          TTree::Class(),
  62                          AliAnalysisManager::kOutputContainer,
  63                          "dstTree_friend.root");*/
  64   /*  cout << "output containers: " << endl
  65        << cOutputHist1 << endl
  66        << cOutputHist2 << endl
  67        << cOutputHist3 << endl;*/
  68
  69   mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
  70   mgr->ConnectOutput(task, 0, coutput1 );
  71   mgr->ConnectOutput(task, 1, cOutputHist1);
  72   mgr->ConnectOutput(task, 2, cOutputHist2);
  73   //mgr->ConnectOutput(task, 3, cOutputHist3);
  74   //mgr->ConnectOutput(task, 2, cOutputHist3);
  75   //mgr->ConnectOutput(task, 1, cOutputHist2);
  76   return task;
  77 }
  78
  79
  80 //_______________________________________________________________________________________________
  81 AliAnalysisCuts* CreateEventFilter() {
  82   //
  83   // Event wise cuts
  84   //
  85   AliDielectronEventCuts *eventCuts=new AliDielectronEventCuts("eventCuts","Vertex Track && |vtxZ|<10 && ncontrib>0");
  86   eventCuts->SetRequireVertex();
  87   eventCuts->SetMinVtxContributors(1);
  88   eventCuts->SetVertexZ(-10.,10.);
  89   return eventCuts;
  90 }
  91
  92
  93 //_______________________________________________________________________________________________
  94 AliAnalysisCuts* CreateGlobalTrackFilter() {
  95   //
  96   // Cuts for tracks to be written in the dst
  97   //
  98   AliDielectronCutGroup* cuts = new AliDielectronCutGroup("cuts","cuts",AliDielectronCutGroup::kCompAND);
  99   // general ESD cuts ---------------------------------
 100   AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
 101   // basic track quality cuts  (basicQ)
 102   esdTrackCuts->SetMaxDCAToVertexZ(10.0);
 103   esdTrackCuts->SetMaxDCAToVertexXY(3.0);
 104   esdTrackCuts->SetEtaRange( -0.9 , 0.9 );
 105   //esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
 106   //esdTrackCuts->SetRequireITSRefit(kTRUE);
 107   esdTrackCuts->SetRequireTPCRefit(kTRUE);
 108   esdTrackCuts->SetPRange(0.8,1e30);
 109   //esdTrackCuts->SetMinNClustersTPC(60);
 110   //esdTrackCuts->SetMaxChi2PerClusterTPC(4);
 111   //  esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
 112   //  esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kFirst);
 113   cuts->AddCut(esdTrackCuts);
 114
 115   return cuts;
 116 }
 117
 118
 119 //_______________________________________________________________________________________________
 120 AliAnalysisCuts* CreateFlowTrackFilter() {
 121   //
 122   // Cuts for tracks to be used for the event plane q-vector
 123   // These cuts are applied aditionally to the global track cuts
 124   //
 125   AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
 126   esdTrackCuts->SetPtRange(0.2,2.0);
 127   esdTrackCuts->SetEtaRange(-0.8, 0.8);
 128   esdTrackCuts->SetMinNClustersTPC(70);
 129   return esdTrackCuts;
 130 }
 131
 132
 133 //_______________________________________________________________________________________________
 134 AliAnalysisCuts* CreateK0sPionCuts() {
 135   //
 136   // Cuts on the K0s pions (tracking cuts, pid cuts, ...)
 137   //
 138   AliESDtrackCuts *pionCuts = new AliESDtrackCuts;
 139   pionCuts->SetPtRange(0.15,100.0);
 140   return pionCuts;
 141 }
 142
 143
 144 //_______________________________________________________________________________________________
 145 AliAnalysisCuts* CreateLambdaPionCuts() {
 146   //
 147   // Cuts on the Lambda pions (tracking cuts, pid cuts, ...)
 148   //
 149   AliESDtrackCuts *pionCuts = new AliESDtrackCuts;
 150   pionCuts->SetPtRange(0.15,100.0);
 151   return pionCuts;
 152 }
 153
 154
 155 //_______________________________________________________________________________________________
 156 AliAnalysisCuts* CreateLambdaProtonCuts() {
 157   //
 158   // Cuts on the Lambda protons (tracking cuts, pid cuts, ...)
 159   //
 160   AliESDtrackCuts *protonCuts = new AliESDtrackCuts;
 161   protonCuts->SetPtRange(0.15,100.0);
 162   return protonCuts;
 163 }
 164
 165
 166 //______________________________________________________________________________________
 167 AliESDv0Cuts* CreateK0sCuts() {
 168   //
 169   // Cuts on the V0s with K0s hypothesis
 170   //
 171
 172   AliESDv0Cuts *cuts=new AliESDv0Cuts();
 173   //cuts->SetMinDcaPosToVertex(-1);
 174   //cuts->SetMinDcaNegToVertex(-1);
 175   //cuts->SetMaxChi2(10);
 176   //cuts->SetMaxDcaV0Daughters(0.3);
 177   //cuts->SetMinRadius(3.0);
 178   //cuts->SetMaxRadius(90.0);
 179   //cuts->SetMinCosinePointingAngle(0.9);
 180   //cuts->SetRequireOnFlyStatus(kTRUE);
 181   //cuts->SetMaxDcaV0ToVertex(0.5);
 182   //cuts->SetPRange(0.,1.0e10);
 183   //cuts->SetPtRange(0.,1.0e10);
 184   return cuts;
 185 }
 186
 187
 188 //______________________________________________________________________________________
 189 AliESDv0Cuts* CreateLambdaCuts() {
 190   //
 191   // Cuts on the V0s with Lambda hypothesis
 192   //
 193
 194   AliESDv0Cuts *cuts=new AliESDv0Cuts();
 195   //cuts->SetMinDcaPosToVertex(-1);
 196   //cuts->SetMinDcaNegToVertex(-1);
 197   //cuts->SetMaxChi2(10);
 198   //cuts->SetMaxDcaV0Daughters(0.3);
 199   //cuts->SetMinRadius(3.0);
 200   //cuts->SetMaxRadius(90.0);
 201   //cuts->SetMinCosinePointingAngle(0.9);
 202   //cuts->SetRequireOnFlyStatus(kTRUE);
 203   //cuts->SetMaxDcaV0ToVertex(0.5);
 204   //cuts->SetPRange(0.,1.0e10);
 205   //cuts->SetPtRange(0.,1.0e10);
 206   return cuts;
 207 }
 208
 209
 210 //______________________________________________________________________________________
 211 AliDielectronHistos* CreateV0Histograms()
 212 {
 213   //
 214   // Initialise the histograms
 215   //
 216
 217   //Setup histogram Manager
 218   AliDielectronHistos *histos=new AliDielectronHistos("V0Histograms","");
 219
 220   //add histograms to Track classes ---------------------------------------------------------------------------
 221   histos->SetReservedWords("V0Track");
 222
 223   //Track classes
 224   histos->AddClass("V0Track_Pos"); histos->AddClass("V0Track_Neg");
 225
 226   // kinematic acceptance histograms
 227   histos->UserHistogram("V0Track","Pt","Pt;Pt (GeV/c);#tracks",400,0,20.,AliDielectronVarManager::kPt);
 228   histos->UserHistogram("V0Track","Eta_P","Eta_P;#eta;P (GeV/c)",40, -1.0, +1.0, 200,0,10., AliDielectronVarManager::kEta, AliDielectronVarManager::kP);
 229   histos->UserHistogram("V0Track","Eta_TRDPhi","Eta Phi Map; Eta; Phi;#tracks",
 230                         40,-1,1,200,-3.15,3.15,AliDielectronVarManager::kEta,AliDielectronVarManager::kTRDphi);
 231
 232   // DCA diagnostic histograms
 233   histos->UserHistogram("V0Track","dXY","dXY;dXY (cm);#tracks",500,-1.,1.,AliDielectronVarManager::kImpactParXY);
 234   histos->UserHistogram("V0Track","dZ","dZ;dZ (cm);#tracks",600,-3.,3.,AliDielectronVarManager::kImpactParZ);
 235   histos->UserHistogram("V0Track","dZ_dXY","dZ_dXY;dZ (cm);dXY (cm)",600,-3.,3.,500,-1.,1.,AliDielectronVarManager::kImpactParZ, AliDielectronVarManager::kImpactParXY);
 236   histos->UserHistogram("V0Track","P_dXY","P_dXY;P (GeV/c);dXY (cm)",200,0.,10.,500,-1.,1.,AliDielectronVarManager::kP, AliDielectronVarManager::kImpactParXY);
 237   histos->UserHistogram("V0Track","P_dZ","P_dZ;P (GeV/c);dZ (cm)",200,0.,10.,600,-3.,3.,AliDielectronVarManager::kP, AliDielectronVarManager::kImpactParZ);
 238
 239   // ITS
 240   histos->UserHistogram("V0Track","ITSsignal","ITS signal;ITS dE/dx",1000,0.0,1000.0, AliDielectronVarManager::kITSsignal);
 241   histos->UserHistogram("V0Track","ITSsignal_P","ITS signal vs P;P (GeV/c); ITS dE/dx",300, 0.0, 3.0, 200,0.0,1000.0, AliDielectronVarManager::kP, AliDielectronVarManager::kITSsignal);
 242
 243   // TPC
 244   histos->UserHistogram("V0Track","TPCchi2","#chi^{2}/cluster;#chi^{2}/cluster;#tracks",200,0.0,5.0, AliDielectronVarManager::kTPCchi2Cl);
 245   histos->UserHistogram("V0Track","Pt_TPCchi2","#chi^{2}/cluster vs pt;p_{T} (GeV/c);#chi^{2}/cluster",400,0.,20., 200,0.0,5.0,
 246                         AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCchi2Cl);
 247   histos->UserHistogram("V0Track","TPCnCls","Number of Clusters TPC;TPC number clusters;#tracks",160,-0.5,159.5, AliDielectronVarManager::kNclsTPC);
 248   histos->UserHistogram("V0Track","Phi_TPCnCls","Number of Clusters TPC vs #phi;#phi (rad.);TPC number clusters",200,0,6.285, 160,-0.5,159.5,
 249                         AliDielectronVarManager::kPhi,AliDielectronVarManager::kNclsTPC);
 250   histos->UserHistogram("V0Track","Pin_TPCnCls","Number of Clusters TPC vs TPC inner param P;p (GeV/c);TPC number clusters",400,0,20., 160,-0.5,159.5,
 251                         AliDielectronVarManager::kPIn, AliDielectronVarManager::kNclsTPC);
 252   histos->UserHistogram("V0Track","TPCnCls_dEdx","Number of Clusters TPC vs dE/dx; TPC number clusters; dE/dx (arb.units)", 160,-0.5,159.5, 200,0,200.,
 253                         AliDielectronVarManager::kNclsTPC, AliDielectronVarManager::kTPCsignal);
 254   histos->UserHistogram("V0Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 255                         400,0.2,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal);
 256   histos->UserHistogram("V0Track","dEdx_Eta","dEdx;#eta;TPC signal (arb units);#tracks",
 257                         110,-1.1,1.1,200,0.,200.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCsignal);
 258   histos->UserHistogram("V0Track","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 259                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle);
 260   histos->UserHistogram("V0Track","TPCnSigmaEle_Eta","TPC number of sigmas Electrons;#eta;TPC number of sigmas Electrons;#tracks",
 261                         110,-1.1,1.1,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaEle);
 262   histos->UserHistogram("V0Track","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 263                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio);
 264   histos->UserHistogram("V0Track","TPCnSigmaPio_Eta","TPC number of sigmas Pions;#eta;TPC number of sigmas Pions;#tracks",
 265                         110,-1.1,1.1,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaPio);
 266   histos->UserHistogram("V0Track","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 267                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao);
 268   histos->UserHistogram("V0Track","TPCnSigmaKao_Eta","TPC number of sigmas Kaons;#eta;TPC number of sigmas Kaons;#tracks",
 269                         110,-1.1,1.1,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaKao);
 270   histos->UserHistogram("V0Track","TPCnSigmaPro_P","TPC number of sigmas Protons;P [GeV];TPC number of sigmas Protons;#tracks",
 271                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPro);
 272   histos->UserHistogram("V0Track","TPCnSigmaPro_Eta","TPC number of sigmas Protons;#eta;TPC number of sigmas Protons;#tracks",
 273                         110,-1.1,1.1,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaPro);
 274   histos->UserHistogram("V0Track","POut","Track POut;P (GeV/c)",100,0.0,10.0, AliDielectronVarManager::kPOut);
 275
 276   // TOF
 277   histos->UserHistogram("V0Track","Pin_TOFsignal","TOF signal vs TPC inner param P; P [GeV]; TOF signal",
 278                         120,0.0,6.,1200,0.,120000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFsignal);
 279   histos->UserHistogram("V0Track","Pin_TOFbeta","TOF #beta vs TPC inner param P; P [GeV]; TOF #beta",
 280                         120,0.0,6.,120,0.,1.2,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFbeta);
 281   histos->UserHistogram("V0Track","TOFnSigmaPio_Pin","TOF number of sigmas Pions;P [GeV];TOF number of sigmas Pions;#tracks",
 282                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPio);
 283   histos->UserHistogram("V0Track","TOFnSigmaPio_eta","TOF number of sigmas Pions vs #eta;#eta;TOF number of sigmas Pions;#tracks",
 284                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaPio);
 285   histos->UserHistogram("V0Track","TOFnSigmaKao_Pin","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 286                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao);
 287   histos->UserHistogram("V0Track","TOFnSigmaKao_eta","TOF number of sigmas Kaons vs #eta;#eta;TOF number of sigmas Kaons;#tracks",
 288                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaKao);
 289   histos->UserHistogram("V0Track","TOFnSigmaPro_Pin","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 290                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro);
 291   histos->UserHistogram("V0Track","TOFnSigmaPro_eta","TOF number of sigmas Protons vs #eta;#eta;TOF number of sigmas Protons;#tracks",
 292                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaPro);
 293   return histos;
 294 }
 295
 296
 297 //____________________________________________________________________________________________
 298 AliDielectron* ConfigDielectron(Int_t cutDefinition)
 299 {
 300   //
 301   // Setup the instance of an AliDielectron
 302   //
 303   const Char_t* dieleNames[2] = {"JpsiToEE", "PhiToKK"};
 304
 305   AliDielectron *die =
 306     new AliDielectron(dieleNames[cutDefinition],
 307                       Form("Track cuts: %s", dieleNames));
 308   die->SetDontClearArrays(kTRUE);
 309   //die->SetEstimatorFilename("$TRAIN_ROOT/util/dielectron/dielectron/estimators.root");
 310   if(cutDefinition==0) {       // J/psi -> e+e-
 311     die->SetLegPdg(11,11);
 312     die->SetMotherPdg(443);
 313   }
 314   if(cutDefinition==1) {       // phi -> K+K-
 315     die->SetLegPdg(321,321);
 316     die->SetMotherPdg(333);
 317   }
 318
 319   // cut setup
 320   SetupDielectronTrackCuts(die,cutDefinition);
 321   SetupDielectronPairCuts(die,cutDefinition);
 322
 323   // Set MC signals
 324   //SetDielectronMCSignals(die);
 325
 326   //
 327   // histogram setup
 328   // only if an AliDielectronHistos object is attached to the
 329   // dielectron framework histograms will be filled
 330   //
 331   InitDielectronHistograms(die,cutDefinition);
 332
 333   //setup eta correction
 334   //SetEtaCorrection();
 335
 336   return die;
 337 }
 338
 339 //______________________________________________________________________________________
 340 void SetupDielectronTrackCuts(AliDielectron *die, Int_t cutDefinition)
 341 {
 342   //
 343   // Setup the track cuts
 344   //
 345
 346   AliDielectronCutGroup* cuts = new AliDielectronCutGroup("cuts","cuts",AliDielectronCutGroup::kCompAND);
 347   die->GetTrackFilter().AddCuts(cuts);
 348   //ESD quality cuts
 349   AliESDtrackCuts *esdTrackCuts = new AliESDtrackCuts;
 350   // basic track quality cuts  (basicQ)
 351   esdTrackCuts->SetMaxDCAToVertexZ(3.0);
 352   esdTrackCuts->SetMaxDCAToVertexXY(1.0);
 353   if(cutDefinition==1) {
 354     esdTrackCuts->SetMaxDCAToVertexZ(0.3);
 355     esdTrackCuts->SetMaxDCAToVertexXY(0.3);
 356   }
 357   if(cutDefinition==1) esdTrackCuts->SetEtaRange(-1.0,1.0);
 358   if(cutDefinition==0) esdTrackCuts->SetEtaRange(-0.9,0.9);
 359   esdTrackCuts->SetAcceptKinkDaughters(kFALSE);
 360   //esdTrackCuts->SetRequireITSRefit(kTRUE);
 361   if(cutDefinition==0) esdTrackCuts->SetRequireTPCRefit(kTRUE);
 362   if(cutDefinition==0) esdTrackCuts->SetPRange(.9,1e30);   // for J/psi
 363   if(cutDefinition==1) esdTrackCuts->SetPRange(.1,1e30);   // for phi
 364   //if(cutDefinition==0) esdTrackCuts->SetMinNClustersTPC(60);
 365   //if(cutDefinition==0) esdTrackCuts->SetMaxChi2PerClusterTPC(4);
 366   //esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kAny);
 367   cuts->AddCut(esdTrackCuts);
 368
 369   Bool_t hasMC=(AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0);
 370
 371   AliDielectronPID *pid = new AliDielectronPID("PID","PID cut");
 372   if(cutDefinition==0) {   // J/psi->ee, phi->ee
 373     pid->AddCut(AliDielectronPID::kTPC,AliPID::kElectron,-4.5, 4.5, 0.0, 0.0,  kFALSE, AliDielectronPID::kRequire); // TPC 3-sigma inclusion for electron
 374     pid->AddCut(AliDielectronPID::kTPC,AliPID::kProton,  -100.0, 2.0, 0.0, 10.0, kTRUE,  AliDielectronPID::kRequire); // TPC 3.0-sigma exclusion for proton
 375 //    pid->AddCut(AliDielectronPID::kTPC,AliPID::kKaon,    -3.0, 3.0, 0.0, 10.0, kTRUE,  AliDielectronPID::kRequire); // TPC 3.0-sigma exclusion for kaon
 376 //    pid->AddCut(AliDielectronPID::kTPC,AliPID::kPion,    -3.0, 3.0, 0.0, 10.0, kTRUE,  AliDielectronPID::kRequire); // TPC 3.0-sigma exclusion for pion
 377   }
 378   if(cutDefinition==1) {     // phi -> KK
 379     pid->AddCut(AliDielectronPID::kTOF,AliPID::kKaon,   -3.5, 3.5, 0.6, 100.0, kFALSE, AliDielectronPID::kRequire, AliDielectronVarManager::kPIn);  // TOF 3-sigma inclusion for kaon
 380     pid->AddCut(AliDielectronPID::kTPC,AliPID::kKaon,   -3.0, 3.5, 0.3, 100.0, kFALSE, AliDielectronPID::kRequire, AliDielectronVarManager::kPIn);  // TPC 3-sigma inclusion for kaon
 381     /*    pid->AddCut(AliDielectronPID::kTPC,AliPID::kProton, -2.0, 2.0, 0.0, 0.0, kTRUE,  AliDielectronPID::kRequire);  // TPC 2.0-sigma exclusion for  proton
 382     pid->AddCut(AliDielectronPID::kTPC,AliPID::kPion,   -2.0, 2.0, 0.0, 0.0, kTRUE,  AliDielectronPID::kRequire);  // TPC 2.0-sigma exclusion for  pion
 383     pid->AddCut(AliDielectronPID::kTOF,AliPID::kProton, -2.0, 2.0, 0.6, 3.0, kTRUE,  AliDielectronPID::kIfAvailable);  // TOF 2.0-sigma exclusion for  proton
 384     pid->AddCut(AliDielectronPID::kTOF,AliPID::kPion,   -2.0, 2.0, 0.6, 3.0, kTRUE,  AliDielectronPID::kIfAvailable);  // TOF 2.0-sigma exclusion for  pion
 385 */
 386   }
 387
 388   // ++++++++++++++++++++++++++++++++++++++++
 389   // shifts for the nSigma electrons
 390   TGraph* nSigmaCorrection = new TGraph();
 391   // b period (LHC10f7a)
 392   nSigmaCorrection->SetPoint(0, 114000., -0.23-(0.0));
 393   nSigmaCorrection->SetPoint(1, 117222., -0.23-(0.0));
 394   // c period (LHC10f7a)
 395   nSigmaCorrection->SetPoint(2, 119000., -0.23-(0.0));
 396   nSigmaCorrection->SetPoint(3, 120829., -0.23-(0.0));
 397   // d period (LHC10f7a)
 398   nSigmaCorrection->SetPoint(4, 121000., -0.353-(-0.111));
 399   nSigmaCorrection->SetPoint(5, 127000., -0.353-(-0.111));
 400   // e period (LHC10f7a)
 401   nSigmaCorrection->SetPoint(6, 127500., -0.351-(-0.116));
 402   nSigmaCorrection->SetPoint(7, 130900., -0.351-(-0.116));
 403   // LHC11a period (LHC11b10b)
 404   nSigmaCorrection->SetPoint(8, 146680., -0.70-(+0.381));   //
 405   nSigmaCorrection->SetPoint(9, 146870., -0.70-(+0.381));   //
 406   if(hasMC)
 407     pid->SetCorrGraph(nSigmaCorrection);
 408
 409   cuts->AddCut(pid);
 410 }
 411
 412 //______________________________________________________________________________________
 413 void SetupDielectronPairCuts(AliDielectron *die, Int_t cutDefinition)
 414 {
 415   //
 416   // Setup the pair cuts
 417   //
 418
 419   //Invariant mass selection
 420   AliDielectronVarCuts *cuts=new AliDielectronVarCuts("|y|<0.9","|Y|<.9");
 421   if(cutDefinition==0) cuts->AddCut(AliDielectronVarManager::kY,-0.9,0.9);
 422   if(cutDefinition==1) cuts->AddCut(AliDielectronVarManager::kY,-1.0,1.0);
 423   if(cutDefinition==1) cuts->AddCut(AliDielectronVarManager::kM, 0.95,1.1);
 424   die->GetPairFilter().AddCuts(cuts);
 425
 426   // Configure prefilter to remove conversions
 427   AliDielectronVarCuts *gammaCut=new AliDielectronVarCuts("gammaCut","gammaCut");
 428   gammaCut->AddCut(AliDielectronVarManager::kM,0.,.05);
 429   if(cutDefinition==0) {
 430     die->GetPairPreFilter().AddCuts(gammaCut);
 431     die->SetPreFilterUnlikeOnly();
 432   }
 433 }
 434
 435 //______________________________________________________________________________________
 436 void InitDielectronHistograms(AliDielectron *die, Int_t cutDefinition)
 437 {
 438   //
 439   // Initialise the histograms
 440   //
 441
 442   //Setup histogram Manager
 443   AliDielectronHistos *histos=new AliDielectronHistos(die->GetName(),die->GetTitle());
 444
 445   //Initialise histogram classes
 446   histos->SetReservedWords("Track;Pair");
 447
 448   //Track classes
 449   //to fill also track info from 2nd event loop until 2
 450   for (Int_t i=0; i<2; ++i){
 451     histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
 452   }
 453
 454   //Pair classes
 455   // to fill also mixed event histograms loop until 10
 456   for (Int_t i=0; i<3; ++i){
 457     histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 458   }
 459
 460   //legs from pair
 461   for (Int_t i=0; i<3; ++i){
 462     histos->AddClass(Form("Track_Legs_%s",AliDielectron::PairClassName(i)));
 463   }
 464
 465   //add histograms to event class
 466   if (cutDefinition==0) {
 467     histos->AddClass("Event");
 468     histos->UserHistogram("Event","VtxZ","Vertex Z;Z[cm]",500,-25.,25.,AliDielectronVarManager::kZvPrim);
 469     histos->UserHistogram("Event","NTrk","NTrk",500,-0.5,499.5,AliDielectronVarManager::kNTrk);
 470     histos->UserHistogram("Event","VtxZ_NaccTrcklts","VtxZ_NaccTrcklts;VtxZ;NaccTrcklts",
 471                           240,-12.,12.,200,-0.5,199.5,AliDielectronVarManager::kZvPrim, AliDielectronVarManager::kNaccTrcklts);
 472
 473     histos->UserHistogram("Event","MultV0A","VZERO mult",2000,0.,2000.,AliDielectronVarManager::kMultV0A);
 474     histos->UserHistogram("Event","MultV0C","VZERO mult",2000,0.,2000.,AliDielectronVarManager::kMultV0C);
 475     histos->UserHistogram("Event","MultV0","VZERO mult",2000,0.,2000.,AliDielectronVarManager::kMultV0);
 476     histos->UserHistogram("Event","AdcV0A","VZERO adc",10000,0.,100000.,AliDielectronVarManager::kAdcV0A);
 477     histos->UserHistogram("Event","AdcV0C","VZERO adc",10000,0.,100000.,AliDielectronVarManager::kAdcV0C);
 478     histos->UserHistogram("Event","AdcV0","VZERO adc",10000,0.,100000.,AliDielectronVarManager::kAdcV0);
 479     histos->UserHistogram("Event","MultV0A_MultV0C","MultV0A_MultV0C;MultV0A;MultV0C",
 480                           2000,0.0,2000.,2000,0.0,2000,AliDielectronVarManager::kMultV0A, AliDielectronVarManager::kMultV0C);
 481     histos->UserHistogram("Event","AdcV0A_AdcV0C","AdcV0A_AdcV0C;AdcV0A;AdcV0C",
 482                           1000,0.0,20000.,1000,0.0,20000,AliDielectronVarManager::kAdcV0A, AliDielectronVarManager::kAdcV0C);
 483     histos->UserHistogram("Event","MultV0A_AdcV0A","MultV0A_AdcV0A;MultV0A;AdcV0A",
 484                           2000,0.0,2000.,2000,0.0,20000,AliDielectronVarManager::kMultV0A, AliDielectronVarManager::kAdcV0A);
 485     histos->UserHistogram("Event","MultV0C_AdcV0C","MultV0C_AdcV0C;MultV0C;AdcV0C",
 486                           2000,0.0,2000.,2000,0.0,20000,AliDielectronVarManager::kMultV0C, AliDielectronVarManager::kAdcV0C);
 487   }
 488
 489   //add histograms to Track classes ---------------------------------------------------------------------------
 490   // kinematic acceptance histograms
 491   histos->UserHistogram("Track","Pt","Pt;Pt (GeV/c);#tracks",400,0,20.,AliDielectronVarManager::kPt);
 492   histos->UserHistogram("Track","Eta_P","Eta_P;#eta;P (GeV/c)",40, -1.0, +1.0, 200,0,10., AliDielectronVarManager::kEta, AliDielectronVarManager::kP);
 493   histos->UserHistogram("Track","Eta_Phi","Eta Phi Map; Eta; Phi;#tracks",
 494                         40,-1,1,200,0,6.285,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
 495
 496   // DCA diagnostic histograms
 497   histos->UserHistogram("Track","dXY","dXY;dXY (cm);#tracks",500,-1.,1.,AliDielectronVarManager::kImpactParXY);
 498   histos->UserHistogram("Track","dZ","dZ;dZ (cm);#tracks",600,-3.,3.,AliDielectronVarManager::kImpactParZ);
 499   histos->UserHistogram("Track","dZ_dXY","dZ_dXY;dZ (cm);dXY (cm)",600,-3.,3.,500,-1.,1.,AliDielectronVarManager::kImpactParZ, AliDielectronVarManager::kImpactParXY);
 500   histos->UserHistogram("Track","P_dXY","P_dXY;P (GeV/c);dXY (cm)",200,0.,10.,500,-1.,1.,AliDielectronVarManager::kP, AliDielectronVarManager::kImpactParXY);
 501   histos->UserHistogram("Track","P_dZ","P_dZ;P (GeV/c);dZ (cm)",200,0.,10.,600,-3.,3.,AliDielectronVarManager::kP, AliDielectronVarManager::kImpactParZ);
 502
 503     // ITS
 504   histos->UserHistogram("Track","ITSsignal","ITS signal;ITS dE/dx",1000,0.0,1000.0, AliDielectronVarManager::kITSsignal);
 505   histos->UserHistogram("Track","ITSsignal_P","ITS signal vs P;P (GeV/c); ITS dE/dx",300, 0.0, 3.0, 400,0.0,1000.0, AliDielectronVarManager::kP, AliDielectronVarManager::kITSsignal);
 506
 507   // TPC
 508   histos->UserHistogram("Track","TPCchi2","#chi^{2}/cluster;#chi^{2}/cluster;#tracks",200,0.0,5.0, AliDielectronVarManager::kTPCchi2Cl);
 509   histos->UserHistogram("Track","Pt_TPCchi2","#chi^{2}/cluster vs pt;p_{T} (GeV/c);#chi^{2}/cluster",400,0.,20., 200,0.0,5.0,
 510                         AliDielectronVarManager::kPt,AliDielectronVarManager::kTPCchi2Cl);
 511   histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusters;#tracks",160,-0.5,159.5, AliDielectronVarManager::kNclsTPC);
 512   histos->UserHistogram("Track","Phi_TPCnCls","Number of Clusters TPC vs #phi;#phi (rad.);TPC number clusters",200,0,6.285, 160,-0.5,159.5,
 513                         AliDielectronVarManager::kPhi,AliDielectronVarManager::kNclsTPC);
 514   histos->UserHistogram("Track","Pin_TPCnCls","Number of Clusters TPC vs TPC inner param P;p (GeV/c);TPC number clusters",400,0,20., 160,-0.5,159.5,
 515                         AliDielectronVarManager::kPIn, AliDielectronVarManager::kNclsTPC);
 516   histos->UserHistogram("Track","TPCnCls_dEdx","Number of Clusters TPC vs dE/dx; TPC number clusters; dE/dx (arb.units)", 160,-0.5,159.5, 200,0,200.,
 517                         AliDielectronVarManager::kNclsTPC, AliDielectronVarManager::kTPCsignal);
 518   histos->UserHistogram("Track","dEdx_P","dEdx;P [GeV];TPC signal (arb units);#tracks",
 519                         400,0.2,20.,200,0.,200.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCsignal);
 520   histos->UserHistogram("Track","TPCnSigmaEle_P","TPC number of sigmas Electrons;P [GeV];TPC number of sigmas Electrons;#tracks",
 521                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaEle);
 522   histos->UserHistogram("Track","TPCnSigmaPio_P","TPC number of sigmas Pions;P [GeV];TPC number of sigmas Pions;#tracks",
 523                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPio);
 524   histos->UserHistogram("Track","TPCnSigmaKao_P","TPC number of sigmas Kaons;P [GeV];TPC number of sigmas Kaons;#tracks",
 525                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaKao);
 526   histos->UserHistogram("Track","TPCnSigmaPro_P","TPC number of sigmas Protons;P [GeV];TPC number of sigmas Protons;#tracks",
 527                         400,0.2,20.,200,-5.,5.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTPCnSigmaPro);
 528   histos->UserHistogram("Track","dEdx_Eta","dEdx;#eta;TPC signal (arb units);#tracks",
 529                         100,-1.0,1.0,200,0.,200.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCsignal);
 530   histos->UserHistogram("Track","TPCnSigmaEle_Eta","TPC number of sigmas Electrons;#eta;TPC number of sigmas Electrons;#tracks",
 531                         100,-1.0,1.0,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaEle);
 532   histos->UserHistogram("Track","TPCnSigmaPio_Eta","TPC number of sigmas Pions;#eta;TPC number of sigmas Pions;#tracks",
 533                         100,-1.0,1.0,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaPio);
 534   histos->UserHistogram("Track","TPCnSigmaKao_Eta","TPC number of sigmas Kaons;#eta;TPC number of sigmas Kaons;#tracks",
 535                         100,-1.0,1.0,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaKao);
 536   histos->UserHistogram("Track","TPCnSigmaPro_Eta","TPC number of sigmas Protons;#eta;TPC number of sigmas Protons;#tracks",
 537                         100,-1.0,1.0,200,-5.,5.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTPCnSigmaPro);
 538
 539   // TOF
 540   histos->UserHistogram("Track","Pin_TOFsignal","TOF signal vs TPC inner param P; P [GeV]; TOF signal",
 541                         120,0.0,6.,1200,0.,120000.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFsignal);
 542   histos->UserHistogram("Track","Pin_TOFbeta","TOF #beta vs TPC inner param P; P [GeV]; TOF #beta",
 543                         120,0.0,6.,120,0.,1.2,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFbeta);
 544   histos->UserHistogram("Track","TOFnSigmaPio_Pin","TOF number of sigmas Pions;P [GeV];TOF number of sigmas Pions;#tracks",
 545                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPio);
 546   histos->UserHistogram("Track","TOFnSigmaPio_eta","TOF number of sigmas Pions vs #eta;#eta;TOF number of sigmas Pions;#tracks",
 547                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaPio);
 548   histos->UserHistogram("Track","TOFnSigmaKao_Pin","TOF number of sigmas Kaons;P [GeV];TOF number of sigmas Kaons;#tracks",
 549                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaKao);
 550   histos->UserHistogram("Track","TOFnSigmaKao_eta","TOF number of sigmas Kaons vs #eta;#eta;TOF number of sigmas Kaons;#tracks",
 551                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaKao);
 552   histos->UserHistogram("Track","TOFnSigmaPro_Pin","TOF number of sigmas Protons;P [GeV];TOF number of sigmas Protons;#tracks",
 553                         400,0.2,20.,200,-10.,10.,AliDielectronVarManager::kPIn,AliDielectronVarManager::kTOFnSigmaPro);
 554   histos->UserHistogram("Track","TOFnSigmaPro_eta","TOF number of sigmas Protons vs #eta;#eta;TOF number of sigmas Protons;#tracks",
 555                         100,-1.2,1.2,200,-10.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kTOFnSigmaPro);
 556
 557   //add histograms to Pair classes
 558   Int_t nMassBins = 10000;
 559   Double_t massRange[2] = {-0.005, 100.005};
 560   if(cutDefinition==1) {   // Phi -> K+K-
 561     nMassBins = 4000;
 562     massRange[0] = 0.8; massRange[1] = 1.2;
 563   }
 564   histos->UserHistogram("Pair","InvMass","Inv.Mass;Inv. Mass [GeV];#pairs",
 565                         nMassBins, massRange[0], massRange[1], AliDielectronVarManager::kM);
 566   histos->UserHistogram("Pair","InvMass_Pt","Inv.Mass vs pt;Inv. Mass [GeV]; pt (GeV/c);#pairs",
 567                         nMassBins/10, massRange[0], massRange[1], 100, 0.0, 50.0, AliDielectronVarManager::kM, AliDielectronVarManager::kPt);
 568   histos->UserHistogram("Pair","InvMass_Cent","Inv.Mass vs centrality;Inv. Mass [GeV]; centrality;#pairs",
 569                         nMassBins/10, massRange[0], massRange[1], 20, 0.0, 100.0, AliDielectronVarManager::kM, AliDielectronVarManager::kCentrality);
 570   histos->UserHistogram("Pair","InvMass_Rap","Inv.Mass vs rapidity;Inv. Mass [GeV]; rapidity;#pairs",
 571                         nMassBins/10, massRange[0], massRange[1], 100, -1.0, 1.0, AliDielectronVarManager::kM, AliDielectronVarManager::kY);
 572   histos->UserHistogram("Pair","InvMass_OpAngle","Inv.Mass vs opening angle;Inv. Mass [GeV]; op. angle (rad.);#pairs",
 573                         nMassBins/10, massRange[0], massRange[1], 100, 0.0, 4.0, AliDielectronVarManager::kM, AliDielectronVarManager::kOpeningAngle);
 574   histos->UserHistogram("Pair","InvMass_ThetaHE","Inv.Mass vs cos #theta^{*}_{HE};Inv. Mass [GeV]; cos #theta^{*}_{HE};#pairs",
 575                         nMassBins/10, massRange[0], massRange[1], 100, -1.0, 1.0, AliDielectronVarManager::kM, AliDielectronVarManager::kThetaHE);
 576   histos->UserHistogram("Pair","InvMass_ThetaCS","Inv.Mass vs cos #theta^{*}_{CS};Inv. Mass [GeV]; cos #theta^{*}_{CS};#pairs",
 577                         nMassBins/10, massRange[0], massRange[1], 100, -1.0, 1.0, AliDielectronVarManager::kM, AliDielectronVarManager::kThetaCS);
 578   histos->UserHistogram("Pair","Pt","Transverse momentum; P_{t} [GeV/c];#pairs",
 579                         1000, 0.0, 50.0, AliDielectronVarManager::kPt);
 580   histos->UserHistogram("Pair","Pt_Y","Transverse momentum vs rapidity; rapidity;  P_{t} [GeV/c];#pairs",
 581                         100, -1.0, 1.0, 200, 0.0, 50.0, AliDielectronVarManager::kY, AliDielectronVarManager::kPt);
 582   histos->UserHistogram("Pair","Pt_ThetaCS","Transverse momentum vs cos #theta^{*}_{CS}; cos #theta^{*}_{CS};  P_{t} [GeV/c];#pairs",
 583                         100, -1.0, 1.0, 200, 0.0, 50.0, AliDielectronVarManager::kThetaCS, AliDielectronVarManager::kPt);
 584   histos->UserHistogram("Pair","Pt_ThetaHE","Transverse momentum vs cos #theta^{*}_{HE}; cos #theta^{*}_{HE};  P_{t} [GeV/c];#pairs",
 585                         100, -1.0, 1.0, 200, 0.0, 50.0, AliDielectronVarManager::kThetaHE, AliDielectronVarManager::kPt);
 586   histos->UserHistogram("Pair","Pt_Phi","Transverse momentum vs phi; #phi (rad.);  P_{t} [GeV/c];#pairs",
 587                         200, -6.3, 6.3, 200, 0.0, 50.0, AliDielectronVarManager::kPhi, AliDielectronVarManager::kPt);
 588   histos->UserHistogram("Pair","Phi","#phi; #phi (rad.);#pairs",
 589                         1000, -6.3, 6.3, AliDielectronVarManager::kPhi);
 590   histos->UserHistogram("Pair","Rapidity","Rapidity;Rapidity;#pairs",
 591                         100,-1.,1.,AliDielectronVarManager::kY);
 592   histos->UserHistogram("Pair","Rapidity_OpAngle","Rapidity vs opening angle;Rapidity; Op.Angle (rad.);#pairs",
 593                         100,-1.,1.,200, 0.0, 4.0, AliDielectronVarManager::kY, AliDielectronVarManager::kOpeningAngle);
 594   histos->UserHistogram("Pair","Rapidity_ThetaCS","Rapidity vs cos #theta^{*}_{CS};Rapidity; cos #theta^{*}_{CS};#pairs",
 595                         100,-1.,1.,200, -1.0, 1.0, AliDielectronVarManager::kY, AliDielectronVarManager::kThetaCS);
 596   histos->UserHistogram("Pair","Rapidity_ThetaHE","Rapidity vs cos #theta^{*}_{HE};Rapidity; cos #theta^{*}_{HE};#pairs",
 597                         100,-1.,1.,200, -1.0, 1.0, AliDielectronVarManager::kY, AliDielectronVarManager::kThetaHE);
 598   histos->UserHistogram("Pair","OpeningAngle","Opening angle;angle",
 599                         100,0.,3.15,AliDielectronVarManager::kOpeningAngle);
 600   histos->UserHistogram("Pair","Radius","Radius;R[cm]",
 601                         1000,0.,300.,AliDielectronVarManager::kR);
 602   histos->UserHistogram("Pair","ThetaHE","cos #theta^{*}_{HE}; cos #theta^{*}_{HE}",
 603                         220,-1.1,1.1,AliDielectronVarManager::kThetaHE);
 604   histos->UserHistogram("Pair","PhiHE","#varphi^{*}_{HE};#varphi^{*}_{HE} (rad.)",
 605                         160,-3.2,3.2,AliDielectronVarManager::kPhiHE);
 606   histos->UserHistogram("Pair","ThetaCS","cos #theta^{*}_{CS}; cos #theta^{*}_{CS}",
 607                         220,-1.1,1.1,AliDielectronVarManager::kThetaCS);
 608   histos->UserHistogram("Pair","PhiCS","#varphi^{*}_{CS};#varphi^{*}_{CS} (rad.)",
 609                         160,-3.2,3.2,AliDielectronVarManager::kPhiCS);
 610   histos->UserHistogram("Pair","Lxy","Pseudo-proper time;Lxy (cm.)",
 611                         1000,0.0,2.0,AliDielectronVarManager::kPseudoProperTime);
 612   histos->UserHistogram("Pair","Chi2NDF","Pair #chi^{2}/NDF; #chi^{2}/NDF",
 613                         1000,0.0,10.0,AliDielectronVarManager::kChi2NDF);
 614
 615
 616   die->SetHistogramManager(histos);
 617 }
 618
 619
 620 //_____________________________________________________________________________________________
 621 void SetDielectronMCSignals(AliDielectron *die)
 622 {
 623   // J/psi sources (direct + feed down from charm higher states)
 624   // 0
 625   AliDielectronSignalMC* promptJpsi = new AliDielectronSignalMC("promptJpsi","Prompt J/psi");   // prompt J/psi (not from beauty decays)
 626   promptJpsi->SetLegPDGs(11,-11);
 627   promptJpsi->SetMotherPDGs(443,443);
 628   promptJpsi->SetGrandMotherPDGs(503,503,kTRUE,kTRUE);   // not from beauty hadrons
 629   promptJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 630   promptJpsi->SetFillPureMCStep(kTRUE);
 631   promptJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 632   promptJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 633   promptJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 634   promptJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 635   die->AddSignalMC(promptJpsi);
 636
 637   // 1
 638   AliDielectronSignalMC* beautyJpsi = new AliDielectronSignalMC("beautyJpsi","beauty hadron -> J/psi");  // J/psi->e+e- from beauty hadron decays
 639   beautyJpsi->SetLegPDGs(11,-11);
 640   beautyJpsi->SetMotherPDGs(443,443);
 641   beautyJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 642   beautyJpsi->SetGrandMotherPDGs(503,503);
 643   beautyJpsi->SetFillPureMCStep(kTRUE);
 644   beautyJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 645   beautyJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 646   beautyJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 647   beautyJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 648   die->AddSignalMC(beautyJpsi);
 649
 650   // 2
 651   AliDielectronSignalMC* beautyMesonJpsi = new AliDielectronSignalMC("beautyMesonJpsi","beauty meson -> J/psi");  // J/psi->e+e- from beauty meson decays
 652   beautyMesonJpsi->SetLegPDGs(11,-11);
 653   beautyMesonJpsi->SetMotherPDGs(443,443);
 654   beautyMesonJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 655   beautyMesonJpsi->SetGrandMotherPDGs(500,500);
 656   beautyMesonJpsi->SetFillPureMCStep(kTRUE);
 657   beautyMesonJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 658   beautyMesonJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 659   beautyMesonJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 660   beautyMesonJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 661   die->AddSignalMC(beautyMesonJpsi);
 662
 663   // 3
 664   AliDielectronSignalMC* openBeautyMesonJpsi = new AliDielectronSignalMC("openBeautyMesonJpsi","open beauty meson -> J/psi");  // J/psi->e+e- from open beauty meson decays
 665   openBeautyMesonJpsi->SetLegPDGs(11,-11);
 666   openBeautyMesonJpsi->SetMotherPDGs(443,443);
 667   openBeautyMesonJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 668   openBeautyMesonJpsi->SetGrandMotherPDGs(501,501);
 669   openBeautyMesonJpsi->SetFillPureMCStep(kTRUE);
 670   openBeautyMesonJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 671   openBeautyMesonJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 672   openBeautyMesonJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 673   openBeautyMesonJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 674   die->AddSignalMC(openBeautyMesonJpsi);
 675
 676   // 4
 677   AliDielectronSignalMC* chic0Jpsi = new AliDielectronSignalMC("chic0Jpsi","chic0 -> J/psi + X");  // J/psi->e+e- from chic0 decays
 678   chic0Jpsi->SetLegPDGs(11,-11);
 679   chic0Jpsi->SetMotherPDGs(443,443);
 680   chic0Jpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 681   chic0Jpsi->SetGrandMotherPDGs(10441,10441);
 682   chic0Jpsi->SetFillPureMCStep(kTRUE);
 683   chic0Jpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 684   chic0Jpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 685   chic0Jpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 686   chic0Jpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 687   die->AddSignalMC(chic0Jpsi);
 688
 689   // 5
 690   AliDielectronSignalMC* chic1Jpsi = new AliDielectronSignalMC("chic1Jpsi","chic1 -> J/psi + X");  // J/psi->e+e- from chic1 decays
 691   chic1Jpsi->SetLegPDGs(11,-11);
 692   chic1Jpsi->SetMotherPDGs(443,443);
 693   chic1Jpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 694   chic1Jpsi->SetGrandMotherPDGs(20443,20443);
 695   chic1Jpsi->SetFillPureMCStep(kTRUE);
 696   chic1Jpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 697   chic1Jpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 698   chic1Jpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 699   chic1Jpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 700   die->AddSignalMC(chic1Jpsi);
 701
 702   // 6
 703   AliDielectronSignalMC* chic2Jpsi = new AliDielectronSignalMC("chic2Jpsi","chic2 -> J/psi + X");  // J/psi->e+e- from chic2 decays
 704   chic2Jpsi->SetLegPDGs(11,-11);
 705   chic2Jpsi->SetMotherPDGs(443,443);
 706   chic2Jpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 707   chic2Jpsi->SetGrandMotherPDGs(445,445);
 708   chic2Jpsi->SetFillPureMCStep(kTRUE);
 709   chic2Jpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 710   chic2Jpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 711   chic2Jpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 712   chic2Jpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 713   die->AddSignalMC(chic2Jpsi);
 714
 715   // 7
 716   AliDielectronSignalMC* psiPrimeJpsi = new AliDielectronSignalMC("psiPrimeJpsi","psi(2S) -> J/psi + X");  // J/psi->e+e- from psi(2S) decays
 717   psiPrimeJpsi->SetLegPDGs(11,-11);
 718   psiPrimeJpsi->SetMotherPDGs(443,443);
 719   psiPrimeJpsi->SetMothersRelation(AliDielectronSignalMC::kSame);
 720   psiPrimeJpsi->SetGrandMotherPDGs(445,445);
 721   psiPrimeJpsi->SetFillPureMCStep(kTRUE);
 722   psiPrimeJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 723   psiPrimeJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 724   psiPrimeJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 725   psiPrimeJpsi->SetCheckBothChargesGrandMothers(kTRUE,kTRUE);
 726   die->AddSignalMC(psiPrimeJpsi);
 727
 728   // physical backgrounds (electrons pairs from other sources and their combinatorics)
 729   // 8
 730   AliDielectronSignalMC* diEleContinuum = new AliDielectronSignalMC("diEleContinuum","di-electron continuum");     // all di-electrons originating in the collision
 731   diEleContinuum->SetLegPDGs(11,-11);
 732   diEleContinuum->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 733   diEleContinuum->SetCheckBothChargesLegs(kTRUE,kTRUE);
 734   diEleContinuum->SetFillPureMCStep(kTRUE);
 735   die->AddSignalMC(diEleContinuum);
 736
 737   // 9
 738   AliDielectronSignalMC* diEleCharm = new AliDielectronSignalMC("diEleCharm","di-electrons from charm");  // dielectrons originating from charm hadrons (not neccessary from same mother)
 739   diEleCharm->SetLegPDGs(11,-11);
 740   diEleCharm->SetMotherPDGs(403,403);
 741   diEleCharm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 742   diEleCharm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 743   diEleCharm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 744   diEleCharm->SetFillPureMCStep(kTRUE);
 745   die->AddSignalMC(diEleCharm);
 746
 747   // 10
 748   AliDielectronSignalMC* diEleOpenCharm = new AliDielectronSignalMC("diEleOpenCharm","di-electrons from open charm");  // dielectrons originating from open charm hadrons
 749   diEleOpenCharm->SetLegPDGs(11,-11);
 750   diEleOpenCharm->SetMotherPDGs(402,402);
 751   diEleOpenCharm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 752   diEleOpenCharm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 753   diEleOpenCharm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 754   diEleOpenCharm->SetFillPureMCStep(kTRUE);
 755   die->AddSignalMC(diEleOpenCharm);
 756
 757   // 11
 758   AliDielectronSignalMC* diEleOpenCharmJpsi = new AliDielectronSignalMC("diEleOpenCharmJpsi","1 leg from open charm + 1 jpsi leg");  // 1 leg from open charm + 1 leg from jpsi
 759   diEleOpenCharmJpsi->SetLegPDGs(11,-11);
 760   diEleOpenCharmJpsi->SetMotherPDGs(402,443);
 761   diEleOpenCharmJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 762   diEleOpenCharmJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 763   diEleOpenCharmJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 764   diEleOpenCharmJpsi->SetFillPureMCStep(kTRUE);
 765   die->AddSignalMC(diEleOpenCharmJpsi);
 766
 767   // 12
 768   AliDielectronSignalMC* diEleBeauty = new AliDielectronSignalMC("diEleBeauty","di-electrons from beauty");  // dielectrons originating from beauty hadrons (not neccessary from same mother)
 769   diEleBeauty->SetLegPDGs(11,-11);
 770   diEleBeauty->SetMotherPDGs(503,503);
 771   diEleBeauty->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 772   diEleBeauty->SetCheckBothChargesLegs(kTRUE,kTRUE);
 773   diEleBeauty->SetCheckBothChargesMothers(kTRUE,kTRUE);
 774   diEleBeauty->SetFillPureMCStep(kTRUE);
 775   die->AddSignalMC(diEleBeauty);
 776
 777   // 13
 778   AliDielectronSignalMC* diEleOpenBeauty = new AliDielectronSignalMC("diEleOpenBeauty","di-electrons from open beauty");  // dielectrons originating from open beauty hadrons
 779   diEleOpenBeauty->SetLegPDGs(11,-11);
 780   diEleOpenBeauty->SetMotherPDGs(502,502);
 781   diEleOpenBeauty->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 782   diEleOpenBeauty->SetCheckBothChargesLegs(kTRUE,kTRUE);
 783   diEleOpenBeauty->SetCheckBothChargesMothers(kTRUE,kTRUE);
 784   diEleOpenBeauty->SetFillPureMCStep(kTRUE);
 785   die->AddSignalMC(diEleOpenBeauty);
 786
 787   // 14
 788   AliDielectronSignalMC* diEleBeautyJpsi = new AliDielectronSignalMC("diEleBeautyJpsi","1 leg from beauty + 1 jpsi leg");  // 1 leg from beauty + 1 leg from jpsi
 789   diEleBeautyJpsi->SetLegPDGs(11,-11);
 790   diEleBeautyJpsi->SetMotherPDGs(503,443);
 791   diEleBeautyJpsi->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 792   diEleBeautyJpsi->SetCheckBothChargesLegs(kTRUE,kTRUE);
 793   diEleBeautyJpsi->SetCheckBothChargesMothers(kTRUE,kTRUE);
 794   diEleBeautyJpsi->SetFillPureMCStep(kTRUE);
 795   die->AddSignalMC(diEleBeautyJpsi);
 796
 797   // 15
 798   AliDielectronSignalMC* diEleBeautyOpenCharm = new AliDielectronSignalMC("diEleBeautyOpenCharm","1 leg from beauty + 1 leg from open charm");  // 1 leg from open charm + 1 leg from beauty
 799   diEleBeautyOpenCharm->SetLegPDGs(11,-11);
 800   diEleBeautyOpenCharm->SetMotherPDGs(503,403);
 801   diEleBeautyOpenCharm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 802   diEleBeautyOpenCharm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 803   diEleBeautyOpenCharm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 804   diEleBeautyOpenCharm->SetFillPureMCStep(kTRUE);
 805   die->AddSignalMC(diEleBeautyOpenCharm);
 806
 807   // 16
 808   AliDielectronSignalMC* diEleStrange = new AliDielectronSignalMC("diEleStrange","di-electrons from strange particles");  // dielectrons originating from strange hadrons (not neccessary from same mother)
 809   diEleStrange->SetLegPDGs(11,-11);
 810   diEleStrange->SetMotherPDGs(300,300);
 811   diEleStrange->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 812   diEleStrange->SetCheckBothChargesLegs(kTRUE,kTRUE);
 813   diEleStrange->SetCheckBothChargesMothers(kTRUE,kTRUE);
 814   diEleStrange->SetFillPureMCStep(kTRUE);
 815   die->AddSignalMC(diEleStrange);
 816
 817   // 17
 818   AliDielectronSignalMC* diEleStrangeCharm = new AliDielectronSignalMC("diEleStrangeCharm","1 leg from strange + 1 leg from charm");  // 1 leg from strange + 1 leg from charm
 819   diEleStrangeCharm->SetLegPDGs(11,-11);
 820   diEleStrangeCharm->SetMotherPDGs(300,403);
 821   diEleStrangeCharm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 822   diEleStrangeCharm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 823   diEleStrangeCharm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 824   diEleStrangeCharm->SetFillPureMCStep(kTRUE);
 825   die->AddSignalMC(diEleStrangeCharm);
 826
 827   // 18
 828   AliDielectronSignalMC* diEleStrangeBeauty = new AliDielectronSignalMC("diEleStrangeBeauty","1 leg from strange + 1 leg from beauty");  // 1 leg from strange + 1 leg from beauty
 829   diEleStrangeBeauty->SetLegPDGs(11,-11);
 830   diEleStrangeBeauty->SetMotherPDGs(300,503);
 831   diEleStrangeBeauty->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 832   diEleStrangeBeauty->SetCheckBothChargesLegs(kTRUE,kTRUE);
 833   diEleStrangeBeauty->SetCheckBothChargesMothers(kTRUE,kTRUE);
 834   diEleStrangeBeauty->SetFillPureMCStep(kTRUE);
 835   die->AddSignalMC(diEleStrangeBeauty);
 836
 837   // 19
 838   AliDielectronSignalMC* diEleLight1 = new AliDielectronSignalMC("diEleLight1","1 leg from light particles + 1 leg from everything");  // 1 leg from light hadrons (100+ PYTHIA codes) + 1 leg from everything
 839   diEleLight1->SetLegPDGs(11,-11);
 840   diEleLight1->SetMotherPDGs(100,0);
 841   diEleLight1->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 842   diEleLight1->SetCheckBothChargesLegs(kTRUE,kTRUE);
 843   diEleLight1->SetCheckBothChargesMothers(kTRUE,kTRUE);
 844   diEleLight1->SetFillPureMCStep(kTRUE);
 845   die->AddSignalMC(diEleLight1);
 846
 847   // 20
 848   AliDielectronSignalMC* diEleLight1Charm = new AliDielectronSignalMC("diEleLight1Charm","1 leg from light particles + 1 leg from charm");  // 1 leg from light hadrons (100+ PYTHIA codes) + 1 leg from charm
 849   diEleLight1Charm->SetLegPDGs(11,-11);
 850   diEleLight1Charm->SetMotherPDGs(100,403);
 851   diEleLight1Charm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 852   diEleLight1Charm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 853   diEleLight1Charm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 854   diEleLight1Charm->SetFillPureMCStep(kTRUE);
 855   die->AddSignalMC(diEleLight1Charm);
 856
 857   // 21
 858   AliDielectronSignalMC* diEleLight1Beauty = new AliDielectronSignalMC("diEleLight1Beauty","1 leg from light particles + 1 leg from beauty");  // 1 leg from light hadrons (100+ PYTHIA codes) + 1 leg from beauty
 859   diEleLight1Beauty->SetLegPDGs(11,-11);
 860   diEleLight1Beauty->SetMotherPDGs(100,503);
 861   diEleLight1Beauty->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 862   diEleLight1Beauty->SetCheckBothChargesLegs(kTRUE,kTRUE);
 863   diEleLight1Beauty->SetCheckBothChargesMothers(kTRUE,kTRUE);
 864   diEleLight1Beauty->SetFillPureMCStep(kTRUE);
 865   die->AddSignalMC(diEleLight1Beauty);
 866
 867   // 22
 868   AliDielectronSignalMC* diEleLight2 = new AliDielectronSignalMC("diEleLight2","1 leg from light particles + 1 leg from everything");  // 1 leg from light hadrons (200+ PYTHIA codes) + 1 leg from everything
 869   diEleLight2->SetLegPDGs(11,-11);
 870   diEleLight2->SetMotherPDGs(200,0);
 871   diEleLight2->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 872   diEleLight2->SetCheckBothChargesLegs(kTRUE,kTRUE);
 873   diEleLight2->SetCheckBothChargesMothers(kTRUE,kTRUE);
 874   diEleLight2->SetFillPureMCStep(kTRUE);
 875   die->AddSignalMC(diEleLight2);
 876
 877   // 23
 878   AliDielectronSignalMC* diEleLight2Charm = new AliDielectronSignalMC("diEleLight2Charm","1 leg from light particles + 1 leg from charm");  // 1 leg from light hadrons (200+ PYTHIA codes) + 1 leg from charm
 879   diEleLight2Charm->SetLegPDGs(11,-11);
 880   diEleLight2Charm->SetMotherPDGs(100,403);
 881   diEleLight2Charm->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 882   diEleLight2Charm->SetCheckBothChargesLegs(kTRUE,kTRUE);
 883   diEleLight2Charm->SetCheckBothChargesMothers(kTRUE,kTRUE);
 884   diEleLight2Charm->SetFillPureMCStep(kTRUE);
 885   die->AddSignalMC(diEleLight2Charm);
 886
 887   // 24
 888   AliDielectronSignalMC* diEleLight2Beauty = new AliDielectronSignalMC("diEleLight2Beauty","1 leg from light particles + 1 leg from beauty");  // 1 leg from light hadrons (100+ PYTHIA codes) + 1 leg from beauty
 889   diEleLight2Beauty->SetLegPDGs(11,-11);
 890   diEleLight2Beauty->SetMotherPDGs(100,503);
 891   diEleLight2Beauty->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kFinalState);
 892   diEleLight2Beauty->SetCheckBothChargesLegs(kTRUE,kTRUE);
 893   diEleLight2Beauty->SetCheckBothChargesMothers(kTRUE,kTRUE);
 894   diEleLight2Beauty->SetFillPureMCStep(kTRUE);
 895   die->AddSignalMC(diEleLight2Beauty);
 896
 897   // background from secondary electrons
 898   // 25
 899   AliDielectronSignalMC* secondaryElectrons = new AliDielectronSignalMC("secondaryElectrons","Secondary electrons");   // all di-electrons from secondary electrons (interaction with detector)
 900   secondaryElectrons->SetLegPDGs(11,-11);
 901   secondaryElectrons->SetLegSources(AliDielectronSignalMC::kSecondary, AliDielectronSignalMC::kSecondary);
 902   secondaryElectrons->SetCheckBothChargesLegs(kTRUE,kTRUE);
 903   die->AddSignalMC(secondaryElectrons);
 904
 905   // 26
 906   AliDielectronSignalMC* primarySecElePairs = new AliDielectronSignalMC("primarySecElePairs","Primary+Secondary electron pairs");  // primary-secondary pairs
 907   primarySecElePairs->SetLegPDGs(11,-11);
 908   primarySecElePairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 909   primarySecElePairs->SetLegSources(AliDielectronSignalMC::kFinalState, AliDielectronSignalMC::kSecondary);
 910   die->AddSignalMC(primarySecElePairs);
 911
 912   // 27
 913   AliDielectronSignalMC* conversionElePairs = new AliDielectronSignalMC("conversionElePairs","conversion electron pairs");      // pairs made from conversion (may be also from 2 different conversions)
 914   conversionElePairs->SetLegPDGs(11,-11);
 915   conversionElePairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 916   conversionElePairs->SetLegSources(AliDielectronSignalMC::kSecondary, AliDielectronSignalMC::kSecondary);
 917   conversionElePairs->SetMotherPDGs(22,22);
 918   die->AddSignalMC(conversionElePairs);
 919
 920   // misidentification
 921   // 28
 922   AliDielectronSignalMC* allEleMisIdPairs = new AliDielectronSignalMC("allEleMisIdPairs","all electron+misid. pairs");  // one true electron + a mis-id electron (all sources included)
 923   allEleMisIdPairs->SetLegPDGs(11,11,kFALSE,kTRUE);
 924   allEleMisIdPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 925   die->AddSignalMC(allEleMisIdPairs);
 926
 927   // 29
 928   AliDielectronSignalMC* allMisIdMisIdPairs = new AliDielectronSignalMC("allMisIdMisIdPairs","all misid.+misid. pairs");  // mis-id + mis-id
 929   allMisIdMisIdPairs->SetLegPDGs(11,11,kTRUE,kTRUE);
 930   allMisIdMisIdPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 931   die->AddSignalMC(allMisIdMisIdPairs);
 932
 933   // 30
 934   AliDielectronSignalMC* elePionPairs = new AliDielectronSignalMC("elePionPairs","electron+pion pairs");    // true electron + mis-id pion
 935   elePionPairs->SetLegPDGs(11,211);
 936   elePionPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 937   die->AddSignalMC(elePionPairs);
 938
 939   // 31
 940   AliDielectronSignalMC* eleProtonPairs = new AliDielectronSignalMC("eleProtonPairs","Electron+proton pairs");  // true electron + mis-id proton
 941   eleProtonPairs->SetLegPDGs(11,2212);
 942   eleProtonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 943   die->AddSignalMC(eleProtonPairs);
 944
 945   // 32
 946   AliDielectronSignalMC* eleKaonPairs = new AliDielectronSignalMC("eleKaonPairs","electron+kaon pairs");   // true electron + mis-id kaon
 947   eleKaonPairs->SetLegPDGs(11,321);
 948   eleKaonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 949   die->AddSignalMC(eleKaonPairs);
 950
 951   // 33
 952   AliDielectronSignalMC* eleProtonPairs = new AliDielectronSignalMC("eleProtonPairs","Electron+proton pairs");  // true electron + mis-id proton
 953   eleProtonPairs->SetLegPDGs(11,2212);
 954   eleProtonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 955   die->AddSignalMC(eleProtonPairs);
 956
 957   // 34
 958   AliDielectronSignalMC* piPiPairs = new AliDielectronSignalMC("piPiPairs","pion+pion pairs");    // mis-id pion + mis-id pion
 959   piPiPairs->SetLegPDGs(211,211);
 960   piPiPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 961   die->AddSignalMC(piPiPairs);
 962
 963   // 35
 964   AliDielectronSignalMC* piKaonPairs = new AliDielectronSignalMC("piKaonPairs","pion+kaon pairs");  // mis-id pion + mis-id kaon
 965   piKaonPairs->SetLegPDGs(211,321);
 966   piKaonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 967   die->AddSignalMC(piKaonPairs);
 968
 969   // 36
 970   AliDielectronSignalMC* piProtonPairs = new AliDielectronSignalMC("piProtonPairs","pion+proton pairs");  // mis-id pion + mis-id proton
 971   piProtonPairs->SetLegPDGs(211,2212);
 972   piProtonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 973   die->AddSignalMC(piProtonPairs);
 974
 975   // 37
 976   AliDielectronSignalMC* kaonKaonPairs = new AliDielectronSignalMC("kaonKaonPairs","kaon+kaon pairs");  // mis-id kaon + mis-id kaon
 977   kaonKaonPairs->SetLegPDGs(321,321);
 978   kaonKaonPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 979   die->AddSignalMC(kaonKaonPairs);
 980
 981   // 38
 982   AliDielectronSignalMC* muonAllPairs = new AliDielectronSignalMC("muonAllPairs","muon+everything pairs");        // mis-id muon + something else (electron, pion, kaon, proton)
 983   muonAllPairs->SetLegPDGs(13,13,kFALSE,kTRUE);
 984   muonAllPairs->SetCheckBothChargesLegs(kTRUE,kTRUE);
 985   die->AddSignalMC(muonAllPairs);
 986 }
 987
 988 //______________________________________________________________________________________
 989 /*void SetEtaCorrection()
 990 {
 991 //
 992 // Eta equalization for the TPC response
 993 //
 994   if (AliDielectronPID::GetEtaCorrFunction()) return;
 995
 996   TString list=gSystem->Getenv("LIST");
 997
 998   TFile f("$TRAIN_ROOT/jpsi_JPSI/EtaCorrMaps.root");
 999   if (!f.IsOpen()) return;
1000   TList *keys=f.GetListOfKeys();
1001
1002   for (Int_t i=0; i<keys->GetEntries(); ++i){
1003     TString kName=keys->At(i)->GetName();
1004     TPRegexp reg(kName);
1005     if (reg.MatchB(list)){
1006       printf("Using Eta Correction Function: %s\n",kName.Data());
1007       AliDielectronPID::SetEtaCorrFunction((TF1*)f.Get(kName.Data()));
1008     }
1009   }
1010 }*/