[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>

class TFile;
class TTree;
class TClonesArray;

class AliRunLoader;
class AliRawReader;

class AliTRD;
class AliTRDcluster;

class AliTRDarrayADC;
class AliTRDarraySignal;
class AliTRDdigitsManager;
class AliTRDSignalIndex;
class AliTRDtransform;
class AliTRDCalROC;
class AliTRDReconstructor;

class AliTRDclusterizer : public TNamed 
{

 public:

  // steering flags
  enum{
    kOwner = BIT(14)
  };
  
  AliTRDclusterizer(AliTRDReconstructor *rec = 0x0);
  AliTRDclusterizer(const Text_t* name, const Text_t* title, AliTRDReconstructor *rec = 0x0);
  AliTRDclusterizer(const AliTRDclusterizer &c);
  virtual         ~AliTRDclusterizer();
  AliTRDclusterizer &operator=(const AliTRDclusterizer &c);

  virtual void     Copy(TObject &c) const;

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

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

  virtual Bool_t   WriteClusters(Int_t det);
          void     ResetRecPoints();
  TClonesArray       *RecPoints();
          Bool_t   WriteTracklets(Int_t det);

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

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

  virtual Bool_t   AddLabels(Int_t idet, Int_t firstClusterROC, Int_t nClusterROC);
  virtual Bool_t   SetAddLabels(Bool_t kset) { fAddLabels = kset; 
            return fAddLabels;  } // should we assign labels to clusters
  virtual void     SetRawVersion(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);

  Bool_t           IsClustersOwner() const {return TestBit(kOwner);}
  void             SetClustersOwner(Bool_t own=kTRUE) {SetBit(kOwner, own); if(!own) fRecPoints = 0x0;}

 protected:

  void             DeConvExp(Double_t *source, Double_t *target
                           , Int_t nTimeTotal, Int_t nexp);
  void             TailCancelation(AliTRDarrayADC *digitsIn
			         , AliTRDarraySignal *digitsOut 
			         , AliTRDSignalIndex *indexesIn
			         , AliTRDSignalIndex *indexesOut
			         , Int_t nTimeTotal
			         , Float_t ADCthreshold
			         , AliTRDCalROC *calGainFactorROC
			         , Float_t calGainFactorDetValue);
  virtual Double_t Unfold(Double_t eps, Int_t layer, Double_t *padSignal);
          Double_t GetCOG(Double_t signal[5]) const; 
  void             FillLUT();
          Double_t LUTposition(Int_t ilayer, Double_t ampL, Double_t ampC, Double_t ampR) const;
  virtual void     ResetHelperIndexes(AliTRDSignalIndex *indexesIn);
  
  void              SetPadStatus(UChar_t status, UChar_t &encoding);
  UChar_t           GetPadStatus(UChar_t encoding);
  Int_t             GetCorruption(UChar_t encoding);

  const AliTRDReconstructor *fReconstructor;       //! reconstructor
  AliRunLoader        *fRunLoader;           //! Run Loader
  TTree               *fClusterTree;         //! Tree with the cluster
  TClonesArray        *fRecPoints;           //! Array of clusters

  TTree               *fTrackletTree;        //! Tree for tracklets

  AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager

  UInt_t              **fTrackletContainer;  //! tracklet container

  Bool_t               fAddLabels;           //  Should clusters have MC labels?
  Int_t                fRawVersion;          //  Expected raw version of the data - default is 2

  AliTRDSignalIndex   *fIndexesOut;          //! Helper indexes for clusterization
  AliTRDSignalIndex   *fIndexesMaxima;       //! Helper indexes for clusterization

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

  Int_t                fLUTbin;              //  Number of bins of the LUT
  Double_t            *fLUT;                 //! The lookup table

  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
793ff80c	16	class TFile;
bdbb05bb	17	class TTree;
66f6bfd9	18	class TClonesArray;
6d50f529	19
88cb7938	20	class AliRunLoader;
3fe61b77	21	class AliRawReader;
6d50f529	22
bdbb05bb	23	class AliTRD;
c85a4951	24	class AliTRDcluster;
b65e5048	25
	26	class AliTRDarrayADC;
	27	class AliTRDarraySignal;
3fe61b77	28	class AliTRDdigitsManager;
	29	class AliTRDSignalIndex;
	30	class AliTRDtransform;
	31	class AliTRDCalROC;
3a039a31	32	class AliTRDReconstructor;
b65e5048	33
3a039a31	34	class AliTRDclusterizer : public TNamed
3a039a31	35	{
b65e5048	36
	37	public:
	38
66f6bfd9	39	// steering flags
	40	enum{
	41	kOwner = BIT(14)
	42	};
	43
3a039a31	44	AliTRDclusterizer(AliTRDReconstructor *rec = 0x0);
3a039a31	45	AliTRDclusterizer(const Text_t* name, const Text_t* title, AliTRDReconstructor *rec = 0x0);
dd9a6ee3	46	AliTRDclusterizer(const AliTRDclusterizer &c);
6d50f529	47	virtual ~AliTRDclusterizer();
dd9a6ee3	48	AliTRDclusterizer &operator=(const AliTRDclusterizer &c);
dd9a6ee3	49
6d50f529	50	virtual void Copy(TObject &c) const;
3fe61b77	51
6d50f529	52	virtual Bool_t Open(const Char_t *name, Int_t nEvent = 0);
	53	virtual Bool_t OpenInput(Int_t nEvent = 0);
	54	virtual Bool_t OpenOutput();
25ca55ce	55	virtual Bool_t OpenOutput(TTree *clusterTree);
3fe61b77	56
	57	virtual Bool_t ReadDigits();
	58	virtual Bool_t ReadDigits(AliRawReader *rawReader);
	59	virtual Bool_t ReadDigits(TTree *digitsTree);
	60
6d50f529	61	virtual Bool_t WriteClusters(Int_t det);
6d50f529	62	void ResetRecPoints();
66f6bfd9	63	TClonesArray *RecPoints();
9c7c9ec1	64	Bool_t WriteTracklets(Int_t det);
bdbb05bb	65
3fe61b77	66	virtual Bool_t Raw2Clusters(AliRawReader *rawReader);
	67	virtual Bool_t Raw2ClustersChamber(AliRawReader *rawReader);
	68
	69	virtual Bool_t MakeClusters();
	70	virtual Bool_t MakeClusters(Int_t det);
	71
	72	virtual Bool_t AddLabels(Int_t idet, Int_t firstClusterROC, Int_t nClusterROC);
	73	virtual Bool_t SetAddLabels(Bool_t kset) { fAddLabels = kset;
3a039a31	74	return fAddLabels; } // should we assign labels to clusters
3fe61b77	75	virtual void SetRawVersion(Int_t iver) { fRawVersion = iver; } // set the expected raw data version
3a039a31	76	void SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
66f6bfd9	77	static UChar_t GetStatus(Short_t &signal);
	78
	79	Bool_t IsClustersOwner() const {return TestBit(kOwner);}
	80	void SetClustersOwner(Bool_t own=kTRUE) {SetBit(kOwner, own); if(!own) fRecPoints = 0x0;}
3a039a31	81
b65e5048	82	protected:
	83
	84	void DeConvExp(Double_t source, Double_t target
	85	, Int_t nTimeTotal, Int_t nexp);
	86	void TailCancelation(AliTRDarrayADC *digitsIn
	87	, AliTRDarraySignal *digitsOut
	88	, AliTRDSignalIndex *indexesIn
	89	, AliTRDSignalIndex *indexesOut
	90	, Int_t nTimeTotal
	91	, Float_t ADCthreshold
	92	, AliTRDCalROC *calGainFactorROC
	93	, Float_t calGainFactorDetValue);
053767a4	94	virtual Double_t Unfold(Double_t eps, Int_t layer, Double_t *padSignal);
b65e5048	95	Double_t GetCOG(Double_t signal[5]) const;
	96	void FillLUT();
	97	Double_t LUTposition(Int_t ilayer, Double_t ampL, Double_t ampC, Double_t ampR) const;
3fe61b77	98	virtual void ResetHelperIndexes(AliTRDSignalIndex *indexesIn);
6d9e79cc	99
	100	void SetPadStatus(UChar_t status, UChar_t &encoding);
	101	UChar_t GetPadStatus(UChar_t encoding);
	102	Int_t GetCorruption(UChar_t encoding);
3fe61b77	103
3a039a31	104	const AliTRDReconstructor *fReconstructor; //! reconstructor
3fe61b77	105	AliRunLoader *fRunLoader; //! Run Loader
3fe61b77	106	TTree *fClusterTree; //! Tree with the cluster
66f6bfd9	107	TClonesArray *fRecPoints; //! Array of clusters
3fe61b77	108
b65e5048	109	TTree *fTrackletTree; //! Tree for tracklets
9c7c9ec1	110
3fe61b77	111	AliTRDdigitsManager *fDigitsManager; //! TRD digits manager
3fe61b77	112
b65e5048	113	UInt_t **fTrackletContainer; //! tracklet container
9c7c9ec1	114
3fe61b77	115	Bool_t fAddLabels; // Should clusters have MC labels?
	116	Int_t fRawVersion; // Expected raw version of the data - default is 2
	117
	118	AliTRDSignalIndex *fIndexesOut; //! Helper indexes for clusterization
	119	AliTRDSignalIndex *fIndexesMaxima; //! Helper indexes for clusterization
	120
	121	AliTRDtransform *fTransform; //! Transforms the reconstructed space points
f7336fa3	122
fc546d21	123	Int_t fLUTbin; // Number of bins of the LUT
	124	Double_t *fLUT; //! The lookup table
	125
	126	ClassDef(AliTRDclusterizer,6) // TRD clusterfinder
f7336fa3	127
	128	};
	129
f7336fa3	130	#endif