// @(#)root/eve:$Id$
// Authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
/*************************************************************************
* Copyright (C) 1995-2007, Rene Brun and Fons Rademakers. *
* All rights reserved. *
* *
* For the licensing terms see $ROOTSYS/LICENSE. *
* For the list of contributors see $ROOTSYS/README/CREDITS. *
*************************************************************************/
#ifndef ROOT_TEveTrack
#define ROOT_TEveTrack
#include <vector>
#include "TEveVector.h"
#include "TEvePathMark.h"
#include "TEveVSDStructs.h"
#include "TEveElement.h"
#include "TEveLine.h"
#include "TPolyMarker3D.h"
#include "TMarker.h"
class TEveTrackPropagator;
class TEveTrackList;
class TEveMCTrack;
class TParticle;
class TEveTrack : public TEveLine
{
friend class TEveTrackPropagator;
friend class TEveTrackList;
friend class TEveTrackGL;
private:
TEveTrack& operator=(const TEveTrack&); // Not implemented
public:
typedef std::vector<TEvePathMarkD> vPathMark_t;
typedef vPathMark_t::iterator vPathMark_i;
typedef vPathMark_t::const_iterator vPathMark_ci;
// Deprecated -- to be removed.
enum EBreakProjectedTracks_e { kBPTDefault, kBPTAlways, kBPTNever };
protected:
TEveVectorD fV; // Starting vertex
TEveVectorD fP; // Starting momentum
TEveVectorD fPEnd; // Momentum at the last point of extrapolation
Double_t fBeta; // Relativistic beta factor
Double_t fDpDs; // Momentum loss over distance
Int_t fPdg; // PDG code
Int_t fCharge; // Charge in units of e0
Int_t fLabel; // Simulation label
Int_t fIndex; // Reconstruction index
Int_t fStatus; // Status-word, user-defined.
Bool_t fLockPoints; // Lock points that are currently in - do nothing in MakeTrack().
vPathMark_t fPathMarks; // TEveVector of known points along the track
Int_t fLastPMIdx; //!Last path-mark index tried in track-propagation.
TEveTrackPropagator *fPropagator; // Pointer to shared render-style
public:
TEveTrack();
TEveTrack(TParticle* t, Int_t label, TEveTrackPropagator* prop=0);
TEveTrack(TEveMCTrack* t, TEveTrackPropagator* prop=0);
TEveTrack(TEveRecTrack* t, TEveTrackPropagator* prop=0);
TEveTrack(TEveRecTrackD* t, TEveTrackPropagator* prop=0);
TEveTrack(const TEveTrack& t);
virtual ~TEveTrack();
virtual void ComputeBBox();
virtual void SetStdTitle();
virtual void SetTrackParams(const TEveTrack& t);
virtual void SetPathMarks (const TEveTrack& t);
virtual void MakeTrack(Bool_t recurse=kTRUE);
TEveTrackPropagator* GetPropagator() const { return fPropagator; }
Int_t GetLastPMIdx() const { return fLastPMIdx; }
void SetPropagator(TEveTrackPropagator* prop);
void SetAttLineAttMarker(TEveTrackList* tl);
const TEveVectorD& GetVertex() const { return fV; }
const TEveVectorD& GetMomentum() const { return fP; }
const TEveVectorD& GetEndMomentum() const { return fPEnd; }
Double_t GetDpDs() const { return fDpDs; }
void SetDpDs(Double_t dpds) { fDpDs = dpds; }
Int_t GetPdg() const { return fPdg; }
void SetPdg(Int_t pdg) { fPdg = pdg; }
Int_t GetCharge() const { return fCharge; }
void SetCharge(Int_t chg) { fCharge = chg; }
Int_t GetLabel() const { return fLabel; }
void SetLabel(Int_t lbl) { fLabel = lbl; }
Int_t GetIndex() const { return fIndex; }
void SetIndex(Int_t idx) { fIndex = idx; }
Int_t GetStatus() const { return fStatus; }
void SetStatus(Int_t idx) { fStatus = idx; }
void AddPathMark(const TEvePathMarkD& pm) { fPathMarks.push_back(pm); }
void AddPathMark(const TEvePathMark& pm) { fPathMarks.push_back(pm); }
void SortPathMarksByTime();
vPathMark_t& RefPathMarks() { return fPathMarks; }
const vPathMark_t& RefPathMarks() const { return fPathMarks; }
void PrintPathMarks(); // *MENU*
void SetLockPoints(Bool_t l) { fLockPoints = l; }
Bool_t GetLockPoints() const { return fLockPoints; }
//-------------------------------------------------------------------
virtual void SecSelected(TEveTrack*); // *SIGNAL*
virtual const TGPicture* GetListTreeIcon(Bool_t open=kFALSE);
virtual void CopyVizParams(const TEveElement* el);
virtual void WriteVizParams(std::ostream& out, const TString& var);
virtual TClass* ProjectedClass(const TEveProjection* p) const;
ClassDef(TEveTrack, 0); // Track with given vertex, momentum and optional referece-points (path-marks) along its path.
};
/******************************************************************************/
// TEveTrackList
/******************************************************************************/
class TEveTrackList : public TEveElementList,
public TAttMarker,
public TAttLine
{
friend class TEveTrackListEditor;
private:
TEveTrackList(const TEveTrackList&); // Not implemented
TEveTrackList& operator=(const TEveTrackList&); // Not implemented
protected:
TEveTrackPropagator* fPropagator; // Basic track rendering parameters, not enforced to elements.
Bool_t fRecurse; // Recurse when propagating marker/line/etc attributes to tracks.
Bool_t fRnrLine; // Render track as line.
Bool_t fRnrPoints; // Render track as points.
Double_t fMinPt; // Minimum track pTfor display selection.
Double_t fMaxPt; // Maximum track pTfor display selection.
Double_t fLimPt; // Highest track pT in the container.
Double_t fMinP; // Minimum track pfor display selection.
Double_t fMaxP; // Maximum track pfor display selection.
Double_t fLimP; // Highest track p in the container.
void FindMomentumLimits(TEveElement* el, Bool_t recurse=kTRUE);
Double_t RoundMomentumLimit(Double_t x);
void SanitizeMinMaxCuts();
public:
TEveTrackList(TEveTrackPropagator* prop=0);
TEveTrackList(const char* name, TEveTrackPropagator* prop=0);
virtual ~TEveTrackList();
void MakeTracks(Bool_t recurse=kTRUE);
void FindMomentumLimits(Bool_t recurse=kTRUE);
void SetPropagator(TEveTrackPropagator* prop);
TEveTrackPropagator* GetPropagator() { return fPropagator; }
Bool_t GetRecurse() const { return fRecurse; }
void SetRecurse(Bool_t x) { fRecurse = x; }
//--------------------------------
virtual void SetMainColor(Color_t c);
virtual void SetLineColor(Color_t c) { SetMainColor(c); }
virtual void SetLineColor(Color_t c, TEveElement* el);
virtual void SetLineWidth(Width_t w);
virtual void SetLineWidth(Width_t w, TEveElement* el);
virtual void SetLineStyle(Style_t s);
virtual void SetLineStyle(Style_t s, TEveElement* el);
virtual void SetMarkerColor(Color_t c);
virtual void SetMarkerColor(Color_t c, TEveElement* el);
virtual void SetMarkerSize(Size_t s);
virtual void SetMarkerSize(Size_t s, TEveElement* el);
virtual void SetMarkerStyle(Style_t s);
virtual void SetMarkerStyle(Style_t s, TEveElement* el);
void SetRnrLine(Bool_t rnr);
void SetRnrLine(Bool_t rnr, TEveElement* el);
Bool_t GetRnrLine() const { return fRnrLine; }
void SetRnrPoints(Bool_t r);
void SetRnrPoints(Bool_t r, TEveElement* el);
Bool_t GetRnrPoints() const { return fRnrPoints; }
void SelectByPt(Double_t min_pt, Double_t max_pt);
void SelectByPt(Double_t min_pt, Double_t max_pt, TEveElement* el);
void SelectByP (Double_t min_p, Double_t max_p);
void SelectByP (Double_t min_p, Double_t max_p, TEveElement* el);
Double_t GetMinPt() const { return fMinPt; }
Double_t GetMaxPt() const { return fMaxPt; }
Double_t GetLimPt() const { return fLimPt; }
Double_t GetMinP() const { return fMinP; }
Double_t GetMaxP() const { return fMaxP; }
Double_t GetLimP() const { return fLimP; }
//-------------------------------------------------------------------
TEveTrack* FindTrackByLabel(Int_t label); // *MENU*
TEveTrack* FindTrackByIndex(Int_t index); // *MENU*
virtual void CopyVizParams(const TEveElement* el);
virtual void WriteVizParams(std::ostream& out, const TString& var);
virtual TClass* ProjectedClass(const TEveProjection* p) const;
ClassDef(TEveTrackList, 0); // A list of tracks supporting change of common attributes and selection based on track parameters.
};
#endif