// @(#)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 #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 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