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