[u/mrichter/AliRoot.git] / PWGPP / TRD / AliTRDinfoGen.h

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

/* $Id: AliTRDinfoGen.h 27496 2008-07-22 08:35:45Z cblume $ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  TRD Performance tender wagon                                          //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

class AliESDEvent;
class AliMCEvent;
class AliESDfriend;
class AliTRDtrackInfo;
class AliTRDeventInfo;
class AliTRDv0Info;
class AliTRDeventCuts;
class AliESDtrackCuts;
class AliESDv0KineCuts;
class TObjArray;
class TString;
class TTreeSRedirector;
class AliTRDReconstructor;
class AliTRDgeometry;
class AliTRDinfoGen : public AliAnalysisTaskSE
{
public:
  enum AliTRDinfoGenSteeringBits{
    kMCdata               = BIT(18)
   ,kUseLocalTrkSelection = BIT(19)
   ,kCollision            = BIT(20)
   ,kOCDB                 = BIT(21)
   ,kTrkPoints            = BIT(22)
  };
  enum AliTRDinfoGenObjects{
     kTracksESD =  0
    ,kTracksMC
    ,kV0
    ,kTPC
    ,kITS
    ,kTRDin
    ,kTRDout
    ,kBarrel
    ,kBarrelMC
    ,kSA
    ,kSAMC
    ,kKink
    ,kKinkMC
    ,kBarrelFriend
    ,kSAFriend
    ,kNObjects
  };
  enum AliTRDinfoGenClasses{
     kStatTrk = 0
    ,kEvType
//    ,kBCtrack
    ,kBC
    ,kTrigger
    ,kChmb
    ,kNclasses
  };
  AliTRDinfoGen();
  AliTRDinfoGen(char* name);
  virtual ~AliTRDinfoGen();
  
  static Float_t GetEndITS() { return fgkITS;}
  static Float_t GetEndTPC() { return fgkTPC;}
  static Float_t GetEndTRD() { return fgkTRD;}
  Int_t   GetNRefFigures() const  { return 1;} 
  const char* GetOCDB() const {return fOCDB.Data();}
  Bool_t  GetRefFigure(Int_t ifig);
  Bool_t  Load(const Char_t *fn="AnalysisResults.root", const Char_t *dir="TRD_Performance", const Char_t *name=NULL);

  Bool_t  HasMCdata() const       { return TestBit(kMCdata);};
  // temporary until check with AliAnalysisTaskSE collision selection mechanism
  Bool_t  IsInitOCDB() const {return TestBit(kOCDB);}
  Bool_t  IsCollision() const {return TestBit(kCollision);}
  Bool_t  HasTrackPoints() const {return TestBit(kTrkPoints);}
  void    MakeSummary();
  static const AliTRDReconstructor* Reconstructor() {return fgReconstructor;}
  static AliTRDgeometry*      Geometry() {return fgGeo;}
  void    SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}
  void    SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}
  //void    SetLocalEvSelection(const AliTRDeventCuts */*cut*/){;} 
  void    SetLocalEvSelection(const AliTRDeventCuts &ec);
  //void    SetLocalTrkSelection(const AliESDtrackCuts */*cut*/){;} 
  void    SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}
  void    SetLocalV0Selection(const AliTRDv0Info &v0);
  void    SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}
  void    SetOCDB(const char *ocdb) {fOCDB=ocdb;}

  Bool_t  UseLocalEvSelection() const {return Bool_t(fEventCut);}
  Bool_t  UseLocalTrkSelection() const {return TestBit(kUseLocalTrkSelection);}
  void    UseTrackPoints(Bool_t use=kTRUE) {SetBit(kTrkPoints, use);}
  void    UserCreateOutputObjects();
  void    UserExec(Option_t *);
  void    Terminate(Option_t* option = "");
private:
  // rough radial limits for TRD
  static const Float_t fgkITS;      // end ITS
  static const Float_t fgkTPC;      // end TPC
  static const Float_t fgkTRD;      // end TRD

  // Track selection
  static const Float_t fgkTrkDCAxy; // cm
  static const Float_t fgkTrkDCAz;  // cm
  static const Int_t   fgkNclTPC;   // N clusters TPC
  static const Float_t fgkPt;       // min. pt
  static const Float_t fgkEta;      // eta range
  static AliTRDReconstructor   *fgReconstructor; // single instance of TRD reconstructor used by all tasks
  static AliTRDgeometry        *fgGeo;           // single instance of TRD geometry used by all tasks

  AliTRDinfoGen(const AliTRDinfoGen&);
  AliTRDinfoGen& operator=(const AliTRDinfoGen&);
  TTreeSRedirector* DebugStream();
  void              MakeChambers();

//  UShort_t          fBF;             //! Bunch Fill of previous event
  AliESDEvent      *fESDev;          //! ESD event
  AliMCEvent       *fMCev;           //! MC event
  // event/track cuts OO - to be used
  AliTRDeventCuts  *fEventCut;       // event cut
  AliESDtrackCuts  *fTrackCut;       // track cut
  AliESDv0KineCuts *fV0Identifier;   // V0 identifier
  AliTRDv0Info     *fV0Cut;          // v0 cut
  TString           fOCDB;           // OCDB location
  AliTRDtrackInfo  *fTrackInfo;      //! Track info
  AliTRDeventInfo  *fEventInfo;	     //! Event info
  AliTRDv0Info     *fV0Info;         //! V0 info
  TObjArray        *fTracksBarrel;   //! Array of barrel(TPC) tracks
  TObjArray        *fTracksITS;      //! Array of ITS seeded tracks
  TObjArray        *fTracksSA;       //! Array of stand alone tracks
  TObjArray        *fTracksKink;     //! Array of kink tracks
  TObjArray        *fV0List;         //! V0 container
  TObjArray        *fClusters;       //! Clusters container
  TObjArray        *fContainer;      //! container to store results
  TObjArray        *fRecos;          //! array of reco params
  TTreeSRedirector *fDebugStream;    //! debug stream

  ClassDef(AliTRDinfoGen, 8)         // entry to TRD analysis train
};
#endif
Commit	Line	Data
	1	#ifndef ALITRDINFOGEN_H
	2	#define ALITRDINFOGEN_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id: AliTRDinfoGen.h 27496 2008-07-22 08:35:45Z cblume $ */
	7
	8	////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// TRD Performance tender wagon //
	11	// //
	12	////////////////////////////////////////////////////////////////////////////
	13
	14	#ifndef ALIANALYSISTASKSE_H
	15	#include "AliAnalysisTaskSE.h"
	16	#endif
	17
	18	class AliESDEvent;
	19	class AliMCEvent;
	20	class AliESDfriend;
	21	class AliTRDtrackInfo;
	22	class AliTRDeventInfo;
	23	class AliTRDv0Info;
	24	class AliTRDeventCuts;
	25	class AliESDtrackCuts;
	26	class AliESDv0KineCuts;
	27	class TObjArray;
	28	class TString;
	29	class TTreeSRedirector;
	30	class AliTRDReconstructor;
	31	class AliTRDgeometry;
	32	class AliTRDinfoGen : public AliAnalysisTaskSE
	33	{
	34	public:
	35	enum AliTRDinfoGenSteeringBits{
	36	kMCdata = BIT(18)
	37	,kUseLocalTrkSelection = BIT(19)
	38	,kCollision = BIT(20)
	39	,kOCDB = BIT(21)
	40	,kTrkPoints = BIT(22)
	41	};
	42	enum AliTRDinfoGenObjects{
	43	kTracksESD = 0
	44	,kTracksMC
	45	,kV0
	46	,kTPC
	47	,kITS
	48	,kTRDin
	49	,kTRDout
	50	,kBarrel
	51	,kBarrelMC
	52	,kSA
	53	,kSAMC
	54	,kKink
	55	,kKinkMC
	56	,kBarrelFriend
	57	,kSAFriend
	58	,kNObjects
	59	};
	60	enum AliTRDinfoGenClasses{
	61	kStatTrk = 0
	62	,kEvType
	63	// ,kBCtrack
	64	,kBC
	65	,kTrigger
	66	,kChmb
	67	,kNclasses
	68	};
	69	AliTRDinfoGen();
	70	AliTRDinfoGen(char* name);
	71	virtual ~AliTRDinfoGen();
	72
	73	static Float_t GetEndITS() { return fgkITS;}
	74	static Float_t GetEndTPC() { return fgkTPC;}
	75	static Float_t GetEndTRD() { return fgkTRD;}
	76	Int_t GetNRefFigures() const { return 1;}
	77	const char* GetOCDB() const {return fOCDB.Data();}
	78	Bool_t GetRefFigure(Int_t ifig);
	79	Bool_t Load(const Char_t fn="AnalysisResults.root", const Char_t dir="TRD_Performance", const Char_t *name=NULL);
	80
	81	Bool_t HasMCdata() const { return TestBit(kMCdata);};
	82	// temporary until check with AliAnalysisTaskSE collision selection mechanism
	83	Bool_t IsInitOCDB() const {return TestBit(kOCDB);}
	84	Bool_t IsCollision() const {return TestBit(kCollision);}
	85	Bool_t HasTrackPoints() const {return TestBit(kTrkPoints);}
	86	void MakeSummary();
	87	static const AliTRDReconstructor* Reconstructor() {return fgReconstructor;}
	88	static AliTRDgeometry* Geometry() {return fgGeo;}
	89	void SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}
	90	void SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}
	91	//void SetLocalEvSelection(const AliTRDeventCuts /cut*/){;}
	92	void SetLocalEvSelection(const AliTRDeventCuts &ec);
	93	//void SetLocalTrkSelection(const AliESDtrackCuts /cut*/){;}
	94	void SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}
	95	void SetLocalV0Selection(const AliTRDv0Info &v0);
	96	void SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}
	97	void SetOCDB(const char *ocdb) {fOCDB=ocdb;}
	98
	99	Bool_t UseLocalEvSelection() const {return Bool_t(fEventCut);}
	100	Bool_t UseLocalTrkSelection() const {return TestBit(kUseLocalTrkSelection);}
	101	void UseTrackPoints(Bool_t use=kTRUE) {SetBit(kTrkPoints, use);}
	102	void UserCreateOutputObjects();
	103	void UserExec(Option_t *);
	104	void Terminate(Option_t* option = "");
	105	private:
	106	// rough radial limits for TRD
	107	static const Float_t fgkITS; // end ITS
	108	static const Float_t fgkTPC; // end TPC
	109	static const Float_t fgkTRD; // end TRD
	110
	111	// Track selection
	112	static const Float_t fgkTrkDCAxy; // cm
	113	static const Float_t fgkTrkDCAz; // cm
	114	static const Int_t fgkNclTPC; // N clusters TPC
	115	static const Float_t fgkPt; // min. pt
	116	static const Float_t fgkEta; // eta range
	117	static AliTRDReconstructor *fgReconstructor; // single instance of TRD reconstructor used by all tasks
	118	static AliTRDgeometry *fgGeo; // single instance of TRD geometry used by all tasks
	119
	120	AliTRDinfoGen(const AliTRDinfoGen&);
	121	AliTRDinfoGen& operator=(const AliTRDinfoGen&);
	122	TTreeSRedirector* DebugStream();
	123	void MakeChambers();
	124
	125	// UShort_t fBF; //! Bunch Fill of previous event
	126	AliESDEvent *fESDev; //! ESD event
	127	AliMCEvent *fMCev; //! MC event
	128	// event/track cuts OO - to be used
	129	AliTRDeventCuts *fEventCut; // event cut
	130	AliESDtrackCuts *fTrackCut; // track cut
	131	AliESDv0KineCuts *fV0Identifier; // V0 identifier
	132	AliTRDv0Info *fV0Cut; // v0 cut
	133	TString fOCDB; // OCDB location
	134	AliTRDtrackInfo *fTrackInfo; //! Track info
	135	AliTRDeventInfo *fEventInfo; //! Event info
	136	AliTRDv0Info *fV0Info; //! V0 info
	137	TObjArray *fTracksBarrel; //! Array of barrel(TPC) tracks
	138	TObjArray *fTracksITS; //! Array of ITS seeded tracks
	139	TObjArray *fTracksSA; //! Array of stand alone tracks
	140	TObjArray *fTracksKink; //! Array of kink tracks
	141	TObjArray *fV0List; //! V0 container
	142	TObjArray *fClusters; //! Clusters container
	143	TObjArray *fContainer; //! container to store results
	144	TObjArray *fRecos; //! array of reco params
	145	TTreeSRedirector *fDebugStream; //! debug stream
	146
	147	ClassDef(AliTRDinfoGen, 8) // entry to TRD analysis train
	148	};
	149	#endif