[u/mrichter/AliRoot.git] / ITS / AliITS.h

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

/* $Id$ */

////////////////////////////////////////////////////////////////////////
//           Manager class for set: ITS                               //
////////////////////////////////////////////////////////////////////////

#include <TObjArray.h> // used in inline function GetModule.
#include <TBranch.h>   // used in inline function SetHitsAddressBranch

#include "AliLoader.h"
#include "AliDetector.h"
#include "AliITSDetType.h"

class TString;
class TTree;
class TFile;

//class AliITSDetType;
class AliITSsimulation;
class AliITSClusterFinder;
class AliITSLoader;
class AliITSsegmentation;
class AliITSresponse;
class AliITShit;
class AliITSgeom;
class AliITSpListItem;
class AliITSdigit;
class AliITSRecPoint;
class AliITSRawCluster;
class AliITSmodule;
class AliVertexer;

const Int_t kNTYPES=3;

class AliITS : public AliDetector {

 public:
    //================= Standard Classes ===============================
    AliITS();  // Default creator.
    AliITS(const char *name, const char *title); // 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) 
	                   {a = GetMajorVersion();b=GetMinorVersion();return;}
    virtual void  SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
    //-------------------- 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));}

    //================ 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 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) -------------
    // 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);

    //=================== 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(TTree *treeH, Int_t mask = 0);
    virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
			     Option_t *opt, const char *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(GetLoader()->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
    void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
                 Option_t *add, Option_t *det, const char *filename);
    // Standard Hits To SDigits function
    void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
                 Option_t *add, Option_t *det, const char *filename);
    // Standard Hits To Digits function
    void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
                 Option_t *add, Option_t *det, const char *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]);}
    void SelectVertexer(TString sel=" "){fSelectedVertexer = sel;}

    //===================== 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, const char *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 AliVertexer* CreateVertexer() const;
    virtual void FillESD(AliESD* esd) const;
     
 protected:
    //================== Data Members ==================================
    AliITSgeom   *fITSgeom;    // Pointer to ITS geometry
    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        *fIdSens;     //[fIdN] layer identifier
    TString      *fIdName;     //[fIdN] layer identifier

    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
    TString fSelectedVertexer; // Vertexer selected in CreateVertexer

    ClassDef(AliITS,4) // Base class for ITS
};

#endif
Commit	Line	Data
3bd79107	1	#ifndef ALIITS_H
3bd79107	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
143d1056	12	#include <TObjArray.h> // used in inline function GetModule.
cd77595e	13	#include <TBranch.h> // used in inline function SetHitsAddressBranch
3bd79107	14
88cb7938	15	#include "AliLoader.h"
fe4da5cc	16	#include "AliDetector.h"
54b8cf3b	17	#include "AliITSDetType.h"
de82972f	18
d5da1ecf	19	class TString;
d5da1ecf	20	class TTree;
55b5a0b3	21	class TFile;
fe4da5cc	22
54b8cf3b	23	//class AliITSDetType;
3bd79107	24	class AliITSsimulation;
3bd79107	25	class AliITSClusterFinder;
88cb7938	26	class AliITSLoader;
de82972f	27	class AliITSsegmentation;
	28	class AliITSresponse;
	29	class AliITShit;
	30	class AliITSgeom;
2aea926d	31	class AliITSpListItem;
de82972f	32	class AliITSdigit;
	33	class AliITSRecPoint;
	34	class AliITSRawCluster;
	35	class AliITSmodule;
2257f27e	36	class AliVertexer;
27a043b1	37
a8a6107b	38	const Int_t kNTYPES=3;
de82972f	39
3bd79107	40	class AliITS : public AliDetector {
58005f18	41
2aea926d	42	public:
	43	//================= Standard Classes ===============================
	44	AliITS(); // Default creator.
	45	AliITS(const char name, const char title); // standard Creator
	46	virtual ~AliITS(); // destructor
ac74f489	47	AliITS(const AliITS &source); // copy constructor. Not to be used!
2aea926d	48	AliITS& operator=(AliITS &source); // = operator. Not to be used!
2aea926d	49	virtual Int_t IsVersion() const {return 1;}
93f82b23	50	virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) const;
2aea926d	51
2aea926d	52	//===================== Simulation Geometry ========================
e8189707	53	// get geometry version - detailed (major) or coarse (minor)
93f82b23	54	virtual Int_t GetMajorVersion() const {return -1;}
93f82b23	55	virtual Int_t GetMinorVersion() const {return -1;}
2aea926d	56	virtual void GetGeometryVersion(Int_t &a,Int_t &b)
2aea926d	57	{a = GetMajorVersion();b=GetMinorVersion();return;}
41b19549	58	virtual void SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
2aea926d	59	//-------------------- Geometry Transformations --------------------
	60	// ITS geometry functions
	61	AliITSgeom *GetITSgeom() const {return fITSgeom;}
f91c3e58	62	// Sets ITS geometry ! be very careful using this function.
f91c3e58	63	void SetITSgeom(AliITSgeom *geom) {fITSgeom = geom;}
2aea926d	64	// return pointer to the array of modules
	65	TObjArray *GetModules() const {return fITSmodules;}
	66	// return pointer to a particular module
	67	AliITSmodule GetModule(Int_t index) {return (AliITSmodule )
	68	(fITSmodules->At(index));}
3bd79107	69
88cb7938	70	//================ Necessary general Classes =======================
2aea926d	71	virtual void Init();
88cb7938	72	virtual AliLoader* MakeLoader(const char* topfoldername);
2aea926d	73	virtual void SetDefaults();
	74	virtual void SetDefaultSimulation();
	75	virtual void SetDefaultClusterFinders();
88cb7938	76	virtual void MakeBranch(Option_t *opt=" ");
2aea926d	77	virtual void SetTreeAddress();
93f82b23	78	// For a given branch from the treeH sets the TClonesArray address.
93f82b23	79	virtual void SetHitsAddressBranch(TBranch *b) {b->SetAddress(&fHits);}
2aea926d	80	// Return pointer to DetType #id
	81	AliITSDetType *DetType(Int_t id);
	82	//Int_t NDetTypes() {return fNDetTypes;}
	83	//---------- Configuration Methods (per detector type) -------------
7d8046e8	84	// Determines which ITS subdetectors will be processed. Effects
	85	// digitization, and Reconstruction only.
	86	void SetDetectors(Option_t *opt="All"){fOpt = opt;}
	87	// Returns the list of ITS subdetectors that will be processed.
	88	Option_t* GetDetectors(){return fOpt;}
3bd79107	89	// Set response
2aea926d	90	virtual void SetResponseModel(Int_t id, AliITSresponse *response);
3bd79107	91	// Set segmentation
2aea926d	92	virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg);
3bd79107	93	// Set simulation - temporary
2aea926d	94	virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim);
54b8cf3b	95	// Set simulation - temporary
	96	virtual AliITSsimulation* GetSimulationModel(Int_t id){
	97	return ((AliITSDetType*)(fDetTypes->At(id)))->GetSimulationModel();}
3bd79107	98	// Set reconstruction
2aea926d	99	virtual void SetReconstructionModel(Int_t id, AliITSClusterFinder *rec);
3bd79107	100	// Set class names for digit and rec point
2aea926d	101	virtual void SetClasses(Int_t id, const char digit, const char cluster);
	102
	103	//=================== Hits =========================================
	104	virtual void StepManager() {} // See Step Manager for specific geometry.
	105	virtual void AddHit(Int_t track, Int_t vol, Float_t hits);
	106	//------------ sort hits by module for Digitisation ----------------
	107	virtual void InitModules(Int_t size,Int_t &nmodules);
b4012daf	108	virtual void FillModules(TTree *treeH, Int_t mask = 0);
2aea926d	109	virtual void FillModules(Int_t evnt,Int_t bgrev,Int_t nmodules,
8e8eae84	110	Option_t opt, const char filename);
2aea926d	111	virtual void ClearModules();
	112
	113	//===================== Digitisation ===============================
	114	void MakeBranchS(const char *file);
	115	void SetTreeAddressS(TTree *treeS);
f91c3e58	116	TClonesArray * GetSDigits() { return fSDigits; }
9ad8b5dd	117	void MakeBranchInTreeD(TTree treeD,const char file=0);
9ad8b5dd	118	void MakeBranchD(const char *file){
8f726cb3	119	MakeBranchInTreeD(GetLoader()->TreeD(),file);}
2aea926d	120	void SetTreeAddressD(TTree *treeD);
	121	void Hits2SDigits(); // Turn hits into SDigits
	122	void Hits2PreDigits(); // Turn hits into SDigits
c92eb8ad	123	AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const;
e0fc0305	124	void SDigits2Digits(){SDigitsToDigits("All");} // Turn SDigits to Digits
e0fc0305	125	void SDigitsToDigits(Option_t *opt="All"); // Turn SDigits to Digits
2aea926d	126	void Hits2Digits(); // Turn hits straight into Digits.
	127	//------------------ Internal functions ----------------------------
	128	// Standard Hits To SDigits function
	129	void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
8e8eae84	130	Option_t add, Option_t det, const char *filename);
2aea926d	131	// Standard Hits To SDigits function
2aea926d	132	void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
8e8eae84	133	Option_t add, Option_t det, const char *filename);
2aea926d	134	// Standard Hits To Digits function
2aea926d	135	void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
8e8eae84	136	Option_t add, Option_t det, const char *filename);
2aea926d	137	void ResetSDigits(); // Resets the Summable digits.
	138	void ResetDigits(); // depending on how the
	139	void ResetDigits(Int_t branch); // tree will be filled only
	140	void AddSumDigit(AliITSpListItem &sdig);
	141	void AddRealDigit(Int_t branch, Int_t *digits);
	142	void AddSimDigit(Int_t branch, AliITSdigit *d);
	143	void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
	144	Int_t* tracks,Int_t hits,Float_t trkcharges);
	145	// Return pointers to digits
	146	TObjArray *Dtype() {return fDtype;}
	147	Int_t *Ndtype() {return fNdtype;}
	148	TClonesArray *DigitsAddress(Int_t id)
	149	{return ((TClonesArray ) (fDtype)[id]);}
2257f27e	150	void SelectVertexer(TString sel=" "){fSelectedVertexer = sel;}
3bd79107	151
2aea926d	152	//===================== Raw Data IO ================================
	153	// Write digits into raw data format
	154	virtual void Digits2RawData() {}
	155	// Decode raw data and store digits
	156	virtual void RawData2Digits() {}
58005f18	157
2aea926d	158	//==================== Clusterization ==============================
	159	// create separate tree for clusters - declustering refining
	160	void MakeTreeC(Option_t *option="C");
	161	void GetTreeC(Int_t event);
	162	void AddCluster(Int_t branch, AliITSRawCluster *c);
	163	void ResetClusters(); // one of the methods in
	164	void ResetClusters(Int_t branch); // the pair will be kept
3bd79107	165	// Return pointers to clusters
2aea926d	166	TObjArray *Ctype() {return fCtype;}
	167	Int_t *Nctype() {return fNctype;}
	168	TClonesArray *ClustersAddress(Int_t id)
3bd79107	169	{return ((TClonesArray ) (fCtype)[id]);}
3bd79107	170
2aea926d	171	//=================== Reconstruction ===============================
ff0e455e	172	void MakeBranchR(const char file, Option_t opt=" ");
ff0e455e	173	void MakeBranchRF(const char *file){MakeBranchR(file,"Fast");}
88cb7938	174	void MakeBranchC();
2aea926d	175	void SetTreeAddressR(TTree *treeR);
	176	void AddRecPoint(const AliITSRecPoint &p);
	177	void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
8e8eae84	178	Option_t add, Option_t det, const char *filename);
2aea926d	179	void Digits2Reco();
	180	void DigitsToRecPoints(Int_t evNumber,Int_t lastEntry,Option_t *det);
	181	void ResetRecPoints();
3bd79107	182	// Return pointer to rec points
3bd79107	183	TClonesArray *RecPoints() {return fRecPoints;}
596a855f	184
	185	virtual void Reconstruct() const;
	186	virtual AliTracker* CreateTracker() const;
2257f27e	187	virtual AliVertexer* CreateVertexer() const;
596a855f	188	virtual void FillESD(AliESD* esd) const;
88cb7938	189
2aea926d	190	protected:
	191	//================== Data Members ==================================
	192	AliITSgeom *fITSgeom; // Pointer to ITS geometry
e0fc0305	193	Bool_t fEuclidOut; // Flag to write geometry in euclid format
2aea926d	194	TObjArray *fITSmodules; //! Pointer to ITS modules
7d8046e8	195	Option_t *fOpt; //! Detector option ="All" unless changed.
3bd79107	196
e0fc0305	197	Int_t fIdN; // the number of layers
2aea926d	198	Int_t *fIdSens; //[fIdN] layer identifier
2aea926d	199	TString *fIdName; //[fIdN] layer identifier
3bd79107	200
e0fc0305	201	Int_t fNDetTypes; // Number of detector types
2aea926d	202	TObjArray *fDetTypes; // List of detector types
e8189707	203
88cb7938	204	TClonesArray *fSDigits; //! List of Summable digits.
2aea926d	205	Int_t fNSDigits; // Number of Summable Digits.
55b5a0b3	206
88cb7938	207	TObjArray *fDtype; //! List of digits
2aea926d	208	Int_t *fNdtype; //[fNDetTypes] Num. of digits per type of det.
3bd79107	209
88cb7938	210	TObjArray *fCtype; //! List of clusters
2aea926d	211	Int_t *fNctype; //[fNDetTypes] Num. of clust. per type of det.
3bd79107	212
88cb7938	213	TClonesArray *fRecPoints; //! List of reconstructed points
2aea926d	214	Int_t fNRecPoints; // Number of rec points
2257f27e	215	TString fSelectedVertexer; // Vertexer selected in CreateVertexer
3bd79107	216
2257f27e	217	ClassDef(AliITS,4) // Base class for ITS
fe4da5cc	218	};
3bd79107	219
fe4da5cc	220	#endif