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

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

/* $Id$ */

////////////////////////////////////////////////////////////////////////////
//                                                                        //
//  TRD cluster finder                                                    //
//                                                                        //
////////////////////////////////////////////////////////////////////////////

#include <TNamed.h>

#include "AliTRDgeometry.h"

class TFile;
class TTree;
class TClonesArray;

class AliRunLoader;
class AliRawReader;

class AliTRD;
class AliTRDcluster;

class AliTRDrawStream;
class AliTRDarrayADC;
class AliTRDarraySignal;
class AliTRDdigitsManager;
class AliTRDSignalIndex;
class AliTRDtransform;
class AliTRDCalROC;
class AliTRDReconstructor;
class AliTRDCalSingleChamberStatus;
class AliTRDrecoParam;
class AliTRDCalOnlineGainTableROC;

class AliTRDclusterizer : public TNamed 
{

 public:

  // steering flags
  // bits from 0-13 are reserved by ROOT (see TObject.h)
  enum{
    kTrOwner = BIT(14)  //  toggle online tracklets ownership
    ,kClOwner= BIT(15)  //  toggle cluster ownership
    ,kLabels = BIT(16)  //  toggle MC labels for clusters
    ,kSkipTrafo = BIT(17)  //  skip the coordinate transformation of clusters?
    ,kLUT    = BIT(18)  //  using look up table for cluster's r-phi position
    ,kGAUS   = BIT(19)  //  using gauss approx. for cluster's r-phi position
    ,knewDM  = BIT(20)  //  was the digitsmanger created by raw2clusters?
    ,kTracksOwner = BIT(21) //  toggle GTU track ownership
    ,kRawSignal = BIT(22) //  toggle storage of raw signals in clusters
  };

  struct MaxStruct
  {
    Int_t       row;           // row of the current cluster candidate
    Int_t       col;           // col of the current cluster candidate
    Int_t       time;          // time -"-
    Float_t     signals[3];    // signals of the maximum pad and it's twon neigbours
    UChar_t     padStatus;     // padStatus of the current cluster candidate
    Bool_t      fivePad;       // is this cluster candidate part of a 5 pad cluster (two overlaping clusters)?
  MaxStruct():row(-1),col(0),time(0),padStatus(0),fivePad(kFALSE)
      {memset(signals, 0, 3*sizeof(Float_t));}
    MaxStruct &operator=(const MaxStruct &a)
    {memcpy(this, &a, sizeof(MaxStruct)); return *this;}
  };
  
  AliTRDclusterizer(const AliTRDReconstructor *const rec = 0x0);
  AliTRDclusterizer(const Text_t* name, const Text_t* title, const AliTRDReconstructor *const rec = 0x0);
  AliTRDclusterizer(const AliTRDclusterizer &c);
  virtual         ~AliTRDclusterizer();
  AliTRDclusterizer &operator=(const AliTRDclusterizer &c);

  void     Copy(TObject &c) const;

  Bool_t   Open(const Char_t *name, Int_t nEvent = 0);
  Bool_t   OpenInput(Int_t nEvent = 0);
  Bool_t   OpenOutput();
  Bool_t   OpenOutput(TTree *const clusterTree);

  Bool_t   ReadDigits();
  Bool_t   ReadDigits(AliRawReader *rawReader);
  Bool_t   ReadDigits(TTree *digitsTree);

  Bool_t   ReadTracklets();
  Bool_t   ReadTracks();

  Bool_t   WriteClusters(Int_t det);
  void     ResetRecPoints();
  virtual TClonesArray    *RecPoints();
  virtual TClonesArray    *TrackletsArray(const TString &trkltype = "");
  virtual TClonesArray    *TracksArray();

  Bool_t   Raw2Clusters(AliRawReader *rawReader);
  Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);

  Bool_t   MakeClusters();
  Bool_t   MakeClusters(Int_t det);

  Bool_t   AddLabels();
  Bool_t   SetUseLabels(const Bool_t kset) { SetBit(kLabels, kset); return TestBit(kLabels);  } // should we assign labels to clusters
  void     SetRawVersion(const Int_t iver) { fRawVersion = iver; } // set the expected raw data version
  void             SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
  static UChar_t   GetStatus(Short_t &signal);
  Int_t            GetAddedClusters() const {return fNoOfClusters;}
  Int_t            GetNTimeBins() const {return fTimeTotal;}

  Bool_t           IsClustersOwner() const {return TestBit(kClOwner);}
  virtual void     SetClustersOwner(Bool_t own=kTRUE) {SetBit(kClOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
  void             SetTrackletsOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fTracklets = 0x0; } }
  void             SetTracksOwner(Bool_t own=kTRUE) {SetBit(kTracksOwner, own); if(!own) {fTracks = 0x0; } }
  void             SetSkipTransform(Bool_t b=kTRUE) {SetBit(kSkipTrafo, b); }
  void             SetStoreRawSignals(Bool_t b=kTRUE) {SetBit(kRawSignal, b); }

  UInt_t   GetTriggerFlags(const Int_t sector) const { return fTrgFlags[sector]; }

protected:

  void             ApplyTCTM(Short_t *const arr, const Int_t nTime, const Int_t nexp);
  void             DeConvExp (Short_t *const arr, const Int_t nTime, const Int_t nexp);
  void             ConvExp(Short_t *const arr, const Int_t nTime);
  void             TailCancelation(const AliTRDrecoParam* const recoParam);

  Float_t          Unfold(Double_t eps, Int_t layer, const Double_t *const padSignal) const;
  
  void             SetPadStatus(const UChar_t status, UChar_t &encoding) const;
  UChar_t          GetPadStatus(UChar_t encoding) const;
  Int_t            GetCorruption(UChar_t encoding) const;

  Bool_t           IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Float_t *const Signals);       //for const correctness reasons not const parameters are given separately
  Bool_t           FivePadCluster(MaxStruct &ThisMax, MaxStruct &NeighbourMax);
  void             CreateCluster(const MaxStruct &Max); 

  virtual void     AddClusterToArray(AliTRDcluster* cluster);

  const AliTRDReconstructor *fReconstructor;     //! reconstructor
  AliRunLoader        *fRunLoader;                //! Run Loader
  TTree               *fClusterTree;              //! Tree with the cluster
  TClonesArray        *fRecPoints;                //! Array of clusters
  TClonesArray        *fTracklets;                //! Array of online tracklets
  TClonesArray        *fTracks;                   //! Array of GTU tracks

  TTree               *fTrackletTree;             //! Tree for tracklets

  AliTRDdigitsManager *fDigitsManager;            //! TRD digits manager

  UInt_t              **fTrackletContainer;       //! tracklet container
					          // legacy code to avoid breakint AliHLTTRDClusterizer
					          // but it's useless

  Int_t                fRawVersion;               //  Expected raw version of the data - default is 2

  AliTRDtransform     *fTransform;                //! Transforms the reconstructed space points

  AliTRDarrayADC      *fDigits;                   // Array holding the digits
  AliTRDarrayADC      *fDigitsRaw;                // Array holding the raw digits if requested
  AliTRDSignalIndex   *fIndexes;                  // Array holding the indexes to the digits
  Float_t              fMaxThresh;                // Threshold value for the maximum
  Float_t              fMaxThreshTest;            // Threshold value for the maximum to test agains
  Float_t              fSigThresh;                // Threshold value for the digit signal
  Float_t              fMinMaxCutSigma;           // Threshold value for the maximum (cut noise)
  Float_t              fMinLeftRightCutSigma;     // Threshold value for the sum pad (cut noise)
  Int_t                fLayer;                    // Current layer of the detector
  Int_t                fDet;                      // Current detecor
  UShort_t             fVolid;                    // Volume ID
  Int_t                fColMax;                   // Number of Colums in one detector
  Int_t                fTimeTotal;                // Number of time bins
  AliTRDCalROC        *fCalGainFactorROC;         // Calibration object with pad wise values for the gain factors
  Float_t              fCalGainFactorDetValue;    // Calibration value for chamber wise noise
  AliTRDCalROC        *fCalNoiseROC;              // Calibration object with pad wise values for the noise
  Float_t              fCalNoiseDetValue;         // Calibration value for chamber wise noise
  AliTRDCalSingleChamberStatus *fCalPadStatusROC; // Calibration object with the pad status
  AliTRDCalOnlineGainTableROC *fCalOnlGainROC;     // Calibration table of online gain factors
  Int_t                fClusterROC;               // The index to the first cluster of a given ROC
  Int_t                firstClusterROC;           // The number of cluster in a given ROC
  Int_t                fNoOfClusters;             // Number of Clusters already processed and still owned by the clusterizer
  Int_t                fBaseline;                 // Baseline of the ADC values
  AliTRDrawStream     *fRawStream;                // Raw data streamer
  UInt_t               fTrgFlags[AliTRDgeometry::kNsector]; // trigger flags

private:
  inline void      CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);

  ClassDef(AliTRDclusterizer,6)                   //  TRD clusterfinder

};

#endif
Commit	Line	Data
8230f242	1	#ifndef ALITRDCLUSTERIZER_H
8230f242	2	#define ALITRDCLUSTERIZER_H
f7336fa3	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	////////////////////////////////////////////////////////////////////////////
6d50f529	9	// //
3fe61b77	10	// TRD cluster finder //
6d50f529	11	// //
	12	////////////////////////////////////////////////////////////////////////////
	13
f7336fa3	14	#include <TNamed.h>
793ff80c	15
e2cea5e3	16	#include "AliTRDgeometry.h"
8eab6a01	17
793ff80c	18	class TFile;
bdbb05bb	19	class TTree;
66f6bfd9	20	class TClonesArray;
6d50f529	21
88cb7938	22	class AliRunLoader;
3fe61b77	23	class AliRawReader;
6d50f529	24
bdbb05bb	25	class AliTRD;
c85a4951	26	class AliTRDcluster;
b65e5048	27
04e58504	28	class AliTRDrawStream;
b65e5048	29	class AliTRDarrayADC;
b65e5048	30	class AliTRDarraySignal;
3fe61b77	31	class AliTRDdigitsManager;
	32	class AliTRDSignalIndex;
	33	class AliTRDtransform;
	34	class AliTRDCalROC;
3a039a31	35	class AliTRDReconstructor;
1b3a719f	36	class AliTRDCalSingleChamberStatus;
fc0882f3	37	class AliTRDrecoParam;
e8e7f17e	38	class AliTRDCalOnlineGainTableROC;
b65e5048	39
3a039a31	40	class AliTRDclusterizer : public TNamed
3a039a31	41	{
b65e5048	42
	43	public:
	44
66f6bfd9	45	// steering flags
4806451b	46	// bits from 0-13 are reserved by ROOT (see TObject.h)
66f6bfd9	47	enum{
4806451b	48	kTrOwner = BIT(14) // toggle online tracklets ownership
	49	,kClOwner= BIT(15) // toggle cluster ownership
	50	,kLabels = BIT(16) // toggle MC labels for clusters
d03bfe2b	51	,kSkipTrafo = BIT(17) // skip the coordinate transformation of clusters?
4806451b	52	,kLUT = BIT(18) // using look up table for cluster's r-phi position
	53	,kGAUS = BIT(19) // using gauss approx. for cluster's r-phi position
	54	,knewDM = BIT(20) // was the digitsmanger created by raw2clusters?
a1c095f1	55	,kTracksOwner = BIT(21) // toggle GTU track ownership
07897df4	56	,kRawSignal = BIT(22) // toggle storage of raw signals in clusters
66f6bfd9	57	};
486c2339	58
	59	struct MaxStruct
	60	{
4dab9275	61	Int_t row; // row of the current cluster candidate
	62	Int_t col; // col of the current cluster candidate
	63	Int_t time; // time -"-
07897df4	64	Float_t signals[3]; // signals of the maximum pad and it's twon neigbours
4dab9275	65	UChar_t padStatus; // padStatus of the current cluster candidate
4dab9275	66	Bool_t fivePad; // is this cluster candidate part of a 5 pad cluster (two overlaping clusters)?
615f0826	67	MaxStruct():row(-1),col(0),time(0),padStatus(0),fivePad(kFALSE)
07897df4	68	{memset(signals, 0, 3*sizeof(Float_t));}
486c2339	69	MaxStruct &operator=(const MaxStruct &a)
6b0d4ad5	70	{memcpy(this, &a, sizeof(MaxStruct)); return *this;}
486c2339	71	};
66f6bfd9	72
486c2339	73	AliTRDclusterizer(const AliTRDReconstructor *const rec = 0x0);
486c2339	74	AliTRDclusterizer(const Text_t* name, const Text_t* title, const AliTRDReconstructor *const rec = 0x0);
dd9a6ee3	75	AliTRDclusterizer(const AliTRDclusterizer &c);
6d50f529	76	virtual ~AliTRDclusterizer();
dd9a6ee3	77	AliTRDclusterizer &operator=(const AliTRDclusterizer &c);
dd9a6ee3	78
0ab5ff49	79	void Copy(TObject &c) const;
3fe61b77	80
6cd9a21f	81	Bool_t Open(const Char_t *name, Int_t nEvent = 0);
	82	Bool_t OpenInput(Int_t nEvent = 0);
	83	Bool_t OpenOutput();
e7295a3a	84	Bool_t OpenOutput(TTree *const clusterTree);
3fe61b77	85
6cd9a21f	86	Bool_t ReadDigits();
	87	Bool_t ReadDigits(AliRawReader *rawReader);
	88	Bool_t ReadDigits(TTree *digitsTree);
3fe61b77	89
cb8b99ee	90	Bool_t ReadTracklets();
	91	Bool_t ReadTracks();
	92
6cd9a21f	93	Bool_t WriteClusters(Int_t det);
3d0c7d6d	94	void ResetRecPoints();
3d0c7d6d	95	virtual TClonesArray *RecPoints();
cb8b99ee	96	virtual TClonesArray *TrackletsArray(const TString &trkltype = "");
a1c095f1	97	virtual TClonesArray *TracksArray();
bdbb05bb	98
6cd9a21f	99	Bool_t Raw2Clusters(AliRawReader *rawReader);
6cd9a21f	100	Bool_t Raw2ClustersChamber(AliRawReader *rawReader);
3fe61b77	101
6cd9a21f	102	Bool_t MakeClusters();
6cd9a21f	103	Bool_t MakeClusters(Int_t det);
3fe61b77	104
6cd9a21f	105	Bool_t AddLabels();
b72f4eaf	106	Bool_t SetUseLabels(const Bool_t kset) { SetBit(kLabels, kset); return TestBit(kLabels); } // should we assign labels to clusters
6cd9a21f	107	void SetRawVersion(const Int_t iver) { fRawVersion = iver; } // set the expected raw data version
3a039a31	108	void SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
66f6bfd9	109	static UChar_t GetStatus(Short_t &signal);
e7295a3a	110	Int_t GetAddedClusters() const {return fNoOfClusters;}
293e8ba7	111	Int_t GetNTimeBins() const {return fTimeTotal;}
66f6bfd9	112
8c499dbf	113	Bool_t IsClustersOwner() const {return TestBit(kClOwner);}
29f3223b	114	virtual void SetClustersOwner(Bool_t own=kTRUE) {SetBit(kClOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
8c499dbf	115	void SetTrackletsOwner(Bool_t own=kTRUE) {SetBit(kTrOwner, own); if(!own) {fTracklets = 0x0; } }
a1c095f1	116	void SetTracksOwner(Bool_t own=kTRUE) {SetBit(kTracksOwner, own); if(!own) {fTracks = 0x0; } }
d03bfe2b	117	void SetSkipTransform(Bool_t b=kTRUE) {SetBit(kSkipTrafo, b); }
07897df4	118	void SetStoreRawSignals(Bool_t b=kTRUE) {SetBit(kRawSignal, b); }
3a039a31	119
e2cea5e3	120	UInt_t GetTriggerFlags(const Int_t sector) const { return fTrgFlags[sector]; }
8eab6a01	121
b72f4eaf	122	protected:
b65e5048	123
9dcc64cc	124	void ApplyTCTM(Short_t *const arr, const Int_t nTime, const Int_t nexp);
1ab4da8c	125	void DeConvExp (Short_t *const arr, const Int_t nTime, const Int_t nexp);
9dcc64cc	126	void ConvExp(Short_t *const arr, const Int_t nTime);
fc0882f3	127	void TailCancelation(const AliTRDrecoParam* const recoParam);
064d808d	128
8c499dbf	129	Float_t Unfold(Double_t eps, Int_t layer, const Double_t *const padSignal) const;
6d9e79cc	130
e7295a3a	131	void SetPadStatus(const UChar_t status, UChar_t &encoding) const;
064d808d	132	UChar_t GetPadStatus(UChar_t encoding) const;
	133	Int_t GetCorruption(UChar_t encoding) const;
	134
07897df4	135	Bool_t IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Float_t *const Signals); //for const correctness reasons not const parameters are given separately
1b3a719f	136	Bool_t FivePadCluster(MaxStruct &ThisMax, MaxStruct &NeighbourMax);
486c2339	137	void CreateCluster(const MaxStruct &Max);
8dbc94c7	138
e7295a3a	139	virtual void AddClusterToArray(AliTRDcluster* cluster);
3fe61b77	140
04e58504	141	const AliTRDReconstructor *fReconstructor; //! reconstructor
	142	AliRunLoader *fRunLoader; //! Run Loader
	143	TTree *fClusterTree; //! Tree with the cluster
	144	TClonesArray *fRecPoints; //! Array of clusters
	145	TClonesArray *fTracklets; //! Array of online tracklets
	146	TClonesArray *fTracks; //! Array of GTU tracks
	147
	148	TTree *fTrackletTree; //! Tree for tracklets
	149
	150	AliTRDdigitsManager *fDigitsManager; //! TRD digits manager
	151
	152	UInt_t **fTrackletContainer; //! tracklet container
	153	// legacy code to avoid breakint AliHLTTRDClusterizer
	154	// but it's useless
	155
	156	Int_t fRawVersion; // Expected raw version of the data - default is 2
	157
	158	AliTRDtransform *fTransform; //! Transforms the reconstructed space points
	159
	160	AliTRDarrayADC *fDigits; // Array holding the digits
07897df4	161	AliTRDarrayADC *fDigitsRaw; // Array holding the raw digits if requested
04e58504	162	AliTRDSignalIndex *fIndexes; // Array holding the indexes to the digits
07897df4	163	Float_t fMaxThresh; // Threshold value for the maximum
	164	Float_t fMaxThreshTest; // Threshold value for the maximum to test agains
	165	Float_t fSigThresh; // Threshold value for the digit signal
04e58504	166	Float_t fMinMaxCutSigma; // Threshold value for the maximum (cut noise)
	167	Float_t fMinLeftRightCutSigma; // Threshold value for the sum pad (cut noise)
	168	Int_t fLayer; // Current layer of the detector
	169	Int_t fDet; // Current detecor
	170	UShort_t fVolid; // Volume ID
	171	Int_t fColMax; // Number of Colums in one detector
	172	Int_t fTimeTotal; // Number of time bins
	173	AliTRDCalROC *fCalGainFactorROC; // Calibration object with pad wise values for the gain factors
	174	Float_t fCalGainFactorDetValue; // Calibration value for chamber wise noise
	175	AliTRDCalROC *fCalNoiseROC; // Calibration object with pad wise values for the noise
	176	Float_t fCalNoiseDetValue; // Calibration value for chamber wise noise
3d0c7d6d	177	AliTRDCalSingleChamberStatus *fCalPadStatusROC; // Calibration object with the pad status
893e2ec6	178	AliTRDCalOnlineGainTableROC *fCalOnlGainROC; // Calibration table of online gain factors
04e58504	179	Int_t fClusterROC; // The index to the first cluster of a given ROC
	180	Int_t firstClusterROC; // The number of cluster in a given ROC
	181	Int_t fNoOfClusters; // Number of Clusters already processed and still owned by the clusterizer
	182	Int_t fBaseline; // Baseline of the ADC values
	183	AliTRDrawStream *fRawStream; // Raw data streamer
e2cea5e3	184	UInt_t fTrgFlags[AliTRDgeometry::kNsector]; // trigger flags
064d808d	185
04e58504	186	private:
	187	inline void CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);
	188
	189	ClassDef(AliTRDclusterizer,6) // TRD clusterfinder
f7336fa3	190
	191	};
	192
f7336fa3	193	#endif