135d49f6d9fb0338f373a8f2ff1077c28ae6c12e
[u/mrichter/AliRoot.git] / TFluka / TFlukaMCGeometry.h
1 #ifndef TFLUKAMCGEOMETRY_H
2 #define TFLUKAMCGEOMETRY_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 //
7 // Class TFlukaMCGeometry
8 // --------------------
9 // Implementation of the TVirtualMCGeometry interface
10 // for defining and using TGeo geometry with FLUKA.
11 //
12 // Author: Andrei Gheata 10/07/2003
13
14 #ifndef ROOT_TNamed
15 #include "TNamed.h"
16 #endif
17
18 class TGeoMaterial;
19
20
21 class TFlukaMCGeometry :public TNamed {
22
23   public:
24     enum EFlukaLatticeTypes {
25        kLttcOutside = 999999999,
26        kLttcVirtual = 1000000000
27     };   
28   
29     TFlukaMCGeometry();
30     TFlukaMCGeometry(const char* name, const char* title);
31     virtual ~TFlukaMCGeometry();
32   
33     // get methods
34     Int_t         GetMedium() const;
35     Int_t        *GetRegionList(Int_t imed, Int_t &nreg);
36     Int_t        *GetMaterialList(Int_t imat, Int_t &nreg);
37     Int_t         GetFlukaMaterial(Int_t imed) const;
38     Int_t         GetLastMaterialIndex() const {return fLastMaterial;}
39     virtual Int_t NofVolumes() const;
40    // FLUKA specific methods
41     void          CreateFlukaMatFile(const char *fname=0);
42     void          CreatePemfFile();
43     void          PrintHeader(ofstream &out, const char *text) const;
44     Bool_t        IsDebugging() const {return fDebug;}
45     void          SetDebugMode(Bool_t flag=kTRUE) {fDebug = flag;}
46     void          SetMreg(Int_t mreg);
47     void          SetCurrentRegion(Int_t mreg, Int_t latt);
48     void          GetCurrentRegion(Int_t &mreg, Int_t &latt) const {mreg=fCurrentRegion; latt=fCurrentLattice;}
49     void          SetNextRegion(Int_t mreg, Int_t latt);
50     void          GetNextRegion(Int_t &mreg, Int_t &latt) const {mreg=fNextRegion; latt=fNextLattice;}
51     TGeoMaterial *GetMakeWrongMaterial(Double_t z);
52     TObjArray    *GetMatList() {return fMatList;}
53     TObjArray    *GetMatNames() {return fMatNames;}
54     Int_t         GetElementIndex(Int_t z) const;
55     Int_t         RegionId() const; 
56     void          ToFlukaString(TString &str) const;
57     void          FlukaMatName(TString &str) const;
58     void          WritePegFile(Int_t imat, Int_t *NoStern, Int_t *ElemError,
59                        Int_t *MixError, Int_t *countGas) const;
60     Double_t *    GetISSB(Double_t rho, Int_t nElem, Double_t *zelem, Double_t *welem ) const;
61
62     Double_t* CreateDoubleArray(Float_t* array, Int_t size) const;
63     void     Vname(const char *name, char *vname) const;
64    
65   private:
66     TFlukaMCGeometry(const TFlukaMCGeometry& rhs);
67     TFlukaMCGeometry& operator=(const TFlukaMCGeometry& rhs);
68
69     static TFlukaMCGeometry*  fgInstance; // singleton instance
70     Bool_t       fDebug;                  // debug flag
71     Int_t        fLastMaterial;           // last FLUKA material index
72     Int_t        fCurrentRegion;          // current region number
73     Int_t        fCurrentLattice;         // current lattice history
74     Int_t        fNextRegion;             // next region number
75     Int_t        fNextLattice;            // next lattice history
76     Int_t       *fRegionList;             //! region list matching a given medium number
77     Int_t        fIndmat;                 // material index where pemf file creation starts
78     TObjArray   *fMatList;                //! material list as known by FLUKA
79     TObjArray   *fMatNames;               //! list of FLUKA material names
80   ClassDef(TFlukaMCGeometry,1)  //Virtual MonteCarlo Interface
81 };
82
83 #endif //ROOT_TFlukaMCGeometry