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 persistency/P01/src/ExP01TrackerSD.c 26 /// \file persistency/P01/src/ExP01TrackerSD.cc 27 /// \brief Implementation of the ExP01TrackerS 27 /// \brief Implementation of the ExP01TrackerSD class 28 // 28 // 29 // 29 // 30 // 30 // 31 //....oooOO0OOooo........oooOO0OOooo........oo 31 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 32 //....oooOO0OOooo........oooOO0OOooo........oo 32 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 33 33 34 #include "ExP01TrackerSD.hh" 34 #include "ExP01TrackerSD.hh" 35 << 36 #include "RootIO.hh" << 37 << 38 #include "G4HCofThisEvent.hh" 35 #include "G4HCofThisEvent.hh" 39 #include "G4SDManager.hh" << 40 #include "G4Step.hh" 36 #include "G4Step.hh" 41 #include "G4ThreeVector.hh" 37 #include "G4ThreeVector.hh" >> 38 #include "G4SDManager.hh" 42 #include "G4ios.hh" 39 #include "G4ios.hh" 43 40 >> 41 #include "RootIO.hh" >> 42 44 //....oooOO0OOooo........oooOO0OOooo........oo 43 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 45 44 46 ExP01TrackerSD::ExP01TrackerSD(G4String name) 45 ExP01TrackerSD::ExP01TrackerSD(G4String name) 47 : G4VSensitiveDetector(name), fTrackerCollec << 46 :G4VSensitiveDetector(name), fTrackerCollection(0), fHCID(0) 48 { 47 { 49 G4String HCname = name + "_HC"; 48 G4String HCname = name + "_HC"; 50 collectionName.insert(HCname); 49 collectionName.insert(HCname); 51 G4cout << collectionName.size() << " Calor << 50 G4cout << collectionName.size() << " CalorimeterSD name: " << name << " collection Name: " 52 << " collection Name: " << HCname << << 51 << HCname << G4endl; 53 fHCID = -1; 52 fHCID = -1; 54 } 53 } 55 54 56 //....oooOO0OOooo........oooOO0OOooo........oo 55 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 57 56 58 ExP01TrackerSD::~ExP01TrackerSD() 57 ExP01TrackerSD::~ExP01TrackerSD() 59 { << 58 { 60 RootIO::GetInstance()->Close(); 59 RootIO::GetInstance()->Close(); 61 } 60 } 62 61 63 //....oooOO0OOooo........oooOO0OOooo........oo 62 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 64 63 65 void ExP01TrackerSD::Initialize(G4HCofThisEven 64 void ExP01TrackerSD::Initialize(G4HCofThisEvent* HCE) 66 { 65 { 67 fTrackerCollection = new ExP01TrackerHitsCol << 66 fTrackerCollection = new ExP01TrackerHitsCollection >> 67 (SensitiveDetectorName,collectionName[0]); 68 if (fHCID < 0) { 68 if (fHCID < 0) { 69 G4cout << "CalorimeterSD::Initialize: " < << 69 G4cout << "CalorimeterSD::Initialize: " << SensitiveDetectorName << " " 70 << G4endl; << 70 << collectionName[0] << G4endl; 71 fHCID = G4SDManager::GetSDMpointer()->GetC 71 fHCID = G4SDManager::GetSDMpointer()->GetCollectionID(collectionName[0]); >> 72 72 } 73 } 73 HCE->AddHitsCollection(fHCID, fTrackerCollec 74 HCE->AddHitsCollection(fHCID, fTrackerCollection); 74 } 75 } 75 76 76 //....oooOO0OOooo........oooOO0OOooo........oo 77 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 77 78 78 G4bool ExP01TrackerSD::ProcessHits(G4Step* aSt << 79 G4bool ExP01TrackerSD::ProcessHits(G4Step* aStep,G4TouchableHistory*) 79 { 80 { 80 G4double edep = aStep->GetTotalEnergyDeposit 81 G4double edep = aStep->GetTotalEnergyDeposit(); 81 82 82 if (edep == 0.) return false; << 83 if(edep==0.) return false; 83 84 84 ExP01TrackerHit* newHit = new ExP01TrackerHi 85 ExP01TrackerHit* newHit = new ExP01TrackerHit(); 85 newHit->SetTrackID(aStep->GetTrack()->GetTra << 86 newHit->SetTrackID (aStep->GetTrack()->GetTrackID()); 86 newHit->SetChamberNb(aStep->GetPreStepPoint( << 87 newHit->SetChamberNb(aStep->GetPreStepPoint()->GetTouchable() 87 newHit->SetEdep(edep); << 88 ->GetReplicaNumber()); 88 newHit->SetPos(aStep->GetPostStepPoint()->Ge << 89 newHit->SetEdep (edep); 89 fTrackerCollection->insert(newHit); << 90 newHit->SetPos (aStep->GetPostStepPoint()->GetPosition()); 90 << 91 fTrackerCollection->insert( newHit ); 91 // newHit->Print(); << 92 92 // newHit->Draw(); << 93 //newHit->Print(); >> 94 //newHit->Draw(); 93 95 94 return true; 96 return true; 95 } 97 } 96 98 97 //....oooOO0OOooo........oooOO0OOooo........oo 99 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 98 100 99 void ExP01TrackerSD::EndOfEvent(G4HCofThisEven 101 void ExP01TrackerSD::EndOfEvent(G4HCofThisEvent*) 100 { 102 { 101 // storing the hits in ROOT file 103 // storing the hits in ROOT file 102 G4int NbHits = fTrackerCollection->entries() 104 G4int NbHits = fTrackerCollection->entries(); 103 std::vector<ExP01TrackerHit*> hitsVector; 105 std::vector<ExP01TrackerHit*> hitsVector; 104 106 105 { << 107 { 106 G4cout << "\n-------->Storing hits in the << 108 G4cout << "\n-------->Storing hits in the ROOT file: in this event there are " << NbHits 107 << " hits in the tracker chambers: 109 << " hits in the tracker chambers: " << G4endl; 108 for (G4int i = 0; i < NbHits; i++) << 110 for (G4int i=0;i<NbHits;i++) (*fTrackerCollection)[i]->Print(); 109 (*fTrackerCollection)[i]->Print(); << 111 } 110 } << 111 112 112 for (G4int i = 0; i < NbHits; i++) << 113 >> 114 for (G4int i=0;i<NbHits;i++) 113 hitsVector.push_back((*fTrackerCollection) 115 hitsVector.push_back((*fTrackerCollection)[i]); 114 << 116 115 RootIO::GetInstance()->Write(&hitsVector); 117 RootIO::GetInstance()->Write(&hitsVector); 116 118 117 // 119 // 118 } 120 } 119 121 120 //....oooOO0OOooo........oooOO0OOooo........oo 122 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... >> 123 121 124