Buffer for correct particle 4-vector after interaction added.

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

diff --git a/TFluka/TFluka.h b/TFluka/TFluka.h
index aadef7b6d3f620b518527cfe3481b3794ac3cb23..d0eea0df31fe0f67ba91c8234504b0d5e4a18f0b 100644 (file)
--- a/TFluka/TFluka.h
+++ b/TFluka/TFluka.h
@@ -168,6 +168,9 @@ class TFluka : public TVirtualMC {
   virtual const  char* VolName(Int_t id) const;
   virtual Int_t  NofVolumes() const {return fNVolumes;}
   virtual Int_t  VolId2Mate(Int_t id) const;
+  // Return the unique numeric identifier for medium name mediumName
+  virtual Int_t MediumId(const Text_t* mediumName) const;
+
   //
   // Methods for physics management
   // ------------------------------------------------
@@ -368,7 +371,7 @@ class TFluka : public TVirtualMC {
   void  SetXsco(Double_t x) {fXsco = x;}
   void  SetYsco(Double_t y) {fYsco = y;}
   void  SetZsco(Double_t z) {fZsco = z;}
-
+  void  SetPint(Double_t px, Double_t py, Double_t pz, Double_t e) {fPint[0] = px; fPint[1] = py; fPint[2] = pz; fPint[3] = e;}
   void  SetTrackIsEntering(){fTrackIsEntering = kTRUE; fTrackIsExiting = kFALSE;}
   void  SetTrackIsExiting() {fTrackIsExiting  = kTRUE; fTrackIsEntering = kFALSE;}
   void  SetTrackIsInside()  {fTrackIsExiting  = kFALSE; fTrackIsEntering = kFALSE;}
@@ -378,10 +381,6 @@ class TFluka : public TVirtualMC {
   Int_t  GetDummyBoundary() const {return fDummyBoundary;}
   Bool_t IsDummyBoundary() const {return (fDummyBoundary==0)?kFALSE:kTRUE;}
   
-
-  void   SetGeneratePemf(Bool_t flag = kTRUE) {fGeneratePemf = flag;}
-  Bool_t IsGeneratePemf() const {return fGeneratePemf;}
-  
   void   EnableField(Bool_t flag=kTRUE) {fFieldFlag = flag;}
   Bool_t IsFieldEnabled() const {return fFieldFlag;}
   
@@ -400,17 +399,18 @@ class TFluka : public TVirtualMC {
   Double_t GetPrimaryElectronKineticEnergy(Int_t i) const;
   void     GetPrimaryElectronPosition(Int_t i, Double_t& x, Double_t& y, Double_t& z) const;
   void     SetCurrentPrimaryElectronIndex(Int_t i)  {fPrimaryElectronIndex = i;}
-
+  void     PrimaryIonisationStepping(Int_t nprim);  
  private:
    
   // Copy constructor and operator= declared but not implemented (-Weff++ flag)
   TFluka(const TFluka &mc); //: TVirtualMC(mc) {;}
   TFluka & operator=(const TFluka &); // {return (*this);}
  
-  void PrintHeader();
-  void AddParticlesToPdgDataBase() const;
-  //
+  void  PrintHeader();
+  void  AddParticlesToPdgDataBase() const;
+  Int_t GetIonPdg(Int_t z, Int_t a, Int_t i = 0) const;
 
+  //
   Int_t   fVerbosityLevel; //Verbosity level (0 lowest - 3 highest)
   Int_t   fNEvent;         //Current event number
   TString fInputFileName;     //Name of the real input file 
@@ -423,11 +423,11 @@ class TFluka : public TVirtualMC {
   Double_t              fXsco;             // Fluka Draw procedures formal parameter
   Double_t              fYsco;             // Fluka Draw procedures formal parameter
   Double_t              fZsco;             // Fluka Draw procedures formal parameter
+  Double_t              fPint[4];          // 4-vector of particle after interaction
   Bool_t   fTrackIsEntering;      // Flag for track entering
   Bool_t   fTrackIsExiting;       // Flag for track exiting  
   Bool_t   fTrackIsNew;           // Flag for new track
   Bool_t   fFieldFlag;            // Flag for magnetic field
-  Bool_t   fGeneratePemf;         // Flag for automatic .pemf generation
   Int_t    fDummyBoundary;        // Flag for crossing dummy boundaries
   Bool_t   fStopped;              // Flag for stopping 
   Bool_t   fStopEvent;            // Flag for stopped event