Patches and updates for fixes to this and other routines.
[u/mrichter/AliRoot.git] / ITS / AliITS.h
CommitLineData
3bd79107 1#ifndef ALIITS_H
2#define ALIITS_H
3da30618 3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
58005f18 8////////////////////////////////////////////////////////////////////////
3bd79107 9// Manager class for set: ITS //
58005f18 10////////////////////////////////////////////////////////////////////////
fe4da5cc 11
3bd79107 12#include <TObjArray.h>
13#include <TBranch.h>
e8189707 14#include <TString.h>
3bd79107 15#include <TTree.h>
16#include <TFile.h>
17#include <TObjectTable.h>
18
fe4da5cc 19#include "AliDetector.h"
58005f18 20#include "AliITSgeom.h"
3bd79107 21#include "AliITSsegmentation.h"
22#include "AliITSresponse.h"
3bd79107 23#include "AliITShit.h"
58005f18 24#include "AliITSdigit.h"
25#include "AliITSmodule.h"
3bd79107 26#include "AliITSRecPoint.h"
27#include "AliITSRawCluster.h"
fe4da5cc 28
e8189707 29class AliITSDetType;
3bd79107 30class AliITSsimulation;
31class AliITSClusterFinder;
3bd79107 32
33class AliITS : public AliDetector {
58005f18 34
35 public:
3bd79107 36 AliITS();
37 AliITS(const char *name, const char *title);
38 virtual ~AliITS();
39 AliITS(AliITS &source);
40 AliITS & operator=(AliITS &source);
41
42 virtual void AddHit(Int_t track, Int_t *vol, Float_t *hits);
43 virtual void AddRealDigit(Int_t branch, Int_t *digits);
e8189707 44 virtual void AddSimDigit(Int_t branch, AliITSdigit *d);
45 virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,Int_t* tracks,Int_t *hits,Float_t* trkcharges);
58005f18 46 //
3bd79107 47 virtual void AddCluster(Int_t branch, AliITSRawCluster *c);
48 virtual void AddRecPoint(const AliITSRecPoint &p);
3bd79107 49
50 virtual void ResetDigits(); // depending on how the
51 virtual void ResetDigits(Int_t branch); // tree will be filled only
52 virtual void ResetClusters(); // one of the methods in
53 virtual void ResetClusters(Int_t branch); // the pair will be kept
54 virtual void ResetRecPoints();
55
e8189707 56 // get geometry version - detailed (major) or coarse (minor)
3bd79107 57 void GetGeometryVersion(Int_t &a,Int_t &b) const
58 {a = fMajorVersion;b=fMinorVersion;return;}
59 virtual Int_t IsVersion() const {return 1;}
60 Int_t DistancetoPrimitive(Int_t px, Int_t py);
61 virtual void Init();
62 virtual void SetDefaults();
e8189707 63 virtual void SetDefaultSimulation();
64 virtual void SetDefaultClusterFinders();
3bd79107 65 // create separate tree for clusters - declustering refining
66 virtual void MakeTreeC(Option_t *option="C");
67 void GetTreeC(Int_t event);
68 virtual void MakeBranch(Option_t *opt=" ");
69 void SetTreeAddress();
70 virtual void SetEUCLID(Bool_t euclid=1) {fEuclidOut = euclid;}
71 virtual void StepManager() {}
72 // sort hits by module
73 virtual void InitModules(Int_t size,Int_t &nmodules);
e8189707 74 virtual void FillModules(Int_t evnt,Int_t bgrev,
3bd79107 75 Int_t nmodules,Option_t *opt,Text_t *filename);
76 virtual void ClearModules();
77 // Digitisation
e8189707 78 void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
3bd79107 79 Option_t *add, Option_t *det, Text_t *filename);
80 // Reconstruct hits
81 void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
82 // Fast simulation of space points from hits
e8189707 83 void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
3bd79107 84 Option_t *add, Option_t *det, Text_t *filename);
85
3bd79107 86 // Write digits into raw data format
87 virtual void Digits2RawData() {}
e8189707 88 // Decode raw data and store digits
89 virtual void RawData2Digits() {}
3bd79107 90
91 // Configuration Methods (per detector type )
3bd79107 92 // Set response
93 virtual void SetResponseModel(Int_t id, AliITSresponse *response);
94 // Set segmentation
95 virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
96 // Set simulation - temporary
97 virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim);
98 // Set reconstruction
99 virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
100 // Set class names for digit and rec point
e8189707 101 virtual void SetClasses(Int_t id, const char *digit, const char *cluster);
3bd79107 102
103
104 // Getters
58005f18 105 // ITS geometry functions
3bd79107 106 virtual AliITSgeom *GetITSgeom() const {return fITSgeom;}
e8189707 107 // return pointer to the array of modules
3bd79107 108 virtual TObjArray *GetModules() const {return fITSmodules;}
e8189707 109 // return pointer to a particular module
3bd79107 110 AliITSmodule *GetModule(Int_t index) {return (AliITSmodule *)
111 (fITSmodules->At(index));}
58005f18 112
3bd79107 113 // Return pointers to digits
114 TObjArray *Dtype() {return fDtype;}
115 Int_t *Ndtype() {return fNdtype;}
116 virtual TClonesArray *DigitsAddress(Int_t id)
117 {return ((TClonesArray *) (*fDtype)[id]);}
118 // Return pointers to clusters
119 TObjArray *Ctype() {return fCtype;}
120 Int_t *Nctype() {return fNctype;}
121 virtual TClonesArray *ClustersAddress(Int_t id)
122 {return ((TClonesArray *) (*fCtype)[id]);}
123
124 // Return pointer to rec points
125 TClonesArray *RecPoints() {return fRecPoints;}
3bd79107 126
127 // Return pointer to DetType #id
128 AliITSDetType *DetType(Int_t id);
e8189707 129 //Int_t NDetTypes() {return fNDetTypes;}
3bd79107 130
131 // Return pointer to the tree of clusters
132 TTree *TreeC() {return fTreeC;}
133
e8189707 134
3bd79107 135 protected:
136
b669392e 137 static const Int_t fgkNTYPES; // Number of detector types
e8189707 138
3bd79107 139 AliITSgeom *fITSgeom; // Pointer to ITS geometry
140 TObjArray *fITSmodules; // Pointer to ITS modules
141
142 Bool_t fEuclidOut; // Flag to write out geometry
e8189707 143 // in euclid format
3bd79107 144 Int_t fIdN; // the number of layers
2b2439bd 145 Int_t *fIdSens; //[fIdN] layer identifier
e8189707 146 TString *fIdName; //[fIdN] layer identifier
3bd79107 147 // Geometry and Stepmanager version numbers used.
148 Int_t fMajorVersion,fMinorVersion; //detailed and coarse(minor) versions
149 //
150 //
e8189707 151
152 // Int_t fNDetTypes; //Number of detector types
3bd79107 153 TObjArray *fDetTypes; // List of detector types
3bd79107 154
e8189707 155 TObjArray *fDtype; //[fgkNTYPES] List of digits
156 Int_t *fNdtype; //[fgkNTYPES] Number of digits per type of
3bd79107 157 // detector
e8189707 158 TObjArray *fCtype; //[fgkNTYPES] List of clusters
159 Int_t *fNctype; //[fgkNTYPES] Number of clusters per type
3bd79107 160 // of detector
161
3bd79107 162 TClonesArray *fRecPoints; // List of reconstructed points
163 Int_t fNRecPoints; // Number of rec points
3bd79107 164 TTree *fTreeC; // Tree for raw clusters
165
166
167 ClassDef(AliITS,1) // Base class for ITS
fe4da5cc 168};
3bd79107 169
fe4da5cc 170#endif