PWGDQ/dielectron/macrosJPSI/ConfigJpsi_mf_PbPb.C

   1 void SetupTrackCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   2 void SetupPairCutsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   3 void InitHistogramsDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   4 void InitCFDieleData(AliDielectron *diele, Int_t cutDefinition, Bool_t isAOD);
   5 void AddMCSignals(AliDielectron *diele);
   6 AliESDtrackCuts *SetupESDtrackCutsDieleData(Int_t cutDefinition);
   7
   8 TString namesDieleData=("EMCal+QA;TPC+QA;EMCal+ITSfirstclass");
   9 TObjArray *arrNamesDieleData=namesDieleData.Tokenize("; ");
  10 Bool_t hasMC=kFALSE;
  11 TString list  = gSystem->Getenv("LIST");
  12 const Int_t nDie=arrNamesDieleData->GetEntries();
  13
  14 AliDielectron* ConfigJpsi_mf_PbPb(Int_t cutDefinition, Bool_t isAOD=kFALSE)
  15 {
  16   //
  17   // Setup the instance of AliDielectron
  18   //
  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;
  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   }
  29
  30
  31   AliDielectron *diele = new AliDielectron(Form("%s",name.Data()),
  32                              Form("Track cuts: %s",name.Data()));
  33
  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
  45   // cut setup
  46 //   SetupEventCutsDieleFilter(diele, cutDefinition, isAOD);
  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!
  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
  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");
 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");
 115   pt->AddCut(AliDielectronVarManager::kPt,1.,100.);
 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);
 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
 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);
 147 pairCut->AddCut(AliDielectronVarManager::kPt,4.5,100.);
 148 diele->GetPairFilter().AddCuts(pairCut);
 149
 150 AliDielectronVarCuts *mycut = new AliDielectronVarCuts("ptCutEMCAL","cut for EMCal");
 151 mycut->AddCut(AliDielectronVarManager::kEMCALE,4.,100.);
 152 mycut->AddCut(AliDielectronVarManager::kEMCALEoverP,0.65,2.);
 153 AliDielectronPairLegCuts *varpair=new AliDielectronPairLegCuts();
 154 varpair->GetLeg1Filter().AddCuts(mycut);
 155 varpair->GetLeg2Filter().AddCuts(mycut);
 156 varpair->SetCutType(AliDielectronPairLegCuts::kAnyLeg);
 157 if(cutDefinition>=2) diele->GetPairFilter().AddCuts(varpair);
 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
 206    if (cutDefinition==2) {
 207      for (Int_t i=0; i<10; ++i){
 208         histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));
 209      }
 210    }
 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
 218   if (cutDefinition==2) {
 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
 232    //add histograms to Track classes
 233   histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",100,0,20.,AliDielectronVarManager::kPt);
 234   histos->UserHistogram("Track","TPCnCls","Number of Clusters TPC;TPC number clusteres;#tracks",160,-0.5,159.5,AliDielectronVarManager::kNclsTPC);
 235
 236   histos->UserHistogram("Track","dXY","dXY;dXY [cm];#tracks",200,-1.,1.,AliDielectronVarManager::kImpactParXY);
 237   histos->UserHistogram("Track","dZ","dZ;dZ [cm];#tracks",200,-3.,3.,AliDielectronVarManager::kImpactParZ);
 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",
 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
 263
 264   histos->UserHistogram("Track","dEdx_Phi","dEdx;#phi [rad];TPC signal (arb units);#tracks",
 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
 271
 272   histos->UserHistogram("Track","dEdx_nSigmaEMCal","dEdx;NsigmaEmcal;TPC signal (arb units);NSigmaEMCAL",
 273                         200,-5.,5.,800,20.,200.,AliDielectronVarManager::kEMCALnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
 274
 275   histos->UserHistogram("Track","dEdx_TPCnSigmaEle","dEdx;TPC signal (arbunits);TPC number of sigmas Electrons;TPC signal (a.u.);#tracks",
 276                         100,-10.,10.,800,20.,200.,AliDielectronVarManager::kTPCnSigmaEle,AliDielectronVarManager::kTPCsignal,kTRUE);
 277
 278   histos->UserHistogram("Track","dEdx_EoverP","dEdx;EoverP;TPC signal (arbunits);E/P",100,0.,5.,800,20.,200.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kTPCsignal,kTRUE);
 279
 280   histos->UserHistogram("Track","nSigmaEMCal_EoverP","NsigmaEmcal;EoverP;NSigmaEMCAL;E/P",100,0.,5.,200,-5.,5.,AliDielectronVarManager::kEMCALEoverP,AliDielectronVarManager::kEMCALnSigmaEle,kTRUE);
 281
 282   histos->UserHistogram("Track","EMCal_E","EmcalE;Cluster Energy [GeV];#Clusters",200,0.,40.,AliDielectronVarManager::kEMCALE,kTRUE);
 283
 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
 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
 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");
 311   cf->AddVariable(AliDielectronVarManager::kM,125,0.,125*.04); //40Mev Steps
 312   cf->AddVariable(AliDielectronVarManager::kPairType,12,0,12);
 313   cf->AddVariable(AliDielectronVarManager::kOpeningAngle,"0.,0.1,0.2,0.4,0.6,0.8,1.0,2.0,3.14");
 314   cf->AddVariable(AliDielectronVarManager::kY,20,-1.,1.);
 315   //leg variables
 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);
 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);
 318   cf->AddVariable(AliDielectronVarManager::kNclsTPC,"80, 85, 90, 95, 100, 110, 120, 160",kTRUE);
 319 //   cf->AddVariable(AliDielectronVarManager::kEta,"-5.0,-1.0,-0.9,-0.7,0.7,0.9,1.0,5.0",kTRUE);
 320 //   cf->AddVariable(AliDielectronVarManager::kPhi,"0.0,1.0,2.0,3.0,4.0,5.0,6.0",kTRUE);
 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);
 322   cf->AddVariable(AliDielectronVarManager::kEMCALnSigmaEle,"-3.5,-3.0,-2.0,-1.0,1.0,2.0,3.0,4.0,5.0",kTRUE);
 323   cf->AddVariable(AliDielectronVarManager::kEMCALNCells,25,0,25,kTRUE);
 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);
 327   //event variables
 328   cf->AddVariable(AliDielectronVarManager::kCentrality,20,0.,100.);
 329   cf->AddVariable(AliDielectronVarManager::kMultV0,26,0.,26000.);
 330   cf->AddVariable(AliDielectronVarManager::kZvPrim,20,-10.,10.);
 331
 332   cf->AddVariable(AliDielectronVarManager::kITSLayerFirstCls,6,0.,6.,kTRUE);
 333
 334   if (hasMC){
 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);
 338   }
 339     //only in this case write MC truth info
 340   if (cutDefinition==0){
 341     cf->SetStepForMCtruth();
 342 //     cf->SetStepsForMCtruthOnly();
 343  }
 344
 345   diele->SetCFManagerPair(cf);
 346
 347 }
 348
 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 }