[u/mrichter/AliRoot.git] / TPC / AliTPC.h

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

/* $Id$ */

////////////////////////////////////////////////
//  Manager class for TPC                     //
////////////////////////////////////////////////

class TFile;
class TTree;
#include <Htypes.h>
#include <TMatrixFfwd.h>
#include <TVector.h>

class AliTPCDigitsArray;
class AliTPCLoader;
class AliTPCParam;
class AliTPCTrackHitsV2; // M.I.
class AliRawReader;
class TTreeSRedirector;

#include "AliDetector.h"
#include "AliDigit.h" 
#include "AliHit.h" 

using std::fstream;

class AliTPC : public AliDetector {

public:
  AliTPC(); 
  AliTPC(const char *name, const char *title);
  
  virtual AliLoader* MakeLoader(const char* topfoldername);
  void          CreateDebugStremer();
  virtual      ~AliTPC();
  virtual void  AddHit(Int_t a1, Int_t *a2, Float_t *a3);
  virtual void  CreateGeometry() {}
  virtual void  CreateMaterials();
  virtual void  AddAlignableVolumes() const {}
  
  virtual AliDigitizer* CreateDigitizer(AliDigitizationInput* digInput) const;
  virtual void  SDigits2Digits(){;} //MI change -cycling to production
  virtual void  SDigits2Digits2(Int_t eventnumber=0);

  virtual void  Hits2SDigits(); // MI - cycling around
  virtual void  Hits2SDigits2(Int_t eventnumber=0);

  virtual void  Hits2Digits();
  virtual void  Hits2Digits(Int_t eventnumber);   //MI change
  virtual void  Hits2DigitsSector(Int_t isec);  //MI change
  virtual void  Init();
  virtual Int_t IsVersion() const =0;
  virtual void  Digits2Raw();
  virtual Bool_t Raw2SDigits(AliRawReader* rawReader);
  Int_t         GetNsectors() const  {return fNsectors;}
  virtual void  ResetDigits();
  virtual void  SetSens(Int_t sens);
  virtual void  SetSide(Float_t side);
  virtual void SetGEM(Int_t isGEM) {fIsGEM=isGEM; }

  virtual void  StepManager()=0;
  AliTPCDigitsArray*  GetDigitsArray() {return fDigitsArray;} //MI change
  AliTPCParam *GetParam(){return fTPCParam;} // M.K, M.I changes
  void SetParam(AliTPCParam *param){fTPCParam=param;} // M.K, M.I changes
  void SetDigitsArray(AliTPCDigitsArray* param) {fDigitsArray=param;}  //MI change

// additional function neccesary for the new hits 
   virtual void MakeBranch(Option_t *opt=" ");  //
   virtual void SetTreeAddress();
   virtual void SetTreeAddress2();
   virtual void AddHit2(Int_t a1,  Int_t *a2, Float_t *a3);  //
   virtual void ResetHits();
   virtual void ResetHits2();     
   virtual AliHit* FirstHit(Int_t track);
   virtual AliHit* NextHit();
   virtual AliHit* FirstHit2(Int_t track);
   virtual AliHit* NextHit2();
   virtual void FinishPrimary();
   virtual void RemapTrackHitIDs(Int_t *map);
   void SetHitType(Int_t type){fHitType =type;} //set type of hit container
   void SetDigitsSwitch(Int_t sw){fDigitsSwitch = sw;}
   void SetDefSwitch(Int_t def){fDefaults = def;}
   Float_t GetNoise();  //get Current noise  
   void    GenerNoise(Int_t tablasize);  // make noise table
   Bool_t  IsSectorActive(Int_t sec) const;    // check if the sector is active
   void    SetActiveSectors(Int_t * sectors, Int_t n);  //set active sectors
   Int_t GetHitType() const {return fHitType;}
   void    SetActiveSectors(Int_t flag=1); //loop over al hits and set active only hitted sectors
   Bool_t  TrackInVolume(Int_t id,Int_t track);  //return true if current track is in volume
   void    SetPrimaryIonisation(Bool_t flag = kTRUE) {fPrimaryIonisation = flag;}
   void    SetGainFactor(Float_t gain){fGainFactor=gain;} //gas gain scaling factor
   Float_t GetGainFactor()const {return fGainFactor;}//gas gain scaling factor
   // LHC clock phase switch 0 - no phase, 1 - random, 2 - from the OCDB
   void SetLHCclockPhase(Int_t sw){fLHCclockPhaseSw = sw;}
// static functions
   static AliTPCParam* LoadTPCParam(TFile *file); 
protected:
   Int_t          fDefaults; // defaults switch
  Int_t          fSens;             // ISENS
  Int_t          fNsectors;         // Number of sectors in TPC
  //MI changes
  AliTPCDigitsArray * fDigitsArray;              //! detector digit object  
  AliTPCParam *fTPCParam;           // pointer to TPC parameters 
  AliTPCTrackHitsV2 *fTrackHits;      //! hits for given track M.I.
  //  AliTPCTrackHits *fTrackHitsOld;      //! hits for given track M.I. MIold -

  Int_t  fHitType; // if fNewHit = 1 old data structure if 2 new hits  if 4  old MI stucture
  //  3 both types 
  Int_t fDigitsSwitch; // digits type, 0->normal, 1->summable

  //MK changes

  Float_t        fSide;  // selects left(-1), right(+1), or both(0) sides of the TPC
  Bool_t     fPrimaryIonisation; //switch between Fluka(true) and geant3(false)
 protected:
  AliTPC(const AliTPC& t);
  AliTPC &operator = (const AliTPC & param);
  //
  void SetDefaults();
  void DigitizeRow(Int_t irow,Int_t isec,TObjArray **rowTriplet);
  Float_t GetSignal(TObjArray *p1, Int_t ntr, TMatrixF *m1, 
                   TMatrixF *m2,Int_t *IndexRange);
  void GetList (Float_t label,Int_t np,TMatrixF *m,Int_t *IndexRange,
                Float_t **pList);
  void MakeSector(Int_t isec,Int_t nrows,TTree *TH,Stat_t ntracks,TObjArray **row);
  void TransportElectron(Float_t *xyz, Int_t *index);
  Int_t fCurrentIndex[4];// index[0] indicates coordinate system, 
                         // index[1] sector number, 
                         // index[2] pad row number  
                         // index[3] pad row number for which signal is calculated
  Int_t      fNoiseDepth;  //! noise table
  Float_t *  fNoiseTable;  //! table with noise
  Int_t      fCurrentNoise; //! index of the noise in  the noise table 
  Bool_t*    fActiveSectors; //! bool indicating which sectors are active
  Float_t    fGainFactor; // scaling factor
  TTreeSRedirector *fDebugStreamer;     //!debug streamer
  Int_t fLHCclockPhaseSw; //! lhc clock phase switch
  Int_t fIsGEM;        // flag isGEM readout
  ClassDef(AliTPC,14)  // Time Projection Chamber class
};


//_____________________________________________________________________________
 
class AliTPChit : public AliHit {
public:
   Int_t     fSector;     //sector number
   Int_t     fPadRow;     //Pad Row number
   Float_t   fQ ;         //charge
   Float_t   fTime;       //hit time
 
public:
   AliTPChit();
   AliTPChit(Int_t shunt, Int_t track, Int_t *vol, Float_t *hits);
   virtual ~AliTPChit() {}
   void SetX(Float_t x){fX = x;}
   void SetY(Float_t y){fY = y;}
   void SetZ(Float_t z){fZ = z;}

   Float_t Time() const {return fTime;}
 
   ClassDef(AliTPChit,3)  // Time Projection Chamber hits
};


#endif
Commit	Line	Data
	1	#ifndef ALITPC_H
	2	#define ALITPC_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
	8	////////////////////////////////////////////////
	9	// Manager class for TPC //
	10	////////////////////////////////////////////////
	11
	12	class TFile;
	13	class TTree;
	14	#include <Htypes.h>
	15	#include <TMatrixFfwd.h>
	16	#include <TVector.h>
	17
	18	class AliTPCDigitsArray;
	19	class AliTPCLoader;
	20	class AliTPCParam;
	21	class AliTPCTrackHitsV2; // M.I.
	22	class AliRawReader;
	23	class TTreeSRedirector;
	24
	25	#include "AliDetector.h"
	26	#include "AliDigit.h"
	27	#include "AliHit.h"
	28
	29	using std::fstream;
	30
	31	class AliTPC : public AliDetector {
	32
	33	public:
	34	AliTPC();
	35	AliTPC(const char name, const char title);
	36
	37	virtual AliLoader* MakeLoader(const char* topfoldername);
	38	void CreateDebugStremer();
	39	virtual ~AliTPC();
	40	virtual void AddHit(Int_t a1, Int_t a2, Float_t a3);
	41	virtual void CreateGeometry() {}
	42	virtual void CreateMaterials();
	43	virtual void AddAlignableVolumes() const {}
	44
	45	virtual AliDigitizer* CreateDigitizer(AliDigitizationInput* digInput) const;
	46	virtual void SDigits2Digits(){;} //MI change -cycling to production
	47	virtual void SDigits2Digits2(Int_t eventnumber=0);
	48
	49	virtual void Hits2SDigits(); // MI - cycling around
	50	virtual void Hits2SDigits2(Int_t eventnumber=0);
	51
	52	virtual void Hits2Digits();
	53	virtual void Hits2Digits(Int_t eventnumber); //MI change
	54	virtual void Hits2DigitsSector(Int_t isec); //MI change
	55	virtual void Init();
	56	virtual Int_t IsVersion() const =0;
	57	virtual void Digits2Raw();
	58	virtual Bool_t Raw2SDigits(AliRawReader* rawReader);
	59	Int_t GetNsectors() const {return fNsectors;}
	60	virtual void ResetDigits();
	61	virtual void SetSens(Int_t sens);
	62	virtual void SetSide(Float_t side);
	63	virtual void SetGEM(Int_t isGEM) {fIsGEM=isGEM; }
	64
	65	virtual void StepManager()=0;
	66	AliTPCDigitsArray* GetDigitsArray() {return fDigitsArray;} //MI change
	67	AliTPCParam *GetParam(){return fTPCParam;} // M.K, M.I changes
	68	void SetParam(AliTPCParam *param){fTPCParam=param;} // M.K, M.I changes
	69	void SetDigitsArray(AliTPCDigitsArray* param) {fDigitsArray=param;} //MI change
	70
	71	// additional function neccesary for the new hits
	72	virtual void MakeBranch(Option_t *opt=" "); //
	73	virtual void SetTreeAddress();
	74	virtual void SetTreeAddress2();
	75	virtual void AddHit2(Int_t a1, Int_t a2, Float_t a3); //
	76	virtual void ResetHits();
	77	virtual void ResetHits2();
	78	virtual AliHit* FirstHit(Int_t track);
	79	virtual AliHit* NextHit();
	80	virtual AliHit* FirstHit2(Int_t track);
	81	virtual AliHit* NextHit2();
	82	virtual void FinishPrimary();
	83	virtual void RemapTrackHitIDs(Int_t *map);
	84	void SetHitType(Int_t type){fHitType =type;} //set type of hit container
	85	void SetDigitsSwitch(Int_t sw){fDigitsSwitch = sw;}
	86	void SetDefSwitch(Int_t def){fDefaults = def;}
	87	Float_t GetNoise(); //get Current noise
	88	void GenerNoise(Int_t tablasize); // make noise table
	89	Bool_t IsSectorActive(Int_t sec) const; // check if the sector is active
	90	void SetActiveSectors(Int_t * sectors, Int_t n); //set active sectors
	91	Int_t GetHitType() const {return fHitType;}
	92	void SetActiveSectors(Int_t flag=1); //loop over al hits and set active only hitted sectors
	93	Bool_t TrackInVolume(Int_t id,Int_t track); //return true if current track is in volume
	94	void SetPrimaryIonisation(Bool_t flag = kTRUE) {fPrimaryIonisation = flag;}
	95	void SetGainFactor(Float_t gain){fGainFactor=gain;} //gas gain scaling factor
	96	Float_t GetGainFactor()const {return fGainFactor;}//gas gain scaling factor
	97	// LHC clock phase switch 0 - no phase, 1 - random, 2 - from the OCDB
	98	void SetLHCclockPhase(Int_t sw){fLHCclockPhaseSw = sw;}
	99	// static functions
	100	static AliTPCParam* LoadTPCParam(TFile *file);
	101	protected:
	102	Int_t fDefaults; // defaults switch
	103	Int_t fSens; // ISENS
	104	Int_t fNsectors; // Number of sectors in TPC
	105	//MI changes
	106	AliTPCDigitsArray * fDigitsArray; //! detector digit object
	107	AliTPCParam *fTPCParam; // pointer to TPC parameters
	108	AliTPCTrackHitsV2 *fTrackHits; //! hits for given track M.I.
	109	// AliTPCTrackHits *fTrackHitsOld; //! hits for given track M.I. MIold -
	110
	111	Int_t fHitType; // if fNewHit = 1 old data structure if 2 new hits if 4 old MI stucture
	112	// 3 both types
	113	Int_t fDigitsSwitch; // digits type, 0->normal, 1->summable
	114
	115	//MK changes
	116
	117	Float_t fSide; // selects left(-1), right(+1), or both(0) sides of the TPC
	118	Bool_t fPrimaryIonisation; //switch between Fluka(true) and geant3(false)
	119	protected:
	120	AliTPC(const AliTPC& t);
	121	AliTPC &operator = (const AliTPC & param);
	122	//
	123	void SetDefaults();
	124	void DigitizeRow(Int_t irow,Int_t isec,TObjArray **rowTriplet);
	125	Float_t GetSignal(TObjArray p1, Int_t ntr, TMatrixF m1,
	126	TMatrixF m2,Int_t IndexRange);
	127	void GetList (Float_t label,Int_t np,TMatrixF m,Int_t IndexRange,
	128	Float_t **pList);
	129	void MakeSector(Int_t isec,Int_t nrows,TTree TH,Stat_t ntracks,TObjArray *row);
	130	void TransportElectron(Float_t xyz, Int_t index);
	131	Int_t fCurrentIndex[4];// index[0] indicates coordinate system,
	132	// index[1] sector number,
	133	// index[2] pad row number
	134	// index[3] pad row number for which signal is calculated
	135	Int_t fNoiseDepth; //! noise table
	136	Float_t * fNoiseTable; //! table with noise
	137	Int_t fCurrentNoise; //! index of the noise in the noise table
	138	Bool_t* fActiveSectors; //! bool indicating which sectors are active
	139	Float_t fGainFactor; // scaling factor
	140	TTreeSRedirector *fDebugStreamer; //!debug streamer
	141	Int_t fLHCclockPhaseSw; //! lhc clock phase switch
	142	Int_t fIsGEM; // flag isGEM readout
	143	ClassDef(AliTPC,14) // Time Projection Chamber class
	144	};
	145
	146
	147	//_____________________________________________________________________________
	148
	149	class AliTPChit : public AliHit {
	150	public:
	151	Int_t fSector; //sector number
	152	Int_t fPadRow; //Pad Row number
	153	Float_t fQ ; //charge
	154	Float_t fTime; //hit time
	155
	156	public:
	157	AliTPChit();
	158	AliTPChit(Int_t shunt, Int_t track, Int_t vol, Float_t hits);
	159	virtual ~AliTPChit() {}
	160	void SetX(Float_t x){fX = x;}
	161	void SetY(Float_t y){fY = y;}
	162	void SetZ(Float_t z){fZ = z;}
	163
	164	Float_t Time() const {return fTime;}
	165
	166	ClassDef(AliTPChit,3) // Time Projection Chamber hits
	167	};
	168
	169
	170	#endif
	171
	172
	173
	174
	175
	176
	177
	178
	179