LXe/src/LXeTrajectory.cc

0001 //
0002 // ********************************************************************
0003 // * License and Disclaimer                                           *
0004 // *                                                                  *
0005 // * The  Geant4 software  is  copyright of the Copyright Holders  of *
0006 // * the Geant4 Collaboration.  It is provided  under  the terms  and *
0007 // * conditions of the Geant4 Software License,  included in the file *
0008 // * LICENSE and available at  http://cern.ch/geant4/license .  These *
0009 // * include a list of copyright holders.                             *
0010 // *                                                                  *
0011 // * Neither the authors of this software system, nor their employing *
0012 // * institutes,nor the agencies providing financial support for this *
0013 // * work  make  any representation or  warranty, express or implied, *
0014 // * regarding  this  software system or assume any liability for its *
0015 // * use.  Please see the license in the file  LICENSE  and URL above *
0016 // * for the full disclaimer and the limitation of liability.         *
0017 // *                                                                  *
0018 // * This  code  implementation is the result of  the  scientific and *
0019 // * technical work of the GEANT4 collaboration.                      *
0020 // * By using,  copying,  modifying or  distributing the software (or *
0021 // * any work based  on the software)  you  agree  to acknowledge its *
0022 // * use  in  resulting  scientific  publications,  and indicate your *
0023 // * acceptance of all terms of the Geant4 Software license.          *
0024 // ********************************************************************
0025 //
0026 //
0027 /// \file optical/LXe/src/LXeTrajectory.cc
0028 /// \brief Implementation of the LXeTrajectory class
0029 //
0030 //
0031 #include "LXeTrajectory.hh"
0032
0033 #include "G4Circle.hh"
0034 #include "G4Colour.hh"
0035 #include "G4ParticleTable.hh"
0036 #include "G4ParticleTypes.hh"
0037 #include "G4Polyline.hh"
0038 #include "G4Polymarker.hh"
0039 #include "G4ThreeVector.hh"
0040 #include "G4Trajectory.hh"
0041 #include "G4TrajectoryPoint.hh"
0042 #include "G4VVisManager.hh"
0043 #include "G4VisAttributes.hh"
0044
0045 G4ThreadLocal G4Allocator<LXeTrajectory>* LXeTrajectoryAllocator = nullptr;
0046
0047 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0048
0049 LXeTrajectory::LXeTrajectory(const G4Track* aTrack) : G4Trajectory(aTrack)
0050 {
0051   fParticleDefinition = aTrack->GetDefinition();
0052 }
0053
0054 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0055
0056 LXeTrajectory::LXeTrajectory(LXeTrajectory& right)
0057   : G4Trajectory(right), fWls(right.fWls), fDrawit(right.fDrawit)
0058 {
0059   fParticleDefinition = right.fParticleDefinition;
0060 }
0061
0062 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
0063
0064 void LXeTrajectory::DrawTrajectory() const
0065 {
0066   // Taken from G4VTrajectory and modified to select colours based on particle
0067   // type and to selectively eliminate drawing of certain trajectories.
0068
0069   if (!fForceDraw && (!fDrawit || fForceNoDraw)) return;
0070
0071   G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
0072   if (!pVVisManager) return;
0073
0074   const G4double markerSize = 0.05;
0075   G4bool lineRequired = true;
0076   G4bool markersRequired = true;
0077
0078   G4Polyline trajectoryLine;
0079   G4Polymarker stepPoints;
0080   G4Polymarker auxiliaryPoints;
0081
0082   for (G4int i = 0; i < GetPointEntries(); ++i) {
0083     G4VTrajectoryPoint* aTrajectoryPoint = GetPoint(i);
0084     const std::vector<G4ThreeVector>* auxiliaries = aTrajectoryPoint->GetAuxiliaryPoints();
0085     if (auxiliaries) {
0086       for (size_t iAux = 0; iAux < auxiliaries->size(); ++iAux) {
0087         const G4ThreeVector pos((*auxiliaries)[iAux]);
0088         if (lineRequired) {
0089           trajectoryLine.push_back(pos);
0090         }
0091         if (markersRequired) {
0092           auxiliaryPoints.push_back(pos);
0093         }
0094       }
0095     }
0096     const G4ThreeVector pos(aTrajectoryPoint->GetPosition());
0097     if (lineRequired) {
0098       trajectoryLine.push_back(pos);
0099     }
0100     if (markersRequired) {
0101       stepPoints.push_back(pos);
0102     }
0103   }
0104
0105   if (lineRequired) {
0106     G4Colour colour;
0107
0108     if (fParticleDefinition == G4OpticalPhoton::OpticalPhotonDefinition()) {
0109       if (fWls)  // WLS photons are red
0110         colour = G4Colour(1., 0., 0.);
0111       else {  // Scintillation and Cerenkov photons are green
0112         colour = G4Colour(0., 1., 0.);
0113       }
0114     }
0115     else  // All other particles are blue
0116       colour = G4Colour(0., 0., 1.);
0117
0118     G4VisAttributes trajectoryLineAttribs(colour);
0119     trajectoryLine.SetVisAttributes(&trajectoryLineAttribs);
0120     pVVisManager->Draw(trajectoryLine);
0121   }
0122   if (markersRequired) {
0123     auxiliaryPoints.SetMarkerType(G4Polymarker::squares);
0124     auxiliaryPoints.SetScreenSize(markerSize);
0125     auxiliaryPoints.SetFillStyle(G4VMarker::filled);
0126     G4VisAttributes auxiliaryPointsAttribs(G4Colour(0., 1., 1.));  // Magenta
0127     auxiliaryPoints.SetVisAttributes(&auxiliaryPointsAttribs);
0128     pVVisManager->Draw(auxiliaryPoints);
0129
0130     stepPoints.SetMarkerType(G4Polymarker::circles);
0131     stepPoints.SetScreenSize(markerSize);
0132     stepPoints.SetFillStyle(G4VMarker::filled);
0133     G4VisAttributes stepPointsAttribs(G4Colour(1., 1., 0.));  // Yellow
0134     stepPoints.SetVisAttributes(&stepPointsAttribs);
0135     pVVisManager->Draw(stepPoints);
0136   }
0137 }