Geant4 Cross Reference

Cross-Referencing   Geant4
Geant4/examples/extended/optical/LXe/src/LXeTrajectory.cc

Version: [ ReleaseNotes ] [ 1.0 ] [ 1.1 ] [ 2.0 ] [ 3.0 ] [ 3.1 ] [ 3.2 ] [ 4.0 ] [ 4.0.p1 ] [ 4.0.p2 ] [ 4.1 ] [ 4.1.p1 ] [ 5.0 ] [ 5.0.p1 ] [ 5.1 ] [ 5.1.p1 ] [ 5.2 ] [ 5.2.p1 ] [ 5.2.p2 ] [ 6.0 ] [ 6.0.p1 ] [ 6.1 ] [ 6.2 ] [ 6.2.p1 ] [ 6.2.p2 ] [ 7.0 ] [ 7.0.p1 ] [ 7.1 ] [ 7.1.p1 ] [ 8.0 ] [ 8.0.p1 ] [ 8.1 ] [ 8.1.p1 ] [ 8.1.p2 ] [ 8.2 ] [ 8.2.p1 ] [ 8.3 ] [ 8.3.p1 ] [ 8.3.p2 ] [ 9.0 ] [ 9.0.p1 ] [ 9.0.p2 ] [ 9.1 ] [ 9.1.p1 ] [ 9.1.p2 ] [ 9.1.p3 ] [ 9.2 ] [ 9.2.p1 ] [ 9.2.p2 ] [ 9.2.p3 ] [ 9.2.p4 ] [ 9.3 ] [ 9.3.p1 ] [ 9.3.p2 ] [ 9.4 ] [ 9.4.p1 ] [ 9.4.p2 ] [ 9.4.p3 ] [ 9.4.p4 ] [ 9.5 ] [ 9.5.p1 ] [ 9.5.p2 ] [ 9.6 ] [ 9.6.p1 ] [ 9.6.p2 ] [ 9.6.p3 ] [ 9.6.p4 ] [ 10.0 ] [ 10.0.p1 ] [ 10.0.p2 ] [ 10.0.p3 ] [ 10.0.p4 ] [ 10.1 ] [ 10.1.p1 ] [ 10.1.p2 ] [ 10.1.p3 ] [ 10.2 ] [ 10.2.p1 ] [ 10.2.p2 ] [ 10.2.p3 ] [ 10.3 ] [ 10.3.p1 ] [ 10.3.p2 ] [ 10.3.p3 ] [ 10.4 ] [ 10.4.p1 ] [ 10.4.p2 ] [ 10.4.p3 ] [ 10.5 ] [ 10.5.p1 ] [ 10.6 ] [ 10.6.p1 ] [ 10.6.p2 ] [ 10.6.p3 ] [ 10.7 ] [ 10.7.p1 ] [ 10.7.p2 ] [ 10.7.p3 ] [ 10.7.p4 ] [ 11.0 ] [ 11.0.p1 ] [ 11.0.p2 ] [ 11.0.p3, ] [ 11.0.p4 ] [ 11.1 ] [ 11.1.1 ] [ 11.1.2 ] [ 11.1.3 ] [ 11.2 ] [ 11.2.1 ] [ 11.2.2 ] [ 11.3.0 ]

  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/LXeTrajectory.cc
 28 /// \brief Implementation of the LXeTrajectory class
 29 //
 30 //
 31 #include "LXeTrajectory.hh"
 32 
 33 #include "G4Circle.hh"
 34 #include "G4Colour.hh"
 35 #include "G4ParticleTable.hh"
 36 #include "G4ParticleTypes.hh"
 37 #include "G4Polyline.hh"
 38 #include "G4Polymarker.hh"
 39 #include "G4ThreeVector.hh"
 40 #include "G4Trajectory.hh"
 41 #include "G4TrajectoryPoint.hh"
 42 #include "G4VVisManager.hh"
 43 #include "G4VisAttributes.hh"
 44 
 45 G4ThreadLocal G4Allocator<LXeTrajectory>* LXeTrajectoryAllocator = nullptr;
 46 
 47 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 48 
 49 LXeTrajectory::LXeTrajectory(const G4Track* aTrack) : G4Trajectory(aTrack)
 50 {
 51   fParticleDefinition = aTrack->GetDefinition();
 52 }
 53 
 54 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 55 
 56 LXeTrajectory::LXeTrajectory(LXeTrajectory& right)
 57   : G4Trajectory(right), fWls(right.fWls), fDrawit(right.fDrawit)
 58 {
 59   fParticleDefinition = right.fParticleDefinition;
 60 }
 61 
 62 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
 63 
 64 void LXeTrajectory::DrawTrajectory() const
 65 {
 66   // Taken from G4VTrajectory and modified to select colours based on particle
 67   // type and to selectively eliminate drawing of certain trajectories.
 68 
 69   if (!fForceDraw && (!fDrawit || fForceNoDraw)) return;
 70 
 71   G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
 72   if (!pVVisManager) return;
 73 
 74   const G4double markerSize = 0.05;
 75   G4bool lineRequired = true;
 76   G4bool markersRequired = true;
 77 
 78   G4Polyline trajectoryLine;
 79   G4Polymarker stepPoints;
 80   G4Polymarker auxiliaryPoints;
 81 
 82   for (G4int i = 0; i < GetPointEntries(); ++i) {
 83     G4VTrajectoryPoint* aTrajectoryPoint = GetPoint(i);
 84     const std::vector<G4ThreeVector>* auxiliaries = aTrajectoryPoint->GetAuxiliaryPoints();
 85     if (auxiliaries) {
 86       for (size_t iAux = 0; iAux < auxiliaries->size(); ++iAux) {
 87         const G4ThreeVector pos((*auxiliaries)[iAux]);
 88         if (lineRequired) {
 89           trajectoryLine.push_back(pos);
 90         }
 91         if (markersRequired) {
 92           auxiliaryPoints.push_back(pos);
 93         }
 94       }
 95     }
 96     const G4ThreeVector pos(aTrajectoryPoint->GetPosition());
 97     if (lineRequired) {
 98       trajectoryLine.push_back(pos);
 99     }
100     if (markersRequired) {
101       stepPoints.push_back(pos);
102     }
103   }
104 
105   if (lineRequired) {
106     G4Colour colour;
107 
108     if (fParticleDefinition == G4OpticalPhoton::OpticalPhotonDefinition()) {
109       if (fWls)  // WLS photons are red
110         colour = G4Colour(1., 0., 0.);
111       else {  // Scintillation and Cerenkov photons are green
112         colour = G4Colour(0., 1., 0.);
113       }
114     }
115     else  // All other particles are blue
116       colour = G4Colour(0., 0., 1.);
117 
118     G4VisAttributes trajectoryLineAttribs(colour);
119     trajectoryLine.SetVisAttributes(&trajectoryLineAttribs);
120     pVVisManager->Draw(trajectoryLine);
121   }
122   if (markersRequired) {
123     auxiliaryPoints.SetMarkerType(G4Polymarker::squares);
124     auxiliaryPoints.SetScreenSize(markerSize);
125     auxiliaryPoints.SetFillStyle(G4VMarker::filled);
126     G4VisAttributes auxiliaryPointsAttribs(G4Colour(0., 1., 1.));  // Magenta
127     auxiliaryPoints.SetVisAttributes(&auxiliaryPointsAttribs);
128     pVVisManager->Draw(auxiliaryPoints);
129 
130     stepPoints.SetMarkerType(G4Polymarker::circles);
131     stepPoints.SetScreenSize(markerSize);
132     stepPoints.SetFillStyle(G4VMarker::filled);
133     G4VisAttributes stepPointsAttribs(G4Colour(1., 1., 0.));  // Yellow
134     stepPoints.SetVisAttributes(&stepPointsAttribs);
135     pVVisManager->Draw(stepPoints);
136   }
137 }
138