X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ITS%2FAliITS.h;h=3524e46c8fcf2e764177ac34a37f56048d8e5d00;hb=696b423a2736586093252ab52be34f71235086d9;hp=e0e892043d1997ad4f8b3097c4a55156991d3085;hpb=596a855f140200041a7a9c6af1d4d99d432a34b4;p=u%2Fmrichter%2FAliRoot.git diff --git a/ITS/AliITS.h b/ITS/AliITS.h index e0e892043d1..3524e46c8fc 100644 --- a/ITS/AliITS.h +++ b/ITS/AliITS.h @@ -9,209 +9,172 @@ // Manager class for set: ITS // //////////////////////////////////////////////////////////////////////// -#include // used in inline function GetModule. -#include // used in inline function SetHitsAddressBranch -#include "AliRun.h" -#include "AliLoader.h" +#include // used in inline function GetModule. #include "AliDetector.h" -#include "AliITSDetType.h" +#include "AliITSTrigger.h" +#include "AliITSDetTypeSim.h" + class TString; class TTree; -class TFile; - -//class AliITSDetType; +class AliITSpListItem; class AliITSsimulation; -class AliITSClusterFinder; -class AliITSLoader; class AliITSsegmentation; -class AliITSresponse; + +class AliITSCalibration; class AliITShit; class AliITSgeom; -class AliITSpListItem; class AliITSdigit; -class AliITSRecPoint; -class AliITSRawCluster; class AliITSmodule; - -const Int_t kNTYPES=3; +class AliDigitizationInput; +class TArrayI; class AliITS : public AliDetector { public: + enum {kSPD,kSDD,kSSD}; + enum {kMaxLabel=3141593}; // default track label assigned to digits/clusters + //================= Standard Classes =============================== AliITS(); // Default creator. - AliITS(const char *name, const char *title); // standard Creator + AliITS(const Char_t *title); // standard Creator + AliITS(const char *name, const char *title); // extended standard Creator virtual ~AliITS(); // destructor - AliITS(const AliITS &source); // copy constructor. Not to be used! - AliITS& operator=(AliITS &source); // = operator. Not to be used! virtual Int_t IsVersion() const {return 1;} - virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) const; //===================== Simulation Geometry ======================== // get geometry version - detailed (major) or coarse (minor) virtual Int_t GetMajorVersion() const {return -1;} virtual Int_t GetMinorVersion() const {return -1;} - virtual void GetGeometryVersion(Int_t &a,Int_t &b) + virtual void GetGeometryVersion(Int_t &a,Int_t &b) const {a = GetMajorVersion();b=GetMinorVersion();return;} virtual void SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;} + virtual Bool_t GetEUCLID()const {return fEuclidOut;} //-------------------- 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 - AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *) - (fITSmodules->At(index));} + // ITS geometry functions From Simulation + AliITSgeom* GetITSgeom() const {return fDetTypeSim->GetITSgeom();} + void SetITSgeom(AliITSgeom *geom) {fDetTypeSim->SetITSgeom(geom);} + // return pointer to the array of modules + TObjArray *GetModules(){return fITSmodules;} + + AliITSmodule *GetModule(Int_t index){ + return (AliITSmodule*)(fITSmodules->At(index));} + virtual void SetSimuParam(AliITSSimuParam *sp){ + fSimuParam=sp; + fDetTypeSim->SetSimuParam(sp); + } + AliITSSimuParam* GetSimuParam() const {return fSimuParam;} + + virtual void SetDetTypeSim(AliITSDetTypeSim* dts) {fDetTypeSim=dts;} + AliITSDetTypeSim* GetDetTypeSim() const {return fDetTypeSim;} //================ Necessary general Classes ======================= virtual void Init(); virtual AliLoader* MakeLoader(const char* topfoldername); virtual void SetDefaults(); virtual void SetDefaultSimulation(); - virtual void SetDefaultClusterFinders(); virtual void MakeBranch(Option_t *opt=" "); + virtual void MakeBranchS(const char* fl); + virtual void MakeBranchD(const char* file); + virtual void MakeBranchInTreeD(TTree* treeD, const char* file=0); virtual void SetTreeAddress(); - // For a given 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) ------------- + //---------- 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 SetClasses(Int_t id, const char *digit, const char *cluster); + // Set calibration + virtual void SetCalibrationModel(Int_t dettype, AliITSCalibration *cal){ + fDetTypeSim->SetCalibrationModel(dettype,cal);} + // Set segmentation for Simulation + virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg){ + fDetTypeSim->SetSegmentationModel(id,seg);} + // Set simulation + virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim){ + fDetTypeSim->SetSimulationModel(id,sim);} + // Set simulation + virtual AliITSsimulation* GetSimulationModel(Int_t id){ + return fDetTypeSim->GetSimulationModel(id);} //=================== Hits ========================================= virtual void StepManager() {} // See Step Manager for specific geometry. - 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(Int_t /* evnt */,Int_t bgrev,Int_t /* nmodules */, + Option_t *opt, const char *filename); + virtual Bool_t 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); - 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 - AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const; - 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 + virtual void ClearModules(){if(fITSmodules) fITSmodules->Delete();} + virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits); + // Trigger + virtual AliTriggerDetector* CreateTriggerDetector() const; + + AliDigitizer* CreateDigitizer(AliDigitizationInput* digInput) const; + virtual void UpdateInternalGeometry(); + virtual void SDigitsToDigits(Option_t *opt="All"); + virtual void SDigits2Digits(){SDigitsToDigits("All");} + virtual void Hits2Digits(); + virtual void Hits2SDigits(); + virtual void Hits2PreDigits(); + virtual void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *option,Option_t *opt, + const char *filename); + virtual void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *option,Option_t *opt, + const char *filename); void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size, - Option_t *add, Option_t *det, Text_t *filename); - // Standard Hits To SDigits function - void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size, - Option_t *add, Option_t *det, Text_t *filename); - // Standard Hits To Digits function - void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size, - Option_t *add, Option_t *det, Text_t *filename); - void ResetSDigits(); // Resets the Summable digits. - void ResetDigits(); // depending on how the - void ResetDigits(Int_t branch); // tree will be filled only - void AddSumDigit(AliITSpListItem &sdig); - void AddRealDigit(Int_t branch, Int_t *digits); - void AddSimDigit(Int_t branch, AliITSdigit *d); - void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits, - Int_t* tracks,Int_t *hits,Float_t* trkcharges); - // Return pointers to digits - TObjArray *Dtype() {return fDtype;} - Int_t *Ndtype() {return fNdtype;} - TClonesArray *DigitsAddress(Int_t id) - {return ((TClonesArray *) (*fDtype)[id]);} + Option_t *add, Option_t *det, const char *filename) + {HitsToPreDigits(evNumber,bgrev,size,add,det,filename);} + + virtual void ResetDigits(); + virtual void ResetDigits(Int_t branch); + virtual void AddSumDigit(AliITSpListItem &sdig); + virtual void AddSimDigit(Int_t branch, AliITSdigit *d); + virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits, + Int_t* tracks,Int_t *hits,Float_t* trkcharges, + Int_t sigexpanded=-1000); + TObjArray* GetDigits() const {return fDetTypeSim->GetDigits();} + Int_t* GetNDigitArray() const {return fDetTypeSim->GetNDigitArray();} + TClonesArray *DigitsAddress(Int_t id) { + return fDetTypeSim->DigitsAddress(id);} + //Fast simulation + virtual void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size, + Option_t *add, Option_t *det, const char *filename); + virtual Int_t Hits2Clusters(TTree *in, TTree *out); + virtual void CheckLabels(Int_t lab[3]) const; //===================== Raw Data IO ================================ // Write digits into raw data format - virtual void Digits2RawData() {} - // Decode raw data and store digits - virtual void RawData2Digits() {} - - //==================== Clusterization ============================== - // create separate tree for clusters - declustering refining - void MakeTreeC(Option_t *option="C"); - void GetTreeC(Int_t event); - void AddCluster(Int_t branch, AliITSRawCluster *c); - void ResetClusters(); // one of the methods in - void ResetClusters(Int_t branch); // the pair will be kept - // Return pointers to clusters - TObjArray *Ctype() {return fCtype;} - Int_t *Nctype() {return fNctype;} - TClonesArray *ClustersAddress(Int_t id) - {return ((TClonesArray *) (*fCtype)[id]);} - - //=================== Reconstruction =============================== - void MakeBranchR(const char *file, Option_t *opt=" "); - void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");} - void MakeBranchC(); - void SetTreeAddressR(TTree *treeR); - void AddRecPoint(const AliITSRecPoint &p); - void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size, - Option_t *add, Option_t *det, Text_t *filename); - void Digits2Reco(); - void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det); - void ResetRecPoints(); - // Return pointer to rec points - TClonesArray *RecPoints() {return fRecPoints;} - - virtual void Reconstruct() const; - virtual AliTracker* CreateTracker() const; - virtual void FillESD(AliESD* esd) const; - + virtual void Digits2Raw(); + virtual Bool_t Raw2SDigits(AliRawReader* rawReader); + + //===================== FO signals ================================ + // Write FO signals in UserInfo of SDigits/Digits tree + void WriteFOSignals(); + void SetRawID2ClusID(const TArrayI* arr, Int_t iDet) { if (iDet>-1&&iDet-1&&iDetCluster assingment in Raw2SDigit (for embedding) + private: + AliITS(const AliITS &source); // copy constructor. Not to be used! + AliITS& operator=(const AliITS &source); // = operator. Not to be used! + ClassDef(AliITS,9) // Base class for ITS - Int_t fNDetTypes; // Number of detector types - TObjArray *fDetTypes; // List of detector types - - TClonesArray *fSDigits; //! List of Summable digits. - Int_t fNSDigits; // Number of Summable Digits. - - TObjArray *fDtype; //! List of digits - Int_t *fNdtype; //[fNDetTypes] Num. of digits per type of det. - - TObjArray *fCtype; //! List of clusters - Int_t *fNctype; //[fNDetTypes] Num. of clust. per type of det. - - TClonesArray *fRecPoints; //! List of reconstructed points - Int_t fNRecPoints; // Number of rec points - - ClassDef(AliITS,3) // Base class for ITS }; #endif