[u/mrichter/AliRoot.git] / TPC / AliTPCcalibV0.h

#ifndef AliTPCCALIBV0_H
#define AliTPCCALIBV0_H


#include <AliTPCcalibBase.h>


class TTreeSRedirector;
class AliTPCROC;
class AliTPCseed;
class AliESDtrack;
class AliESDEvent;
class TH3F;
class TH1F;
class TH2F;
class TH1I;
class TDatabasePDG;
class AliKFParticle;
class AliKFVertex;
class AliESDv0;
class TArrayI;
class TTree;
class AliStack;
class  AliExternalTrackParam;
class  AliTPCCorrection;

class AliTPCcalibV0 : public AliTPCcalibBase {
public :

   // List of branches

  AliTPCcalibV0();
  AliTPCcalibV0(const Text_t *name, const Text_t *title);
  virtual ~AliTPCcalibV0();
  virtual void     Process(AliESDEvent *event) {return ProcessESD(event);}
  void FilterV0s(AliESDEvent* event);
  Long64_t Merge(TCollection *const li);
  void AddTree(TTree * treeInput);
  void AddTreeHPT(TTree * treeInput);
  static AliExternalTrackParam * RefitTrack(AliTPCseed *seed, AliTPCCorrection * corr, Double_t xstart, Double_t xstop, Double_t mass);
  static void MakeFitTreeTrack(const TObjArray * corrArray, Double_t ptCut, Int_t run);
  static void MakeFitTreeV0(const TObjArray * corrArray, Double_t ptCut, Int_t run);
  //
  //
  //
  void ProcessESD(AliESDEvent *esd);
  void DumpToTree(AliESDEvent *esd);
  void DumpToTreeHPT(AliESDEvent *esd);
  TTree * GetV0Tree(){return fV0Tree;}
  TTree * GetHPTTree(){return fHPTTree;}
  //  
  //  void MakeMC();
  //  void MakeV0s();
  //void ProcessV0(Int_t ftype);
  //void ProcessPI0();
  void BinLogX(TH2F * h);
  //
  //
  //  
  static AliKFParticle * Fit(AliKFVertex & primVtx, AliESDv0 *v0, Int_t PDG1, Int_t PDG2);

protected:
private:

  AliTPCcalibV0(const AliTPCcalibV0&); // Not implemented
  AliTPCcalibV0& operator=(const AliTPCcalibV0&); // Not implemented
  TTree          *fV0Tree;      // tree with full V0 information
  TTree          *fHPTTree;      // tree with high mometa tracks - full calib info
  //
  AliStack       *fStack;        // pointer to kinematic tree        
  AliESDEvent    *fESD;              //! current ED to proccess - NOT OWNER
  TDatabasePDG   *fPdg;              //! particle database
  TObjArray      *fParticles;         // array of selected MC particles
  TObjArray      *fV0s;               // array of V0s
  TObjArray      *fGammas;           // gamma conversion candidates
  //
  void     Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
  void     Process(AliTPCseed *track){return AliTPCcalibBase::Process(track);}  
  //       
  ClassDef(AliTPCcalibV0,3);
};


#endif
Commit	Line	Data
10757ee9	1	#ifndef AliTPCCALIBV0_H
	2	#define AliTPCCALIBV0_H
	3
	4
57e4988a	5	#include <AliTPCcalibBase.h>
10757ee9	6
	7
	8	class TTreeSRedirector;
	9	class AliTPCROC;
	10	class AliTPCseed;
	11	class AliESDtrack;
57e4988a	12	class AliESDEvent;
10757ee9	13	class TH3F;
	14	class TH1F;
	15	class TH2F;
	16	class TH1I;
	17	class TDatabasePDG;
	18	class AliKFParticle;
	19	class AliKFVertex;
	20	class AliESDv0;
	21	class TArrayI;
	22	class TTree;
	23	class AliStack;
46e89793	24	class AliExternalTrackParam;
46e89793	25	class AliTPCCorrection;
10757ee9	26
57e4988a	27	class AliTPCcalibV0 : public AliTPCcalibBase {
10757ee9	28	public :
	29
	30	// List of branches
	31
57e4988a	32	AliTPCcalibV0();
46e89793	33	AliTPCcalibV0(const Text_t name, const Text_t title);
10757ee9	34	virtual ~AliTPCcalibV0();
30e59eac	35	virtual void Process(AliESDEvent *event) {return ProcessESD(event);}
30e59eac	36	void FilterV0s(AliESDEvent* event);
46e89793	37	Long64_t Merge(TCollection *const li);
	38	void AddTree(TTree * treeInput);
	39	void AddTreeHPT(TTree * treeInput);
	40	static AliExternalTrackParam * RefitTrack(AliTPCseed seed, AliTPCCorrection corr, Double_t xstart, Double_t xstop, Double_t mass);
	41	static void MakeFitTreeTrack(const TObjArray * corrArray, Double_t ptCut, Int_t run);
	42	static void MakeFitTreeV0(const TObjArray * corrArray, Double_t ptCut, Int_t run);
57e4988a	43	//
	44	//
	45	//
30e59eac	46	void ProcessESD(AliESDEvent *esd);
46e89793	47	void DumpToTree(AliESDEvent *esd);
	48	void DumpToTreeHPT(AliESDEvent *esd);
	49	TTree * GetV0Tree(){return fV0Tree;}
	50	TTree * GetHPTTree(){return fHPTTree;}
	51	//
30e59eac	52	// void MakeMC();
	53	// void MakeV0s();
	54	//void ProcessV0(Int_t ftype);
	55	//void ProcessPI0();
10757ee9	56	void BinLogX(TH2F * h);
	57	//
	58	//
	59	//
0d2c6644	60	static AliKFParticle * Fit(AliKFVertex & primVtx, AliESDv0 *v0, Int_t PDG1, Int_t PDG2);
1aa15e8d	61
10757ee9	62	protected:
10757ee9	63	private:
77f88633	64
46e89793	65	AliTPCcalibV0(const AliTPCcalibV0&); // Not implemented
	66	AliTPCcalibV0& operator=(const AliTPCcalibV0&); // Not implemented
	67	TTree *fV0Tree; // tree with full V0 information
	68	TTree *fHPTTree; // tree with high mometa tracks - full calib info
	69	//
	70	AliStack *fStack; // pointer to kinematic tree
	71	AliESDEvent *fESD; //! current ED to proccess - NOT OWNER
	72	TDatabasePDG *fPdg; //! particle database
	73	TObjArray *fParticles; // array of selected MC particles
	74	TObjArray *fV0s; // array of V0s
	75	TObjArray *fGammas; // gamma conversion candidates
	76	//
	77	void Process(AliESDtrack *track, Int_t runNo=-1){AliTPCcalibBase::Process(track,runNo);};
	78	void Process(AliTPCseed *track){return AliTPCcalibBase::Process(track);}
	79	//
	80	ClassDef(AliTPCcalibV0,3);
10757ee9	81	};
	82
	83
	84	#endif