[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h

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

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD cluster                                                              //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliCluster.h"

class AliTRDcluster : public AliCluster {
public:
  enum ETRDclusterStatus { 
    kInChamber = BIT(16) // Out of fiducial volume of chamber (signal tails)
   ,kFivePad   = BIT(17) // Deconvoluted clusters
   ,kLUT       = BIT(18)
   ,kGAUS      = BIT(19)
   ,kCOG       = BIT(20)
  };
  enum ETRDclusterMask { 
    kMaskedLeft   = 0
   ,kMaskedCenter = 1
   ,kMaskedRight  = 2
  };

  AliTRDcluster();
  AliTRDcluster(Int_t det, UChar_t col, UChar_t row, UChar_t time, const Short_t *sig, UShort_t volid);
  AliTRDcluster(
    Int_t det, Float_t q, Float_t *pos, Float_t *sig,
    Int_t *tracks, Char_t npads, Short_t *signals,
    UChar_t col, UChar_t row, UChar_t time,
    Char_t timebin, Float_t center, UShort_t volid);
  AliTRDcluster(const AliTRDcluster &c);
  virtual ~AliTRDcluster() {};

  virtual void     AddTrackIndex(Int_t *i); 
  void     Clear(Option_t *o="");
  
  Bool_t   IsEqual(const TObject *otherCluster) const;
  Bool_t   IsInChamber() const             { return TestBit(kInChamber);       }
  Bool_t   IsMasked() const                { return fClusterMasking ? kTRUE : kFALSE; }
  Bool_t   IsShared() const                { return IsClusterShared();}
  Bool_t   IsUsed() const                  { return IsClusterUsed(); }
  Bool_t   IsFivePad() const               { return TestBit(kFivePad);}
  inline Bool_t IsRPhiMethod(ETRDclusterStatus m) const;

  UChar_t  GetPadMaskedPosition() const    { return fClusterMasking & 7; }
  UChar_t  GetPadMaskedStatus() const      { return fClusterMasking >> 3; }
  Int_t    GetDetector() const             { return fDetector;      }
  Int_t    GetLocalTimeBin() const         { return fLocalTimeBin;  }
  Float_t  GetQ() const                    { return fQ;             }
  Int_t    GetNPads() const                { return fNPads;         }
  Float_t  GetCenter() const               { return fCenter;        }
  Int_t    GetPadCol() const               { return fPadCol;        }
  Int_t    GetPadRow() const               { return fPadRow;        }
  Int_t    GetPadTime() const              { return fPadTime;       }
  Short_t *GetSignals()                    { return fSignals;       }
  Float_t  GetSumS() const;
  static Double_t  GetSX(Int_t tb, Double_t z=-1);
  static Double_t  GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
  static Double_t  GetSYcharge(Float_t q);
  static Double_t  GetSYprf(Int_t ly, Double_t center, Double_t s2);
  static Double_t  GetXcorr(Int_t tb, Double_t z=-1);
  static Double_t  GetYcorr(Int_t ly, Float_t y);
  Float_t  GetXloc(Double_t t0, Double_t vd, Double_t *const q=0x0, Double_t *const xq=0x0, Double_t z = 0.2);
  Float_t  GetYloc(Double_t y0, Double_t s2, Double_t W, Double_t *const y1=0x0, Double_t *const y2=0x0);

  void     Print(Option_t* o="") const;

  void     SetLocalTimeBin(Char_t t)       { fLocalTimeBin = t;     }
  void     SetInChamber(Bool_t in = kTRUE) { SetBit(kInChamber,in); }
  void     SetNPads(Int_t n)               { fNPads = n; }
  void     SetPadMaskedPosition(UChar_t position);
  void     SetPadMaskedStatus(UChar_t status);
  void     SetPadCol(UChar_t inPadCol){ fPadCol = inPadCol;}
  void     SetPadRow(UChar_t inPadRow){ fPadRow = inPadRow;}
  void     SetPadTime(UChar_t inPadTime){ fPadTime = inPadTime;}
  inline void SetRPhiMethod(ETRDclusterStatus m);
  void     SetDetector(Short_t inDetector){ fDetector = inDetector;}
  void     SetQ(Float_t inQ){ fQ = inQ;}
  void     SetClusterMasking(UChar_t inClusterMasking){ fClusterMasking = inClusterMasking;}
  void     SetShared(Bool_t sh  = kTRUE)   { SetBit(AliCluster::kShared,sh);    }
  void     SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
  void     Use(Int_t u = 1)                  { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE);              }
  void     SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b);}

protected:
  UChar_t fPadCol;         //  Central pad number in column direction
  UChar_t fPadRow;         //  Central pad number in row direction
  UChar_t fPadTime;        //  Uncalibrated time bin number
  Char_t  fLocalTimeBin;   //  T0-calibrated time bin number
  UChar_t fNPads;          //  Number of pads in cluster
  UChar_t fClusterMasking; //  Bit field containing cluster status information;
  Short_t fDetector;       //  TRD detector number
  Short_t fSignals[7];     //  Signals in the cluster
  Float_t fQ;              //  Amplitude 
  Float_t fCenter;         //  Center of the cluster relative to the pad 

private:
  Float_t  GetDYcog(Double_t *const y1=0x0, Double_t *const y2=0x0);
  Float_t  GetDYlut(Double_t *const y1=0x0, Double_t *const y2=0x0);
  Float_t  GetDYgauss(Double_t sw, Double_t *const y1=0x0, Double_t *const y2=0x0);
  static void      FillLUT();

  static const Int_t   fgkNlut;              //!  Number of bins of the LUT
  static Double_t     *fgLUT;                //! The lookup table

  ClassDef(AliTRDcluster, 7)        //  Cluster for the TRD
};

//________________________________________________
inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
{
  if(m==kLUT && TestBit(kLUT)) return kTRUE;
  else if(m==kGAUS && TestBit(kGAUS)) return kTRUE;
  else if(m==kCOG && (!TestBit(kLUT)&&!TestBit(kGAUS))) return kTRUE;

  return kFALSE;
}

//________________________________________________
inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
{
  SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
  switch(m){
  case kCOG: SetBit(kCOG); break;
  case kLUT: SetBit(kLUT); break;
  case kGAUS: SetBit(kGAUS); break;
  default: SetBit(kLUT); break;
  }
}

#endif
Commit	Line	Data
46d29e70	1	#ifndef ALITRDCLUSTER_H
	2	#define ALITRDCLUSTER_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
6d50f529	8	///////////////////////////////////////////////////////////////////////////////
	9	// //
	10	// TRD cluster //
	11	// //
	12	///////////////////////////////////////////////////////////////////////////////
5443e65e	13
5843e420	14	#include "AliCluster.h"
46d29e70	15
5443e65e	16	class AliTRDcluster : public AliCluster {
5843e420	17	public:
5843e420	18	enum ETRDclusterStatus {
	19	kInChamber = BIT(16) // Out of fiducial volume of chamber (signal tails)
	20	,kFivePad = BIT(17) // Deconvoluted clusters
b72f4eaf	21	,kLUT = BIT(18)
	22	,kGAUS = BIT(19)
	23	,kCOG = BIT(20)
5843e420	24	};
	25	enum ETRDclusterMask {
	26	kMaskedLeft = 0
	27	,kMaskedCenter = 1
	28	,kMaskedRight = 2
f5375dcb	29	};
0e09df31	30
bdb68f8c	31	AliTRDcluster();
b72f4eaf	32	AliTRDcluster(Int_t det, UChar_t col, UChar_t row, UChar_t time, const Short_t *sig, UShort_t volid);
203967fc	33	AliTRDcluster(
	34	Int_t det, Float_t q, Float_t pos, Float_t sig,
	35	Int_t tracks, Char_t npads, Short_t signals,
	36	UChar_t col, UChar_t row, UChar_t time,
	37	Char_t timebin, Float_t center, UShort_t volid);
a2b90f83	38	AliTRDcluster(const AliTRDcluster &c);
acc49af9	39	virtual ~AliTRDcluster() {};
a2b90f83	40
6d50f529	41	virtual void AddTrackIndex(Int_t *i);
203967fc	42	void Clear(Option_t *o="");
	43
	44	Bool_t IsEqual(const TObject *otherCluster) const;
	45	Bool_t IsInChamber() const { return TestBit(kInChamber); }
	46	Bool_t IsMasked() const { return fClusterMasking ? kTRUE : kFALSE; }
de777ccb	47	Bool_t IsShared() const { return IsClusterShared();}
de777ccb	48	Bool_t IsUsed() const { return IsClusterUsed(); }
5843e420	49	Bool_t IsFivePad() const { return TestBit(kFivePad);}
b72f4eaf	50	inline Bool_t IsRPhiMethod(ETRDclusterStatus m) const;
a2b90f83	51
203967fc	52	UChar_t GetPadMaskedPosition() const { return fClusterMasking & 7; }
	53	UChar_t GetPadMaskedStatus() const { return fClusterMasking >> 3; }
	54	Int_t GetDetector() const { return fDetector; }
	55	Int_t GetLocalTimeBin() const { return fLocalTimeBin; }
	56	Float_t GetQ() const { return fQ; }
	57	Int_t GetNPads() const { return fNPads; }
	58	Float_t GetCenter() const { return fCenter; }
	59	Int_t GetPadCol() const { return fPadCol; }
	60	Int_t GetPadRow() const { return fPadRow; }
	61	Int_t GetPadTime() const { return fPadTime; }
	62	Short_t *GetSignals() { return fSignals; }
	63	Float_t GetSumS() const;
b72f4eaf	64	static Double_t GetSX(Int_t tb, Double_t z=-1);
1fd9389f	65	static Double_t GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
	66	static Double_t GetSYcharge(Float_t q);
	67	static Double_t GetSYprf(Int_t ly, Double_t center, Double_t s2);
b72f4eaf	68	static Double_t GetXcorr(Int_t tb, Double_t z=-1);
b72f4eaf	69	static Double_t GetYcorr(Int_t ly, Float_t y);
834ac2c9	70	Float_t GetXloc(Double_t t0, Double_t vd, Double_t const q=0x0, Double_t const xq=0x0, Double_t z = 0.2);
b72f4eaf	71	Float_t GetYloc(Double_t y0, Double_t s2, Double_t W, Double_t const y1=0x0, Double_t const y2=0x0);
5843e420	72
203967fc	73	void Print(Option_t* o="") const;
f5375dcb	74
203967fc	75	void SetLocalTimeBin(Char_t t) { fLocalTimeBin = t; }
203967fc	76	void SetInChamber(Bool_t in = kTRUE) { SetBit(kInChamber,in); }
b72f4eaf	77	void SetNPads(Int_t n) { fNPads = n; }
203967fc	78	void SetPadMaskedPosition(UChar_t position);
203967fc	79	void SetPadMaskedStatus(UChar_t status);
bee2b41e	80	void SetPadCol(UChar_t inPadCol){ fPadCol = inPadCol;}
	81	void SetPadRow(UChar_t inPadRow){ fPadRow = inPadRow;}
	82	void SetPadTime(UChar_t inPadTime){ fPadTime = inPadTime;}
b72f4eaf	83	inline void SetRPhiMethod(ETRDclusterStatus m);
bee2b41e	84	void SetDetector(Short_t inDetector){ fDetector = inDetector;}
	85	void SetQ(Float_t inQ){ fQ = inQ;}
	86	void SetClusterMasking(UChar_t inClusterMasking){ fClusterMasking = inClusterMasking;}
de777ccb	87	void SetShared(Bool_t sh = kTRUE) { SetBit(AliCluster::kShared,sh); }
1fd9389f	88	void SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
07bbc13c	89	void Use(Int_t u = 1) { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE); }
5843e420	90	void SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b);}
af26ce80	91
5843e420	92	protected:
	93	UChar_t fPadCol; // Central pad number in column direction
	94	UChar_t fPadRow; // Central pad number in row direction
	95	UChar_t fPadTime; // Uncalibrated time bin number
	96	Char_t fLocalTimeBin; // T0-calibrated time bin number
	97	UChar_t fNPads; // Number of pads in cluster
	98	UChar_t fClusterMasking; // Bit field containing cluster status information;
	99	Short_t fDetector; // TRD detector number
	100	Short_t fSignals[7]; // Signals in the cluster
	101	Float_t fQ; // Amplitude
	102	Float_t fCenter; // Center of the cluster relative to the pad
0e09df31	103
b72f4eaf	104	private:
	105	Float_t GetDYcog(Double_t const y1=0x0, Double_t const y2=0x0);
	106	Float_t GetDYlut(Double_t const y1=0x0, Double_t const y2=0x0);
	107	Float_t GetDYgauss(Double_t sw, Double_t const y1=0x0, Double_t const y2=0x0);
	108	static void FillLUT();
	109
	110	static const Int_t fgkNlut; //! Number of bins of the LUT
	111	static Double_t *fgLUT; //! The lookup table
	112
	113	ClassDef(AliTRDcluster, 7) // Cluster for the TRD
46d29e70	114	};
b72f4eaf	115
	116	//________________________________________________
	117	inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
	118	{
	119	if(m==kLUT && TestBit(kLUT)) return kTRUE;
	120	else if(m==kGAUS && TestBit(kGAUS)) return kTRUE;
	121	else if(m==kCOG && (!TestBit(kLUT)&&!TestBit(kGAUS))) return kTRUE;
	122
	123	return kFALSE;
	124	}
	125
	126	//________________________________________________
	127	inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
	128	{
a8d5c885	129	SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
b72f4eaf	130	switch(m){
	131	case kCOG: SetBit(kCOG); break;
	132	case kLUT: SetBit(kLUT); break;
	133	case kGAUS: SetBit(kGAUS); break;
a8d5c885	134	default: SetBit(kLUT); break;
b72f4eaf	135	}
	136	}
	137
46d29e70	138	#endif