TVirtualGeoTrack.cxx

Go to the documentation of this file.

// @(#)root/geom:$Id$
// Author: Andrei Gheata  2003/04/10
 
/*************************************************************************
 * Copyright (C) 1995-2000, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#include "TVirtualGeoTrack.h"
 
#include "Rtypes.h"
#include "TGeoManager.h"
#include "TObject.h"
 
/** \class TVirtualGeoTrack
\ingroup Geometry_classes
 
Base class for user-defined tracks attached to a geometry.
Tracks are 3D objects made of points and they store a
pointer to a TParticle. The geometry manager holds a list
of all tracks that will be deleted on destruction of
gGeoManager.
*/
 
 
////////////////////////////////////////////////////////////////////////////////
/// Virtual tracks default constructor
 
TVirtualGeoTrack::TVirtualGeoTrack()
{
   fPDG = 0;
   fId = -1;
   fParent = nullptr;
   fParticle = nullptr;
   fTracks = nullptr;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Constructor providing ID for parent track (-1 for primaries), ID of this
/// track and related particle pointer.
 
TVirtualGeoTrack::TVirtualGeoTrack(Int_t id, Int_t pdgcode, TVirtualGeoTrack *parent, TObject *particle)
{
   fPDG = pdgcode;
   fId = id;
   fParent = parent;
   fParticle = particle;
   fTracks = nullptr;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Destructor.
 
TVirtualGeoTrack::~TVirtualGeoTrack()
{
   if (fTracks) {
      fTracks->Delete();
      delete fTracks;
   }
}
 
////////////////////////////////////////////////////////////////////////////////
/// Returns daughter id.
 
Int_t TVirtualGeoTrack::GetDaughterId(Int_t index) const
{
   TVirtualGeoTrack *daughter = GetDaughter(index);
   if (!daughter) {
      Error("GetDaughterId", "No daughter track with index %d", index);
      return -1;
   }
   return daughter->GetId();
}
 
////////////////////////////////////////////////////////////////////////////////
/// Recursively search through this track for a daughter
/// particle (at any depth) with the specified id
 
TVirtualGeoTrack *TVirtualGeoTrack::FindTrackWithId(Int_t id) const
{
   TVirtualGeoTrack *trk = nullptr;
   if (GetId() == id) {
      trk = (TVirtualGeoTrack *)this;
      return trk;
   }
   TVirtualGeoTrack *kid = nullptr;
   Int_t nd = GetNdaughters();
   for (Int_t i = 0; i < nd; i++)
      if (GetDaughterId(i) == id)
         return GetDaughter(i);
   for (Int_t i = 0; i < nd; i++) {
      kid = GetDaughter(i);
      if (kid != nullptr) {
         trk = kid->FindTrackWithId(id);
         if (trk != nullptr)
            break;
      }
   }
   return trk;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Get the PDG name.
 
const char *TVirtualGeoTrack::GetName() const
{
   return gGeoManager->GetPdgName(fPDG);
}
 
////////////////////////////////////////////////////////////////////////////////
/// True if track TOF range overlaps with time interval of TGeoManager
 
Bool_t TVirtualGeoTrack::IsInTimeRange() const
{
   Double_t tmin, tmax;
   Bool_t timecut = gGeoManager->GetTminTmax(tmin, tmax);
   if (!timecut)
      return kTRUE;
   const Double_t *point = GetFirstPoint();
   if (!point)
      return kFALSE;
   if (point[3] > tmax)
      return kFALSE;
   point = GetLastPoint();
   if (point[3] < tmin)
      return kFALSE;
   return kTRUE;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Set a default name for this track.
 
void TVirtualGeoTrack::SetName(const char *name)
{
   gGeoManager->SetPdgName(fPDG, name);
   if (!strcmp(name, "gamma")) {
      SetLineColor(kGreen);
      SetMarkerColor(kGreen);
      SetLineWidth(1);
      SetLineStyle(kDotted);
      return;
   }
   if (!strcmp(name, "pi+") || !strcmp(name, "proton") || !strcmp(name, "K+")) {
      SetLineColor(kRed);
      SetMarkerColor(kRed);
      SetLineWidth(2);
      return;
   }
   if (!strcmp(name, "pi-") || !strcmp(name, "K-")) {
      SetLineColor(30);
      SetMarkerColor(30);
      SetLineWidth(2);
      return;
   }
   if (!strcmp(name, "pi0") || !strcmp(name, "K0")) {
      SetLineColor(kCyan);
      SetMarkerColor(kCyan);
      SetLineWidth(2);
      return;
   }
   if (!strcmp(name, "neutron")) {
      SetLineColor(16);
      SetMarkerColor(16);
      SetLineWidth(1);
      SetLineStyle(kDotted);
      return;
   }
   if (!strcmp(name, "Alpha") || !strcmp(name, "Deuteron") || !strcmp(name, "Triton")) {
      SetLineColor(kMagenta);
      SetMarkerColor(kMagenta);
      SetLineWidth(3);
      return;
   }
   if (!strcmp(name, "e-") || !strcmp(name, "mu-")) {
      SetLineColor(kBlue);
      SetMarkerColor(kBlue);
      SetLineWidth(1);
      SetLineStyle(kDotted);
      return;
   }
   if (!strcmp(name, "e+") || !strcmp(name, "mu+")) {
      SetLineColor(kMagenta);
      SetMarkerColor(kMagenta);
      SetLineWidth(1);
      SetLineStyle(kDotted);
      return;
   }
}