Gstpar implemented.

[u/mrichter/AliRoot.git] / TFluka / TFlukaGeo.h

diff --git a/TFluka/TFlukaGeo.h b/TFluka/TFlukaGeo.h
index f10052f8a8412959378d58421f73136242aaf51d..1692926827915575563553d274c839a023be4beb 100644 (file)
--- a/TFluka/TFlukaGeo.h
+++ b/TFluka/TFlukaGeo.h
@@ -15,10 +15,11 @@
 
 
 #include "TVirtualMC.h"
-#include "TMCProcess.h" 
+#include "TMCProcess.h"
 
 //Forward declaration
 class TFlukaMCGeometry;
+class TGeoMaterial;
 
 class TFluka : public TVirtualMC {
   
@@ -119,7 +120,9 @@ class TFluka : public TVirtualMC {
   
   // set methods
   virtual void     SetProcess(const char* flagName, Int_t flagValue);
+    virtual void     SetProcess(const char* flagName, Int_t flagValue, Int_t imed);
   virtual void     SetCut(const char* cutName, Double_t cutValue);
+  virtual void     SetCut(const char* cutName, Double_t cutValue, Int_t imed);
   virtual Double_t Xsec(char*, Double_t, Int_t, Int_t);
   
   // particle table usage         
@@ -266,10 +269,10 @@ class TFluka : public TVirtualMC {
   void SetInputFileName(const char* n) {sInputFileName = n;}
 
   // - SetProcess and SetCut
-  Int_t GetProcessNb() const {return iNbOfProc;}
-  void SetProcessNb(Int_t l) {iNbOfProc = l;}
-  Int_t GetCutNb() const {return iNbOfProc;}
-  void SetCutNb(Int_t l) {iNbOfCut = l;}
+  Int_t GetProcessNb() const {return fNbOfProc;}
+  void SetProcessNb(Int_t l) {fNbOfProc = l;}
+  Int_t GetCutNb() const {return fNbOfProc;}
+  void SetCutNb(Int_t l) {fNbOfCut = l;}
 
   // - Verbosity level
   Int_t GetVerbosityLevel() const {return fVerbosityLevel;}
@@ -293,7 +296,7 @@ class TFluka : public TVirtualMC {
   // in the case of sodraw fIcode=0
 
   Int_t GetMreg() const {return fCurrentFlukaRegion;}
-  void SetMreg(Int_t l) {fCurrentFlukaRegion = l;}
+  void SetMreg(Int_t l);
 
   Int_t GetNewreg() const {return iNewreg;}
   void SetNewreg(Int_t l) {iNewreg = l;}
@@ -314,20 +317,17 @@ class TFluka : public TVirtualMC {
   Int_t GetCurrentFlukaRegion() const {return fCurrentFlukaRegion;}
 
   void SetTrackIsEntering(){fTrackIsEntering = kTRUE; fTrackIsExiting = kFALSE;}
-  void SetTrackIsExiting() {fTrackIsExiting  = kTRUE;}
+  void SetTrackIsExiting() {fTrackIsExiting  = kTRUE; fTrackIsEntering = kFALSE;}
   void SetTrackIsInside()  {fTrackIsExiting  = kFALSE; fTrackIsEntering = kFALSE;}
+  void SetTrackIsNew(Bool_t flag=kTRUE) {fTrackIsNew = flag;}
+  Int_t GetMaterialIndex(Int_t idmat) {return fMaterials[idmat];}
   
-  //
-  // test
-  // ------------------------------------------------
-  //
-  virtual void     FutoTest() ;
-
  private:
   TFluka(const TFluka &mc): TVirtualMC(mc) {;}
   TFluka & operator=(const TFluka &) {return (*this);}
 
  protected:
+  
   Int_t   fVerbosityLevel; //Verbosity level (0 lowest - 3 highest)
 
   TString sInputFileName;     //Name of the real input file (e.g. alice.inp)
@@ -342,17 +342,20 @@ class TFluka : public TVirtualMC {
   Double_t fZsco;   //Fluka Draw procedures formal parameter
   Bool_t   fTrackIsEntering;  // Flag for track entering
   Bool_t   fTrackIsExiting;   // Flag for track exiting  
-
+  Bool_t   fTrackIsNew;       // Flag for new track
   //variables for SetProcess and SetCut
-  Int_t    iNbOfProc;
-  Int_t    iProcessValue[100];
-  Char_t   sProcessFlag[100][5];
-  Int_t    iNbOfCut;
-  Double_t fCutValue[100];
-  Char_t   sCutFlag[100][7];
+  Int_t    fNbOfProc;
+  Int_t    fProcessValue[1000];
+  Int_t    fProcessMedium[1000];
+  Char_t   fProcessFlag[1000][5];
+  Int_t    fNbOfCut;
+  Double_t fCutValue[1000];
+  Char_t   fCutFlag[1000][7];
+  Int_t    fCutMedium[1000];
 
-  //Geometry through Geant4 for the time being!!!
-  Int_t                fNVolumes;        //!Current number of volumes
+  //Geometry through TGeo
+  Int_t*               fMaterials;         //!Array of indices
+  Int_t                fNVolumes;           //!Current number of volumes
   Int_t                fCurrentFlukaRegion; //Index of fluka region at each step
   TFlukaMCGeometry    *fGeom;               // TGeo-FLUKA interface