[u/mrichter/AliRoot.git] / TRD / qaAnalysis / AliTRDqaJPsi.h

#ifndef ALITRDQAJPSI_H
#define ALITRDQAJPSI_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */

//
// This class is a part of a package of high level QA monitoring for TRD.
//
// S. Radomski
// radomski@physi.uni-heidelberg.de
// March 2008
//

#include "AliAnalysisTask.h"  

class TTree; 
class TH1D; 
class TH2D;
class TLorentzVector;

class AliExternalTrackParam;
class AliKFParticle;
class AliESDEvent; 
class AliESDtrack;

class AliTRDqaJPsi : public AliAnalysisTask {

 public:

  AliTRDqaJPsi();
  AliTRDqaJPsi(const char *name);
  AliTRDqaJPsi(const AliTRDqaJPsi &trd);
  AliTRDqaJPsi &operator=(const AliTRDqaJPsi & /*g*/) { return *this; };
  virtual ~AliTRDqaJPsi() {}
   
  virtual void Exec(Option_t * opt = "");
  virtual void ConnectInputData(Option_t *);
  virtual void CreateOutputObjects();
  virtual void Terminate(Option_t * opt = "");

 private:
 
  TTree        * fChain;             //!pointer to the analyzed TTree or TChain
  AliESDEvent  * fESD;               //! Declaration of leave types

  TObjArray * fOutputContainer; //! output data container
  
  // histograms

  static const Int_t fgknSteps = 5;   // number of analysis steps (arbitrary)

  TH1D *fStatus[fgknSteps];      // track status
  TH1D *fnTracks[2*fgknSteps];   // number of tracks
  TH1D *fPt[2*fgknSteps];        // transverse momentum
  TH1D *fPID[2*fgknSteps];       // PID LQ
  TH1D *fAngleSM[fgknSteps];     // difference in SM ID
  
  //TH2D *fnGoodTracks;          // correlation of the final number of Pos and Neg tracks
  TH1D *fInvMass[fgknSteps];     // invariant mass using different cuts
  TH1D *fInvMassVec[fgknSteps];  // invariant mass
  TH1D *fInvMassDiff[fgknSteps]; // invariant mass difference

  TH2D *fPtAngle[fgknSteps];     // pt angle 

  // tracks
  AliKFParticle *fTracks[1000];  // tracks
  TLorentzVector *fVec[1000];    // Lorentz vector for the tracks
  Int_t fInSample[1000][fgknSteps]; // in sample?
  Int_t fSM[1000];                  // TRD sector
  Int_t fnKFtracks;                 //[2];          
  
  // helper functions
  void FillHist(AliESDtrack *track, Int_t step);
  TLorentzVector *CreateVector(AliESDtrack *track);


  ClassDef(AliTRDqaJPsi, 0); // a TRD analysis task 
};
#endif // ALITRDQAJPSI_H
Commit	Line	Data
8e2f611a	1	#ifndef ALITRDQAJPSI_H
	2	#define ALITRDQAJPSI_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//
	7	// This class is a part of a package of high level QA monitoring for TRD.
	8	//
	9	// S. Radomski
	10	// radomski@physi.uni-heidelberg.de
	11	// March 2008
	12	//
	13
	14	#include "AliAnalysisTask.h"
	15
	16	class TTree;
8e2f611a	17	class TH1D;
8e2f611a	18	class TH2D;
c336eab0	19	class TLorentzVector;
c336eab0	20
8e2f611a	21	class AliExternalTrackParam;
8e2f611a	22	class AliKFParticle;
c336eab0	23	class AliESDEvent;
c336eab0	24	class AliESDtrack;
8e2f611a	25
	26	class AliTRDqaJPsi : public AliAnalysisTask {
	27
	28	public:
	29
	30	AliTRDqaJPsi();
	31	AliTRDqaJPsi(const char *name);
c336eab0	32	AliTRDqaJPsi(const AliTRDqaJPsi &trd);
c336eab0	33	AliTRDqaJPsi &operator=(const AliTRDqaJPsi & /g/) { return *this; };
8e2f611a	34	virtual ~AliTRDqaJPsi() {}
	35
	36	virtual void Exec(Option_t * opt = "");
	37	virtual void ConnectInputData(Option_t *);
	38	virtual void CreateOutputObjects();
	39	virtual void Terminate(Option_t * opt = "");
	40
	41	private:
	42
	43	TTree * fChain; //!pointer to the analyzed TTree or TChain
	44	AliESDEvent * fESD; //! Declaration of leave types
	45
	46	TObjArray * fOutputContainer; //! output data container
	47
	48	// histograms
	49
c336eab0	50	static const Int_t fgknSteps = 5; // number of analysis steps (arbitrary)
8e2f611a	51
c336eab0	52	TH1D *fStatus[fgknSteps]; // track status
	53	TH1D fnTracks[2fgknSteps]; // number of tracks
	54	TH1D fPt[2fgknSteps]; // transverse momentum
	55	TH1D fPID[2fgknSteps]; // PID LQ
	56	TH1D *fAngleSM[fgknSteps]; // difference in SM ID
8e2f611a	57
8e2f611a	58	//TH2D *fnGoodTracks; // correlation of the final number of Pos and Neg tracks
c336eab0	59	TH1D *fInvMass[fgknSteps]; // invariant mass using different cuts
	60	TH1D *fInvMassVec[fgknSteps]; // invariant mass
	61	TH1D *fInvMassDiff[fgknSteps]; // invariant mass difference
8e2f611a	62
c336eab0	63	TH2D *fPtAngle[fgknSteps]; // pt angle
8e2f611a	64
8e2f611a	65	// tracks
c336eab0	66	AliKFParticle *fTracks[1000]; // tracks
	67	TLorentzVector *fVec[1000]; // Lorentz vector for the tracks
	68	Int_t fInSample[1000][fgknSteps]; // in sample?
	69	Int_t fSM[1000]; // TRD sector
	70	Int_t fnKFtracks; //[2];
8e2f611a	71
	72	// helper functions
	73	void FillHist(AliESDtrack *track, Int_t step);
	74	TLorentzVector CreateVector(AliESDtrack track);
	75
	76
	77	ClassDef(AliTRDqaJPsi, 0); // a TRD analysis task
	78	};
	79	#endif // ALITRDQAJPSI_H