[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 "AliDetector.h"
#include "AliITSTrigger.h"
#include "AliITSDetTypeSim.h"


class TString;
class TTree;
class AliITSpListItem;
class AliITSsimulation;
class AliITSsegmentation;

class AliITSCalibration;
class AliITShit;
class AliITSgeom;
class AliITSdigit;
class AliITSmodule;
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_t *title); // standard Creator
    AliITS(const char *name, const char *title); // extended standard Creator
    virtual ~AliITS(); // destructor
    virtual Int_t IsVersion() const {return 1;}

    //===================== 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) 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 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 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();
     //---------- 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 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.
    //------------ sort hits by module for Digitisation ----------------
    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 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, 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   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<fgkNTYPES) fkRawID2ClusID[iDet] = arr;}
    const TArrayI* GetRawID2ClusID(Int_t iDet) const {return (iDet>-1&&iDet<fgkNTYPES) ? fkRawID2ClusID[iDet]:0;}

 protected:
    static const Int_t fgkNTYPES=3; //number of detector types
    //================== Data Members ==================================
    AliITSDetTypeSim *fDetTypeSim; //detector type for simulation
    Bool_t        fEuclidOut;  // Flag to write geometry in euclid format
    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
    TObjArray    *fITSmodules; //! Pointer to ITS modules
    Bool_t        fTiming;     // flag to turn on/off timers.
    AliITSSimuParam* fSimuParam; //simulation parameters
    TClonesArray** fModA;      //! Used by Raw2SDigits (one TC per module)
    TClonesArray* fpSDigits;   //! Branch address to build SD from raw data 
    const TArrayI* fkRawID2ClusID[fgkNTYPES]; //! optional array for SDigit->Cluster 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

};

#endif
Commit	Line	Data
	1	#ifndef ALIITS_H
	2	#define ALIITS_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 set: ITS //
	10	////////////////////////////////////////////////////////////////////////
	11
	12
	13	#include <TObjArray.h> // used in inline function GetModule.
	14	#include "AliDetector.h"
	15	#include "AliITSTrigger.h"
	16	#include "AliITSDetTypeSim.h"
	17
	18
	19	class TString;
	20	class TTree;
	21	class AliITSpListItem;
	22	class AliITSsimulation;
	23	class AliITSsegmentation;
	24
	25	class AliITSCalibration;
	26	class AliITShit;
	27	class AliITSgeom;
	28	class AliITSdigit;
	29	class AliITSmodule;
	30	class AliDigitizationInput;
	31	class TArrayI;
	32
	33	class AliITS : public AliDetector {
	34
	35	public:
	36	enum {kSPD,kSDD,kSSD};
	37	enum {kMaxLabel=3141593}; // default track label assigned to digits/clusters
	38
	39	//================= Standard Classes ===============================
	40	AliITS(); // Default creator.
	41	AliITS(const Char_t *title); // standard Creator
	42	AliITS(const char name, const char title); // extended standard Creator
	43	virtual ~AliITS(); // destructor
	44	virtual Int_t IsVersion() const {return 1;}
	45
	46	//===================== Simulation Geometry ========================
	47	// get geometry version - detailed (major) or coarse (minor)
	48	virtual Int_t GetMajorVersion() const {return -1;}
	49	virtual Int_t GetMinorVersion() const {return -1;}
	50	virtual void GetGeometryVersion(Int_t &a,Int_t &b) const
	51	{a = GetMajorVersion();b=GetMinorVersion();return;}
	52	virtual void SetEUCLID(Bool_t euclid=kTRUE) {fEuclidOut = euclid;}
	53	virtual Bool_t GetEUCLID()const {return fEuclidOut;}
	54	//-------------------- Geometry Transformations --------------------
	55
	56	// ITS geometry functions From Simulation
	57	AliITSgeom* GetITSgeom() const {return fDetTypeSim->GetITSgeom();}
	58	void SetITSgeom(AliITSgeom *geom) {fDetTypeSim->SetITSgeom(geom);}
	59	// return pointer to the array of modules
	60	TObjArray *GetModules(){return fITSmodules;}
	61
	62	AliITSmodule *GetModule(Int_t index){
	63	return (AliITSmodule*)(fITSmodules->At(index));}
	64	virtual void SetSimuParam(AliITSSimuParam *sp){
	65	fSimuParam=sp;
	66	fDetTypeSim->SetSimuParam(sp);
	67	}
	68	AliITSSimuParam* GetSimuParam() const {return fSimuParam;}
	69
	70	virtual void SetDetTypeSim(AliITSDetTypeSim* dts) {fDetTypeSim=dts;}
	71	AliITSDetTypeSim* GetDetTypeSim() const {return fDetTypeSim;}
	72	//================ Necessary general Classes =======================
	73	virtual void Init();
	74	virtual AliLoader* MakeLoader(const char* topfoldername);
	75	virtual void SetDefaults();
	76	virtual void SetDefaultSimulation();
	77	virtual void MakeBranch(Option_t *opt=" ");
	78	virtual void MakeBranchS(const char* fl);
	79	virtual void MakeBranchD(const char* file);
	80	virtual void MakeBranchInTreeD(TTree* treeD, const char* file=0);
	81	virtual void SetTreeAddress();
	82	//---------- Configuration Methods (per detector type) -------------
	83	// Determines which ITS subdetectors will be processed. Effects
	84	// digitization, and Reconstruction only.
	85	void SetDetectors(Option_t *opt="All"){fOpt = opt;}
	86	// Returns the list of ITS subdetectors that will be processed.
	87	Option_t* GetDetectors(){return fOpt;}
	88
	89	// Set calibration
	90	virtual void SetCalibrationModel(Int_t dettype, AliITSCalibration *cal){
	91	fDetTypeSim->SetCalibrationModel(dettype,cal);}
	92	// Set segmentation for Simulation
	93	virtual void SetSegmentationModel(Int_t id, AliITSsegmentation *seg){
	94	fDetTypeSim->SetSegmentationModel(id,seg);}
	95	// Set simulation
	96	virtual void SetSimulationModel(Int_t id, AliITSsimulation *sim){
	97	fDetTypeSim->SetSimulationModel(id,sim);}
	98	// Set simulation
	99	virtual AliITSsimulation* GetSimulationModel(Int_t id){
	100	return fDetTypeSim->GetSimulationModel(id);}
	101	//=================== Hits =========================================
	102	virtual void StepManager() {} // See Step Manager for specific geometry.
	103	//------------ sort hits by module for Digitisation ----------------
	104	virtual void FillModules(Int_t /* evnt /,Int_t bgrev,Int_t / nmodules */,
	105	Option_t opt, const char filename);
	106	virtual Bool_t InitModules(Int_t size,Int_t &nmodules);
	107	virtual void FillModules(TTree *treeH, Int_t mask = 0);
	108	virtual void ClearModules(){if(fITSmodules) fITSmodules->Delete();}
	109	virtual void AddHit(Int_t track, Int_t vol, Float_t hits);
	110	// Trigger
	111	virtual AliTriggerDetector* CreateTriggerDetector() const;
	112
	113	AliDigitizer* CreateDigitizer(AliDigitizationInput* digInput) const;
	114	virtual void UpdateInternalGeometry();
	115	virtual void SDigitsToDigits(Option_t *opt="All");
	116	virtual void SDigits2Digits(){SDigitsToDigits("All");}
	117	virtual void Hits2Digits();
	118	virtual void Hits2SDigits();
	119	virtual void Hits2PreDigits();
	120	virtual void HitsToDigits(Int_t evNumber,Int_t bgrev,Int_t size,
	121	Option_t option,Option_t opt,
	122	const char *filename);
	123	virtual void HitsToPreDigits(Int_t evNumber,Int_t bgrev,Int_t size,
	124	Option_t option,Option_t opt,
	125	const char *filename);
	126	void HitsToSDigits(Int_t evNumber,Int_t bgrev,Int_t size,
	127	Option_t add, Option_t det, const char *filename)
	128	{HitsToPreDigits(evNumber,bgrev,size,add,det,filename);}
	129
	130	virtual void ResetDigits();
	131	virtual void ResetDigits(Int_t branch);
	132	virtual void AddSumDigit(AliITSpListItem &sdig);
	133	virtual void AddSimDigit(Int_t branch, AliITSdigit *d);
	134	virtual void AddSimDigit(Int_t branch,Float_t phys,Int_t* digits,
	135	Int_t* tracks,Int_t hits,Float_t trkcharges,
	136	Int_t sigexpanded=-1000);
	137	TObjArray* GetDigits() const {return fDetTypeSim->GetDigits();}
	138	Int_t* GetNDigitArray() const {return fDetTypeSim->GetNDigitArray();}
	139	TClonesArray *DigitsAddress(Int_t id) {
	140	return fDetTypeSim->DigitsAddress(id);}
	141	//Fast simulation
	142	virtual void HitsToFastRecPoints(Int_t evNumber,Int_t bgrev,Int_t size,
	143	Option_t add, Option_t det, const char *filename);
	144	virtual Int_t Hits2Clusters(TTree in, TTree out);
	145	virtual void CheckLabels(Int_t lab[3]) const;
	146
	147	//===================== Raw Data IO ================================
	148	// Write digits into raw data format
	149	virtual void Digits2Raw();
	150	virtual Bool_t Raw2SDigits(AliRawReader* rawReader);
	151
	152	//===================== FO signals ================================
	153	// Write FO signals in UserInfo of SDigits/Digits tree
	154	void WriteFOSignals();
	155	void SetRawID2ClusID(const TArrayI* arr, Int_t iDet) { if (iDet>-1&&iDet<fgkNTYPES) fkRawID2ClusID[iDet] = arr;}
	156	const TArrayI* GetRawID2ClusID(Int_t iDet) const {return (iDet>-1&&iDet<fgkNTYPES) ? fkRawID2ClusID[iDet]:0;}
	157
	158	protected:
	159	static const Int_t fgkNTYPES=3; //number of detector types
	160	//================== Data Members ==================================
	161	AliITSDetTypeSim *fDetTypeSim; //detector type for simulation
	162	Bool_t fEuclidOut; // Flag to write geometry in euclid format
	163	Option_t *fOpt; //! Detector option ="All" unless changed.
	164	Int_t fIdN; // the number of layers
	165	Int_t *fIdSens; //[fIdN] layer identifier
	166	TString *fIdName; //[fIdN] layer identifier
	167	TObjArray *fITSmodules; //! Pointer to ITS modules
	168	Bool_t fTiming; // flag to turn on/off timers.
	169	AliITSSimuParam* fSimuParam; //simulation parameters
	170	TClonesArray** fModA; //! Used by Raw2SDigits (one TC per module)
	171	TClonesArray* fpSDigits; //! Branch address to build SD from raw data
	172	const TArrayI* fkRawID2ClusID[fgkNTYPES]; //! optional array for SDigit->Cluster assingment in Raw2SDigit (for embedding)
	173	private:
	174	AliITS(const AliITS &source); // copy constructor. Not to be used!
	175	AliITS& operator=(const AliITS &source); // = operator. Not to be used!
	176	ClassDef(AliITS,9) // Base class for ITS
	177
	178	};
	179
	180	#endif