[u/mrichter/AliRoot.git] / PWGCF / Correlations / DPhi / AliDptDptInMC.h

#ifndef AliDptDptInMC_H_Included
#define AliDptDptInMC_H_Included

#include "AliAnalysisTaskSE.h"
#include "TString.h"
#include "AliLog.h"

#include "AliPID.h"
#include "AliPIDResponse.h"

class AliAODEvent;
class AliESDEvent;
class AliInputEventHandler;
class TH1;
class TH2;
class TH2;
class TH3;
class TH1F;
class TH2F;
class TH2F;
class TH3F;
class TH1D;
class TH2D;
class TH2D;
class TH3D;
class TProfile;


class AliDptDptInMC : public AliAnalysisTaskSE
{
public:
  AliDptDptInMC();  
  AliDptDptInMC(const TString & name);
  
private:
  AliDptDptInMC(const  AliDptDptInMC&);
  const AliDptDptInMC& operator=(const  AliDptDptInMC&);

public:
  virtual ~AliDptDptInMC();

  // Implementation of interace methods
  //virtual void   ConnectInputData(Option_t *);
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   FinishTaskOutput();
  virtual void   Terminate(Option_t* );   
  virtual void   createHistograms();
  virtual void   finalizeHistograms();
  
  virtual void   addToList(TH1 *h);

  TH1D * createHisto1D(const TString &  name, const TString &  title,int n, double xmin, double xmax,const TString &  xTitle, const TString &  yTitle);
  TH1D * createHisto1D(const TString &  name, const TString &  title,int n, double * bins,const TString &  xTitle, const TString &  yTitle);
  TH2D * createHisto2D(const TString &  name, const TString &  title, 
                       int nx, double xmin, double xmax, int ny, double ymin, double ymax, const TString &  xTitle, const TString &  yTitle, const TString &  zTitle);
  TH2D * createHisto2D(const TString &  name, const TString &  title, int nx, double* xbins, int ny, double ymin, double ymax, 
                       const TString &  xTitle, const TString &  yTitle, const TString &  zTitle);

  TH1F * createHisto1F(const TString &  name, const TString &  title,int n, double xmin, double xmax,const TString &  xTitle, const TString &  yTitle);
  TH1F * createHisto1F(const TString &  name, const TString &  title,int n, double * bins,const TString &  xTitle, const TString &  yTitle);
  TH2F * createHisto2F(const TString &  name, const TString &  title, 
                       int nx, double xmin, double xmax, int ny, double ymin, double ymax, const TString &  xTitle, const TString &  yTitle, const TString &  zTitle);
  TH2F * createHisto2F(const TString &  name, const TString &  title, int nx, double* xbins, int ny, double ymin, double ymax, 
                       const TString &  xTitle, const TString &  yTitle, const TString &  zTitle);
  TH3F * createHisto3F(const TString &  name, const TString &  title, 
                       int nx, double xmin, double xmax, int ny, double ymin, double ymax, int nz, double zmin, double zmax, 
                       const TString &  xTitle, const TString &  yTitle, const TString &  zTitle);  
  TProfile * createProfile(const TString &  title,const TString &  description, int n,  double xMin,double xMax,
                           const TString &  xTitle, const TString &  yTitle);
  TProfile * createProfile(const TString &  name,const TString &  description, 
                           int nx,  double* bins,
                           const TString &  xTitle, const TString &  yTitle);
  
  //________________________________________________________________________
  
  float  * getFloatArray(int size, float v);
  double * getDoubleArray(int size, double v);
  void fillHistoWithArray(TH1 * h, double * array, int size);
  void fillHistoWithArray(TH2 * h, double * array, int size1, int size2);
  void fillHistoWithArray(TH3 * h, double * array, int size1, int size2, int size3);
  void fillHistoWithArray(TH1 * h, float * array, int size);
  void fillHistoWithArray(TH2 * h, float * array, int size1, int size2);
  void fillHistoWithArray(TH3 * h, float * array, int size1, int size2, int size3);


  virtual     void    SetDebugLevel( int v )              { _debugLevel   = v; }
  virtual     void    SetSinglesOnly(int v)               { _singlesOnly  = v; } 
  virtual     void    SetUseWeights(int v)                { _useWeights   = v; } 
  virtual     void    SetSameFilter(int v)                { _sameFilter   = v; }
  
  virtual     void    SetRejectPileup(int v)              { _rejectPileup         = v; } 
  virtual     void    SetRejectPairConversion(int v)      { _rejectPairConversion = v; } 
  virtual     void    SetVertexZMin(double v)             { _vertexZMin           = v; } 
  virtual     void    SetVertexZMax(double v)             { _vertexZMax           = v; } 
  virtual     void    SetVertexXYMin(double v)            { _vertexXYMin          = v; } 
  virtual     void    SetVertexXYMax(double v)            { _vertexXYMax          = v; } 
  virtual     void    SetCentralityMethod(int v)          { _centralityMethod     = v; }
  virtual     void    SetCentrality(double centralityMin, double centralityMax)
  {
    _centralityMin = centralityMin;
    _centralityMax = centralityMax;
  }
  virtual     void    SetRequestedCharge_1(int v)     { _requestedCharge_1 = v; }
  virtual     void    SetRequestedCharge_2(int v)     { _requestedCharge_2 = v; }
  virtual     void    SetPtMin1( double v)            { _min_pt_1          = v; } 
  virtual     void    SetPtMax1( double v)            { _max_pt_1          = v; } 
  virtual     void    SetEtaMin1(double v)            { _min_eta_1         = v; } 
  virtual     void    SetEtaMax1(double v)            { _max_eta_1         = v; } 
  virtual     void    SetPtMin2( double v)            { _min_pt_2          = v; } 
  virtual     void    SetPtMax2( double v)            { _max_pt_2          = v; } 
  virtual     void    SetEtaMin2(double v)            { _min_eta_2         = v; } 
  virtual     void    SetEtaMax2(double v)            { _max_eta_2         = v; } 
  virtual     void    SetDcaZMin(double v)            { _dcaZMin           = v; } 
  virtual     void    SetDcaZMax(double v)            { _dcaZMax           = v; } 
  virtual     void    SetDcaXYMin(double v)           { _dcaXYMin          = v; } 
  virtual     void    SetDcaXYMax(double v)           { _dcaXYMax          = v; } 
  virtual     void    SetDedxMin(double v)            { _dedxMin           = v; } 
  virtual     void    SetDedxMax(double v)            { _dedxMax           = v; } 
  virtual     void    SetNClusterMin(int v)           { _nClusterMin       = v; } 
  virtual     void    SetTrackFilterBit(int v)        { _trackFilterBit    = v; }
  virtual     void    SetWeigth_1(TH3F * v)           { _weight_1          = v; }
  virtual     void    SetWeigth_2(TH3F * v)           { _weight_2          = v; }
  
  void SetNSigmaCut(Double_t nsigma){ fNSigmaCut = nsigma;}

  void SetAnalysisType(const char* analysisType) {fAnalysisType = analysisType;}
  void SetResonancesCut( Bool_t NoResonances ){ fExcludeResonancesInMC = NoResonances;}
  void SetElectronCut( Bool_t NoElectron ){ fExcludeElectronsInMC = NoElectron;}
  
protected:      
  
  // Handlers and events
  AliAODEvent*             fAODEvent;             //! AOD Event 
  AliESDEvent*             fESDEvent;             //! ESD Event 
  AliInputEventHandler*    fInputHandler;    //! Generic InputEventHandler 
  
  AliPIDResponse*          fPIDResponse;

  // Histogram settings
  //TList*              _inputHistoList;
  TList*              _outputHistoList;
  //int _outputSlot;
  
  TClonesArray          *fArrayMC;  
  double   _twoPi;
  long     _eventCount;
  
  //configuration variables and filters
  int      _debugLevel;
  int      _singlesOnly; 
  int      _useWeights; 
  int      _sameFilter;
  int      _rejectPileup; 
  int      _rejectPairConversion; 
  double   _vertexZMin; 
  double   _vertexZMax; 
  double   _vertexXYMin; 
  double   _vertexXYMax; 
  int      _centralityMethod;
  double   _centralityMin;
  double   _centralityMax;
  int      _requestedCharge_1;
  int      _requestedCharge_2;
  double   _dcaZMin; 
  double   _dcaZMax; 
  double   _dcaXYMin; 
  double   _dcaXYMax;
  double   _dedxMin;
  double   _dedxMax;
  int      _nClusterMin; 
  int      _trackFilterBit;
  Double_t fNSigmaCut;  

  //double _min_eta_1;
  //double _max_eta_1;
  //double _min_eta_2;
  //double _max_eta_2;
  TString      fAnalysisType;

  Bool_t fExcludeResonancesInMC;
  Bool_t fExcludeElectronsInMC;
 
  // event and track wise variables
  
  double _field;
  int    _nTracks;
  double _mult0;
  double _mult1;
  double _mult2;
  double _mult3;
  double _mult4;
  double _mult4a;
  double _mult5;
  double _mult6;
  
  //particle 1
  int     arraySize;
  int    *_id_1;               //!
  int    *_charge_1;           //! 
  //int  *  _iPhi_1;            //!
  //int  *  _iEta_1;            //!
  int    *_iEtaPhi_1;            //!
  int    *_iPt_1;             //!
  float  *_pt_1;               //!
  float  *_px_1;              //!
  float  *_py_1;              //!
  float  *_pz_1;              //!
 //float * _phi_1;             //!
  //float*  _eta_1;             //!
  float  *_correction_1;           //!
  float  *_dedx_1;           //!
  
  //particle 2
  int    *_id_2;              //!
  int    *_charge_2;            //!
  //int    *_iPhi_2;            //!
  //int    *_iEta_2;            //!
  int    *_iEtaPhi_2;            //!
  int    *_iPt_2;             //!
  float  *_pt_2;              //!
  float  *_px_2;              //!
  float  *_py_2;              //!
  float  *_pz_2;              //!
  //float  *_phi_2;             //!
  //float  *_eta_2;             //!
  float  *_correction_2;           //!
  float  *_dedx_2;           //!
  
  float * _correctionWeight_1;           //!
  float * _correctionWeight_2;           //!
  
  //histograming
  int _nBins_M0;       double _min_M0;       double _max_M0;       double _width_M0;
  int _nBins_M1;       double _min_M1;       double _max_M1;       double _width_M1;
  int _nBins_M2;       double _min_M2;       double _max_M2;       double _width_M2;
  int _nBins_M3;       double _min_M3;       double _max_M3;       double _width_M3;
  int _nBins_M4;       double _min_M4;       double _max_M4;       double _width_M4;
  int _nBins_M5;       double _min_M5;       double _max_M5;       double _width_M5;
  int _nBins_M6;       double _min_M6;       double _max_M6;       double _width_M6;
  
  int _nBins_vertexZ;  double _min_vertexZ;  double _max_vertexZ;  double _width_vertexZ;
  
  int _nBins_pt_1;     double _min_pt_1;     double _max_pt_1;     double _width_pt_1;
  int _nBins_phi_1;    double _min_phi_1;    double _max_phi_1;    double _width_phi_1;
  int _nBins_eta_1;    double _min_eta_1;    double _max_eta_1;    double _width_eta_1;
  int _nBins_etaPhi_1; 
  int _nBins_etaPhiPt_1;
  int _nBins_zEtaPhiPt_1;
  
  int _nBins_pt_2;     double _min_pt_2;     double _max_pt_2;     double _width_pt_2;
  int _nBins_phi_2;    double _min_phi_2;    double _max_phi_2;    double _width_phi_2;
  int _nBins_eta_2;    double _min_eta_2;    double _max_eta_2;    double _width_eta_2;
  int _nBins_etaPhi_2; 
  int _nBins_etaPhiPt_2;
  int _nBins_zEtaPhiPt_2;
  
  int _nBins_etaPhi_12;
    
  double __n1_1;
  double __n1_2;
  double __n2_12;   
  double __s1pt_1;
  double __s1pt_2;
  double __s2ptpt_12;
  double __s2NPt_12;
  double __s2PtN_12;
  
  double __n1Nw_1;
  double __n1Nw_2;
  double __n2Nw_12;   
  double __s1ptNw_1;
  double __s1ptNw_2;
  double __s2ptptNw_12;
  double __s2NPtNw_12;
  double __s2PtNNw_12;
  
  double * __n1_1_vsPt;   //!
  double * __n1_1_vsEtaPhi;     //! 
  double * __s1pt_1_vsEtaPhi;    //!
  float  * __n1_1_vsZEtaPhiPt;    //!
    
  double * __n1_2_vsPt;   //!
  double * __n1_2_vsEtaPhi;     //! 
  double * __s1pt_2_vsEtaPhi;    //!
  float  * __n1_2_vsZEtaPhiPt;    //!
  
  //double * __n2_12_vsPtPt;
  //double * __n2_12_vsEtaPhi;
  //double * __s2ptpt_12_vsEtaPhi;
  //double * __s2PtN_12_vsEtaPhi;
  //double * __s2NPt_12_vsEtaPhi;
  
  double * __n2_12_vsPtPt;   //!
  float  * __n2_12_vsEtaPhi;   //!
  float  * __s2ptpt_12_vsEtaPhi;   //!
  float  * __s2PtN_12_vsEtaPhi;   //!
  float  * __s2NPt_12_vsEtaPhi;   //!
  
  TH3F * _weight_1;
  TH3F * _weight_2;
  TH1D * _eventAccounting;
  TH1D * _m0;
  TH1D * _m1;
  TH1D * _m2;
  TH1D * _m3;
  TH1D * _m4;
  TH1D * _m5;
  TH1D * _m6;
  TH1D * _vertexZ;

  TH1F * _Ncluster1;
  TH1F * _Ncluster2;
  TH1F * _etadis;
  TH1F * _phidis;
  TH1F * _dcaz;
  TH1F * _dcaxy;  


  // PARTICLE 1 (satisfies filter 1)
  // Primary filled quantities
  TH1F      *  _n1_1_vsPt;         
  TH2F      *  _n1_1_vsEtaVsPhi;
  TH2F      *  _s1pt_1_vsEtaVsPhi; 
  TH3F      *  _n1_1_vsZVsEtaVsPhiVsPt;
  TProfile *  _n1_1_vsM;  // w/ weight
  TProfile *  _s1pt_1_vsM;
  TProfile *  _n1Nw_1_vsM; // w/o weight
  TProfile *  _s1ptNw_1_vsM;
  TH2D      *  _dedxVsP_1;
  TH2D      *  _corrDedxVsP_1;
  TH2F      *  _betaVsP_1;
  
  // PARTICLE 2 (satisfies filter 2)
  // Primary filled quantities
  TH1F      *  _n1_2_vsPt;         
  TH2F      *  _n1_2_vsEtaVsPhi;
  TH2F      *  _s1pt_2_vsEtaVsPhi;
  TH3F      *  _n1_2_vsZVsEtaVsPhiVsPt; 
  TProfile *  _n1_2_vsM;
  TProfile *  _s1pt_2_vsM;
  TProfile *  _n1Nw_2_vsM; // w/o weight
  TProfile *  _s1ptNw_2_vsM;
  TH2D      *  _dedxVsP_2;
  TH2D      *  _corrDedxVsP_2;
  TH2F      *  _betaVsP_2;
  
  // Pairs 1 & 2  
  TH1F      * _n2_12_vsEtaPhi;        
  TH2F      * _n2_12_vsPtVsPt;
  TH1F      * _s2PtPt_12_vsEtaPhi;    
  TH1F      * _s2PtN_12_vsEtaPhi;       
  TH1F      * _s2NPt_12_vsEtaPhi;     
  
  TProfile * _n2_12_vsM;        
  TProfile * _s2PtPt_12_vsM;    
  TProfile * _s2PtN_12_vsM;       
  TProfile * _s2NPt_12_vsM; 
  TProfile * _n2Nw_12_vsM;        
  TProfile * _s2PtPtNw_12_vsM;    
  TProfile * _s2PtNNw_12_vsM;       
  TProfile * _s2NPtNw_12_vsM; 
  
  TH1F      * _invMass;
  TH1F      * _invMassElec;
  
  TString n1Name;
  TString n1NwName;
  TString n2Name;
  TString n2NwName;
  TString n3Name;
  TString n1n1Name;
  TString n1n1n1Name;
  TString n2n1Name;
  TString r1Name;
  TString r2Name;
  TString r3Name;
  TString r2r1Name;
  TString c2Name;
  TString c3Name;
  TString d3Name;
  TString p3Name;
  TString cName;
  
  TString intR2Name;
  TString binCorrName;
  TString intBinCorrName;
  
  TString countsName;
  TString part_1_Name;
  TString part_2_Name;
  TString part_3_Name;
  TString pair_12_Name;
  TString pair_13_Name;
  TString pair_23_Name;
  TString tripletName;
  
  TString avg;
  TString avgName;
  TString sumName;
  TString s1ptName;
  TString s1ptNwName;
  TString s1DptName;
  
  TString s2PtPtName;
  TString s2NPtName;
  TString s2PtNName;
  TString s2DptDptName;
 
  TString s2PtPtNwName;
  TString s2NPtNwName;
  TString s2PtNNwName;
  
  TString ptName;
  TString ptptName;
  TString pt1pt1Name;
  TString DptName;
  TString DptDptName;
  TString RDptDptName;
  TString nPtName;
  TString ptNName;
  TString seanName;
  
  TString _title_counts;
  
  TString _title_m0;
  TString _title_m1;
  TString _title_m2;
  TString _title_m3;
  TString _title_m4;
  TString _title_m5;
  TString _title_m6;
  
  TString _title_eta_1;
  TString _title_phi_1;
  TString _title_pt_1;
  TString _title_etaPhi_1;
  TString _title_n_1;
  TString _title_SumPt_1;
  TString _title_AvgPt_1;
  TString _title_AvgN_1;
  TString _title_AvgSumPt_1;
  
  TString _title_eta_2;
  TString _title_phi_2;
  TString _title_pt_2;
  TString _title_etaPhi_2;
  TString _title_n_2;
  TString _title_SumPt_2;
  TString _title_AvgPt_2;
  TString _title_AvgN_2;
  TString _title_AvgSumPt_2;
  
  TString _title_etaPhi_12;
  
  TString _title_AvgN2_12;
  TString _title_AvgSumPtPt_12;
  TString _title_AvgSumPtN_12;
  TString _title_AvgNSumPt_12;
  
  TString vsZ;
  TString vsM;
  TString vsPt;
  TString vsPhi; 
  TString vsEta; 
  TString vsEtaPhi; 
  TString vsPtVsPt;

  
  ClassDef(AliDptDptInMC,1)
}; 


#endif
Commit	Line	Data
1486c432 MW	1	#ifndef AliDptDptInMC_H_Included
	2	#define AliDptDptInMC_H_Included
	3
	4	#include "AliAnalysisTaskSE.h"
	5	#include "TString.h"
	6	#include "AliLog.h"
	7
	8	#include "AliPID.h"
	9	#include "AliPIDResponse.h"
	10
	11	class AliAODEvent;
	12	class AliESDEvent;
	13	class AliInputEventHandler;
	14	class TH1;
	15	class TH2;
	16	class TH2;
	17	class TH3;
	18	class TH1F;
	19	class TH2F;
	20	class TH2F;
	21	class TH3F;
	22	class TH1D;
	23	class TH2D;
	24	class TH2D;
	25	class TH3D;
	26	class TProfile;
	27
	28
	29
	30	class AliDptDptInMC : public AliAnalysisTaskSE
	31	{
	32	public:
	33	AliDptDptInMC();
	34	AliDptDptInMC(const TString & name);
	35
	36	private:
	37	AliDptDptInMC(const AliDptDptInMC&);
	38	const AliDptDptInMC& operator=(const AliDptDptInMC&);
	39
	40	public:
	41	virtual ~AliDptDptInMC();
	42
	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();
	51
	52	virtual void addToList(TH1 *h);
	53
	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);
	60
	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,
73	int nx, double* bins,
74	const TString & xTitle, const TString & yTitle);
75
76	//________________________________________________________________________
77
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);
86
87
88
89
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; }
94
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)
103	{
104	_centralityMin = centralityMin;
105	_centralityMax = centralityMax;
106	}
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; }
127
128	void SetNSigmaCut(Double_t nsigma){ fNSigmaCut = nsigma;}
129
130	void SetAnalysisType(const char* analysisType) {fAnalysisType = analysisType;}
131	void SetResonancesCut( Bool_t NoResonances ){ fExcludeResonancesInMC = NoResonances;}
132	void SetElectronCut( Bool_t NoElectron ){ fExcludeElectronsInMC = NoElectron;}
133
134	protected:
135
136	// Handlers and events
137	AliAODEvent* fAODEvent; //! AOD Event
138	AliESDEvent* fESDEvent; //! ESD Event
139	AliInputEventHandler* fInputHandler; //! Generic InputEventHandler
140
141	AliPIDResponse* fPIDResponse;
142
143	// Histogram settings
144	//TList* _inputHistoList;
145	TList* _outputHistoList;
146	//int _outputSlot;
147
148	TClonesArray *fArrayMC;
149	double _twoPi;
150	long _eventCount;
151
152	//configuration variables and filters
153	int _debugLevel;
154	int _singlesOnly;
155	int _useWeights;
156	int _sameFilter;
157	int _rejectPileup;
158	int _rejectPairConversion;
159	double _vertexZMin;
160	double _vertexZMax;
161	double _vertexXYMin;
162	double _vertexXYMax;
163	int _centralityMethod;
164	double _centralityMin;
165	double _centralityMax;
166	int _requestedCharge_1;
167	int _requestedCharge_2;
168	double _dcaZMin;
169	double _dcaZMax;
170	double _dcaXYMin;
171	double _dcaXYMax;
172	double _dedxMin;
173	double _dedxMax;
174	int _nClusterMin;
175	int _trackFilterBit;
176	Double_t fNSigmaCut;
177
178	//double _min_eta_1;
179	//double _max_eta_1;
180	//double _min_eta_2;
181	//double _max_eta_2;
182	TString fAnalysisType;
183
184	Bool_t fExcludeResonancesInMC;
185	Bool_t fExcludeElectronsInMC;
186
187	// event and track wise variables
188
189	double _field;
190	int _nTracks;
191	double _mult0;
192	double _mult1;
193	double _mult2;
194	double _mult3;
195	double _mult4;
196	double _mult4a;
197	double _mult5;
198	double _mult6;
199
200	//particle 1
201	int arraySize;
202	int *_id_1; //!
203	int *_charge_1; //!
204	//int * _iPhi_1; //!
205	//int * _iEta_1; //!
206	int *_iEtaPhi_1; //!
207	int *_iPt_1; //!
208	float *_pt_1; //!
209	float *_px_1; //!
210	float *_py_1; //!
211	float *_pz_1; //!
212	//float * _phi_1; //!
213	//float* _eta_1; //!
214	float *_correction_1; //!
215	float *_dedx_1; //!
216
217	//particle 2
218	int *_id_2; //!
219	int *_charge_2; //!
220	//int *_iPhi_2; //!
221	//int *_iEta_2; //!
222	int *_iEtaPhi_2; //!
223	int *_iPt_2; //!
224	float *_pt_2; //!
225	float *_px_2; //!
226	float *_py_2; //!
227	float *_pz_2; //!
228	//float *_phi_2; //!
229	//float *_eta_2; //!
230	float *_correction_2; //!
231	float *_dedx_2; //!
232
233	float * _correctionWeight_1; //!
234	float * _correctionWeight_2; //!
235
236	//histograming
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;
244
245	int _nBins_vertexZ; double _min_vertexZ; double _max_vertexZ; double _width_vertexZ;
246
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;
250	int _nBins_etaPhi_1;
251	int _nBins_etaPhiPt_1;
252	int _nBins_zEtaPhiPt_1;
253
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;
257	int _nBins_etaPhi_2;
258	int _nBins_etaPhiPt_2;
259	int _nBins_zEtaPhiPt_2;
260
261	int _nBins_etaPhi_12;
262
263	double __n1_1;
264	double __n1_2;
265	double __n2_12;
266	double __s1pt_1;
267	double __s1pt_2;
268	double __s2ptpt_12;
269	double __s2NPt_12;
270	double __s2PtN_12;
271
272	double __n1Nw_1;
273	double __n1Nw_2;
274	double __n2Nw_12;
275	double __s1ptNw_1;
276	double __s1ptNw_2;
277	double __s2ptptNw_12;
278	double __s2NPtNw_12;
279	double __s2PtNNw_12;
280
281	double * __n1_1_vsPt; //!
282	double * __n1_1_vsEtaPhi; //!
283	double * __s1pt_1_vsEtaPhi; //!
284	float * __n1_1_vsZEtaPhiPt; //!
285
286	double * __n1_2_vsPt; //!
287	double * __n1_2_vsEtaPhi; //!
288	double * __s1pt_2_vsEtaPhi; //!
289	float * __n1_2_vsZEtaPhiPt; //!
290
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;
296
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; //!
302
303	TH3F * _weight_1;
304	TH3F * _weight_2;
305	TH1D * _eventAccounting;
306	TH1D * _m0;
307	TH1D * _m1;
308	TH1D * _m2;
309	TH1D * _m3;
310	TH1D * _m4;
311	TH1D * _m5;
312	TH1D * _m6;
313	TH1D * _vertexZ;
314
315	TH1F * _Ncluster1;
316	TH1F * _Ncluster2;
317	TH1F * _etadis;
318	TH1F * _phidis;
319	TH1F * _dcaz;
320	TH1F * _dcaxy;
321
322
323	// PARTICLE 1 (satisfies filter 1)
324	// Primary filled quantities
325	TH1F * _n1_1_vsPt;
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;
333	TH2D * _dedxVsP_1;
334	TH2D * _corrDedxVsP_1;
335	TH2F * _betaVsP_1;
336
337	// PARTICLE 2 (satisfies filter 2)
338	// Primary filled quantities
339	TH1F * _n1_2_vsPt;
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;
347	TH2D * _dedxVsP_2;
348	TH2D * _corrDedxVsP_2;
349	TH2F * _betaVsP_2;
350
351	// Pairs 1 & 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;
357
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;
366
367	TH1F * _invMass;
368	TH1F * _invMassElec;
369
370	TString n1Name;
371	TString n1NwName;
372	TString n2Name;
373	TString n2NwName;
374	TString n3Name;
375	TString n1n1Name;
376	TString n1n1n1Name;
377	TString n2n1Name;
378	TString r1Name;
379	TString r2Name;
380	TString r3Name;
381	TString r2r1Name;
382	TString c2Name;
383	TString c3Name;
384	TString d3Name;
385	TString p3Name;
386	TString cName;
387
388	TString intR2Name;
389	TString binCorrName;
390	TString intBinCorrName;
391
392	TString countsName;
393	TString part_1_Name;
394	TString part_2_Name;
395	TString part_3_Name;
396	TString pair_12_Name;
397	TString pair_13_Name;
398	TString pair_23_Name;
399	TString tripletName;
400
401	TString avg;
402	TString avgName;
403	TString sumName;
404	TString s1ptName;
405	TString s1ptNwName;
406	TString s1DptName;
407
408	TString s2PtPtName;
409	TString s2NPtName;
410	TString s2PtNName;
411	TString s2DptDptName;
412
413	TString s2PtPtNwName;
414	TString s2NPtNwName;
415	TString s2PtNNwName;
416
417	TString ptName;
418	TString ptptName;
419	TString pt1pt1Name;
420	TString DptName;
421	TString DptDptName;
422	TString RDptDptName;
423	TString nPtName;
424	TString ptNName;
425	TString seanName;
426
427	TString _title_counts;
428
429	TString _title_m0;
430	TString _title_m1;
431	TString _title_m2;
432	TString _title_m3;
433	TString _title_m4;
434	TString _title_m5;
435	TString _title_m6;
436
437	TString _title_eta_1;
438	TString _title_phi_1;
439	TString _title_pt_1;
440	TString _title_etaPhi_1;
441	TString _title_n_1;
442	TString _title_SumPt_1;
443	TString _title_AvgPt_1;
444	TString _title_AvgN_1;
445	TString _title_AvgSumPt_1;
446
447	TString _title_eta_2;
448	TString _title_phi_2;
449	TString _title_pt_2;
450	TString _title_etaPhi_2;
451	TString _title_n_2;
452	TString _title_SumPt_2;
453	TString _title_AvgPt_2;
454	TString _title_AvgN_2;
455	TString _title_AvgSumPt_2;
456
457	TString _title_etaPhi_12;
458
459	TString _title_AvgN2_12;
460	TString _title_AvgSumPtPt_12;
461	TString _title_AvgSumPtN_12;
462	TString _title_AvgNSumPt_12;
463
464	TString vsZ;
465	TString vsM;
466	TString vsPt;
467	TString vsPhi;
468	TString vsEta;
469	TString vsEtaPhi;
470	TString vsPtVsPt;
471
472
473	ClassDef(AliDptDptInMC,1)
474	};
475
476
477	#endif
478
479