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 //--------------------------------------------------------------------- 27 //* |\___/| * 28 //* ) ( * 29 //* =\ /= * 30 //* )===( * 31 //* / \ CaTS: Calorimeter and Tracker Simulation * 32 //* | | is a flexible and extend-able framework * 33 //* / \ for the simulation of various detector * 34 //* \ / systems * 35 //* \__ _/ https://github.com/hanswenzel/CaTS * 36 //* ( ( * 37 //* ) ) * 38 //* (_( * 39 //* CaTS also serves as an example that demonstrates how to use * 40 //* opticks from within Geant4 for the creation and propagation of * 41 //* optical photons. * 42 //* see https://bitbucket.org/simoncblyth/opticks.git). * 43 //* Ascii Art by Joan Stark: https://www.asciiworld.com/-Cats-2-.html * 44 //--------------------------------------------------------------------- 45 // 46 /// \file readMscHits.cc 47 /// \brief example how to read the CaTS::MscHits 48 // 49 // Root headers 50 #include "TFile.h" 51 #include "TSystem.h" 52 #include "TTree.h" 53 #include "TH1.h" 54 #include <string> 55 // project headers 56 #include "Event.hh" 57 #include "MscHit.hh" 58 #include "G4ThreeVector.hh" 59 60 int main(int argc, char** argv) 61 { 62 // initialize ROOT 63 TSystem ts; 64 gSystem->Load("libCaTSClassesDict"); 65 if(argc < 4) 66 { 67 G4cout << "Program requires 3 arguments: name of input file, name of " 68 "output file, Volume that sensitive detector is attached to" 69 << G4endl; 70 exit(1); 71 } 72 TFile* outfile = new TFile(argv[2], "RECREATE"); 73 TFile fo(argv[1]); 74 Event* event = new Event(); 75 TTree* Tevt = (TTree*) fo.Get("Events"); 76 Tevt->SetBranchAddress("event.", &event); 77 TBranch* fevtbranch = Tevt->GetBranch("event."); 78 Int_t nevent = fevtbranch->GetEntries(); 79 outfile->cd(); 80 TH1F* theta = new TH1F("angle", "angle", 100, 0.0, 0.15); 81 std::string CollectionName = argv[3]; 82 CollectionName = CollectionName + "_Msc_HC"; 83 for(Int_t i = 0; i < nevent; i++) 84 { 85 fevtbranch->GetEntry(i); 86 auto* hcmap = event->GetHCMap(); 87 for(const auto& ele : *hcmap) 88 { 89 if(ele.first.compare(CollectionName) == 0) 90 { 91 auto hits = ele.second; 92 G4int NbHits = hits.size(); 93 for(G4int ii = 0; ii < NbHits; ii++) 94 { 95 MscHit* mscHit = dynamic_cast<MscHit*>(hits.at(ii)); 96 G4ThreeVector invec(0, 0, 1.); 97 std::cout << std::setprecision(12) << mscHit->GetKinE() << " " 98 << std::setprecision(12) << mscHit->GetMomentum().getX() 99 << " " << std::setprecision(12) 100 << mscHit->GetMomentum().getY() << " " 101 << std::setprecision(12) << mscHit->GetMomentum().getZ() 102 << std::endl; 103 theta->Fill(invec.angle(mscHit->GetMomentum()) * 57.3); 104 } 105 } 106 } 107 } 108 outfile->Write(); 109 } 110