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/include/G4PhononTrackMap.hh 27 /// \brief Definition of the G4PhononTrackMap base class 28 // 29 // 30 // 20131111 Move implementation of Clear() to .cc file 31 32 #ifndef G4PhononTrackMap_h 33 #define G4PhononTrackMap_h 1 34 35 #include "G4ThreeVector.hh" 36 #include <map> 37 38 class G4Track; 39 40 class G4PhononTrackMap { 41 public: 42 typedef std::map<const G4Track*, G4ThreeVector> TrkIDKmap; 43 static G4ThreadLocal G4PhononTrackMap* theTrackMap; 44 45 public: 46 static G4PhononTrackMap* GetPhononTrackMap(); // Synonyms for access 47 static G4PhononTrackMap* GetInstance() { return GetPhononTrackMap(); } 48 49 // Update the wavevector for specified track, add track if not found 50 void SetK(const G4Track* track, const G4ThreeVector& K); 51 void SetK(const G4Track& track, const G4ThreeVector& K) { SetK(&track, K); } 52 53 // Access current wavevector for specified track (NULL if doesn't exist) 54 const G4ThreeVector& GetK(const G4Track* track) const; 55 const G4ThreeVector& GetK(const G4Track& track) const { return GetK(&track); } 56 57 // Check if specified track is already loaded 58 G4bool Find(const G4Track* track) const; 59 G4bool Find(const G4Track& track) const { return Find(&track); } 60 61 // Remove specified track from map (used by EndTracking) 62 void RemoveTrack(const G4Track* track); 63 64 void Clear(); // Remove all entries from map 65 66 private: 67 TrkIDKmap theMap; // Associate track ID numbers with vectors 68 69 private: 70 G4PhononTrackMap() { Clear(); } // Ensure map is empty 71 ~G4PhononTrackMap() {;} 72 }; 73 #endif /* G4PhononTrackMap_h */ 74