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 // G4HEPEvtInterface class implementation << 27 // 26 // 28 // Author: Makoto Asai, 1997 << 27 // $Id$ >> 28 // >> 29 // 29 // ------------------------------------------- 30 // -------------------------------------------------------------------- 30 31 31 #include "G4HEPEvtInterface.hh" 32 #include "G4HEPEvtInterface.hh" 32 33 33 #include "G4Types.hh" 34 #include "G4Types.hh" 34 #include "G4SystemOfUnits.hh" 35 #include "G4SystemOfUnits.hh" 35 36 36 #include "G4ios.hh" 37 #include "G4ios.hh" 37 #include "G4PrimaryVertex.hh" 38 #include "G4PrimaryVertex.hh" 38 #include "G4PrimaryParticle.hh" 39 #include "G4PrimaryParticle.hh" 39 #include "G4HEPEvtParticle.hh" 40 #include "G4HEPEvtParticle.hh" 40 #include "G4Event.hh" 41 #include "G4Event.hh" 41 42 42 G4HEPEvtInterface::G4HEPEvtInterface(const cha << 43 G4HEPEvtInterface::G4HEPEvtInterface(char* evfile) 43 : vLevel(vl) << 44 { 44 { 45 inputFile.open((char*)evfile); << 45 inputFile.open(evfile); 46 if (inputFile.is_open()) << 46 if (inputFile) { 47 { << 48 fileName = evfile; 47 fileName = evfile; 49 if(vl>0) << 50 G4cout << "G4HEPEvtInterface - " << file << 51 } 48 } 52 else << 49 else { 53 { << 50 G4Exception("G4HEPEvtInterface::G4HEPEvtInterface","Event0201",FatalException, 54 G4Exception("G4HEPEvtInterface::G4HEPEvtIn << 51 "G4HEPEvtInterface:: cannot open file."); 55 FatalException, "G4HEPEvtInter << 56 } 52 } 57 G4ThreeVector zero; 53 G4ThreeVector zero; 58 particle_position = zero; 54 particle_position = zero; 59 particle_time = 0.0; 55 particle_time = 0.0; >> 56 60 } 57 } 61 58 62 void G4HEPEvtInterface::GeneratePrimaryVertex( << 59 G4HEPEvtInterface::G4HEPEvtInterface(G4String evfile) 63 { 60 { 64 G4int NHEP = 0; // number of entries << 61 const char* fn = evfile.data(); 65 if (inputFile.is_open()) << 62 inputFile.open((char*)fn); 66 { << 63 if (inputFile) { 67 inputFile >> NHEP; << 64 fileName = evfile; 68 } 65 } 69 else << 66 else { 70 { << 67 G4Exception("G4HEPEvtInterface::G4HEPEvtInterface","Event0201",FatalException, 71 G4Exception("G4HEPEvtInterface::G4HEPEvtIn << 68 "G4HEPEvtInterface:: cannot open file."); 72 FatalException, "G4HEPEvtInter << 73 } 69 } >> 70 G4ThreeVector zero; >> 71 particle_position = zero; >> 72 particle_time = 0.0; >> 73 } >> 74 >> 75 G4HEPEvtInterface::~G4HEPEvtInterface() >> 76 {;} >> 77 >> 78 void G4HEPEvtInterface::GeneratePrimaryVertex(G4Event* evt) >> 79 { >> 80 G4int NHEP; // number of entries >> 81 inputFile >> NHEP; 74 if( inputFile.eof() ) 82 if( inputFile.eof() ) 75 { 83 { 76 G4Exception("G4HEPEvtInterface::GeneratePr << 84 G4Exception("G4HEPEvtInterface::GeneratePrimaryVertex","Event0202", 77 RunMustBeAborted, << 85 JustWarning,"End-Of-File : HEPEvt input file"); 78 "End-Of-File: HEPEvt input fil << 79 return; 86 return; 80 } 87 } 81 88 82 if(vLevel > 0) << 89 for( G4int IHEP=0; IHEP<NHEP; IHEP++ ) 83 { << 84 G4cout << "G4HEPEvtInterface - reading " < << 85 << " HEPEvt particles from " << fil << 86 } << 87 for( G4int IHEP=0; IHEP<NHEP; ++IHEP ) << 88 { 90 { 89 G4int ISTHEP; // status code 91 G4int ISTHEP; // status code 90 G4int IDHEP; // PDG code 92 G4int IDHEP; // PDG code 91 G4int JDAHEP1; // first daughter 93 G4int JDAHEP1; // first daughter 92 G4int JDAHEP2; // last daughter 94 G4int JDAHEP2; // last daughter 93 G4double PHEP1; // px in GeV 95 G4double PHEP1; // px in GeV 94 G4double PHEP2; // py in GeV 96 G4double PHEP2; // py in GeV 95 G4double PHEP3; // pz in GeV 97 G4double PHEP3; // pz in GeV 96 G4double PHEP5; // mass in GeV 98 G4double PHEP5; // mass in GeV 97 99 98 inputFile >> ISTHEP >> IDHEP >> JDAHEP1 >> 100 inputFile >> ISTHEP >> IDHEP >> JDAHEP1 >> JDAHEP2 99 >> PHEP1 >> PHEP2 >> PHEP3 >> PH << 101 >> PHEP1 >> PHEP2 >> PHEP3 >> PHEP5; 100 if( inputFile.eof() ) << 101 { << 102 G4Exception("G4HEPEvtInterface::Generate << 103 FatalException, << 104 "Unexpected End-Of-File in t << 105 } << 106 if(vLevel > 1) << 107 { << 108 G4cout << " " << ISTHEP << " " << IDHEP << 109 << " " << JDAHEP2 << " " << PHEP1 << 110 << " " << PHEP3 << " " << PHEP5 < << 111 } << 112 102 113 // Create G4PrimaryParticle object << 103 // create G4PrimaryParticle object 114 // << 104 G4PrimaryParticle* particle 115 auto* particle = new G4PrimaryParticle( ID << 105 = new G4PrimaryParticle( IDHEP ); 116 particle->SetMass( PHEP5*GeV ); 106 particle->SetMass( PHEP5*GeV ); 117 particle->SetMomentum(PHEP1*GeV, PHEP2*GeV 107 particle->SetMomentum(PHEP1*GeV, PHEP2*GeV, PHEP3*GeV ); 118 108 119 // Create G4HEPEvtParticle object << 109 // create G4HEPEvtParticle object 120 // << 110 G4HEPEvtParticle* hepParticle 121 auto* hepParticle << 122 = new G4HEPEvtParticle( particle, ISTHEP 111 = new G4HEPEvtParticle( particle, ISTHEP, JDAHEP1, JDAHEP2 ); 123 112 124 // Store 113 // Store 125 // << 126 HPlist.push_back( hepParticle ); 114 HPlist.push_back( hepParticle ); 127 } 115 } 128 116 129 // Check if there is at least one particle << 117 // check if there is at least one particle 130 // << 118 if( HPlist.size() == 0 ) return; 131 if( HPlist.empty() ) return; << 119 132 << 120 // make connection between daughter particles decayed from 133 // Make connection between daughter particle << 121 // the same mother 134 // << 122 for( size_t i=0; i<HPlist.size(); i++ ) 135 for(auto & i : HPlist) << 136 { 123 { 137 if( i->GetJDAHEP1() > 0 ) // it has daugh << 124 if( HPlist[i]->GetJDAHEP1() > 0 ) // it has daughters 138 { 125 { 139 G4int jda1 = i->GetJDAHEP1()-1; // FORTR << 126 G4int jda1 = HPlist[i]->GetJDAHEP1()-1; // FORTRAN index starts from 1 140 G4int jda2 = i->GetJDAHEP2()-1; // but C << 127 G4int jda2 = HPlist[i]->GetJDAHEP2()-1; // but C++ starts from 0. 141 G4PrimaryParticle* mother = i->GetThePar << 128 G4PrimaryParticle* mother = HPlist[i]->GetTheParticle(); 142 for( G4int j=jda1; j<=jda2; ++j ) << 129 for( G4int j=jda1; j<=jda2; j++ ) 143 { 130 { 144 G4PrimaryParticle* daughter = HPlist[j 131 G4PrimaryParticle* daughter = HPlist[j]->GetTheParticle(); 145 if(HPlist[j]->GetISTHEP()>0) 132 if(HPlist[j]->GetISTHEP()>0) 146 { 133 { 147 mother->SetDaughter( daughter ); 134 mother->SetDaughter( daughter ); 148 HPlist[j]->Done(); 135 HPlist[j]->Done(); 149 } 136 } 150 } 137 } 151 } 138 } 152 } 139 } 153 140 154 // Create G4PrimaryVertex object << 141 // create G4PrimaryVertex object 155 // << 142 G4PrimaryVertex* vertex = new G4PrimaryVertex(particle_position,particle_time); 156 auto* vertex = new G4PrimaryVertex(particle_ << 143 157 << 144 // put initial particles to the vertex 158 // Put initial particles to the vertex << 145 for( size_t ii=0; ii<HPlist.size(); ii++ ) 159 // << 160 for(auto & ii : HPlist) << 161 { 146 { 162 if( ii->GetISTHEP() > 0 ) // ISTHEP of dau << 147 if( HPlist[ii]->GetISTHEP() > 0 ) // ISTHEP of daughters had been 163 // set t << 148 // set to negative 164 { 149 { 165 G4PrimaryParticle* initialParticle = ii- << 150 G4PrimaryParticle* initialParticle = HPlist[ii]->GetTheParticle(); 166 vertex->SetPrimary( initialParticle ); 151 vertex->SetPrimary( initialParticle ); 167 } 152 } 168 } 153 } 169 154 170 // Clear G4HEPEvtParticles << 155 // clear G4HEPEvtParticles 171 // << 156 //HPlist.clearAndDestroy(); 172 for(auto & iii : HPlist) << 157 for(size_t iii=0;iii<HPlist.size();iii++) 173 { << 158 { delete HPlist[iii]; } 174 delete iii; << 175 } << 176 HPlist.clear(); 159 HPlist.clear(); 177 160 178 // Put the vertex to G4Event object 161 // Put the vertex to G4Event object 179 // << 180 evt->AddPrimaryVertex( vertex ); 162 evt->AddPrimaryVertex( vertex ); 181 } 163 } >> 164 182 165