added enum TG4StepStatus type, attribute fStepStatus and argument to SetStep of this...

diff --git a/TGeant4/TG4StepManager.h b/TGeant4/TG4StepManager.h
index 34625f548585b2cd4f423d6d380810423bcd7762..2bd40a5ad7cf52b1aef253df002999078be7344d 100644 (file)
--- a/TGeant4/TG4StepManager.h
+++ b/TGeant4/TG4StepManager.h
@@ -20,6 +20,8 @@ class G4SteppingManager;
 
 class TLorentzVector;
 
+enum TG4StepStatus { kPreStepPoint, kPostStepPoint};
+
 class TG4StepManager
 {
   public:
@@ -37,7 +39,7 @@ class TG4StepManager
     void Rndm(Float_t* array, const Int_t size) const;
     
     // set methods
-    void SetStep(G4Step* step);                          // G4 specific
+    void SetStep(G4Step* step, TG4StepStatus status);    // G4 specific
     void SetSteppingManager(G4SteppingManager* manager); // G4 specific
     void SetMaxStep(Float_t step);
     void SetMaxNStep(Int_t maxNofSteps);  //??
@@ -45,6 +47,7 @@ class TG4StepManager
     
     // get methods
     G4Step* GetStep() const;                              // G4 specific
+    TG4StepStatus GetStepStatus() const;                  // G4 specific
     
         // tracking volume(s) 
     Int_t CurrentVolID(Int_t& copyNo) const;
@@ -107,6 +110,9 @@ class TG4StepManager
     
     // data members
     G4Step*             fStep;            //current step
+    TG4StepStatus       fStepStatus;      //step status that decides whether
+                                          //track properties will be returned
+                                         //from PreStepPoint or PostStepPoint
     G4SteppingManager*  fSteppingManager; //G4SteppingManager
 };
 
@@ -115,8 +121,8 @@ class TG4StepManager
 inline TG4StepManager* TG4StepManager::Instance() 
 { return fgInstance; }
 
-inline void TG4StepManager::SetStep(G4Step* step)
-{ fStep = step; }
+inline void TG4StepManager::SetStep(G4Step* step, TG4StepStatus status)
+{ fStep = step; fStepStatus = status; }
 
 inline void TG4StepManager::SetSteppingManager(G4SteppingManager* manager)
 { fSteppingManager = manager; }
@@ -124,5 +130,8 @@ inline void TG4StepManager::SetSteppingManager(G4SteppingManager* manager)
 inline G4Step* TG4StepManager::GetStep() const
 { return fStep; }
 
+inline TG4StepStatus TG4StepManager::GetStepStatus() const
+{ return fStepStatus; }
+
 #endif //TG4_STEP_MANAGER_H