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 eventgenerator/HepMC/MCTruth/include/MCTruthManager.hh 27 /// \brief Definition of the MCTruthManager class 28 // 29 // 30 // 31 // 32 // -------------------------------------------------------------- 33 // GEANT 4 - MCTruthManager class 34 // -------------------------------------------------------------- 35 // 36 // Author: Witold POKORSKI (Witold.Pokorski@cern.ch) 37 // Date : 2006-02-28 38 // 39 // -------------------------------------------------------------- 40 #ifndef INCLUDE_MCTRUTHMANAGER_H 41 #define INCLUDE_MCTRUTHMANAGER_H 1 42 43 #include "HepMC/GenEvent.h" 44 #include "HepMC/GenParticle.h" 45 #include "MCTruthConfig.hh" 46 47 #include "G4LorentzVector.hh" 48 #include "G4Types.hh" 49 50 class MCTruthManager 51 { 52 public: 53 static MCTruthManager* GetInstance(); 54 55 void NewEvent(); 56 HepMC::GenEvent* GetCurrentEvent() const { return fEvent; } 57 void PrintEvent(); 58 59 void AddParticle(G4LorentzVector&, G4LorentzVector&, G4LorentzVector&, G4int, G4int, G4int, 60 G4bool); 61 62 void SetConfig(MCTruthConfig* c) { fConfig = c; } 63 MCTruthConfig* GetConfig() const { return fConfig; } 64 65 protected: 66 MCTruthManager(); 67 68 virtual ~MCTruthManager(); 69 70 private: 71 HepMC::GenEvent* fEvent; 72 73 // vector containing barcodes of primary particles (not having any mother) 74 // 75 std::vector<G4int> fPrimarybarcodes; 76 77 // map containing number of 'segmentations' for each particle (i.e. number 78 // of additional vertices introduced in order to attach secondary particles 79 // which were created 'in-flight', for instance bremstrahlung gammas, etc) 80 // 81 std::map<G4int, G4int> fSegmentations; 82 83 // different criteria for storing (or not) particles 84 // 85 MCTruthConfig* fConfig; 86 87 // recursive printing of the tree 88 // 89 void PrintTree(HepMC::GenParticle*, G4String); 90 }; 91 #endif // INCLUDE_MCTRUTHMANAGER_H 92