| 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 /// \file UserMoleculeGun.hh
28 /// \brief Definition of the UserMoleculeGun class
29
30 #ifndef UserMoleculeGun_h
31 #define UserMoleculeGun_h 1
32
33 #include "G4DNAChemistryManager.hh"
34 #include "G4MoleculeGun.hh"
35
36 class UserMoleculeGun;
37
38 class UserMoleculeShoot : public G4enable_shared_from_this<UserMoleculeShoot>,
39 public G4MoleculeShoot
40 {
41 public:
42
43 UserMoleculeShoot();
44 ~UserMoleculeShoot() override;
45 void Shoot(G4MoleculeGun*) override {};
46 virtual void MyShoot(UserMoleculeGun*) = 0;
47
48 template<typename TYPE> G4shared_ptr<UserMoleculeShoot> ChangeType();
49
50 G4int fCopyNumber{-1};
51 G4int fStrand{-1};
52 };
53
54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
55
56 template<typename TYPE>
57 class TUserMoleculeShoot : public UserMoleculeShoot
58 {
59 public:
60 TUserMoleculeShoot() : UserMoleculeShoot(){;}
61 ~TUserMoleculeShoot() override {;}
62 void MyShoot(UserMoleculeGun*) override;
63 protected:
64 void ShootAtRandomPosition(UserMoleculeGun*);
65 void ShootAtFixedPosition(UserMoleculeGun*);
66 };
67
68 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
69
70 template<typename TYPE>
71 G4shared_ptr<UserMoleculeShoot> UserMoleculeShoot::ChangeType()
72 {
73 G4shared_ptr<UserMoleculeShoot> output(new TUserMoleculeShoot<TYPE>);
74 output->fMoleculeName = fMoleculeName;
75 output->fPosition = fPosition;
76 output->fTime = fTime;
77 output->fNumber = fNumber;
78 output->fBoxSize = fBoxSize;
79 output->fCopyNumber = fCopyNumber;
80 output->fStrand = fStrand;
81 return output;
82 }
83
84 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
85
86 class UserMoleculeGun : public G4ITGun
87 {
88 public:
89 UserMoleculeGun() = default;
90 ~UserMoleculeGun() override = default;
91
92 void DefineTracks() override;
93
94 void AddMolecule(const G4String& name,
95 const G4ThreeVector& position,
96 G4double time,
97 G4int copyNumber,
98 G4int strand);
99
100 void AddWaterMolecule(const G4ThreeVector& position,
101 G4int trackId,
102 ElectronicModification elecModif,
103 G4int electronicLevel);
104
105 protected:
106 void BuildAndPushTrack(const G4String& name,
107 const G4ThreeVector& position,
108 G4double time = 0);
109
110 void BuildAndPushTrack(const G4String& name,
111 const G4ThreeVector& position,
112 G4double time,
113 G4int copyNumber,
114 G4int strand);
115
116 std::vector<G4shared_ptr<UserMoleculeShoot> > fShoots;
117 friend class UserMoleculeShoot;
118 template<class T> friend class TUserMoleculeShoot;
119 };
120
121 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
122
123 #endif
124