| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 //
27
28 // ********************************************************************
29 //
30 // CaTS (Calorimetry and Tracking Simulation)
31 //
32 // Authors : Hans Wenzel
33 // Soon Yung Jun
34 // (Fermi National Accelerator Laboratory)
35 //
36 // History
37 // October 18th, 2021 : first implementation
38 //
39 // ********************************************************************
40 //
41 /// \file PhotonHit.hh
42 /// \brief Definition of the CaTS::PhotonHit class
43
44 #pragma once
45
46 #include "G4VHit.hh"
47 #include "G4THitsCollection.hh"
48 #include "G4Allocator.hh"
49 #include "G4ThreeVector.hh"
50
51 class PhotonHit : public G4VHit
52 {
53 public:
54 PhotonHit();
55 ~PhotonHit() = default;
56 PhotonHit(const PhotonHit&);
57 const PhotonHit& operator=(const PhotonHit&);
58 G4bool operator==(const PhotonHit&) const;
59 inline void* operator new(size_t);
60 inline void operator delete(void*);
61 void Draw() final;
62
63 inline void Print() final
64 {
65 G4cout << "PhotonHit id: " << fid << " pid: " << fpid
66 << " wavelength: " << fwavelength << " time: " << ftime
67 << " position X: " << fposition.getX() << " Y: " << fposition.getY()
68 << " Z: " << fposition.getZ()
69 << " direction X: " << fdirection.getX()
70 << " Y: " << fdirection.getY() << " Z: " << fdirection.getZ()
71 << " polarization: X:" << fpolarization.getX()
72 << " Y: " << fpolarization.getY() << " Z: " << fpolarization.getZ()
73 << G4endl;
74 }
75
76 PhotonHit(unsigned id, unsigned pid, G4double wavelength, G4double time,
77 G4ThreeVector position, G4ThreeVector direction,
78 G4ThreeVector polarization);
79 inline void SetWavelength(G4double wavelength) { fwavelength = wavelength; }
80 inline G4double GetWavelength() { return fwavelength; }
81 inline void SetPolarization(G4ThreeVector polarization)
82 {
83 fpolarization = polarization;
84 }
85 inline G4ThreeVector GetPolarization() const { return fpolarization; }
86 inline void SetDirection(G4ThreeVector direction) { fdirection = direction; }
87 inline G4ThreeVector GetDirection() const { return fdirection; }
88 inline void SetPosition(G4ThreeVector position) { fposition = position; }
89 inline G4ThreeVector GetPosition() const { return fposition; }
90 inline void SetTime(G4double time) { ftime = time; }
91 inline G4double GetTime() const { return ftime; }
92 inline void SetPid(unsigned pid) { fpid = pid; }
93 inline unsigned GetPid() const { return fpid; }
94 inline void SetId(unsigned id) { fid = id; }
95 inline unsigned GetId() const { return fid; }
96
97 private:
98 unsigned int fid{ 0 };
99 unsigned int fpid{ 0 };
100 G4double fwavelength{ 0 };
101 G4double ftime{ 0 };
102 G4ThreeVector fposition{ 0, 0, 0 };
103 G4ThreeVector fdirection{ 0, 0, 0 };
104 G4ThreeVector fpolarization{ 0, 0, 0 };
105 };
106
107 using PhotonHitsCollection = G4THitsCollection<PhotonHit>;
108 extern G4ThreadLocal G4Allocator<PhotonHit>* PhotonHitAllocator;
109
110 inline void* PhotonHit::operator new(size_t)
111 {
112 if(!PhotonHitAllocator)
113 {
114 PhotonHitAllocator = new G4Allocator<PhotonHit>;
115 }
116 return (void*) PhotonHitAllocator->MallocSingle();
117 }
118
119 inline void PhotonHit::operator delete(void* aHit)
120 {
121 PhotonHitAllocator->FreeSingle((PhotonHit*) aHit);
122 }
123