1 #ifndef AliDptDptInMC_H_Included
2 #define AliDptDptInMC_H_Included
4 #include "AliAnalysisTaskSE.h"
9 #include "AliPIDResponse.h"
13 class AliInputEventHandler;
30 class AliDptDptInMC : public AliAnalysisTaskSE
34 AliDptDptInMC(const TString & name);
37 AliDptDptInMC(const AliDptDptInMC&);
38 const AliDptDptInMC& operator=(const AliDptDptInMC&);
41 virtual ~AliDptDptInMC();
43 // Implementation of interace methods
44 //virtual void ConnectInputData(Option_t *);
45 virtual void UserCreateOutputObjects();
46 virtual void UserExec(Option_t *option);
47 virtual void FinishTaskOutput();
48 virtual void Terminate(Option_t* );
49 virtual void createHistograms();
50 virtual void finalizeHistograms();
52 virtual void addToList(TH1 *h);
54 TH1D * createHisto1D(const TString & name, const TString & title,int n, double xmin, double xmax,const TString & xTitle, const TString & yTitle);
55 TH1D * createHisto1D(const TString & name, const TString & title,int n, double * bins,const TString & xTitle, const TString & yTitle);
56 TH2D * createHisto2D(const TString & name, const TString & title,
57 int nx, double xmin, double xmax, int ny, double ymin, double ymax, const TString & xTitle, const TString & yTitle, const TString & zTitle);
58 TH2D * createHisto2D(const TString & name, const TString & title, int nx, double* xbins, int ny, double ymin, double ymax,
59 const TString & xTitle, const TString & yTitle, const TString & zTitle);
61 TH1F * createHisto1F(const TString & name, const TString & title,int n, double xmin, double xmax,const TString & xTitle, const TString & yTitle);
62 TH1F * createHisto1F(const TString & name, const TString & title,int n, double * bins,const TString & xTitle, const TString & yTitle);
63 TH2F * createHisto2F(const TString & name, const TString & title,
64 int nx, double xmin, double xmax, int ny, double ymin, double ymax, const TString & xTitle, const TString & yTitle, const TString & zTitle);
65 TH2F * createHisto2F(const TString & name, const TString & title, int nx, double* xbins, int ny, double ymin, double ymax,
66 const TString & xTitle, const TString & yTitle, const TString & zTitle);
67 TH3F * createHisto3F(const TString & name, const TString & title,
68 int nx, double xmin, double xmax, int ny, double ymin, double ymax, int nz, double zmin, double zmax,
69 const TString & xTitle, const TString & yTitle, const TString & zTitle);
70 TProfile * createProfile(const TString & title,const TString & description, int n, double xMin,double xMax,
71 const TString & xTitle, const TString & yTitle);
72 TProfile * createProfile(const TString & name,const TString & description,
74 const TString & xTitle, const TString & yTitle);
76 //________________________________________________________________________
78 float * getFloatArray(int size, float v);
79 double * getDoubleArray(int size, double v);
80 void fillHistoWithArray(TH1 * h, double * array, int size);
81 void fillHistoWithArray(TH2 * h, double * array, int size1, int size2);
82 void fillHistoWithArray(TH3 * h, double * array, int size1, int size2, int size3);
83 void fillHistoWithArray(TH1 * h, float * array, int size);
84 void fillHistoWithArray(TH2 * h, float * array, int size1, int size2);
85 void fillHistoWithArray(TH3 * h, float * array, int size1, int size2, int size3);
90 virtual void SetDebugLevel( int v ) { _debugLevel = v; }
91 virtual void SetSinglesOnly(int v) { _singlesOnly = v; }
92 virtual void SetUseWeights(int v) { _useWeights = v; }
93 virtual void SetSameFilter(int v) { _sameFilter = v; }
95 virtual void SetRejectPileup(int v) { _rejectPileup = v; }
96 virtual void SetRejectPairConversion(int v) { _rejectPairConversion = v; }
97 virtual void SetVertexZMin(double v) { _vertexZMin = v; }
98 virtual void SetVertexZMax(double v) { _vertexZMax = v; }
99 virtual void SetVertexXYMin(double v) { _vertexXYMin = v; }
100 virtual void SetVertexXYMax(double v) { _vertexXYMax = v; }
101 virtual void SetCentralityMethod(int v) { _centralityMethod = v; }
102 virtual void SetCentrality(double centralityMin, double centralityMax)
104 _centralityMin = centralityMin;
105 _centralityMax = centralityMax;
107 virtual void SetRequestedCharge_1(int v) { _requestedCharge_1 = v; }
108 virtual void SetRequestedCharge_2(int v) { _requestedCharge_2 = v; }
109 virtual void SetPtMin1( double v) { _min_pt_1 = v; }
110 virtual void SetPtMax1( double v) { _max_pt_1 = v; }
111 virtual void SetEtaMin1(double v) { _min_eta_1 = v; }
112 virtual void SetEtaMax1(double v) { _max_eta_1 = v; }
113 virtual void SetPtMin2( double v) { _min_pt_2 = v; }
114 virtual void SetPtMax2( double v) { _max_pt_2 = v; }
115 virtual void SetEtaMin2(double v) { _min_eta_2 = v; }
116 virtual void SetEtaMax2(double v) { _max_eta_2 = v; }
117 virtual void SetDcaZMin(double v) { _dcaZMin = v; }
118 virtual void SetDcaZMax(double v) { _dcaZMax = v; }
119 virtual void SetDcaXYMin(double v) { _dcaXYMin = v; }
120 virtual void SetDcaXYMax(double v) { _dcaXYMax = v; }
121 virtual void SetDedxMin(double v) { _dedxMin = v; }
122 virtual void SetDedxMax(double v) { _dedxMax = v; }
123 virtual void SetNClusterMin(int v) { _nClusterMin = v; }
124 virtual void SetTrackFilterBit(int v) { _trackFilterBit = v; }
125 virtual void SetWeigth_1(TH3F * v) { _weight_1 = v; }
126 virtual void SetWeigth_2(TH3F * v) { _weight_2 = v; }
128 void SetNSigmaCut(Double_t nsigma){ fNSigmaCut = nsigma;}
130 void SetAnalysisType(const char* analysisType) {fAnalysisType = analysisType;}
131 void SetResonancesCut( Bool_t NoResonances ){ fExcludeResonancesInMC = NoResonances;}
132 void SetElectronCut( Bool_t NoElectron ){ fExcludeElectronsInMC = NoElectron;}
136 // Handlers and events
137 AliAODEvent* fAODEvent; //! AOD Event
138 AliESDEvent* fESDEvent; //! ESD Event
139 AliInputEventHandler* fInputHandler; //! Generic InputEventHandler
141 AliPIDResponse* fPIDResponse;
143 // Histogram settings
144 //TList* _inputHistoList;
145 TList* _outputHistoList;
148 TClonesArray *fArrayMC;
152 //configuration variables and filters
158 int _rejectPairConversion;
163 int _centralityMethod;
164 double _centralityMin;
165 double _centralityMax;
166 int _requestedCharge_1;
167 int _requestedCharge_2;
182 TString fAnalysisType;
184 Bool_t fExcludeResonancesInMC;
185 Bool_t fExcludeElectronsInMC;
187 // event and track wise variables
212 //float * _phi_1; //!
214 float *_correction_1; //!
230 float *_correction_2; //!
233 float * _correctionWeight_1; //!
234 float * _correctionWeight_2; //!
237 int _nBins_M0; double _min_M0; double _max_M0; double _width_M0;
238 int _nBins_M1; double _min_M1; double _max_M1; double _width_M1;
239 int _nBins_M2; double _min_M2; double _max_M2; double _width_M2;
240 int _nBins_M3; double _min_M3; double _max_M3; double _width_M3;
241 int _nBins_M4; double _min_M4; double _max_M4; double _width_M4;
242 int _nBins_M5; double _min_M5; double _max_M5; double _width_M5;
243 int _nBins_M6; double _min_M6; double _max_M6; double _width_M6;
245 int _nBins_vertexZ; double _min_vertexZ; double _max_vertexZ; double _width_vertexZ;
247 int _nBins_pt_1; double _min_pt_1; double _max_pt_1; double _width_pt_1;
248 int _nBins_phi_1; double _min_phi_1; double _max_phi_1; double _width_phi_1;
249 int _nBins_eta_1; double _min_eta_1; double _max_eta_1; double _width_eta_1;
251 int _nBins_etaPhiPt_1;
252 int _nBins_zEtaPhiPt_1;
254 int _nBins_pt_2; double _min_pt_2; double _max_pt_2; double _width_pt_2;
255 int _nBins_phi_2; double _min_phi_2; double _max_phi_2; double _width_phi_2;
256 int _nBins_eta_2; double _min_eta_2; double _max_eta_2; double _width_eta_2;
258 int _nBins_etaPhiPt_2;
259 int _nBins_zEtaPhiPt_2;
261 int _nBins_etaPhi_12;
277 double __s2ptptNw_12;
281 double * __n1_1_vsPt; //!
282 double * __n1_1_vsEtaPhi; //!
283 double * __s1pt_1_vsEtaPhi; //!
284 float * __n1_1_vsZEtaPhiPt; //!
286 double * __n1_2_vsPt; //!
287 double * __n1_2_vsEtaPhi; //!
288 double * __s1pt_2_vsEtaPhi; //!
289 float * __n1_2_vsZEtaPhiPt; //!
291 //double * __n2_12_vsPtPt;
292 //double * __n2_12_vsEtaPhi;
293 //double * __s2ptpt_12_vsEtaPhi;
294 //double * __s2PtN_12_vsEtaPhi;
295 //double * __s2NPt_12_vsEtaPhi;
297 double * __n2_12_vsPtPt; //!
298 float * __n2_12_vsEtaPhi; //!
299 float * __s2ptpt_12_vsEtaPhi; //!
300 float * __s2PtN_12_vsEtaPhi; //!
301 float * __s2NPt_12_vsEtaPhi; //!
305 TH1D * _eventAccounting;
323 // PARTICLE 1 (satisfies filter 1)
324 // Primary filled quantities
326 TH2F * _n1_1_vsEtaVsPhi;
327 TH2F * _s1pt_1_vsEtaVsPhi;
328 TH3F * _n1_1_vsZVsEtaVsPhiVsPt;
329 TProfile * _n1_1_vsM; // w/ weight
330 TProfile * _s1pt_1_vsM;
331 TProfile * _n1Nw_1_vsM; // w/o weight
332 TProfile * _s1ptNw_1_vsM;
334 TH2D * _corrDedxVsP_1;
337 // PARTICLE 2 (satisfies filter 2)
338 // Primary filled quantities
340 TH2F * _n1_2_vsEtaVsPhi;
341 TH2F * _s1pt_2_vsEtaVsPhi;
342 TH3F * _n1_2_vsZVsEtaVsPhiVsPt;
343 TProfile * _n1_2_vsM;
344 TProfile * _s1pt_2_vsM;
345 TProfile * _n1Nw_2_vsM; // w/o weight
346 TProfile * _s1ptNw_2_vsM;
348 TH2D * _corrDedxVsP_2;
352 TH1F * _n2_12_vsEtaPhi;
353 TH2F * _n2_12_vsPtVsPt;
354 TH1F * _s2PtPt_12_vsEtaPhi;
355 TH1F * _s2PtN_12_vsEtaPhi;
356 TH1F * _s2NPt_12_vsEtaPhi;
358 TProfile * _n2_12_vsM;
359 TProfile * _s2PtPt_12_vsM;
360 TProfile * _s2PtN_12_vsM;
361 TProfile * _s2NPt_12_vsM;
362 TProfile * _n2Nw_12_vsM;
363 TProfile * _s2PtPtNw_12_vsM;
364 TProfile * _s2PtNNw_12_vsM;
365 TProfile * _s2NPtNw_12_vsM;
390 TString intBinCorrName;
396 TString pair_12_Name;
397 TString pair_13_Name;
398 TString pair_23_Name;
411 TString s2DptDptName;
413 TString s2PtPtNwName;
427 TString _title_counts;
437 TString _title_eta_1;
438 TString _title_phi_1;
440 TString _title_etaPhi_1;
442 TString _title_SumPt_1;
443 TString _title_AvgPt_1;
444 TString _title_AvgN_1;
445 TString _title_AvgSumPt_1;
447 TString _title_eta_2;
448 TString _title_phi_2;
450 TString _title_etaPhi_2;
452 TString _title_SumPt_2;
453 TString _title_AvgPt_2;
454 TString _title_AvgN_2;
455 TString _title_AvgSumPt_2;
457 TString _title_etaPhi_12;
459 TString _title_AvgN2_12;
460 TString _title_AvgSumPtPt_12;
461 TString _title_AvgSumPtN_12;
462 TString _title_AvgNSumPt_12;
473 ClassDef(AliDptDptInMC,1)