| Geant4 Cross Reference |
1 // 1 //
2 // ******************************************* 2 // ********************************************************************
3 // * License and Disclaimer 3 // * License and Disclaimer *
4 // * 4 // * *
5 // * The Geant4 software is copyright of th 5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided 6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License 7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/ 8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. 9 // * include a list of copyright holders. *
10 // * 10 // * *
11 // * Neither the authors of this software syst 11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing fin 12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warran 13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assum 14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file 15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitatio 16 // * for the full disclaimer and the limitation of liability. *
17 // * 17 // * *
18 // * This code implementation is the result 18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboratio 19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distri 20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you ag 21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publicati 22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Sof 23 // * acceptance of all terms of the Geant4 Software license. *
24 // ******************************************* 24 // ********************************************************************
25 // 25 //
26 /// \file processes/phonon/src/G4PhononTrackMa 26 /// \file processes/phonon/src/G4PhononTrackMap.hh
27 /// \brief Implementation of the G4PhononTrack 27 /// \brief Implementation of the G4PhononTrackMap base class
28 // 28 //
29 // 29 //
30 // 20131111 Move Clear() function to .cc file 30 // 20131111 Move Clear() function to .cc file
31 31
32 #include "G4PhononTrackMap.hh" 32 #include "G4PhononTrackMap.hh"
33 #include "G4ThreeVector.hh" 33 #include "G4ThreeVector.hh"
34 #include "G4Track.hh" 34 #include "G4Track.hh"
35 #include <map> 35 #include <map>
36 36
37 namespace { const G4ThreeVector nullVec(0.,0., 37 namespace { const G4ThreeVector nullVec(0.,0.,0.); } // For convenience below
38 38
39 // Singleton instance, must be created at firs 39 // Singleton instance, must be created at first request
40 40
41 G4ThreadLocal G4PhononTrackMap* G4PhononTrackM 41 G4ThreadLocal G4PhononTrackMap* G4PhononTrackMap::theTrackMap = 0;
42 42
43 43
44 // Pseudo-constructor creates singleton instan 44 // Pseudo-constructor creates singleton instance
45 45
46 G4PhononTrackMap* G4PhononTrackMap::GetPhononT 46 G4PhononTrackMap* G4PhononTrackMap::GetPhononTrackMap() {
47 if (!theTrackMap) theTrackMap = new G4Phonon 47 if (!theTrackMap) theTrackMap = new G4PhononTrackMap;
48 return theTrackMap; 48 return theTrackMap;
49 } 49 }
50 50
51 void G4PhononTrackMap::Clear() { 51 void G4PhononTrackMap::Clear() {
52 theMap.clear(); // Remove all entries fr 52 theMap.clear(); // Remove all entries from map
53 } 53 }
54 54
55 55
56 56
57 // Check if specified track is already loaded 57 // Check if specified track is already loaded
58 58
59 G4bool G4PhononTrackMap::Find(const G4Track* t 59 G4bool G4PhononTrackMap::Find(const G4Track* track) const {
60 return (!track || theMap.find(track) != theM 60 return (!track || theMap.find(track) != theMap.end());
61 } 61 }
62 62
63 63
64 // Remove specified track from map (used by En 64 // Remove specified track from map (used by EndTracking)
65 65
66 void G4PhononTrackMap::RemoveTrack(const G4Tra 66 void G4PhononTrackMap::RemoveTrack(const G4Track* track) {
67 TrkIDKmap::iterator entry = theMap.find(trac 67 TrkIDKmap::iterator entry = theMap.find(track);
68 if (entry != theMap.end()) theMap.erase(entr 68 if (entry != theMap.end()) theMap.erase(entry);
69 } 69 }
70 70
71 71
72 // Update the wavevector for specified track, 72 // Update the wavevector for specified track, add track if non-existent
73 73
74 void G4PhononTrackMap::SetK(const G4Track* tra 74 void G4PhononTrackMap::SetK(const G4Track* track, const G4ThreeVector& K) {
75 if (track) theMap[track] = K; 75 if (track) theMap[track] = K;
76 } 76 }
77 77
78 78
79 // Access current wavevector for specified tra 79 // Access current wavevector for specified track (NULL if doesn't exist)
80 80
81 const G4ThreeVector& G4PhononTrackMap::GetK(co 81 const G4ThreeVector& G4PhononTrackMap::GetK(const G4Track* track) const {
82 TrkIDKmap::const_iterator entry = theMap.fin 82 TrkIDKmap::const_iterator entry = theMap.find(track);
83 return (entry != theMap.end()) ? entry->seco 83 return (entry != theMap.end()) ? entry->second : nullVec;
84 } 84 }
85 85
86 86