[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSupgrade.h

#ifndef ALIITSUPGRADE_H
#define ALIITSUPGRADE_H

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

/* $Id$ */

//========================================================================
//
//      Geometry of the Inner Tracking System Upgrade based on TGeo
//
//
//   Authors: A. Mastroserio
//            C. Terrevoli
//            annalisa.mastroserio@cern.ch
//	      cristina.terrevoli@ba.infn.it
//
//========================================================================

//
//ITS class for new geometry based on TGeo
//

#include "AliITS.h"             //base class 
#include "AliITSupgradeDigitizer.h"
#include <TArrayS.h>
    
class TGeoVolume;
class TGeoVolumeAssembly;
class TArrayD;
class TClonesArray;
class TObjArray;
class AiITShit;
class AliITSsegmentationUpgrade;

class AliITSupgrade : public AliITS //TObject-TNamed-AliModule-AliDetector-AliITS-AliITSupgrade
{
 public:
  AliITSupgrade();                                                              //default ctor
  AliITSupgrade(const char *name, const char *title, Bool_t isBeamPipe=kFALSE);  //ctor for standard ITS 
  AliITSupgrade(const char *name, const char *title, TArrayD widths, TArrayD radii,TArrayD halfLengths, TArrayD radiiCu, TArrayD widthsCu, TArrayS copper,Bool_t bp, Double_t radiusBP, Double_t widthPB, Double_t halfLengthsBP);  //ctor
  virtual       ~AliITSupgrade();                                       //dtor
  Int_t IsVersion() const {return 110;}// same as ITSv11Hybrid to comply with the AliITSInitGeom cases
 
  
  //framework part+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 
  void    AddAlignableVolumes() const;             //from AliModule invoked from AliMC           
  void    CreateMaterials  ();                     //from AliModule invoked from AliMC
  void    CreateGeometry   ();                     //from AliModule invoked from AliMC                     
  void    StepManager      ();                     //from AliModule invoked from AliMC::Stepping()
  void    Hits2SDigits();
  void    MakeBranch      (Option_t *opt="");        //from AliModule invokde from AliRun::Tree2Tree() to make requested ITSupgrade branch
  void    SetTreeAddress  (                );        //from AliModule invoked from AliRun::GetEvent(), AliLoader::SetTAddrInDet()
          
  //create containers++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
  void     HitCreate()  	   {if(fHits)return; fHits=new TClonesArray("AliITShit"); fNhits=0;     }//create hits list 

  TObjArray*      SDigitsList()            const{return fSdigits;} //get digits list for all layers
  TClonesArray*   SDigitsList(Int_t layer) const{return fSdigits ? (TClonesArray *)fSdigits->At(layer):0;}//get sdigits list for a layer
  void            SDigitsCreate()            {if (fSdigits) return; 
    fSdigits=new  TObjArray(fNlayers);
    for(Int_t i=0;i<fNlayers;i++)fSdigits->AddAt(new TClonesArray("AliITSDigitUpgrade"),i);}//create digits list
  void           SDigitsReset ()              {if(fSdigits) for(int i=0;i<fNlayers;i++)fSdigits->At(i)->Clear();                     }//clean sdigits list  
     
     
     
  TObjArray*      DigitsList()            const{return fDigitArray;}//get digits list for all layers
  TClonesArray*   DigitsList(Int_t layer) const{return fDigitArray ? (TClonesArray *)fDigitArray->At(layer):0;}//get digits list for chamber
  void            DigitsCreate()            {if (fDigitArray) return; 
    fDigitArray=new TObjArray(fNlayers);
    for(Int_t i=0;i<fNlayers;i++)fDigitArray->AddAt(new TClonesArray("AliITSDigitUpgrade"),i);}//create digits list
  void          DigitsReset ()              {if(fDigitArray)for(int i=0;i<fNlayers;i++)fDigitArray->At(i)->Clear();                     }//clean digits list

  AliDigitizer*   CreateDigitizer  (AliDigitizationInput *m) const {return new AliITSupgradeDigitizer(m);}  //from AliModule invoked from AliSimulation::RunDigitization()
           
  void Hit2SumDig(TClonesArray *hits,const TObjArray *pSDig, Int_t *nSdigit);

  enum EMedia {kAir=0,kSi=1,kBe=2, kCu=3};                         //media ids used in CreateMaterials  
    

  // Geometry/segmentation creation part
  TGeoVolumeAssembly * CreateVol();
  TGeoVolumeAssembly * CreateVol(Bool_t cil);
  
  void SetFullSegmentation(TArrayD xsize, TArrayD zsize);

  void SetRadius(Double_t r, Int_t lay) {fRadii.AddAt(r,lay);}
  void SetWidth(Double_t w, Int_t lay) {fWidths.AddAt(w,lay);}
  void SetRadiusCu(Double_t rCu, Int_t lay) {fRadiiCu.AddAt(rCu,lay);}
  void SetWidthCu(Double_t wCu, Int_t lay) {fWidthsCu.AddAt(wCu,lay);}
  void SetSegmentationX(Double_t x, Int_t lay);
  void SetSegmentationZ(Double_t z, Int_t lay);
 void SetNSectors(Int_t nsect) {fNSectors=nsect;}
  void StepHistory();
  void PrintSummary();

 protected:
  TArrayD fWidths;   			      // thicknesses of silicon cylinders
  TArrayD fRadii;    			      // radii of silicon cylinders
  Int_t fNSectors;   			      // number of sectors
  TArrayD fRadiiCu;  			      // radii of copper cylinders 
  TArrayD fWidthsCu; 			      // thicknesses of copper cylinders
  TArrayS   fCopper; 			      // check to set the copper cylinder 
  Bool_t fBeampipe;                           // boolean to set beam pipe vol in geometry
  Double_t fRadiusBP;                         // beam pipe radius
  Double_t fWidthBP; 			      // thickness of the beam pipe		
  Double_t fHalfLengthBP;		      // 1/2 beam pipe cyclinder length
  Int_t   fNlayers;			      // # of layers
  TArrayD fHalfLength;			      // 1/2 silicon and copper cylinder lengths
  TObjArray            *fSdigits;             //![fNlayers] list of sdigits
  TObjArray            *fDigitArray;          //![fNlayers] list of digits per layer
  AliITSsegmentationUpgrade *fSegmentation;   //virtual segmentation class
  
 private:
  AliITSupgrade(const AliITSupgrade& u);              //dummy copy constructor
  AliITSupgrade &operator=(const AliITSupgrade& u);   //dummy assignment operator

  ClassDef(AliITSupgrade,2)                                                                //ITS Upgrade full version for simulation
  
    };

#endif
Commit	Line	Data
1d9af2d5	1	#ifndef ALIITSUPGRADE_H
	2	#define ALIITSUPGRADE_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
	8
	9	//========================================================================
	10	//
	11	// Geometry of the Inner Tracking System Upgrade based on TGeo
	12	//
	13	//
	14	// Authors: A. Mastroserio
	15	// C. Terrevoli
	16	// annalisa.mastroserio@cern.ch
	17	// cristina.terrevoli@ba.infn.it
	18	//
	19	//========================================================================
	20
	21	//
	22	//ITS class for new geometry based on TGeo
	23	//
	24
	25	#include "AliITS.h" //base class
1d9af2d5	26	#include "AliITSupgradeDigitizer.h"
1d9af2d5	27	#include <TArrayS.h>
0ac80088	28
1d9af2d5	29	class TGeoVolume;
	30	class TGeoVolumeAssembly;
	31	class TArrayD;
	32	class TClonesArray;
	33	class TObjArray;
	34	class AiITShit;
a19b4e8c	35	class AliITSsegmentationUpgrade;
a19b4e8c	36
1d9af2d5	37	class AliITSupgrade : public AliITS //TObject-TNamed-AliModule-AliDetector-AliITS-AliITSupgrade
	38	{
	39	public:
387f1a9e	40	AliITSupgrade(); //default ctor
ebaa7604	41	AliITSupgrade(const char name, const char title, Bool_t isBeamPipe=kFALSE); //ctor for standard ITS
1d9af2d5	42	AliITSupgrade(const char name, const char title, TArrayD widths, TArrayD radii,TArrayD halfLengths, TArrayD radiiCu, TArrayD widthsCu, TArrayS copper,Bool_t bp, Double_t radiusBP, Double_t widthPB, Double_t halfLengthsBP); //ctor
1d9af2d5	43	virtual ~AliITSupgrade(); //dtor
84ab81bf	44	Int_t IsVersion() const {return 110;}// same as ITSv11Hybrid to comply with the AliITSInitGeom cases
84ab81bf	45
1d9af2d5	46
	47	//framework part+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	48	void AddAlignableVolumes() const; //from AliModule invoked from AliMC
	49	void CreateMaterials (); //from AliModule invoked from AliMC
	50	void CreateGeometry (); //from AliModule invoked from AliMC
1d9af2d5	51	void StepManager (); //from AliModule invoked from AliMC::Stepping()
	52	void Hits2SDigits();
	53	void MakeBranch (Option_t *opt=""); //from AliModule invokde from AliRun::Tree2Tree() to make requested ITSupgrade branch
	54	void SetTreeAddress ( ); //from AliModule invoked from AliRun::GetEvent(), AliLoader::SetTAddrInDet()
	55
	56	//create containers++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	57	void HitCreate() {if(fHits)return; fHits=new TClonesArray("AliITShit"); fNhits=0; }//create hits list
	58
9b615954	59	TObjArray* SDigitsList() const{return fSdigits;} //get digits list for all layers
1d9af2d5	60	TClonesArray* SDigitsList(Int_t layer) const{return fSdigits ? (TClonesArray *)fSdigits->At(layer):0;}//get sdigits list for a layer
	61	void SDigitsCreate() {if (fSdigits) return;
	62	fSdigits=new TObjArray(fNlayers);
	63	for(Int_t i=0;i<fNlayers;i++)fSdigits->AddAt(new TClonesArray("AliITSDigitUpgrade"),i);}//create digits list
	64	void SDigitsReset () {if(fSdigits) for(int i=0;i<fNlayers;i++)fSdigits->At(i)->Clear(); }//clean sdigits list
	65
	66
	67
a19b4e8c	68	TObjArray* DigitsList() const{return fDigitArray;}//get digits list for all layers
	69	TClonesArray* DigitsList(Int_t layer) const{return fDigitArray ? (TClonesArray *)fDigitArray->At(layer):0;}//get digits list for chamber
	70	void DigitsCreate() {if (fDigitArray) return;
	71	fDigitArray=new TObjArray(fNlayers);
	72	for(Int_t i=0;i<fNlayers;i++)fDigitArray->AddAt(new TClonesArray("AliITSDigitUpgrade"),i);}//create digits list
	73	void DigitsReset () {if(fDigitArray)for(int i=0;i<fNlayers;i++)fDigitArray->At(i)->Clear(); }//clean digits list
1d9af2d5	74
f21fc003	75	AliDigitizer* CreateDigitizer (AliDigitizationInput *m) const {return new AliITSupgradeDigitizer(m);} //from AliModule invoked from AliSimulation::RunDigitization()
1d9af2d5	76
a19b4e8c	77	void Hit2SumDig(TClonesArray hits,const TObjArray pSDig, Int_t *nSdigit);
1d9af2d5	78
	79	enum EMedia {kAir=0,kSi=1,kBe=2, kCu=3}; //media ids used in CreateMaterials
	80
	81
	82	// Geometry/segmentation creation part
	83	TGeoVolumeAssembly * CreateVol();
0ac80088	84	TGeoVolumeAssembly * CreateVol(Bool_t cil);
0ac80088	85
1d9af2d5	86	void SetFullSegmentation(TArrayD xsize, TArrayD zsize);
1d9af2d5	87
387f1a9e	88	void SetRadius(Double_t r, Int_t lay) {fRadii.AddAt(r,lay);}
	89	void SetWidth(Double_t w, Int_t lay) {fWidths.AddAt(w,lay);}
	90	void SetRadiusCu(Double_t rCu, Int_t lay) {fRadiiCu.AddAt(rCu,lay);}
	91	void SetWidthCu(Double_t wCu, Int_t lay) {fWidthsCu.AddAt(wCu,lay);}
	92	void SetSegmentationX(Double_t x, Int_t lay);
	93	void SetSegmentationZ(Double_t z, Int_t lay);
0ac80088	94	void SetNSectors(Int_t nsect) {fNSectors=nsect;}
1d9af2d5	95	void StepHistory();
387f1a9e	96	void PrintSummary();
1d9af2d5	97
1d9af2d5	98	protected:
a19b4e8c	99	TArrayD fWidths; // thicknesses of silicon cylinders
	100	TArrayD fRadii; // radii of silicon cylinders
	101	Int_t fNSectors; // number of sectors
	102	TArrayD fRadiiCu; // radii of copper cylinders
	103	TArrayD fWidthsCu; // thicknesses of copper cylinders
	104	TArrayS fCopper; // check to set the copper cylinder
	105	Bool_t fBeampipe; // boolean to set beam pipe vol in geometry
	106	Double_t fRadiusBP; // beam pipe radius
	107	Double_t fWidthBP; // thickness of the beam pipe
	108	Double_t fHalfLengthBP; // 1/2 beam pipe cyclinder length
	109	Int_t fNlayers; // # of layers
	110	TArrayD fHalfLength; // 1/2 silicon and copper cylinder lengths
	111	TObjArray *fSdigits; //![fNlayers] list of sdigits
	112	TObjArray *fDigitArray; //![fNlayers] list of digits per layer
	113	AliITSsegmentationUpgrade *fSegmentation; //virtual segmentation class
1d9af2d5	114
	115	private:
	116	AliITSupgrade(const AliITSupgrade& u); //dummy copy constructor
	117	AliITSupgrade &operator=(const AliITSupgrade& u); //dummy assignment operator
	118
	119	ClassDef(AliITSupgrade,2) //ITS Upgrade full version for simulation
	120
	121	};
	122
	123	#endif
	124