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 eventgenerator/HepMC/HepMCEx01/src/H 26 /// \file eventgenerator/HepMC/HepMCEx01/src/HepMCG4Interface.cc 27 /// \brief Implementation of the HepMCG4Interf 27 /// \brief Implementation of the HepMCG4Interface class 28 // 28 // 29 // 29 // 30 30 31 #include "HepMCG4Interface.hh" 31 #include "HepMCG4Interface.hh" 32 32 33 #include "G4Event.hh" << 33 #include "G4RunManager.hh" 34 #include "G4LorentzVector.hh" 34 #include "G4LorentzVector.hh" 35 #include "G4PhysicalConstants.hh" << 35 #include "G4Event.hh" 36 #include "G4PrimaryParticle.hh" 36 #include "G4PrimaryParticle.hh" 37 #include "G4PrimaryVertex.hh" 37 #include "G4PrimaryVertex.hh" 38 #include "G4RunManager.hh" << 39 #include "G4SystemOfUnits.hh" << 40 #include "G4TransportationManager.hh" 38 #include "G4TransportationManager.hh" >> 39 #include "G4PhysicalConstants.hh" >> 40 #include "G4SystemOfUnits.hh" 41 41 42 //....oooOO0OOooo........oooOO0OOooo........oo 42 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 43 HepMCG4Interface::HepMCG4Interface() : hepmcEv << 43 HepMCG4Interface::HepMCG4Interface() >> 44 : hepmcEvent(0) >> 45 { >> 46 } 44 47 45 //....oooOO0OOooo........oooOO0OOooo........oo 48 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 46 HepMCG4Interface::~HepMCG4Interface() 49 HepMCG4Interface::~HepMCG4Interface() 47 { 50 { 48 delete hepmcEvent; 51 delete hepmcEvent; 49 } 52 } 50 53 51 //....oooOO0OOooo........oooOO0OOooo........oo 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 52 G4bool HepMCG4Interface::CheckVertexInsideWorl << 55 G4bool HepMCG4Interface::CheckVertexInsideWorld >> 56 (const G4ThreeVector& pos) const 53 { 57 { 54 G4Navigator* navigator = << 58 G4Navigator* navigator= G4TransportationManager::GetTransportationManager() 55 G4TransportationManager::GetTransportation << 59 -> GetNavigatorForTracking(); >> 60 >> 61 G4VPhysicalVolume* world= navigator-> GetWorldVolume(); >> 62 G4VSolid* solid= world-> GetLogicalVolume()-> GetSolid(); >> 63 EInside qinside= solid-> Inside(pos); 56 64 57 G4VPhysicalVolume* world = navigator->GetWor << 65 if( qinside != kInside) return false; 58 G4VSolid* solid = world->GetLogicalVolume()- << 66 else return true; 59 EInside qinside = solid->Inside(pos); << 60 << 61 if (qinside != kInside) << 62 return false; << 63 else << 64 return true; << 65 } 67 } 66 68 67 //....oooOO0OOooo........oooOO0OOooo........oo 69 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 68 void HepMCG4Interface::HepMC2G4(const HepMC::G << 70 void HepMCG4Interface::HepMC2G4(const HepMC::GenEvent* hepmcevt, >> 71 G4Event* g4event) 69 { 72 { 70 for (HepMC::GenEvent::vertex_const_iterator << 73 for(HepMC::GenEvent::vertex_const_iterator vitr= hepmcevt->vertices_begin(); 71 vitr != hepmcevt->vertices_end(); ++vit << 74 vitr != hepmcevt->vertices_end(); ++vitr ) { // loop for vertex ... 72 { // loop for vertex ... << 73 75 74 // real vertex? 76 // real vertex? 75 G4bool qvtx = false; << 77 G4bool qvtx=false; 76 for (HepMC::GenVertex::particle_iterator p << 78 for (HepMC::GenVertex::particle_iterator 77 pitr != (*vitr)->particles_end(HepMC: << 79 pitr= (*vitr)->particles_begin(HepMC::children); 78 { << 80 pitr != (*vitr)->particles_end(HepMC::children); ++pitr) { 79 if (!(*pitr)->end_vertex() && (*pitr)->s << 81 80 qvtx = true; << 82 if (!(*pitr)->end_vertex() && (*pitr)->status()==1) { >> 83 qvtx=true; 81 break; 84 break; 82 } 85 } 83 } 86 } 84 if (!qvtx) continue; 87 if (!qvtx) continue; 85 88 86 // check world boundary 89 // check world boundary 87 HepMC::FourVector pos = (*vitr)->position( << 90 HepMC::FourVector pos= (*vitr)-> position(); 88 G4LorentzVector xvtx(pos.x(), pos.y(), pos 91 G4LorentzVector xvtx(pos.x(), pos.y(), pos.z(), pos.t()); 89 if (!CheckVertexInsideWorld(xvtx.vect() * << 92 if (! CheckVertexInsideWorld(xvtx.vect()*mm)) continue; 90 93 91 // create G4PrimaryVertex and associated G 94 // create G4PrimaryVertex and associated G4PrimaryParticles 92 G4PrimaryVertex* g4vtx = << 95 G4PrimaryVertex* g4vtx= 93 new G4PrimaryVertex(xvtx.x() * mm, xvtx. << 96 new G4PrimaryVertex(xvtx.x()*mm, xvtx.y()*mm, xvtx.z()*mm, >> 97 xvtx.t()*mm/c_light); >> 98 >> 99 for (HepMC::GenVertex::particle_iterator >> 100 vpitr= (*vitr)->particles_begin(HepMC::children); >> 101 vpitr != (*vitr)->particles_end(HepMC::children); ++vpitr) { 94 102 95 for (HepMC::GenVertex::particle_iterator v << 103 if( (*vpitr)->status() != 1 ) continue; 96 vpitr != (*vitr)->particles_end(HepMC << 97 { << 98 if ((*vpitr)->status() != 1) continue; << 99 104 100 G4int pdgcode = (*vpitr)->pdg_id(); << 105 G4int pdgcode= (*vpitr)-> pdg_id(); 101 pos = (*vpitr)->momentum(); << 106 pos= (*vpitr)-> momentum(); 102 G4LorentzVector p(pos.px(), pos.py(), po 107 G4LorentzVector p(pos.px(), pos.py(), pos.pz(), pos.e()); 103 G4PrimaryParticle* g4prim = << 108 G4PrimaryParticle* g4prim= 104 new G4PrimaryParticle(pdgcode, p.x() * << 109 new G4PrimaryParticle(pdgcode, p.x()*GeV, p.y()*GeV, p.z()*GeV); 105 110 106 g4vtx->SetPrimary(g4prim); << 111 g4vtx-> SetPrimary(g4prim); 107 } 112 } 108 g4event->AddPrimaryVertex(g4vtx); << 113 g4event-> AddPrimaryVertex(g4vtx); 109 } 114 } 110 } 115 } 111 116 112 //....oooOO0OOooo........oooOO0OOooo........oo 117 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 113 HepMC::GenEvent* HepMCG4Interface::GenerateHep 118 HepMC::GenEvent* HepMCG4Interface::GenerateHepMCEvent() 114 { 119 { 115 HepMC::GenEvent* aevent = new HepMC::GenEven << 120 HepMC::GenEvent* aevent= new HepMC::GenEvent(); 116 return aevent; 121 return aevent; 117 } 122 } 118 123 119 //....oooOO0OOooo........oooOO0OOooo........oo 124 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 120 void HepMCG4Interface::GeneratePrimaryVertex(G 125 void HepMCG4Interface::GeneratePrimaryVertex(G4Event* anEvent) 121 { 126 { 122 // delete previous event object 127 // delete previous event object 123 delete hepmcEvent; 128 delete hepmcEvent; 124 129 125 // generate next event 130 // generate next event 126 hepmcEvent = GenerateHepMCEvent(); << 131 hepmcEvent= GenerateHepMCEvent(); 127 if (!hepmcEvent) { << 132 if(! hepmcEvent) { 128 G4cout << "HepMCInterface: no generated pa << 133 G4cout << "HepMCInterface: no generated particles. run terminated..." 129 G4RunManager::GetRunManager()->AbortRun(); << 134 << G4endl; >> 135 G4RunManager::GetRunManager()-> AbortRun(); 130 return; 136 return; 131 } 137 } 132 HepMC2G4(hepmcEvent, anEvent); 138 HepMC2G4(hepmcEvent, anEvent); 133 } 139 } 134 140