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

#ifndef ALITRDDIGITSMANAGER_H
#define ALITRDDIGITSMANAGER_H

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

/* $Id$ */

/////////////////////////////////////////////////////////////
//  Manages the TRD digits                                 //
/////////////////////////////////////////////////////////////

#include <TObject.h>

class TFile;
class TTree;
class TBranch;  
class AliTRDdigit;
class AliTRDSignalIndex;
class AliTRDarrayADC;  
class AliTRDarraySignal; 
class AliTRDarrayDictionary;
class AliTRDReconstructor;

class AliTRDdigitsManager : public TObject {

 public:

  enum { kNDict = 3 };

  AliTRDdigitsManager(const AliTRDReconstructor *const rec = 0x0);  //if rec is given, we are reading raw data, so the TObjArrays may (and will) contain only one entry
  AliTRDdigitsManager(const AliTRDdigitsManager &m);
  virtual ~AliTRDdigitsManager();
  AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);

  virtual void                Copy(TObject &m) const;

  virtual void                CreateArrays();
  virtual void                ResetArrays();
  virtual void                ResetArrays(Int_t det);
  virtual Bool_t              BuildIndexes(Int_t det);

  virtual Bool_t              MakeBranch(TTree *tree);
  virtual Bool_t              ReadDigits(TTree *tree);
  virtual Bool_t              WriteDigits();

  virtual void                SetEvent(Int_t evt)             { fEvent           = evt;  };
  virtual void                SetSDigits(Int_t v = 1)         { fHasSDigits      = v;    };
  virtual void                SetUseDictionaries(Bool_t kval) { fUseDictionaries = kval; };

  virtual Bool_t              UsesDictionaries() const        { return fUseDictionaries; };
  virtual Bool_t              HasSDigits() const              { return fHasSDigits;      };
  static  Int_t               NDict()                         { return fgkNDict;         }; 

  virtual TObjArray          *GetDigits() const               { return fDigits;          };  
  virtual TObjArray          *GetDictionary(Int_t i) const    { return fDict[i];         }; 

  AliTRDdigit                *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det) const;
  Int_t                       GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det) const;
  
  AliTRDarrayADC             *GetDigits(Int_t det)  const;
  AliTRDarraySignal          *GetSDigits(Int_t det) const;    
  AliTRDarrayDictionary      *GetDictionary(Int_t det, Int_t i) const;  
  
  AliTRDSignalIndex          *GetIndexes(Int_t det);
  TObjArray                  *GetIndexes()                    { return fSignalIndexes;   };

  void                        RemoveDigits(Int_t det);
  void                        RemoveDictionaries(Int_t det);
  void                        RemoveIndexes(Int_t det);
  void                        ClearIndexes(Int_t det);
  
  Int_t                       GetTrack(Int_t track, AliTRDdigit *digit) const;
  Short_t                     GetDigitAmp(Int_t row, Int_t col, Int_t time, Int_t det) const;
  UChar_t                     GetPadStatus(Int_t row, Int_t col, Int_t time, Int_t det) const;

  Bool_t                      LoadArray(TObjArray *object, const Char_t *branchname, TTree *tree=0);  
  Bool_t                      LoadArrayDict(TObjArray *object, const Char_t *branchname, TTree *tree=0);  
  Bool_t                      StoreArray(TObjArray *array1, const Char_t *branchname, TTree *tree=0); 
  Bool_t                      StoreArrayDict(TObjArray *array3, const Char_t *branchname, TTree *tree=0); 
  
 protected:
  
  static const Int_t  fgkNDict;            //  Number of track dictionary arrays
  Int_t               fEvent;              //  Event number
  TTree              *fTree;               //! Tree for the digits arrays
  TObjArray          *fDigits;             //  Digits data array               
  TObjArray          *fDict[kNDict];       //  Track dictionary data array   
  Bool_t              fHasSDigits;         //  Switch for the summable digits
  TObjArray          *fSignalIndexes;      //  Provides access to the active pads and tbins
  Bool_t              fUseDictionaries;    //  Use dictionaries or not (case of real data)
  TTree              *fTreeD;              //  Tree with detector objects
  TBranch            *fBranch;             //  Branchaddress
  Int_t               fDets;               //  No of Detectors
  Bool_t              fRawRec;             //  Reconstruct from raw data. If its kTRUE then the TObjArrays have only one entry.
                                           //  If kFALSE then they have (AliTRDgeometry::Ndet()) entries (default).

  ClassDef(AliTRDdigitsManager,7)          //  Manages the TRD digits

};
#endif
Commit	Line	Data
8230f242	1	#ifndef ALITRDDIGITSMANAGER_H
8230f242	2	#define ALITRDDIGITSMANAGER_H
6f1e466d	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
ca21baaa	7	/* $Id$ */
6f1e466d	8
	9	/////////////////////////////////////////////////////////////
	10	// Manages the TRD digits //
	11	/////////////////////////////////////////////////////////////
	12
793ff80c	13	#include <TObject.h>
6f1e466d	14
2ab0c725	15	class TFile;
abaf1f1d	16	class TTree;
b65e5048	17	class TBranch;
793ff80c	18	class AliTRDdigit;
ca21baaa	19	class AliTRDSignalIndex;
b65e5048	20	class AliTRDarrayADC;
	21	class AliTRDarraySignal;
	22	class AliTRDarrayDictionary;
486c2339	23	class AliTRDReconstructor;
6f1e466d	24
	25	class AliTRDdigitsManager : public TObject {
	26
	27	public:
	28
dd56b762	29	enum { kNDict = 3 };
dd56b762	30
486c2339	31	AliTRDdigitsManager(const AliTRDReconstructor *const rec = 0x0); //if rec is given, we are reading raw data, so the TObjArrays may (and will) contain only one entry
dd9a6ee3	32	AliTRDdigitsManager(const AliTRDdigitsManager &m);
8230f242	33	virtual ~AliTRDdigitsManager();
dd9a6ee3	34	AliTRDdigitsManager &operator=(const AliTRDdigitsManager &m);
6f1e466d	35
625f5260	36	virtual void Copy(TObject &m) const;
625f5260	37
abaf1f1d	38	virtual void CreateArrays();
17eee78a	39	virtual void ResetArrays();
486c2339	40	virtual void ResetArrays(Int_t det);
625f5260	41	virtual Bool_t BuildIndexes(Int_t det);
88cb7938	42
88cb7938	43	virtual Bool_t MakeBranch(TTree *tree);
88cb7938	44	virtual Bool_t ReadDigits(TTree *tree);
6f1e466d	45	virtual Bool_t WriteDigits();
6f1e466d	46
625f5260	47	virtual void SetEvent(Int_t evt) { fEvent = evt; };
	48	virtual void SetSDigits(Int_t v = 1) { fHasSDigits = v; };
	49	virtual void SetUseDictionaries(Bool_t kval) { fUseDictionaries = kval; };
6f1e466d	50
625f5260	51	virtual Bool_t UsesDictionaries() const { return fUseDictionaries; };
	52	virtual Bool_t HasSDigits() const { return fHasSDigits; };
	53	static Int_t NDict() { return fgkNDict; };
793ff80c	54
b65e5048	55	virtual TObjArray *GetDigits() const { return fDigits; };
	56	virtual TObjArray *GetDictionary(Int_t i) const { return fDict[i]; };
	57
	58	AliTRDdigit *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det) const;
	59	Int_t GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det) const;
	60
	61	AliTRDarrayADC *GetDigits(Int_t det) const;
	62	AliTRDarraySignal *GetSDigits(Int_t det) const;
	63	AliTRDarrayDictionary *GetDictionary(Int_t det, Int_t i) const;
	64
	65	AliTRDSignalIndex *GetIndexes(Int_t det);
	66	TObjArray *GetIndexes() { return fSignalIndexes; };
	67
	68	void RemoveDigits(Int_t det);
	69	void RemoveDictionaries(Int_t det);
486c2339	70	void RemoveIndexes(Int_t det);
b65e5048	71	void ClearIndexes(Int_t det);
	72
	73	Int_t GetTrack(Int_t track, AliTRDdigit *digit) const;
	74	Short_t GetDigitAmp(Int_t row, Int_t col, Int_t time, Int_t det) const;
	75	UChar_t GetPadStatus(Int_t row, Int_t col, Int_t time, Int_t det) const;
	76
	77	Bool_t LoadArray(TObjArray object, const Char_t branchname, TTree *tree=0);
	78	Bool_t LoadArrayDict(TObjArray object, const Char_t branchname, TTree *tree=0);
	79	Bool_t StoreArray(TObjArray array1, const Char_t branchname, TTree *tree=0);
	80	Bool_t StoreArrayDict(TObjArray array3, const Char_t branchname, TTree *tree=0);
	81
6f1e466d	82	protected:
b65e5048	83
793ff80c	84	static const Int_t fgkNDict; // Number of track dictionary arrays
abaf1f1d	85	Int_t fEvent; // Event number
abaf1f1d	86	TTree *fTree; //! Tree for the digits arrays
b65e5048	87	TObjArray *fDigits; // Digits data array
b65e5048	88	TObjArray *fDict[kNDict]; // Track dictionary data array
625f5260	89	Bool_t fHasSDigits; // Switch for the summable digits
ca21baaa	90	TObjArray *fSignalIndexes; // Provides access to the active pads and tbins
001be664	91	Bool_t fUseDictionaries; // Use dictionaries or not (case of real data)
b65e5048	92	TTree *fTreeD; // Tree with detector objects
b65e5048	93	TBranch *fBranch; // Branchaddress
486c2339	94	Int_t fDets; // No of Detectors
	95	Bool_t fRawRec; // Reconstruct from raw data. If its kTRUE then the TObjArrays have only one entry.
	96	// If kFALSE then they have (AliTRDgeometry::Ndet()) entries (default).
625f5260	97
625f5260	98	ClassDef(AliTRDdigitsManager,7) // Manages the TRD digits
6f1e466d	99
6f1e466d	100	};
6f1e466d	101	#endif