| 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 /// \file processes/phonon/src/G4PhononTrackMap.hh
27 /// \brief Implementation of the G4PhononTrackMap base class
28 //
29 //
30 // 20131111 Move Clear() function to .cc file
31
32 #include "G4PhononTrackMap.hh"
33 #include "G4ThreeVector.hh"
34 #include "G4Track.hh"
35 #include <map>
36
37 namespace { const G4ThreeVector nullVec(0.,0.,0.); } // For convenience below
38
39 // Singleton instance, must be created at first request
40
41 G4ThreadLocal G4PhononTrackMap* G4PhononTrackMap::theTrackMap = 0;
42
43
44 // Pseudo-constructor creates singleton instance
45
46 G4PhononTrackMap* G4PhononTrackMap::GetPhononTrackMap() {
47 if (!theTrackMap) theTrackMap = new G4PhononTrackMap;
48 return theTrackMap;
49 }
50
51 void G4PhononTrackMap::Clear() {
52 theMap.clear(); // Remove all entries from map
53 }
54
55
56
57 // Check if specified track is already loaded
58
59 G4bool G4PhononTrackMap::Find(const G4Track* track) const {
60 return (!track || theMap.find(track) != theMap.end());
61 }
62
63
64 // Remove specified track from map (used by EndTracking)
65
66 void G4PhononTrackMap::RemoveTrack(const G4Track* track) {
67 TrkIDKmap::iterator entry = theMap.find(track);
68 if (entry != theMap.end()) theMap.erase(entry);
69 }
70
71
72 // Update the wavevector for specified track, add track if non-existent
73
74 void G4PhononTrackMap::SetK(const G4Track* track, const G4ThreeVector& K) {
75 if (track) theMap[track] = K;
76 }
77
78
79 // Access current wavevector for specified track (NULL if doesn't exist)
80
81 const G4ThreeVector& G4PhononTrackMap::GetK(const G4Track* track) const {
82 TrkIDKmap::const_iterator entry = theMap.find(track);
83 return (entry != theMap.end()) ? entry->second : nullVec;
84 }
85
86