| Geant4 Cross Reference |
1 // 1
2 // *******************************************
3 // * License and Disclaimer
4 // *
5 // * The Geant4 software is copyright of th
6 // * the Geant4 Collaboration. It is provided
7 // * conditions of the Geant4 Software License
8 // * LICENSE and available at http://cern.ch/
9 // * include a list of copyright holders.
10 // *
11 // * Neither the authors of this software syst
12 // * institutes,nor the agencies providing fin
13 // * work make any representation or warran
14 // * regarding this software system or assum
15 // * use. Please see the license in the file
16 // * for the full disclaimer and the limitatio
17 // *
18 // * This code implementation is the result
19 // * technical work of the GEANT4 collaboratio
20 // * By using, copying, modifying or distri
21 // * any work based on the software) you ag
22 // * use in resulting scientific publicati
23 // * acceptance of all terms of the Geant4 Sof
24 // *******************************************
25 //
26 //
27 //
28 //
29 // Class Description:
30 //
31 // Base class for the trajectory state
32
33 // History:
34 // - Created: P. Arce
35 // -------------------------------------------
36
37 #ifndef G4ErrorTrajState_hh
38 #define G4ErrorTrajState_hh
39
40 #include "globals.hh"
41 #include "G4Track.hh"
42 #include "G4Point3D.hh"
43 #include "G4Vector3D.hh"
44 #include "G4ErrorTrajErr.hh"
45
46 enum G4eTSType
47 {
48 G4eTS_FREE,
49 G4eTS_OS
50 };
51
52 class G4ErrorTrajState
53 {
54 public: // with description
55 G4ErrorTrajState()
56 : fCharge(0.)
57 , theG4Track(0)
58 , iverbose(0)
59 {}
60
61 G4ErrorTrajState(const G4String& partType, c
62 const G4Vector3D& mom,
63 const G4ErrorTrajErr& errma
64 // Constructor by providing particle, positi
65
66 G4ErrorTrajState(const G4ErrorTrajState&);
67 G4ErrorTrajState(G4ErrorTrajState&&);
68 // The copy and move constructors
69
70 virtual ~G4ErrorTrajState() {}
71
72 G4ErrorTrajState& operator=(const G4ErrorTra
73 G4ErrorTrajState& operator=(G4ErrorTrajState
74 // The copy and move assignment operators
75
76 void SetData(const G4String& partType, const
77 const G4Vector3D& mom);
78 // Set particle, position and momentum
79
80 void BuildCharge();
81 // Build charge based on particle type
82
83 friend std::ostream& operator<<(std::ostream
84
85 virtual G4int PropagateError(const G4Track*)
86 // Propagate the error along the step
87
88 virtual G4int Update(const G4Track*) { retur
89 // Update parameters from G4Track
90
91 void UpdatePosMom(const G4Point3D& pos, cons
92 // Update position and momentum
93
94 void DumpPosMomError(std::ostream& out = G4c
95 // Dump position, momentum and error
96
97 virtual void Dump(std::ostream& out = G4cout
98 // Abstract method to dump all TrajState par
99
100 // Set and Get methods
101
102 const G4String& GetParticleType() const { re
103 void SetParticleType(const G4String& partTyp
104
105 G4Point3D GetPosition() const { return fPosi
106 virtual void SetPosition(const G4Point3D pos
107
108 G4Vector3D GetMomentum() const { return fMom
109 virtual void SetMomentum(const G4Vector3D& m
110
111 G4ErrorTrajErr GetError() const { return fEr
112 virtual void SetError(G4ErrorTrajErr em) { f
113
114 G4Track* GetG4Track() const { return theG4Tr
115 void SetG4Track(G4Track* trk) { theG4Track =
116
117 G4double GetCharge() const { return fCharge;
118 void SetCharge(G4double ch) { fCharge = ch;
119
120 virtual G4eTSType GetTSType() const { return
121
122 protected:
123 G4String fParticleType;
124 G4Point3D fPosition;
125 G4Vector3D fMomentum;
126 G4double fCharge = 0.;
127 G4ErrorTrajErr fError;
128
129 G4eTSType theTSType;
130
131 G4Track* theG4Track = nullptr;
132
133 G4int iverbose = 0;
134 };
135
136 #endif
137