[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSURecoDet.h

#ifndef ALIITSURECODET
#define ALIITSURECODET

#include <TNamed.h>
#include <TObjArray.h>
#include "AliITSURecoLayer.h"
class AliITSUGeomTGeo;
class TTree;

///////////////////////////////////////////////////////////////////////
//                                                                   //
//  Class AliITSURecoDet                                             //
//  Interface between the framework and reconstruction for ITS       //
//                                                                   //
///////////////////////////////////////////////////////////////////////


class AliITSURecoDet : public TNamed
{
 public:
  //
  AliITSURecoDet();
  AliITSURecoDet(AliITSUGeomTGeo* geom, const char* name="");
  virtual ~AliITSURecoDet();
  //
  Double_t           GetRMin()                     const {return fRMin;}
  Double_t           GetRMax()                     const {return fRMax;}
  Double_t           GetRITSTPCRef()               const {return fRITSTPCRef;}
  Int_t              GetNLayers()                  const {return fNLayers;}
  Int_t              GetNLayersActive()            const {return fNLayersActive;}
  Int_t              GetLrIDActive(Int_t lrActID)  const;
  Int_t              FindLastLayerID(Double_t r, int dir)  const;
  Int_t              FindFirstLayerID(Double_t r, int dir) const;
  AliITSURecoLayer*  GetLayer(Int_t i)             const;
  AliITSURecoLayer*  GetLayerActive(Int_t i)       const;
  AliITSUGeomTGeo*   GetGeom()                     const {return fGeom;}
  //
  void               SetRMin(Double_t r)                 {fRMin = r;}
  void               SetRMax(Double_t r)                 {fRMax = r;}
  void               SetRITSTPCRef(Double_t r)           {fRITSTPCRef = r;}
  //
  void               AddLayer(const AliITSURecoLayer* lr);
  //
  void               ProcessClusters(Int_t mode=0);
  void               CreateClusterArrays();
  Int_t              LoadClusters(TTree* treeRP);
  //
  virtual void       Print(Option_t* option = "")  const;
  //
 protected:
  Bool_t             Build();
  void               IndexLayers();
  //
 protected:
  Int_t              fNLayers;        // total number of layers
  Int_t              fNLayersActive;  // N of active layers
  Double_t           fRMax;           // max  R
  Double_t           fRMin;           // min  R
  Double_t           fRITSTPCRef;     // reference radius for ITS/TPC matching check
  TObjArray          fLayers;         // layers
  TObjArray          fLayersActive;   // active layers
  AliITSUGeomTGeo*   fGeom;           // ITS geometry
  //
 protected:
  static const Char_t*     fgkBeamPipeVolName;    // name of the beam pipe volume

 private:
  AliITSURecoDet(const AliITSURecoDet &source); 
  AliITSURecoDet& operator=(const AliITSURecoDet &source); 
  //
  ClassDef(AliITSURecoDet,1); // helper for ITS data in reco
};


//_____________________________________________________________
inline Int_t AliITSURecoDet::GetLrIDActive(Int_t lrActID) const 
{
  // get global layer id from active id
  return (lrActID<fNLayersActive) ? ((AliITSURecoLayer*)fLayersActive.UncheckedAt(lrActID))->GetID() 
    : GetLayerActive(fNLayersActive-1)->GetID()+1;
}

//_____________________________________________________________
inline AliITSURecoLayer* AliITSURecoDet::GetLayer(Int_t i) const 
{
  // get layer with global id=i
  return i>=0&&i<fNLayers ? (AliITSURecoLayer*)fLayers.UncheckedAt(i):0;
}

//_____________________________________________________________
inline AliITSURecoLayer* AliITSURecoDet::GetLayerActive(Int_t i) const
{
  // get layer with activeID=i
  return i>=0&&i<fNLayersActive ? (AliITSURecoLayer*)fLayersActive.UncheckedAt(i):0;
}

//______________________________________________________
inline void AliITSURecoDet::ProcessClusters(Int_t mode)
{
  // prepare clsuters for reconstrunction
  for (int ilr=fNLayersActive;ilr--;) GetLayerActive(ilr)->ProcessClusters(mode);
}

#endif
Commit	Line	Data
a11ef2e4	1	#ifndef ALIITSURECODET
	2	#define ALIITSURECODET
	3
	4	#include <TNamed.h>
	5	#include <TObjArray.h>
	6	#include "AliITSURecoLayer.h"
	7	class AliITSUGeomTGeo;
ee52e7b5	8	class TTree;
a11ef2e4	9
	10	///////////////////////////////////////////////////////////////////////
	11	// //
	12	// Class AliITSURecoDet //
	13	// Interface between the framework and reconstruction for ITS //
	14	// //
	15	///////////////////////////////////////////////////////////////////////
	16
	17
	18	class AliITSURecoDet : public TNamed
	19	{
	20	public:
	21	//
6121e5c5	22	AliITSURecoDet();
32d38de2	23	AliITSURecoDet(AliITSUGeomTGeo* geom, const char* name="");
a11ef2e4	24	virtual ~AliITSURecoDet();
	25	//
	26	Double_t GetRMin() const {return fRMin;}
	27	Double_t GetRMax() const {return fRMax;}
70cb7fe4	28	Double_t GetRITSTPCRef() const {return fRITSTPCRef;}
a11ef2e4	29	Int_t GetNLayers() const {return fNLayers;}
	30	Int_t GetNLayersActive() const {return fNLayersActive;}
	31	Int_t GetLrIDActive(Int_t lrActID) const;
8b16dbae	32	Int_t FindLastLayerID(Double_t r, int dir) const;
8b16dbae	33	Int_t FindFirstLayerID(Double_t r, int dir) const;
a11ef2e4	34	AliITSURecoLayer* GetLayer(Int_t i) const;
a11ef2e4	35	AliITSURecoLayer* GetLayerActive(Int_t i) const;
32d38de2	36	AliITSUGeomTGeo* GetGeom() const {return fGeom;}
a11ef2e4	37	//
	38	void SetRMin(Double_t r) {fRMin = r;}
	39	void SetRMax(Double_t r) {fRMax = r;}
70cb7fe4	40	void SetRITSTPCRef(Double_t r) {fRITSTPCRef = r;}
a11ef2e4	41	//
a11ef2e4	42	void AddLayer(const AliITSURecoLayer* lr);
32d38de2	43	//
32d38de2	44	void ProcessClusters(Int_t mode=0);
ee52e7b5	45	void CreateClusterArrays();
ee52e7b5	46	Int_t LoadClusters(TTree* treeRP);
a11ef2e4	47	//
	48	virtual void Print(Option_t* option = "") const;
	49	//
32d38de2	50	protected:
32d38de2	51	Bool_t Build();
dde91d5d	52	void IndexLayers();
32d38de2	53	//
a11ef2e4	54	protected:
	55	Int_t fNLayers; // total number of layers
	56	Int_t fNLayersActive; // N of active layers
	57	Double_t fRMax; // max R
	58	Double_t fRMin; // min R
70cb7fe4	59	Double_t fRITSTPCRef; // reference radius for ITS/TPC matching check
a11ef2e4	60	TObjArray fLayers; // layers
a11ef2e4	61	TObjArray fLayersActive; // active layers
32d38de2	62	AliITSUGeomTGeo* fGeom; // ITS geometry
a11ef2e4	63	//
173b3073	64	protected:
	65	static const Char_t* fgkBeamPipeVolName; // name of the beam pipe volume
	66
a11ef2e4	67	private:
	68	AliITSURecoDet(const AliITSURecoDet &source);
	69	AliITSURecoDet& operator=(const AliITSURecoDet &source);
	70	//
	71	ClassDef(AliITSURecoDet,1); // helper for ITS data in reco
	72	};
	73
	74
	75
	76	//_____________________________________________________________
	77	inline Int_t AliITSURecoDet::GetLrIDActive(Int_t lrActID) const
	78	{
	79	// get global layer id from active id
70cb7fe4	80	return (lrActID<fNLayersActive) ? ((AliITSURecoLayer*)fLayersActive.UncheckedAt(lrActID))->GetID()
70cb7fe4	81	: GetLayerActive(fNLayersActive-1)->GetID()+1;
a11ef2e4	82	}
	83
	84	//_____________________________________________________________
	85	inline AliITSURecoLayer* AliITSURecoDet::GetLayer(Int_t i) const
	86	{
	87	// get layer with global id=i
8b16dbae	88	return i>=0&&i<fNLayers ? (AliITSURecoLayer*)fLayers.UncheckedAt(i):0;
a11ef2e4	89	}
	90
	91	//_____________________________________________________________
	92	inline AliITSURecoLayer* AliITSURecoDet::GetLayerActive(Int_t i) const
	93	{
	94	// get layer with activeID=i
8b16dbae	95	return i>=0&&i<fNLayersActive ? (AliITSURecoLayer*)fLayersActive.UncheckedAt(i):0;
a11ef2e4	96	}
a11ef2e4	97
32d38de2	98	//______________________________________________________
	99	inline void AliITSURecoDet::ProcessClusters(Int_t mode)
	100	{
	101	// prepare clsuters for reconstrunction
	102	for (int ilr=fNLayersActive;ilr--;) GetLayerActive(ilr)->ProcessClusters(mode);
	103	}
a11ef2e4	104
a11ef2e4	105	#endif