AtlasTrajectory Class Reference

Class to store G4 trajectory information. More...

#include <AtlasTrajectory.h>

Inheritance diagram for AtlasTrajectory:

Collaboration diagram for AtlasTrajectory:

Public Member Functions
	AtlasTrajectory (const G4Track *aTrack, int subDetVolLevel)
	Constructor. More...
void	AppendStep (const G4Step *aStep)
	Overriden from G4 in order to do vertex analysis. More...
void	DrawTrajectory (G4int) const
	Visualization stuff. More...

Private Attributes
int	m_subDetVolLevel
	The level in the G4 volume hierarchy at which can we find the sub-detector name. More...

Detailed Description

Class to store G4 trajectory information.

NEEDS DOCUMENTATION

Definition at line 14 of file AtlasTrajectory.h.

Constructor & Destructor Documentation

AtlasTrajectory()

AtlasTrajectory::AtlasTrajectory	(	const G4Track *	aTrack,
		int	subDetVolLevel
	)

Constructor.

Definition at line 19 of file AtlasTrajectory.cxx.

  : G4Trajectory(track), m_subDetVolLevel(subDetVolLevel)
{}

Member Function Documentation

AppendStep()

void AtlasTrajectory::AppendStep

(

const G4Step *

aStep

)

Overriden from G4 in order to do vertex analysis.

Definition at line 24 of file AtlasTrajectory.cxx.

{
  // only use truth service if there are new any secondaries
  const int numSecondaries = aStep->GetSecondaryInCurrentStep()->size();
  // This method not available until G4 10.2
  //const int numSecondaries = aStep->GetNumberOfSecondariesInCurrentStep();
 
  if (numSecondaries) {
      // OK, there was an interation. look at the track, if it
      // is not a secondary (i.e. we have a connected tree) we
      // apply the MC truth machinery...
      TrackHelper tHelper(aStep->GetTrack());
      if (!tHelper.IsSecondary())
        {
          const TruthStrategyManager& sManager =
            TruthStrategyManager::GetStrategyManager();
          sManager.CreateTruthIncident(aStep, m_subDetVolLevel);
        }
    }
 
  // Call the base class
  G4Trajectory::AppendStep(aStep);
}

DrawTrajectory()

void AtlasTrajectory::DrawTrajectory

(

G4int

i_mode = 0

)

const

Visualization stuff.

Definition at line 49 of file AtlasTrajectory.cxx.

{
  // If i_mode>=0, draws a trajectory as a polyline (blue for
  // positive, red for negative, green for neutral) and, if i_mode!=0,
  // adds markers - yellow circles for step points and magenta squares
  // for auxiliary points, if any - whose screen size in pixels is
  // given by abs(i_mode)/1000.  E.g: i_mode = 5000 gives easily
  // visible markers.
 
  G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
  if (!pVVisManager) return;
 
  const G4double markerSize = abs(i_mode)*0.001;
  G4bool lineRequired (i_mode >= 0);
  G4bool markersRequired (markerSize > 0.);
 
  G4Polyline trajectoryLine;
  G4Polymarker stepPoints;
  G4Polymarker auxiliaryPoints;
 
  for (G4int i = 0; i < GetPointEntries() ; i++)
    {
      G4VTrajectoryPoint* aTrajectoryPoint = GetPoint(i);
      const std::vector<G4ThreeVector>* auxiliaries
        = aTrajectoryPoint->GetAuxiliaryPoints();
      if (auxiliaries)
        {
          for (size_t iAux = 0; iAux < auxiliaries->size(); ++iAux)
            {
              const G4ThreeVector pos((*auxiliaries)[iAux]);
              if (lineRequired)
                {
                  trajectoryLine.push_back(pos);
                }
              if (markersRequired)
                {
                  auxiliaryPoints.push_back(pos);
                }
            }
        }
      const G4ThreeVector pos(aTrajectoryPoint->GetPosition());
      if (lineRequired)
        {
          trajectoryLine.push_back(pos);
        }
      if (markersRequired)
        {
          stepPoints.push_back(pos);
        }
    }
 
  if (lineRequired)
    {
      int visScheme=TruthController::getTruthController()->
        getVisualizationHelper()->trackVisualizationScheme();
      G4Colour colour;
      const G4double charge = GetCharge();
      if (visScheme==1)
        {
          // Geant3-like color code - neutrals are dashed
          const G4String pname=GetParticleName();
          if (pname=="e+" || pname=="e-") colour = G4Colour(1.,1.,0.);
          else if (pname=="mu+" || pname=="mu-") colour = G4Colour(0.,1.,0.);
          else if (pname=="gamma") colour = G4Colour(0.,0.,1.);
          else if (pname=="neutron") colour = G4Colour(0.,0.,0.);
          else if (charge) colour = G4Colour(1.,0.,0.);
        }
      else if (visScheme==2)
        {
          if(charge>0.)      colour = G4Colour(0.,0.,1.); // Blue = positive.
          else if(charge<0.) colour = G4Colour(1.,0.,0.); // Red = negative.
          else               colour = G4Colour(0.,1.,0.); // Green = neutral.
        }
      else if (visScheme==3)
        {
          const G4String pname=GetParticleName();
          if (pname=="e+" || pname=="e-") colour = G4Colour(1.,1.,0.);
          else if (pname=="mu+" || pname=="mu-") colour = G4Colour(0.,0.,1.);
          else if (pname=="gamma") colour = G4Colour(60./256.,79./256.,113./256.);
          else if (pname=="neutron") colour = G4Colour(1.,1.,1.);
          else if (pname=="pi-"|| pname=="pi+")
            colour = G4Colour(250/256.,128/256.,114/256.);
          else colour = G4Colour(176/256.,48/256.,96/256.);
        }
      G4VisAttributes trajectoryLineAttribs(colour);
      if(visScheme==1 && charge==0)
        {
          G4VisAttributes::LineStyle style=G4VisAttributes::dotted;
          trajectoryLineAttribs.SetLineStyle(style);
        }
      trajectoryLine.SetVisAttributes(&trajectoryLineAttribs);
      pVVisManager->Draw(trajectoryLine);
    }
  if (markersRequired)
    {
      auxiliaryPoints.SetMarkerType(G4Polymarker::squares);
      auxiliaryPoints.SetScreenSize(markerSize);
      auxiliaryPoints.SetFillStyle(G4VMarker::filled);
      G4VisAttributes auxiliaryPointsAttribs(G4Colour(0.,1.,1.));  // Magenta
      auxiliaryPoints.SetVisAttributes(&auxiliaryPointsAttribs);
      pVVisManager->Draw(auxiliaryPoints);
 
      stepPoints.SetMarkerType(G4Polymarker::circles);
      stepPoints.SetScreenSize(markerSize);
      stepPoints.SetFillStyle(G4VMarker::filled);
      G4VisAttributes stepPointsAttribs(G4Colour(1.,1.,0.));  // Yellow.
      stepPoints.SetVisAttributes(&stepPointsAttribs);
      pVVisManager->Draw(stepPoints);
    }
}

Member Data Documentation

m_subDetVolLevel

int AtlasTrajectory::m_subDetVolLevel

private

The level in the G4 volume hierarchy at which can we find the sub-detector name.

Definition at line 36 of file AtlasTrajectory.h.

---

The documentation for this class was generated from the following files:

• AtlasTrajectory.h
• AtlasTrajectory.cxx