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