TFluka/TFluka.h

   1 #ifndef TFLUKA
   2 #define TFLUKA
   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 //                                                                           //
  10 //                                                                           //
  11 // FLUKA implementation of the AliMC Interface                               //
  12 //                                                                           //
  13 //                                                                           //
  14 ///////////////////////////////////////////////////////////////////////////////
  15
  16
  17 #include "TVirtualMC.h"
  18 #include "TMCProcess.h"
  19
  20 #include <map>
  21 #include <vector>
  22
  23 //Forward declaration
  24 class TG4GeometryManager;
  25 class TG4DetConstruction;
  26 class TClonesArray;
  27
  28
  29 class TFluka : public TVirtualMC {
  30
  31  public:
  32   TFluka(const char *title, Int_t verbosity = 0);
  33   TFluka();
  34   virtual ~TFluka();
  35
  36   //
  37   // methods for building/management of geometry
  38   // ------------------------------------------------
  39   //
  40
  41   // functions from GCONS
  42   virtual void  Gfmate(Int_t imat, char *name, Float_t &a, Float_t &z,
  43                        Float_t &dens, Float_t &radl, Float_t &absl,
  44                        Float_t* ubuf, Int_t& nbuf);
  45   virtual void  Gfmate(Int_t imat, char *name, Double_t &a, Double_t &z,
  46                        Double_t &dens, Double_t &radl, Double_t &absl,
  47                        Double_t* ubuf, Int_t& nbuf);
  48
  49   // detector composition
  50   virtual void  Material(Int_t& kmat, const char* name, Double_t a,
  51                          Double_t z, Double_t dens, Double_t radl, Double_t absl,
  52                          Float_t* buf, Int_t nwbuf);
  53   virtual void  Material(Int_t& kmat, const char* name, Double_t a,
  54                          Double_t z, Double_t dens, Double_t radl, Double_t absl,
  55                          Double_t* buf, Int_t nwbuf);
  56   virtual void  Mixture(Int_t& kmat, const char *name, Float_t *a,
  57                         Float_t *z, Double_t dens, Int_t nlmat, Float_t *wmat);
  58   virtual void  Mixture(Int_t& kmat, const char *name, Double_t *a,
  59                         Double_t *z, Double_t dens, Int_t nlmat, Double_t *wmat);
  60   virtual void  Medium(Int_t& kmed, const char *name, Int_t nmat,
  61                        Int_t isvol, Int_t ifield, Double_t fieldm, Double_t tmaxfd,
  62                        Double_t stemax, Double_t deemax, Double_t epsil,
  63                        Double_t stmin, Float_t* ubuf, Int_t nbuf);
  64   virtual void  Medium(Int_t& kmed, const char *name, Int_t nmat,
  65                        Int_t isvol, Int_t ifield, Double_t fieldm, Double_t tmaxfd,
  66                        Double_t stemax, Double_t deemax, Double_t epsil,
  67                        Double_t stmin, Double_t* ubuf, Int_t nbuf);
  68   virtual void  Matrix(Int_t& krot, Double_t thetaX, Double_t phiX,
  69                        Double_t thetaY, Double_t phiY, Double_t thetaZ,
  70                        Double_t phiZ);
  71   virtual void  Gstpar(Int_t itmed, const char *param, Double_t parval);
  72
  73   // functions from GGEOM
  74   virtual Int_t  Gsvolu(const char *name, const char *shape, Int_t nmed,
  75                         Float_t *upar, Int_t np);
  76   virtual Int_t  Gsvolu(const char *name, const char *shape, Int_t nmed,
  77                         Double_t *upar, Int_t np);
  78   virtual void  Gsdvn(const char *name, const char *mother, Int_t ndiv,
  79                       Int_t iaxis);
  80   virtual void  Gsdvn2(const char *name, const char *mother, Int_t ndiv,
  81                        Int_t iaxis, Double_t c0i, Int_t numed);
  82   virtual void  Gsdvt(const char *name, const char *mother, Double_t step,
  83                       Int_t iaxis, Int_t numed, Int_t ndvmx);
  84   virtual void  Gsdvt2(const char *name, const char *mother, Double_t step,
  85                        Int_t iaxis, Double_t c0, Int_t numed, Int_t ndvmx);
  86   virtual void  Gsord(const char *name, Int_t iax);
  87   virtual void  Gspos(const char *name, Int_t nr, const char *mother,
  88                       Double_t x, Double_t y, Double_t z, Int_t irot,
  89                       const char *konly="ONLY");
  90   virtual void  Gsposp(const char *name, Int_t nr, const char *mother,
  91                        Double_t x, Double_t y, Double_t z, Int_t irot,
  92                        const char *konly, Float_t *upar, Int_t np);
  93   virtual void  Gsposp(const char *name, Int_t nr, const char *mother,
  94                        Double_t x, Double_t y, Double_t z, Int_t irot,
  95                        const char *konly, Double_t *upar, Int_t np);
  96   virtual void  Gsbool(const char* onlyVolName, const char* manyVolName);
  97
  98   virtual void  SetCerenkov(Int_t itmed, Int_t npckov, Float_t *ppckov,
  99                             Float_t *absco, Float_t *effic, Float_t *rindex);
 100   virtual void  SetCerenkov(Int_t itmed, Int_t npckov, Double_t *ppckov,
 101                             Double_t *absco, Double_t *effic, Double_t *rindex);
 102
 103
 104   // functions for drawing
 105   virtual void  DrawOneSpec(const char* name)
 106     {printf("WARNING: DrawOneSpec not yet implemented !\n");}
 107   virtual void  Gsatt(const char* name, const char* att, Int_t val)
 108     {printf("WARNING: Gsatt not yet implemented !\n");}
 109   virtual void  Gdraw(const char*,Double_t theta = 30, Double_t phi = 30,
 110                       Double_t psi = 0, Double_t u0 = 10, Double_t v0 = 10,
 111                       Double_t ul = 0.01, Double_t vl = 0.01)
 112     {printf("WARNING: Gdraw not yet implemented !\n");}
 113
 114   // Euclid
 115   virtual void  WriteEuclid(const char*, const char*, Int_t, Int_t);
 116
 117   // get methods
 118   virtual Int_t VolId(const Text_t* volName) const;
 119   virtual const char* VolName(Int_t id) const;
 120   virtual Int_t NofVolumes() const
 121     {printf("NofVolumes %d\n", fNVolumes); return fNVolumes;}
 122   virtual Int_t VolId2Mate(Int_t id) const;
 123   //
 124   // methods for physics management
 125   // ------------------------------------------------
 126   //
 127
 128   // set methods
 129   virtual void     SetCut(const char* cutName, Double_t cutValue)
 130     {printf("WARNING: SetCut not yet implemented !\n");}
 131   virtual void     SetProcess(const char* flagName, Int_t flagValue)
 132     {printf("WARNING: SetProcess not yet implemented !\n");}
 133   virtual Double_t Xsec(char*, Double_t, Int_t, Int_t)
 134     {printf("WARNING: Xsec not yet implemented !\n"); return -1.;}
 135
 136   // particle table usage
 137   virtual Int_t   IdFromPDG(Int_t id) const;
 138   virtual Int_t   PDGFromId(Int_t pdg) const;
 139   virtual void    DefineParticles()
 140   {printf("WARNING: DefineParticles not yet implemented !\n");}
 141
 142   //
 143   // methods for step management
 144   // ------------------------------------------------
 145   //
 146
 147   // action methods
 148   virtual void StopTrack()
 149     {printf("WARNING: StopTrack not yet implemented !\n");}
 150   virtual void StopEvent()
 151     {printf("WARNING: StopEvent not yet implemented !\n");}
 152
 153   // set methods
 154   virtual void SetMaxStep(Double_t);
 155   virtual void SetMaxNStep(Int_t);
 156   virtual void SetUserDecay(Int_t);
 157
 158   // get methods
 159   // tracking volume(s)
 160   virtual Int_t    CurrentVolID(Int_t& copyNo) const;
 161   virtual Int_t    CurrentVolOffID(Int_t off, Int_t& copyNo) const;
 162   virtual const char* CurrentVolName() const;
 163   virtual const char* CurrentVolOffName(Int_t off) const;
 164   virtual Int_t    CurrentMaterial(Float_t &a, Float_t &z,
 165                                    Float_t &dens, Float_t &radl, Float_t &absl) const;
 166   virtual Int_t    CurrentEvent() const
 167       {printf("WARNING: CurrentEvent not yet implemented !\n"); return -1;}
 168   virtual void     Gmtod(Float_t* xm, Float_t* xd, Int_t iflag)
 169     {printf("WARNING: Gmtod not yet implemented !\n");}
 170   virtual void     Gmtod(Double_t* xm, Double_t* xd, Int_t iflag)
 171     {printf("WARNING: Gmtod not yet implemented !\n");}
 172   virtual void     Gdtom(Float_t* xd, Float_t* xm, Int_t iflag)
 173     {printf("WARNING: Gdtom not yet implemented !\n");}
 174   virtual void     Gdtom(Double_t* xd, Double_t* xm, Int_t iflag)
 175     {printf("WARNING: Gdtom not yet implemented !\n");}
 176   virtual Double_t MaxStep() const
 177     {printf("WARNING: MaxStep not yet implemented !\n"); return -1.;}
 178   virtual Int_t    GetMaxNStep() const
 179     {printf("WARNING: GetMaxNStep not yet implemented !\n"); return -1;}
 180   virtual Int_t    GetMedium() const;
 181
 182   // tracking particle
 183   // dynamic properties
 184   virtual void     TrackPosition(TLorentzVector& position) const;
 185   virtual void     TrackMomentum(TLorentzVector& momentum) const;
 186   virtual Double_t TrackStep() const;
 187   virtual Double_t TrackLength() const;
 188   virtual Double_t TrackTime() const;
 189   virtual Double_t Edep() const;
 190   // static properties
 191   virtual Int_t    TrackPid() const;
 192   virtual Double_t TrackCharge() const;
 193   virtual Double_t TrackMass() const;
 194   virtual Double_t Etot() const;
 195   // track status
 196   virtual Bool_t   IsNewTrack() const;
 197   virtual Bool_t   IsTrackInside() const;
 198   virtual Bool_t   IsTrackEntering() const;
 199   virtual Bool_t   IsTrackExiting() const;
 200   virtual Bool_t   IsTrackOut() const;
 201   virtual Bool_t   IsTrackDisappeared() const;
 202   virtual Bool_t   IsTrackStop() const;
 203   virtual Bool_t   IsTrackAlive() const;
 204
 205   // secondaries
 206   virtual Int_t    NSecondaries() const ;
 207   virtual void     GetSecondary(Int_t isec, Int_t& particleId,
 208                         TLorentzVector& position, TLorentzVector& momentum);
 209   virtual TMCProcess ProdProcess(Int_t isec) const ;
 210   virtual Int_t    StepProcesses(TArrayI &proc) const
 211     {printf("WARNING: StepProcesses not yet implemented !\n"); return -1;}
 212
 213
 214   //
 215   // Geant3 specific methods
 216   // !!! need to be transformed to common interface
 217   //
 218   virtual void Gdopt(const char*,const char*)
 219     {printf("WARNING: Gdopt not yet implemented !\n");}
 220   virtual void SetClipBox(const char*,Double_t=-9999,Double_t=0, Double_t=-9999,
 221                           Double_t=0,Double_t=-9999,Double_t=0)
 222     {printf("WARNING: SetClipBox not yet implemented !\n");}
 223   virtual void DefaultRange()
 224     {printf("WARNING: DefaultRange not yet implemented !\n");}
 225   virtual void Gdhead(Int_t, const char*, Double_t=0)
 226     {printf("WARNING: Gdhead not yet implemented !\n");}
 227   virtual void Gdman(Double_t, Double_t, const char*)
 228     {printf("WARNING: Gdman not yet implemented !\n");}
 229   virtual void SetColors()
 230     {printf("WARNING: SetColors not yet implemented !\n");}
 231   virtual void Gtreve()
 232     {printf("WARNING: Gtreve not yet implemented !\n");}
 233   virtual void GtreveRoot()
 234     {printf("WARNING: GtreveRoot not yet implemented !\n");}
 235   virtual void Gckmat(Int_t, char*)
 236     {printf("WARNING: Gckmat not yet implemented !\n");}
 237   virtual void InitLego()
 238     {printf("WARNING: InitLego not yet implemented !\n");}
 239   virtual void Gfpart(Int_t, char*, Int_t&, Float_t&, Float_t&, Float_t&)
 240     {printf("WARNING: Gfpart not yet implemented !\n");}
 241   virtual void Gspart(Int_t, const char*, Int_t, Double_t, Double_t, Double_t)
 242     {printf("WARNING: Gspart not yet implemented !\n");}
 243
 244   //
 245   // control methods
 246   // ------------------------------------------------
 247   //
 248
 249   virtual void Init();
 250   virtual void FinishGeometry();
 251   virtual void BuildPhysics();
 252   virtual void ProcessEvent();
 253   virtual void ProcessRun(Int_t nevent);
 254
 255
 256   //
 257   //New Getter and Setters
 258   // ------------------------------------------------
 259   //
 260   // - Input file name
 261   TString GetInputFileName() const {return fInputFileName;}
 262   void SetInputFileName(const char* n) {fInputFileName = n;}
 263   // - Verbosity level
 264   Int_t GetVerbosityLevel() const {return fVerbosityLevel;}
 265   void SetVerbosityLevel(Int_t l) {fVerbosityLevel = l;}
 266
 267   // - Fluka Draw procedures formal parameters
 268   Int_t GetIcode() const {return fIcode;}
 269   void SetIcode(Int_t l) {fIcode = l;}
 270
 271   Int_t GetMreg() const {return fCurrentFlukaRegion;}
 272   void SetMreg(Int_t l) {fCurrentFlukaRegion = l;}
 273
 274   Int_t GetNewreg() const {return fNewreg;}
 275   void SetNewreg(Int_t l) {fNewreg = l;}
 276
 277   Double_t GetRull() const {return fRull;}
 278   void SetRull(Double_t r) {fRull = r;}
 279
 280   Double_t GetXsco() const {return fXsco;}
 281   void SetXsco(Double_t x) {fXsco = x;}
 282
 283   Double_t GetYsco() const {return fYsco;}
 284   void SetYsco(Double_t y) {fYsco = y;}
 285
 286   Double_t GetZsco() const {return fZsco;}
 287   void SetZsco(Double_t z) {fZsco = z;}
 288
 289   void SetCurrentFlukaRegion(Int_t reg) {fCurrentFlukaRegion=reg;}
 290   Int_t GetCurrentFlukaRegion() const {return fCurrentFlukaRegion;}
 291   //
 292   // test
 293   // ------------------------------------------------
 294   //
 295   virtual void     FutoTest() ;
 296
 297  private:
 298   TFluka(const TFluka &mc){}
 299   TFluka & operator=(const TFluka &) {return (*this);}
 300
 301  protected:
 302   Int_t   fVerbosityLevel; //Verbosity level (0 lowest - 3 highest)
 303
 304   TString fInputFileName; //Name of the input file (f.e. mu.inp)
 305
 306   Int_t    fIcode;  //Fluka Draw procedures formal parameter
 307   Int_t    fNewreg; //Fluka Draw procedures formal parameter
 308   Double_t fRull;   //Fluka Draw procedures formal parameter
 309   Double_t fXsco;   //Fluka Draw procedures formal parameter
 310   Double_t fYsco;   //Fluka Draw procedures formal parameter
 311   Double_t fZsco;   //Fluka Draw procedures formal parameter
 312
 313
 314
 315   //Geometry through Geant4 for the time being!!!
 316   TG4GeometryManager*  fGeometryManager; //Geometry manager
 317   TG4DetConstruction*  fDetector;        //Detector
 318
 319   TClonesArray*        fVolumeMediaMap;  //!Transient list of volumes
 320   Int_t                fNVolumes;        //!Current number of volumes
 321   Int_t*               fMediaByRegion;   //!Media by Fluka region
 322
 323
 324   Int_t fCurrentFlukaRegion; //Index of fluka region at each step
 325   ClassDef(TFluka,1)  //C++ interface to Fluka montecarlo
 326 };
 327
 328 #endif //TFLUKA
 329