LBNETrajectory.hh

Go to the documentation of this file.

//
// LBNETrajectory.hh
//

#ifndef LBNETrajectory_h
#define LBNETrajectory_h 1

#include "G4VTrajectory.hh"
#include "G4Allocator.hh"
#include <stdlib.h>
#include "G4ThreeVector.hh"
#include "G4ios.hh"     
#include <vector>
#include <iostream>
#include "globals.hh" 
#include "G4ParticleDefinition.hh" 
#include "G4TrajectoryPoint.hh"   
#include "G4Track.hh"
#include "G4Step.hh"
#include "G4UserSteppingAction.hh"
#include "G4Colour.hh"

class LBNETrajectory;
class G4Polyline;

typedef std::vector<G4VTrajectoryPoint*> LBNETrajectoryPointContainer;
typedef std::vector<G4ThreeVector> LBNETrajectoryMomentumContainer;
typedef std::vector<G4String> LBNETrajectoryVolumeName;
typedef std::vector<G4String> LBNETrajectoryMaterialName;
typedef std::vector<G4double> DVec;

class LBNETrajectory : public G4VTrajectory
{
 public:
   LBNETrajectory();
   LBNETrajectory(const G4Track* aTrack);
   LBNETrajectory(LBNETrajectory &);
   LBNETrajectory(const LBNETrajectory* right);
   virtual ~LBNETrajectory();

   inline void* operator new(size_t);
   inline void  operator delete(void*);
   inline int operator == (const LBNETrajectory& right) const
   {return (this==&right);} 

   inline G4int GetTrackID() const
   { return fTrackID; }
   inline G4int GetParentID() const
   { return fParentID; }
   inline G4String GetParticleName() const
   { return fParticleName; }
   inline G4double GetCharge() const
   { return fPDGCharge; }
   inline G4double GetMass() const
   { return fParticleMass; }
   inline G4int GetPDGEncoding() const
   { return fPDGEncoding; }
   inline const G4ThreeVector& GetVertexPosition() const
   { return fVertexPosition; }
   virtual int GetPointEntries() const
   { return fPositionRecord->size(); }
   virtual G4VTrajectoryPoint* GetPoint(G4int i) const 
   { return (*fPositionRecord)[i]; }
   virtual G4ThreeVector GetMomentum(G4int i) const
   { return (*fMomentumRecord)[i]; }
   virtual G4String GetPreStepVolumeName(G4int i) const
   { return (*fPreStepVolume)[i]; }
   inline G4ThreeVector GetInitialMomentum() const 
   { return fMomentum; }                            
   virtual G4int GetTgen() const
   { return fTgen;}
   inline G4int GetDecayCode() const
   { return fDecayCode;}
   virtual G4double GetNImpWt() const
   { return fNImpWt;}
   virtual G4double GetStepLength(G4int i) const
      {return (*fStepLength)[i];}
      
   inline G4int GetMaterialNumber1rst() const 
   {return fMaterialNumber1rst;}
   inline void SetMaterialNumber1rst(G4double matNum)
   {fMaterialNumber1rst = matNum;}
   
   inline G4int GetMaterialNumberLast() const 
   {return fMaterialNumberLast;}
   inline void SetMaterialNumberLast(G4double matNum)
    {fMaterialNumberLast = matNum;}
   
   inline G4String GetVolName1rst() const 
   {return fVolName1rst;}
   
   inline G4double GetTimeStart() const 
   {return fTimeStart;}
   
   inline G4ThreeVector GetPolarization() const
   { return fPolarization;}
   
   inline G4int GetPDGNucleus() const {
    return fPDGNucleus; }
                                                
   inline G4String  GetProcessName() const 
   { return fProcessName; } 
   
   inline G4String GetMaterialName1rst() const 
   { return fMaterialName1rst; }
   virtual G4String GetMaterialName(G4int i)const
   {return (*fMaterialName)[i];}
   inline const G4ThreeVector GetParentMomentumAtThisProduction() const {
        return fParentMomentumAtThisProduction;
    }  
   virtual void ShowTrajectory() const;
   virtual void ShowTrajectory(std::ostream& o) const;
   virtual void DrawTrajectory() const;

   virtual const std::map<G4String,G4AttDef>* GetAttDefs() const;
   virtual std::vector<G4AttValue>* CreateAttValues() const;

   virtual void AppendStep(const G4Step* aStep);
   virtual void MergeTrajectory(G4VTrajectory* secondTrajectory);

   G4ParticleDefinition* GetParticleDefinition();

 private:
   LBNETrajectoryMomentumContainer* fMomentumRecord;
   LBNETrajectoryPointContainer*    fPositionRecord;
   G4int                            fDecayCode;
   G4int                            fEventNo;
   G4int                            fTgen;
   G4double                         fNImpWt;
   G4int                            fTrackID;
   G4int                            fParentID;
   G4ParticleDefinition*            fParticleDefinition;
   G4String                         fParticleName;
   G4double                         fPDGCharge;
   G4int                            fPDGEncoding;
   G4ThreeVector                    fMomentum;
   G4ThreeVector                    fVertexPosition;
   G4double                         fParticleMass;
   LBNETrajectoryVolumeName*        fPreStepVolume;
   DVec*                            fStepLength;
// April 2014 : added to support Dk2Nu Ntuple output 
   G4int                            fMaterialNumber1rst; // at the first point
   G4String                         fVolName1rst; // at the first point
   G4int                            fMaterialNumberLast; // at the last recorded point
   G4double                         fTimeStart;
   G4ThreeVector                    fPolarization;
   G4int                            fPDGNucleus; // The nucleus that caused the scatter.. 
                                                 // if I can find it.. 
   G4String                         fProcessName;  // Process name at creation time.
   G4ThreeVector      fParentMomentumAtThisProduction; //for ppfx...A.Bashyal
   LBNETrajectoryMaterialName*      fMaterialName;  //for ppf....A.Bashyal
   G4String                         fMaterialName1rst; // First material name where the corresponding track 
                                                       // has been created.                         

};
extern G4Allocator<LBNETrajectory> myTrajectoryAllocator;

inline void* LBNETrajectory::operator new(size_t)
{
  void* aTrajectory;
  aTrajectory = (void*)myTrajectoryAllocator.MallocSingle();
  return aTrajectory;
}

inline void LBNETrajectory::operator delete(void* aTrajectory)
{
  myTrajectoryAllocator.FreeSingle((LBNETrajectory*)aTrajectory);
}
#endif