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 /// \file optical/LXe/src/LXeTrackingAction.cc 28 /// \brief Implementation of the LXeTrackingAction class 29 // 30 // 31 #include "LXeTrackingAction.hh" 32 33 #include "LXeDetectorConstruction.hh" 34 #include "LXeTrajectory.hh" 35 #include "LXeUserTrackInformation.hh" 36 37 #include "G4OpticalPhoton.hh" 38 #include "G4Track.hh" 39 #include "G4TrackingManager.hh" 40 41 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 42 43 void LXeTrackingAction::PreUserTrackingAction(const G4Track* aTrack) 44 { 45 // Let this be up to the user via vis.mac 46 // fpTrackingManager->SetStoreTrajectory(true); 47 48 // Use custom trajectory class 49 fpTrackingManager->SetTrajectory(new LXeTrajectory(aTrack)); 50 51 // This user track information is only relevant to the photons 52 fpTrackingManager->SetUserTrackInformation(new LXeUserTrackInformation); 53 } 54 55 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... 56 57 void LXeTrackingAction::PostUserTrackingAction(const G4Track* aTrack) 58 { 59 auto trajectory = (LXeTrajectory*)fpTrackingManager->GimmeTrajectory(); 60 auto trackInformation = (LXeUserTrackInformation*)aTrack->GetUserInformation(); 61 62 // Let's choose to draw only the photons that hit the sphere and a pmt 63 if (aTrack->GetDefinition() == G4OpticalPhoton::OpticalPhotonDefinition()) { 64 const G4VProcess* creator = aTrack->GetCreatorProcess(); 65 if (creator && creator->GetProcessName() == "OpWLS") { 66 trajectory->WLS(); 67 trajectory->SetDrawTrajectory(true); 68 } 69 70 if (LXeDetectorConstruction::GetSphereOn()) { 71 if ((trackInformation->GetTrackStatus() & hitPMT) 72 && (trackInformation->GetTrackStatus() & hitSphere)) 73 { 74 trajectory->SetDrawTrajectory(true); 75 } 76 } 77 else { 78 if (trackInformation->GetTrackStatus() & hitPMT) trajectory->SetDrawTrajectory(true); 79 } 80 } 81 // draw all other (not optical photon) trajectories 82 else 83 trajectory->SetDrawTrajectory(true); 84 85 if (trackInformation->GetForceDrawTrajectory()) trajectory->SetDrawTrajectory(true); 86 } 87