[u/mrichter/AliRoot.git] / PWG1 / AliGenInfoMaker.h

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


//////////////////////////////////////////////////////////////////////////////
//                          Class AliGenInfoMaker                           //
//   collect together MC info for comparison purposes - effieciency studies and so on//                                                                 //
//   marian.ivanov@cern.ch                                                  //
//////////////////////////////////////////////////////////////////////////////


#include <TParticle.h>
#include "AliAnalysisTask.h"
#include "AliTrackReference.h"

class TFile;
class AliRunLoader;
class AliStack;
class AliTPCParam;
class AliMCEventHandler;
class AliMCInfo;

////////////////////////////////////////////////////////////////////////
// 
// Start of implementation of the class AliGenInfoMaker
//
////////////////////////////////////////////////////////////////////////

class AliGenInfoMaker : public TObject {

public:
  AliGenInfoMaker();
  virtual ~AliGenInfoMaker();
  // 
  //
  AliGenInfoMaker(const char * fnGalice, const char* fnRes,
		   Int_t nEvents=1, Int_t firstEvent=0);
  //event by event function - used in the analysis task
  Int_t ProcessEvent(AliMCEventHandler* mcinfo);

  Int_t ProcessEvent();   // process event
  Int_t TreeKLoop();      // process kinamatics
  Int_t TreeTRLoop();     // process track refereces
  Int_t TreeDLoop();      // process digits tree
  Int_t BuildKinkInfo();  // build information about MC kinks
  Int_t BuildV0Info();    // build information about MC kinks
  //
  //
  Int_t Exec();  
  void CreateTreeGenTracks();
  void CloseOutputFile();
  void SetFirstEventNr(Int_t i) {fFirstEventNr = i;}
  void SetNEvents(Int_t i) {fNEvents = i;}
  void SetDebug(Int_t level) {fDebug = level;}
  Int_t SetIO(Int_t eventNr);
  Int_t CloseIOEvent();
  Int_t CloseIO();
  Int_t SetIO();

protected:
  AliGenInfoMaker(const AliGenInfoMaker& /*info*/);
  AliGenInfoMaker& operator=(const AliGenInfoMaker& /*info*/) { return *this;}

  AliMCInfo * MakeInfo(UInt_t i);
  AliMCInfo * GetInfo(UInt_t i) const {return (i<fNParticles)? fGenInfo[i]:0;}
  Float_t TR2LocalX(AliTrackReference *trackRef,
		    AliTPCParam *paramTPC) const;
  AliTPCParam * GetTPCParam();
  //
  TObjArray *fGenTracksArray;  //clones array with filtered particles
  TObjArray *fGenKinkArray;    //clones array with filtered Kinks
  TObjArray *fGenV0Array;      //clones array with filtered V0s
  //
  Int_t  fDebug;                   //! debug flag  
  Int_t  fEventNr;                 //! current event number
  Int_t  fLabel;                   //! track label
  Int_t  fNEvents;                 //! number of events to process
  Int_t  fFirstEventNr;            //! first event to process
  UInt_t fNParticles;              //! number of particles in TreeK
  TTree *fTreeGenTracks;          //! output tree with generated tracks
  TTree *fTreeKinks;             //!  output tree with Kinks
  TTree *fTreeV0;                //!  output tree with V0
  char   fFnRes[1000];             //! output file name with stored tracks
  TFile *fFileGenTracks;          //! output file with stored fTreeGenTracks
  //
  AliRunLoader * fLoader;         //! pointer to the run loader
  TTree * fTreeD;                 //! current tree with digits
  TTree * fTreeTR;                //! current tree with TR
  AliStack *fStack;               //! current stack
  // 
  AliMCInfo **   fGenInfo;    //! array with pointers to gen info
  Int_t   fNInfos;                  //! number of tracks with infos
  //
  AliTPCParam* fParamTPC;         //! AliTPCParam
  Float_t  fVPrim[3];             //! primary vertex position                                  // the fVDist[3] contains size of the 3-vector
  // cuts
  //
  Double_t fTPCPtCut; // do not store particles with generated pT less than this
  Double_t fITSPtCut; // do not store particles with generated pT less than this
  Double_t fTRDPtCut; // do not store particles with generated pT less than this
  Double_t fTOFPtCut; // do not store particles with generated pT less than this
 
  ClassDef(AliGenInfoMaker,0)    // class which creates and fills tree with TPCGenTrack objects
};


#endif
Commit	Line	Data
d92975ba	1	#ifndef ALIGENINFOMAKER_H
	2	#define ALIGENINFOMAKER_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6
	7
	8	//////////////////////////////////////////////////////////////////////////////
d390cc7e	9	// Class AliGenInfoMaker //
d92975ba	10	// collect together MC info for comparison purposes - effieciency studies and so on// //
	11	// marian.ivanov@cern.ch //
	12	//////////////////////////////////////////////////////////////////////////////
	13
	14
	15
d92975ba	16
d92975ba	17	#include <TParticle.h>
d390cc7e	18	#include "AliAnalysisTask.h"
d92975ba	19	#include "AliTrackReference.h"
	20
	21	class TFile;
	22	class AliRunLoader;
	23	class AliStack;
	24	class AliTPCParam;
d390cc7e	25	class AliMCEventHandler;
d390cc7e	26	class AliMCInfo;
d92975ba	27
	28	////////////////////////////////////////////////////////////////////////
	29	//
	30	// Start of implementation of the class AliGenInfoMaker
	31	//
	32	////////////////////////////////////////////////////////////////////////
	33
d390cc7e	34	class AliGenInfoMaker : public TObject {
d92975ba	35
	36	public:
	37	AliGenInfoMaker();
d92975ba	38	virtual ~AliGenInfoMaker();
d390cc7e	39	//
	40	//
	41	AliGenInfoMaker(const char * fnGalice, const char* fnRes,
	42	Int_t nEvents=1, Int_t firstEvent=0);
	43	//event by event function - used in the analysis task
	44	Int_t ProcessEvent(AliMCEventHandler* mcinfo);
	45
	46	Int_t ProcessEvent(); // process event
	47	Int_t TreeKLoop(); // process kinamatics
	48	Int_t TreeTRLoop(); // process track refereces
	49	Int_t TreeDLoop(); // process digits tree
	50	Int_t BuildKinkInfo(); // build information about MC kinks
	51	Int_t BuildV0Info(); // build information about MC kinks
	52	//
	53	//
	54	Int_t Exec();
d92975ba	55	void CreateTreeGenTracks();
d92975ba	56	void CloseOutputFile();
d92975ba	57	void SetFirstEventNr(Int_t i) {fFirstEventNr = i;}
	58	void SetNEvents(Int_t i) {fNEvents = i;}
	59	void SetDebug(Int_t level) {fDebug = level;}
	60	Int_t SetIO(Int_t eventNr);
	61	Int_t CloseIOEvent();
	62	Int_t CloseIO();
	63	Int_t SetIO();
d390cc7e	64
d390cc7e	65	protected:
cd875161	66	AliGenInfoMaker(const AliGenInfoMaker& /info/);
	67	AliGenInfoMaker& operator=(const AliGenInfoMaker& /info/) { return *this;}
	68
d390cc7e	69	AliMCInfo * MakeInfo(UInt_t i);
d390cc7e	70	AliMCInfo * GetInfo(UInt_t i) const {return (i<fNParticles)? fGenInfo[i]:0;}
d92975ba	71	Float_t TR2LocalX(AliTrackReference *trackRef,
d92975ba	72	AliTPCParam *paramTPC) const;
d92975ba	73	AliTPCParam * GetTPCParam();
d390cc7e	74	//
	75	TObjArray *fGenTracksArray; //clones array with filtered particles
	76	TObjArray *fGenKinkArray; //clones array with filtered Kinks
	77	TObjArray *fGenV0Array; //clones array with filtered V0s
	78	//
d92975ba	79	Int_t fDebug; //! debug flag
	80	Int_t fEventNr; //! current event number
	81	Int_t fLabel; //! track label
	82	Int_t fNEvents; //! number of events to process
	83	Int_t fFirstEventNr; //! first event to process
	84	UInt_t fNParticles; //! number of particles in TreeK
	85	TTree *fTreeGenTracks; //! output tree with generated tracks
	86	TTree *fTreeKinks; //! output tree with Kinks
	87	TTree *fTreeV0; //! output tree with V0
d92975ba	88	char fFnRes[1000]; //! output file name with stored tracks
	89	TFile *fFileGenTracks; //! output file with stored fTreeGenTracks
	90	//
	91	AliRunLoader * fLoader; //! pointer to the run loader
	92	TTree * fTreeD; //! current tree with digits
	93	TTree * fTreeTR; //! current tree with TR
	94	AliStack *fStack; //! current stack
	95	//
	96	AliMCInfo ** fGenInfo; //! array with pointers to gen info
	97	Int_t fNInfos; //! number of tracks with infos
	98	//
	99	AliTPCParam* fParamTPC; //! AliTPCParam
d390cc7e	100	Float_t fVPrim[3]; //! primary vertex position // the fVDist[3] contains size of the 3-vector
d92975ba	101	// cuts
	102	//
	103	Double_t fTPCPtCut; // do not store particles with generated pT less than this
	104	Double_t fITSPtCut; // do not store particles with generated pT less than this
	105	Double_t fTRDPtCut; // do not store particles with generated pT less than this
	106	Double_t fTOFPtCut; // do not store particles with generated pT less than this
	107
	108	ClassDef(AliGenInfoMaker,0) // class which creates and fills tree with TPCGenTrack objects
	109	};
	110
	111
	112
	113
	114
	115
	116	#endif