Minor corrections (Sun)

[u/mrichter/AliRoot.git] / TGeant4 / TG4StepManager.h

diff --git a/TGeant4/TG4StepManager.h b/TGeant4/TG4StepManager.h
index 9bf68bfdb94033c0c10c4ffb0fddb68b09f01314..fdd97f266ab85ff975bc324764ecdcf0addebfe7 100644 (file)
--- a/TGeant4/TG4StepManager.h
+++ b/TGeant4/TG4StepManager.h
@@ -1,22 +1,32 @@
 // $Id$
 // Category: event
 //
-// Geant4 implementation of the MonteCarlo interface methods                    
-// for access to Geant4 at step level
+// Author: I.Hrivnacova
 //
+// Class TG4StepManager
+// --------------------
+// Geant4 implementation of the MonteCarlo interface methods                    
+// for access to Geant4 at step level.
 // The public methods that do not implement AliMC methods
 // are commented as G4 specific
 
 #ifndef TG4_STEP_MANAGER_H
 #define TG4_STEP_MANAGER_H
 
+#include <Rtypes.h>
+#include "AliMCProcess.h"
+
+#include "TG4StepStatus.h"
+
+#include <G4Step.hh>
 #include <G4ThreeVector.hh>
 #include <globals.hh>
 
-#include <Rtypes.h>
+#include <TArrayI.h>
 
-class G4Step;
+class G4Track;
 class G4SteppingManager;
+class G4VPhysicalVolume;
 
 class TLorentzVector;
 
@@ -32,32 +42,35 @@ class TG4StepManager
     static TG4StepManager* Instance();
         
     // methods
-    void StopTrack(); //new
-    void StopEvent(); //new
-    void Rndm(Float_t* array, const Int_t size) const;
+    void StopTrack();
+    void StopEvent();
     
     // set methods
-    void SetStep(G4Step* step);                          // G4 specific
+    void SetStep(G4Step* step, TG4StepStatus status);    // G4 specific
+    void SetStep(G4Track* track, TG4StepStatus status);  // G4 specific
     void SetSteppingManager(G4SteppingManager* manager); // G4 specific
     void SetMaxStep(Float_t step);
     void SetMaxNStep(Int_t maxNofSteps);  //??
-    void SetUserDecay(Int_t);  //NEW
+    void SetUserDecay(Int_t pdg);  //NEW
     
     // get methods
-    G4Step* GetStep() const;                              // G4 specific
-    
+    G4Track* GetTrack() const;                            // G4 specific
+    G4Step*  GetStep() const;                             // G4 specific
+    TG4StepStatus GetStepStatus() const;                  // G4 specific
+        
         // tracking volume(s) 
+    G4VPhysicalVolume* GetCurrentPhysicalVolume() const;  // G4 specific
     Int_t CurrentVolID(Int_t& copyNo) const;
     Int_t CurrentVolOffID(Int_t off, Int_t& copyNo) const;
     const char* CurrentVolName() const;
     const char* CurrentVolOffName(Int_t off) const;
     Int_t CurrentMaterial(Float_t &a, Float_t &z, Float_t &dens, 
                     Float_t &radl, Float_t &absl) const;
-    void Gmtod(Float_t* xm, Float_t* xd, Int_t iflag);  //new
-    void Gdtom(Float_t* xd, Float_t* xm, Int_t iflag); //new
+    void Gmtod(Float_t* xm, Float_t* xd, Int_t iflag);
+    void Gdtom(Float_t* xd, Float_t* xm, Int_t iflag);
     Float_t MaxStep() const;
     Int_t GetMaxNStep() const;  //??                       
-    Int_t GetMedium() const;  //??
+    Int_t GetMedium() const;
 
         // tracking particle 
         // dynamic properties
@@ -89,7 +102,8 @@ class TG4StepManager
     Int_t NSecondaries() const;
     void GetSecondary(Int_t isec, Int_t& particleId,
                       TLorentzVector& position, TLorentzVector& momentum);      
-    const char* ProdProcess() const; 
+    AliMCProcess ProdProcess(Int_t isec) const; 
+    Int_t StepProcesses(TArrayI &proc) const;
 
   protected:
     TG4StepManager(const TG4StepManager& right);
@@ -99,13 +113,22 @@ class TG4StepManager
 
   private:
     // methods
-    void SetTLorentzVector(G4ThreeVector, G4double, TLorentzVector&) const;
-    
+    void CheckTrack() const;
+    void CheckStep(const G4String& method) const;
+    void CheckSteppingManager() const;
+    void SetTLorentzVector(G4ThreeVector xyz, G4double t, 
+                           TLorentzVector& lv) const;    
+    G4VPhysicalVolume* GetCurrentOffPhysicalVolume(G4int off) const;
+
     // static data members
     static TG4StepManager*  fgInstance;   //this instance
     
     // data members
+    G4Track*            fTrack;           //current track
     G4Step*             fStep;            //current step
+    TG4StepStatus       fStepStatus;      //step status that decides whether
+                                          //track properties will be returned
+                                         //from PreStepPoint or PostStepPoint
     G4SteppingManager*  fSteppingManager; //G4SteppingManager
 };
 
@@ -114,14 +137,23 @@ class TG4StepManager
 inline TG4StepManager* TG4StepManager::Instance() 
 { return fgInstance; }
 
-inline void TG4StepManager::SetStep(G4Step* step)
-{ fStep = step; }
+inline void TG4StepManager::SetStep(G4Step* step, TG4StepStatus status)
+{ fTrack = step->GetTrack(); fStep = step; fStepStatus = status; }
+
+inline void TG4StepManager::SetStep(G4Track* track, TG4StepStatus status)
+{ fTrack = track; fStep = 0; fStepStatus = status; }
 
 inline void TG4StepManager::SetSteppingManager(G4SteppingManager* manager)
 { fSteppingManager = manager; }
 
+inline G4Track* TG4StepManager::GetTrack() const
+{ return fTrack; }
+
 inline G4Step* TG4StepManager::GetStep() const
 { return fStep; }
 
+inline TG4StepStatus TG4StepManager::GetStepStatus() const
+{ return fStepStatus; }
+
 #endif //TG4_STEP_MANAGER_H