WLSTrajectory Class Reference

#include <WLSTrajectory.hh>

Inheritance diagram for WLSTrajectory:

Public Member Functions
	WLSTrajectory ()
	WLSTrajectory (const G4Track *)
	WLSTrajectory (WLSTrajectory &)
virtual	~WLSTrajectory ()
void *	operator new (size_t)
void	operator delete (void *)
int	operator== (const WLSTrajectory &right) const
virtual G4int	GetTrackID () const
virtual G4int	GetParentID () const
virtual G4String	GetParticleName () const
virtual G4double	GetCharge () const
virtual G4int	GetPDGEncoding () const
virtual G4ThreeVector	GetInitialMomentum () const
virtual void	ShowTrajectory (std::ostream &os=G4cout) const
virtual void	DrawTrajectory () const
virtual void	AppendStep (const G4Step *aStep)
virtual void	MergeTrajectory (G4VTrajectory *secondTrajectory)
G4ParticleDefinition *	GetParticleDefinition ()
virtual int	GetPointEntries () const
virtual G4VTrajectoryPoint *	GetPoint (G4int i) const
virtual const std::map < G4String, G4AttDef > *	GetAttDefs () const
virtual std::vector< G4AttValue > *	CreateAttValues () const
void	SetDrawTrajectory (G4bool b)
void	WLS ()
void	SetForceDrawTrajectory (G4bool b)
void	SetForceNoDrawTrajectory (G4bool b)
Public Member Functions inherited from G4VTrajectory
	G4VTrajectory ()
virtual	~G4VTrajectory ()
G4bool	operator== (const G4VTrajectory &right) const

Detailed Description

Definition at line 53 of file WLSTrajectory.hh.

Constructor & Destructor Documentation

WLSTrajectory::WLSTrajectory

(

)

Definition at line 59 of file WLSTrajectory.cc.

    : fpPointsContainer(0), fTrackID(0), fParentID(0),
      fPDGCharge(0.0), fPDGEncoding(0), fParticleName(""),
      fInitialMomentum(G4ThreeVector())
{
    fWLS         = false;
    fDrawIt      = false;
    fForceNoDraw = false;
    fForceDraw   = false;

    fParticleDefinition = NULL;
}

WLSTrajectory::WLSTrajectory

(

const G4Track *

aTrack

)

Definition at line 74 of file WLSTrajectory.cc.

References G4Track::GetDefinition(), G4Track::GetMomentum(), G4Track::GetParentID(), G4ParticleDefinition::GetParticleName(), G4ParticleDefinition::GetPDGCharge(), G4ParticleDefinition::GetPDGEncoding(), and G4Track::GetTrackID().

{
    fParticleDefinition = aTrack->GetDefinition();
    fParticleName = fParticleDefinition->GetParticleName();
    fPDGCharge = fParticleDefinition->GetPDGCharge();
    fPDGEncoding = fParticleDefinition->GetPDGEncoding();
    fTrackID = aTrack->GetTrackID();
    fParentID = aTrack->GetParentID();
    fInitialMomentum = aTrack->GetMomentum();
    fpPointsContainer = new WLSTrajectoryPointContainer();
    // Following is for the first trajectory point
    fpPointsContainer->push_back(new WLSTrajectoryPoint(aTrack));

    fWLS         = false;
    fDrawIt      = false;
    fForceNoDraw = false;
    fForceDraw   = false;
}

WLSTrajectory::WLSTrajectory

(

WLSTrajectory &

right

)

Definition at line 95 of file WLSTrajectory.cc.

                                                  : G4VTrajectory()
{
    fParticleDefinition=right.fParticleDefinition;
    fParticleName = right.fParticleName;
    fPDGCharge = right.fPDGCharge;
    fPDGEncoding = right.fPDGEncoding;
    fTrackID = right.fTrackID;
    fParentID = right.fParentID;
    fInitialMomentum = right.fInitialMomentum;
    fpPointsContainer = new WLSTrajectoryPointContainer();

    for(size_t i=0;i<right.fpPointsContainer->size();++i) {
        WLSTrajectoryPoint* rightPoint
            = (WLSTrajectoryPoint*)((*(right.fpPointsContainer))[i]);
        fpPointsContainer->push_back(new WLSTrajectoryPoint(*rightPoint));
    }

    fWLS         = right.fWLS;
    fDrawIt      = right.fDrawIt;
    fForceNoDraw = right.fForceNoDraw;
    fForceDraw   = right.fForceDraw;
}

WLSTrajectory::~WLSTrajectory ( )

virtual

Definition at line 120 of file WLSTrajectory.cc.

{
    for(size_t i=0;i<fpPointsContainer->size();++i){
        delete  (*fpPointsContainer)[i];
    }
    fpPointsContainer->clear();

    delete fpPointsContainer;
}

Member Function Documentation

void WLSTrajectory::AppendStep ( const G4Step *  aStep )

virtual

Implements G4VTrajectory.

Definition at line 234 of file WLSTrajectory.cc.

{
    fpPointsContainer->push_back(new WLSTrajectoryPoint(aStep));
}

std::vector< G4AttValue > * WLSTrajectory::CreateAttValues ( ) const

virtual

Reimplemented from G4VTrajectory.

Definition at line 306 of file WLSTrajectory.cc.

References G4UIcommand::ConvertToString(), G4BestUnit, G4cout, GetAttDefs(), GetPointEntries(), and CLHEP::Hep3Vector::mag().

{
  std::vector<G4AttValue>* values = new std::vector<G4AttValue>;

  values->push_back
    (G4AttValue("ID",G4UIcommand::ConvertToString(fTrackID),""));

  values->push_back
    (G4AttValue("PID",G4UIcommand::ConvertToString(fParentID),""));

  values->push_back(G4AttValue("PN",fParticleName,""));

  values->push_back
    (G4AttValue("Ch",G4UIcommand::ConvertToString(fPDGCharge),""));

  values->push_back
    (G4AttValue("PDG",G4UIcommand::ConvertToString(fPDGEncoding),""));

  values->push_back
    (G4AttValue("IMom",G4BestUnit(fInitialMomentum,"Energy"),""));

  values->push_back
    (G4AttValue("IMag",G4BestUnit(fInitialMomentum.mag(),"Energy"),""));

  values->push_back
    (G4AttValue("NTP",G4UIcommand::ConvertToString(GetPointEntries()),""));

#ifdef G4ATTDEBUG
  G4cout << G4AttCheck(values,GetAttDefs());
#endif
    return values;
}

void WLSTrajectory::DrawTrajectory ( ) const

virtual

Reimplemented from G4VTrajectory.

Definition at line 141 of file WLSTrajectory.cc.

References G4Polymarker::circles, G4VVisManager::Draw(), G4VMarker::filled, G4VTrajectoryPoint::GetAuxiliaryPoints(), G4VVisManager::GetConcreteInstance(), GetPoint(), GetPointEntries(), G4VTrajectoryPoint::GetPosition(), G4OpticalPhoton::OpticalPhotonDefinition(), G4VMarker::SetFillStyle(), G4Polymarker::SetMarkerType(), G4VMarker::SetScreenSize(), G4Visible::SetVisAttributes(), and G4Polymarker::squares.

Referenced by WLSEventAction::EndOfEventAction().

{
    // i_mode is no longer available as an argument of G4VTrajectory.
    // In this exampple it was always called with an argument of 50.
    const G4int i_mode = 50;
    // Consider using commands /vis/modeling/trajectories.

    // Invoke the default implementation in G4VTrajectory...
    // G4VTrajectory::DrawTrajectory(i_mode);
    // ... or override with your own code here.

    //Taken from G4VTrajectory and modified to select colours based on particle
    //type and to selectively eliminate drawing of certain trajectories.

    if (!fForceDraw && (!fDrawIt || fForceNoDraw)) return;

    // If i_mode>=0, draws a trajectory as a polyline 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 std::abs(i_mode)/1000.  E.g: i_mode = 5000 gives easily
    // visible markers.

    G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
    if (!pVVisManager) return;

    const G4double markerSize = std::abs(i_mode)/1000;
    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) {
      G4Colour colour;

      if(fParticleDefinition==G4OpticalPhoton::OpticalPhotonDefinition()){
        if(fWLS) //WLS photons are red
          colour = G4Colour(1.,0.,0.);
        else{ //Scintillation and Cerenkov photons are green
          colour = G4Colour(0.,1.,0.);
        }
      }
      else //All other particles are blue
        colour = G4Colour(0.,0.,1.);

      G4VisAttributes trajectoryLineAttribs(colour);
      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);
    }
}

const std::map< G4String, G4AttDef > * WLSTrajectory::GetAttDefs ( ) const

virtual

Reimplemented from G4VTrajectory.

Definition at line 264 of file WLSTrajectory.cc.

References G4AttDefStore::GetInstance().

Referenced by CreateAttValues().

{
    G4bool isNew;
    std::map<G4String,G4AttDef>* store
        = G4AttDefStore::GetInstance("Trajectory",isNew);

    if (isNew) {

      G4String ID("ID");
      (*store)[ID] = G4AttDef(ID,"Track ID","Bookkeeping","","G4int");

      G4String PID("PID");
      (*store)[PID] = G4AttDef(PID,"Parent ID","Bookkeeping","","G4int");

      G4String PN("PN");
      (*store)[PN] = G4AttDef(PN,"Particle Name","Physics","","G4String");

      G4String Ch("Ch");
      (*store)[Ch] = G4AttDef(Ch,"Charge","Physics","e+","G4double");

      G4String PDG("PDG");
      (*store)[PDG] = G4AttDef(PDG,"PDG Encoding","Physics","","G4int");

      G4String IMom("IMom");
      (*store)[IMom] = G4AttDef(IMom,
                       "Momentum of track at start of trajectory",
                       "Physics","G4BestUnit","G4ThreeVector");

      G4String IMag("IMag");
      (*store)[IMag] = G4AttDef(IMag,
                       "Magnitude of momentum of track at start of trajectory",
                       "Physics","G4BestUnit","G4double");

        G4String NTP("NTP");
        (*store)[NTP] = G4AttDef(NTP,"No. of points","Bookkeeping","","G4int");

    }
    return store;
}

virtual G4double WLSTrajectory::GetCharge ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 79 of file WLSTrajectory.hh.

Referenced by WLSEventAction::EndOfEventAction().

{ return fPDGCharge; }

virtual G4ThreeVector WLSTrajectory::GetInitialMomentum ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 81 of file WLSTrajectory.hh.

                                                    {return fInitialMomentum;}

virtual G4int WLSTrajectory::GetParentID ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 77 of file WLSTrajectory.hh.

{ return fParentID; }

G4ParticleDefinition * WLSTrajectory::GetParticleDefinition

(

)

Definition at line 241 of file WLSTrajectory.cc.

References G4ParticleTable::GetParticleTable().

{
    return (G4ParticleTable::GetParticleTable()->FindParticle(fParticleName));
}

virtual G4String WLSTrajectory::GetParticleName ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 78 of file WLSTrajectory.hh.

Referenced by WLSEventAction::EndOfEventAction().

{ return fParticleName; }

virtual G4int WLSTrajectory::GetPDGEncoding ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 80 of file WLSTrajectory.hh.

{ return fPDGEncoding; }

virtual G4VTrajectoryPoint* WLSTrajectory::GetPoint ( G4int  i ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 95 of file WLSTrajectory.hh.

Referenced by DrawTrajectory().

     { return (*fpPointsContainer)[i]; }

virtual int WLSTrajectory::GetPointEntries ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 93 of file WLSTrajectory.hh.

Referenced by CreateAttValues(), DrawTrajectory(), and MergeTrajectory().

     { return fpPointsContainer->size(); }

virtual G4int WLSTrajectory::GetTrackID ( ) const

inlinevirtual

Implements G4VTrajectory.

Definition at line 76 of file WLSTrajectory.hh.

{ return fTrackID; }

void WLSTrajectory::MergeTrajectory ( G4VTrajectory *  secondTrajectory )

virtual

Implements G4VTrajectory.

Definition at line 248 of file WLSTrajectory.cc.

References GetPointEntries(), and python.hepunit::second.

{
    if(!secondTrajectory) return;

    WLSTrajectory* second = (WLSTrajectory*)secondTrajectory;
    G4int ent = second->GetPointEntries();
    // initial point of the second trajectory should not be merged
    for(G4int i=1; i<ent; ++i) {
        fpPointsContainer->push_back((*(second->fpPointsContainer))[i]);
    }
    delete (*second->fpPointsContainer)[0];
    second->fpPointsContainer->clear();
}

void WLSTrajectory::operator delete ( void *  aTrajectory )

inline

Definition at line 138 of file WLSTrajectory.hh.

References WLSTrajectoryAllocator.

                                                            {
    WLSTrajectoryAllocator->FreeSingle((WLSTrajectory*)aTrajectory);
}

void * WLSTrajectory::operator new ( size_t  )

inline

Definition at line 132 of file WLSTrajectory.hh.

References WLSTrajectoryAllocator.

                                               {
    if(!WLSTrajectoryAllocator)
      WLSTrajectoryAllocator = new G4Allocator<WLSTrajectory>;
    return (void*) WLSTrajectoryAllocator->MallocSingle();
}

int WLSTrajectory::operator== ( const WLSTrajectory &  right ) const

inline

Definition at line 71 of file WLSTrajectory.hh.

     { return (this==&right); }

void WLSTrajectory::SetDrawTrajectory ( G4bool  b )

inline

Definition at line 101 of file WLSTrajectory.hh.

References test::b.

Referenced by WLSTrackingAction::PostUserTrackingAction().

{fDrawIt=b;}

void WLSTrajectory::SetForceDrawTrajectory ( G4bool  b )

inline

Definition at line 103 of file WLSTrajectory.hh.

References test::b.

Referenced by WLSEventAction::EndOfEventAction().

{fForceDraw=b;}

void WLSTrajectory::SetForceNoDrawTrajectory ( G4bool  b )

inline

Definition at line 104 of file WLSTrajectory.hh.

References test::b.

Referenced by WLSEventAction::EndOfEventAction().

{fForceNoDraw=b;}

void WLSTrajectory::ShowTrajectory ( std::ostream &  os = G4cout ) const

virtual

Reimplemented from G4VTrajectory.

Definition at line 132 of file WLSTrajectory.cc.

References G4VTrajectory::ShowTrajectory().

{
    // Invoke the default implementation in G4VTrajectory...
    G4VTrajectory::ShowTrajectory(os);
    // ... or override with your own code here.
}

void WLSTrajectory::WLS ( )

inline

Definition at line 102 of file WLSTrajectory.hh.

Referenced by WLSTrackingAction::PostUserTrackingAction().

{fWLS=true;}

---

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

• WLSTrajectory.hh
• WLSTrajectory.cc