////////////////////////////////////////////////////////////////////////
#include <TObjArray.h> // used in inline function GetModule.
+#include <TBranch.h> // used in inline function SetHitsAddressBranch
+#include "AliRun.h"
#include "AliDetector.h"
+#include "AliITSDetType.h"
class TString;
class TTree;
class TFile;
-class AliITSDetType;
+//class AliITSDetType;
class AliITSsimulation;
class AliITSClusterFinder;
class AliITSsegmentation;
//-------------------- Geometry Transformations --------------------
// ITS geometry functions
AliITSgeom *GetITSgeom() const {return fITSgeom;}
+ // Sets ITS geometry ! be very careful using this function.
+ void SetITSgeom(AliITSgeom *geom) {fITSgeom = geom;}
// return pointer to the array of modules
TObjArray *GetModules() const {return fITSmodules;}
// return pointer to a particular module
virtual void SetDefaultClusterFinders();
virtual void MakeBranch(Option_t *opt=" ", const char *file=0);
virtual void SetTreeAddress();
+ // For a give branch from the treeH sets the TClonesArray address.
+ virtual void SetHitsAddressBranch(TBranch *b){b->SetAddress(&fHits);}
// Return pointer to DetType #id
AliITSDetType *DetType(Int_t id);
//Int_t NDetTypes() {return fNDetTypes;}
//---------- Configuration Methods (per detector type) -------------
+ // Determines which ITS subdetectors will be processed. Effects
+ // digitization, and Reconstruction only.
+ void SetDetectors(Option_t *opt="All"){fOpt = opt;}
+ // Returns the list of ITS subdetectors that will be processed.
+ Option_t* GetDetectors(){return fOpt;}
// Set response
virtual void SetResponseModel(Int_t id, AliITSresponse *response);
// Set segmentation
virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
// Set simulation - temporary
virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim);
+ // Set simulation - temporary
+ virtual AliITSsimulation* GetSimulationModel(Int_t id){
+ return ((AliITSDetType*)(fDetTypes->At(id)))->GetSimulationModel();}
// Set reconstruction
virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
// Set class names for digit and rec point
virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits);
//------------ sort hits by module for Digitisation ----------------
virtual void InitModules(Int_t size,Int_t &nmodules);
+ virtual void FillModules(TTree *treeH, Int_t mask = 0);
virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
Option_t *opt,Text_t *filename);
virtual void ClearModules();
//===================== Digitisation ===============================
void MakeBranchS(const char *file);
void SetTreeAddressS(TTree *treeS);
- void MakeBranchD(const char *file);
+ TClonesArray * GetSDigits() { return fSDigits; }
+ void MakeBranchInTreeD(TTree *treeD,const char *file=0);
+ void MakeBranchD(const char *file){
+ MakeBranchInTreeD(gAlice->TreeD(),file);}
void SetTreeAddressD(TTree *treeD);
void Hits2SDigits(); // Turn hits into SDigits
void Hits2PreDigits(); // Turn hits into SDigits
- void SDigits2Digits(); // Turn SDigits to Digits
+ void SDigits2Digits(){SDigitsToDigits("All");} // Turn SDigits to Digits
+ void SDigitsToDigits(Option_t *opt="All"); // Turn SDigits to Digits
void Hits2Digits(); // Turn hits straight into Digits.
//------------------ Internal functions ----------------------------
// Standard Hits To SDigits function
{return ((TClonesArray *) (*fCtype)[id]);}
//=================== Reconstruction ===============================
- void MakeBranchR(const char *file);
+ void MakeBranchR(const char *file, Option_t *opt=" ");
+ void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");}
void SetTreeAddressR(TTree *treeR);
void AddRecPoint(const AliITSRecPoint &p);
void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
protected:
//================== Data Members ==================================
AliITSgeom *fITSgeom; // Pointer to ITS geometry
- Bool_t fEuclidOut; // Flag to write geometry in euclid format
+ Bool_t fEuclidOut; // Flag to write geometry in euclid format
TObjArray *fITSmodules; //! Pointer to ITS modules
+ Option_t *fOpt; //! Detector option ="All" unless changed.
- Int_t fIdN; // the number of layers
+ Int_t fIdN; // the number of layers
Int_t *fIdSens; //[fIdN] layer identifier
TString *fIdName; //[fIdN] layer identifier
- Int_t fNDetTypes; // Number of detector types
+ Int_t fNDetTypes; // Number of detector types
TObjArray *fDetTypes; // List of detector types
-// TObjArray *fSDigits; // List of Summable digits.
- TClonesArray *fSDigits; // List of Summable digits.
+ TClonesArray *fSDigits; // List of Summable digits.
Int_t fNSDigits; // Number of Summable Digits.
TObjArray *fDtype; // List of digits
TObjArray *fCtype; // List of clusters
Int_t *fNctype; //[fNDetTypes] Num. of clust. per type of det.
- TTree *fTreeC; // Tree for raw clusters
+ TTree *fTreeC; //! Tree for raw clusters
TClonesArray *fRecPoints; // List of reconstructed points
Int_t fNRecPoints; // Number of rec points
- ClassDef(AliITS,1) // Base class for ITS
+ ClassDef(AliITS,2) // Base class for ITS
};
#endif