Belle II Software development
EVEVisualization Class Reference

Produces visualisation for MCParticles, simhits, genfit::Tracks, geometry and other things. More...

#include <EVEVisualization.h>

Classes

struct  ElementGroup
 Group of TEveElements, remembers whether user wants it visible or not. More...
 
struct  MCTrack
 Hold MC tracks and associated visualisation objects. More...
 

Public Member Functions

 EVEVisualization ()
 Constructor.
 
 EVEVisualization (const EVEVisualization &)=delete
 disabled.
 
EVEVisualizationoperator= (const EVEVisualization &)=delete
 disabled assignment
 
 ~EVEVisualization ()
 Destructor.
 
void addTrack (const Belle2::Track *belle2Track)
 Add this genfit::Track to event data.
 
void addTrackCandidate (const std::string &collectionName, const RecoTrack &recoTrack)
 Add a RecoTrack, to evaluate track finding.
 
void addTrackCandidateImproved (const std::string &collectionName, const RecoTrack &recoTrack)
 Add a RecoTrack, but use stored genfit track representation to make visualisation objects.
 
void addCDCTriggerTrack (const std::string &collectionName, const CDCTriggerTrack &track)
 Add a CDCTriggerTrack.
 
template<class T>
void addSimHits (const StoreArray< T > &hits)
 Add all entries in the given 'hits' array (and the corresponding MCParticles) to the event scene.
 
void addSimHit (const CDCSimHit *hit, const MCParticle *particle)
 Add a CDCSimHit.
 
void addSimHit (const PXDSimHit *hit, const MCParticle *particle)
 Add a PXDSimHit.
 
void addSimHit (const SVDSimHit *hit, const MCParticle *particle)
 Add a SVDSimHit.
 
void addSimHit (const KLMSimHit *hit, const MCParticle *particle)
 Add a KLMSimHit.
 
void addSimHit (const ROOT::Math::XYZVector &v, const MCParticle *particle)
 Add simhit as a simple point.
 
MCTrackaddMCParticle (const MCParticle *particle)
 Return MCTrack for given particle, add it if it doesn't exist yet.
 
void addVertex (const genfit::GFRaveVertex *vertex)
 Add a vertex point and its covariance matrix.
 
void addECLCluster (const ECLCluster *cluster)
 Add a reconstructed cluster in the ECL.
 
void addKLMCluster (const KLMCluster *cluster)
 Add a reconstructed cluster in the KLM.
 
void addBKLMHit2d (const KLMHit2d *bklm2dhit)
 Add a reconstructed 2d hit in the BKLM.
 
void addEKLMHit2d (const KLMHit2d *eklm2dhit)
 Add a reconstructed 2d hit in the EKLM.
 
void addARICHHit (const ARICHHit *hit)
 Add reconstructed hit in ARICH.
 
void addROI (const ROIid *roi)
 Add a Region Of Interest, computed by the PXDDataReduction module.
 
template<class T>
void addUnassignedRecoHits (const StoreArray< T > &hits)
 After adding recohits for tracks/candidates, this function adds the remaining hits in a global collection.
 
void addCDCHit (const CDCHit *hit, bool showTriggerHits=false)
 show CDCHits directly.
 
void addCDCTriggerSegmentHit (const std::string &collectionName, const CDCTriggerSegmentHit *hit)
 show outline of track segments.
 
void addTOPDigits (const StoreArray< TOPDigit > &digits)
 Add TOPDigits (shown aggregated per module).
 
void showUserData (const DisplayData &displayData)
 Add user-defined data (labels, points, etc.)
 
void makeTracks ()
 Create visual representation of all tracks.
 
void clearEvent ()
 clear event data.
 
void setOptions (const std::string &opts)
 Set the display options.
 
void setErrScale (double errScale=1.)
 Set the scaling factor for the visualization of track hit errors.
 
void setAssignToPrimaries (bool on)
 If true, hits created by secondary particles (e.g.
 
void setHideSecondaries (bool on)
 If true, secondary MCParticles (and hits created by them) will not be shown.
 

Static Public Member Functions

static void addObject (const TObject *dataStoreObject, TEveElement *visualRepresentation)
 Generic function to keep track of which objects have which visual representation.
 

Private Types

enum  eFitterType {
  SimpleKalman ,
  RefKalman ,
  DafSimple ,
  DafRef ,
  Gbl
}
 Fitter type to be used for addTrack(). More...
 

Private Member Functions

template<class SomeVXDHit>
void addRecoHit (const SomeVXDHit *hit, TEveStraightLineSet *lines)
 adds given VXD hit to lines.
 
void addRecoHit (const SVDCluster *hit, TEveStraightLineSet *lines)
 specialisation for SVDCluster
 
void addRecoHit (const CDCHit *hit, TEveStraightLineSet *lines)
 specialisation for CDCHit.
 
void addToGroup (const std::string &name, TEveElement *elem)
 Add 'elem' to the element group 'name' (created if necessary).
 

Static Private Member Functions

static TEveBox * boxCreator (const ROOT::Math::XYZVector &o, ROOT::Math::XYZVector u, ROOT::Math::XYZVector v, float ud, float vd, float depth)
 Create a box around o, oriented along u and v with widths ud, vd and depth and return a pointer to the box object.
 
static void makeLines (TEveTrack *eveTrack, const genfit::StateOnPlane *prevState, const genfit::StateOnPlane *state, const genfit::AbsTrackRep *rep, TEvePathMark::EType_e markType, bool drawErrors, int markerPos=1)
 Create hit visualisation for the given options, and add them to 'eveTrack'.
 

Private Attributes

TEveCalo3D * m_calo3d
 Object for the energy bar visualisation.
 
double m_errorScale
 Rescale PXD/SVD errors with this factor to ensure visibility.
 
std::string m_options
 Option string for genfit::Track visualisation.
 
bool m_assignToPrimaries
 If true, hits created by secondary particles (e.g.
 
bool m_hideSecondaries {false}
 If true, secondary MCParticles (and hits created by them) will not be shown.
 
std::map< const MCParticle *, MCTrackm_mcparticleTracks
 map MCParticles to MCTrack (so hits can be added to the correct track).
 
std::map< std::string, ElementGroupm_groups
 name -> grouping element.
 
TEveTrackList * m_tracklist
 parent object for MC tracks.
 
TEveTrackPropagator * m_trackpropagator
 Track propagator for MCParticles.
 
TEveTrackPropagator * m_gftrackpropagator
 Track propagator for genfit::Tracks (different mainly because of drawing options)
 
TEveTrackPropagator * m_consttrackpropagator
 Track propagator for CDCTriggerTracks (uses constant B field)
 
TEveCaloDataVec * m_eclData
 ECL cluster data.
 
EveVisBFieldm_bfield
 The global magnetic field.
 
std::set< const TObject * > m_shownRecohits
 List of shown recohits (PXDCluster, SVDCluster, CDCHit).
 
TEveStraightLineSet * m_unassignedRecoHits = nullptr
 Unassigned recohits.
 
bool m_unassignedRecoHitsVisibility = true
 is m_unassignedRecoHits visible?
 
bool m_drawCardinalRep = true
 Draw cardinal representation in addTrack.
 
bool m_drawErrors = false
 Draw errors in addTrack.
 
bool m_drawRefTrack = false
 Draw reference track in addTrack.
 
bool m_drawForward = false
 draw forward in addTrack
 
bool m_drawBackward = false
 draw backward in addTrack
 

Static Private Attributes

static const int c_recoHitColor = getTColorID("Orange", 1)
 Color for reco hits.
 
static const int c_recoTrackColor = getTColorID("Sky Blue", 1)
 Color for TrackCandidates.
 
static const int c_trackColor = getTColorID("Sky Blue", 2)
 Color for tracks.
 
static const int c_trackMarkerColor = getTColorID("Chameleon", 3)
 Color for track markers.
 
static const int c_unassignedHitColor = getTColorID("Plum", 1)
 Color for unassigned (reco)hits.
 
static const int c_klmClusterColor = getTColorID("Chameleon", 1)
 Color for KLMCluster objects.
 
static constexpr double c_minPCut = 0.00
 don't show MCParticles with momentum below this cutoff.
 

Detailed Description

Produces visualisation for MCParticles, simhits, genfit::Tracks, geometry and other things.

Creates TEve objects from the given data, and adds them to the global or event scene.

See also
DisplayModule

Definition at line 63 of file EVEVisualization.h.

Member Enumeration Documentation

◆ eFitterType

enum eFitterType
private

Fitter type to be used for addTrack().

Definition at line 65 of file EVEVisualization.h.

65 {
66 SimpleKalman,
67 RefKalman,
68 DafSimple,
69 DafRef,
70 Gbl
71 };

Constructor & Destructor Documentation

◆ EVEVisualization()

EVEVisualization ( )
explicit

Constructor.

Definition at line 134 of file EVEVisualization.cc.

134 :
135 m_assignToPrimaries(false),
136 m_eclData(0),
137 m_bfield(new EveVisBField())
138{
139 setErrScale();
140
141 TGLLogicalShape::SetIgnoreSizeForCameraInterest(kTRUE); // Allows the visualization of the "small" error ellipsoid.
142
143 //create new containers
144 m_trackpropagator = new TEveTrackPropagator();
145 m_trackpropagator->IncDenyDestroy();
146 m_trackpropagator->SetMagFieldObj(m_bfield, false);
147 m_trackpropagator->SetFitDaughters(false); //most secondaries are no longer immediate daughters since we might discard those!
148 m_trackpropagator->SetMaxR(EveGeometry::getMaxR()); //don't draw tracks outside detector
149 //TODO is this actually needed?
150 m_trackpropagator->SetMaxStep(1.0); //make sure to reeval magnetic field often enough
151
152 m_tracklist = new TEveTrackList(m_trackpropagator);
153 m_tracklist->IncDenyDestroy();
154 m_tracklist->SetName("MCParticles");
155 m_tracklist->SelectByP(c_minPCut, FLT_MAX); //don't show too many particles by default...
156
157 m_gftrackpropagator = new TEveTrackPropagator();
158 m_gftrackpropagator->IncDenyDestroy();
159 m_gftrackpropagator->SetMagFieldObj(m_bfield, false);
160 m_gftrackpropagator->SetMaxOrbs(0.5); //stop after track markers
161
162 m_consttrackpropagator = new TEveTrackPropagator();
163 m_consttrackpropagator->IncDenyDestroy();
164 m_consttrackpropagator->SetMagField(0, 0, -1.5);
166
167 m_calo3d = new TEveCalo3D(NULL, "ECLClusters");
168 m_calo3d->SetBarrelRadius(125.80); //inner radius of ECL barrel
169 m_calo3d->SetForwardEndCapPos(196.5); //inner edge of forward endcap
170 m_calo3d->SetBackwardEndCapPos(-102.0); //inner edge of backward endcap
171 m_calo3d->SetMaxValAbs(2.1);
172 m_calo3d->SetRnrFrame(false, false); //don't show crystal grid
173 m_calo3d->IncDenyDestroy();
174
175 //Stop eve from deleting contents... (which might already be deleted)
176 gEve->GetSelection()->IncDenyDestroy();
177 gEve->GetHighlight()->IncDenyDestroy();
178
179 clearEvent();
180}
void clearEvent()
clear event data.
static constexpr double c_minPCut
don't show MCParticles with momentum below this cutoff.
EveVisBField * m_bfield
The global magnetic field.
void setErrScale(double errScale=1.)
Set the scaling factor for the visualization of track hit errors.
TEveCalo3D * m_calo3d
Object for the energy bar visualisation.
TEveTrackPropagator * m_trackpropagator
Track propagator for MCParticles.
TEveTrackList * m_tracklist
parent object for MC tracks.
TEveTrackPropagator * m_gftrackpropagator
Track propagator for genfit::Tracks (different mainly because of drawing options)
TEveTrackPropagator * m_consttrackpropagator
Track propagator for CDCTriggerTracks (uses constant B field)
bool m_assignToPrimaries
If true, hits created by secondary particles (e.g.
TEveCaloDataVec * m_eclData
ECL cluster data.
double getMaxR()
find a point that is inside the top node.

◆ ~EVEVisualization()

Destructor.

Definition at line 186 of file EVEVisualization.cc.

187{
188 if (!gEve)
189 return; //objects are probably already freed by Eve
190
191 //Eve objects
192 destroyEveElement(m_eclData);
193 destroyEveElement(m_unassignedRecoHits);
194 destroyEveElement(m_tracklist);
195 destroyEveElement(m_trackpropagator);
196 destroyEveElement(m_gftrackpropagator);
197 destroyEveElement(m_consttrackpropagator);
198 destroyEveElement(m_calo3d);
199 delete m_bfield;
200}
TEveStraightLineSet * m_unassignedRecoHits
Unassigned recohits.

Member Function Documentation

◆ addARICHHit()

void addARICHHit ( const ARICHHit * hit)

Add reconstructed hit in ARICH.

Definition at line 1773 of file EVEVisualization.cc.

1774{
1775 DBObjPtr<ARICHGeometryConfig> arichGeo;
1776
1777 int hitModule = hit->getModule();
1778 float fi = arichGeo->getDetectorPlane().getSlotPhi(hitModule);
1779
1780 ROOT::Math::XYZVector centerPos3D = hit->getPosition();
1781
1782 ROOT::Math::RotationZ rotZ(fi);
1783 ROOT::Math::XYZVector channelX(1, 0, 0);
1784 ROOT::Math::XYZVector channelY(0, 1, 0);
1785 channelX = rotZ * channelX;
1786 channelY = rotZ * channelY;
1787
1788 auto* arichbox = boxCreator(centerPos3D,
1789 arichGeo->getMasterVolume().momentumToGlobal(channelX),
1790 arichGeo->getMasterVolume().momentumToGlobal(channelY),
1791 0.49, 0.49, 0.05);
1792 arichbox->SetMainColor(kOrange + 10);
1793 arichbox->SetName((std::to_string(hitModule)).c_str());
1794
1795 addToGroup("ARICHHits", arichbox);
1796 addObject(hit, arichbox);
1797}
int getModule() const
Get module ID.
Definition ARICHHit.h:60
ROOT::Math::XYZVector getPosition() const
Get photon hit position.
Definition ARICHHit.h:54
static TEveBox * boxCreator(const ROOT::Math::XYZVector &o, ROOT::Math::XYZVector u, ROOT::Math::XYZVector v, float ud, float vd, float depth)
Create a box around o, oriented along u and v with widths ud, vd and depth and return a pointer to th...
void addToGroup(const std::string &name, TEveElement *elem)
Add 'elem' to the element group 'name' (created if necessary).
static void addObject(const TObject *dataStoreObject, TEveElement *visualRepresentation)
Generic function to keep track of which objects have which visual representation.

◆ addBKLMHit2d()

void addBKLMHit2d ( const KLMHit2d * bklm2dhit)

Add a reconstructed 2d hit in the BKLM.

Definition at line 1495 of file EVEVisualization.cc.

1496{
1497 const bklm::GeometryPar* m_GeoPar = Belle2::bklm::GeometryPar::instance();
1498 const bklm::Module* module = m_GeoPar->findModule(bklm2dhit->getSection(), bklm2dhit->getSector(), bklm2dhit->getLayer());
1499
1500 CLHEP::Hep3Vector global;
1501 //+++ global coordinates of the hit
1502 global[0] = bklm2dhit->getPositionX();
1503 global[1] = bklm2dhit->getPositionY();
1504 global[2] = bklm2dhit->getPositionZ();
1505
1506 //+++ local coordinates of the hit
1507 CLHEP::Hep3Vector local = module->globalToLocal(global);
1508 //double localU = local[1]; //phi
1509 //double localV = local[2]; //z
1510 int Nphistrip = bklm2dhit->getPhiStripMax() - bklm2dhit->getPhiStripMin() + 1;
1511 int Nztrip = bklm2dhit->getZStripMax() - bklm2dhit->getZStripMin() + 1;
1512 double du = module->getPhiStripWidth() * Nphistrip;
1513 double dv = module->getZStripWidth() * Nztrip;
1514
1515 //Let's do some simple thing
1516 CLHEP::Hep3Vector localU(local[0], local[1] + 1.0, local[2]);
1517 CLHEP::Hep3Vector localV(local[0], local[1], local[2] + 1.0);
1518
1519 CLHEP::Hep3Vector globalU = module->localToGlobal(localU);
1520 CLHEP::Hep3Vector globalV = module->localToGlobal(localV);
1521
1522 ROOT::Math::XYZVector o(global[0], global[1], global[2]);
1523 ROOT::Math::XYZVector u(globalU[0], globalU[1], globalU[2]);
1524 ROOT::Math::XYZVector v(globalV[0], globalV[1], globalV[2]);
1525
1526 //Lest's just assign the depth is 1.0 cm (thickness of a layer), better to update
1527 TEveBox* bklmbox = boxCreator(o, u - o, v - o, du, dv, 1.0);
1528
1529 bklmbox->SetMainColor(kGreen);
1530 //bklmbox->SetName((std::to_string(hitModule)).c_str());
1531 bklmbox->SetName("BKLMHit2d");
1532
1533 addToGroup("BKLM2dHits", bklmbox);
1534 addObject(bklm2dhit, bklmbox);
1535}
int getLayer() const
Get layer number.
Definition KLMHit2d.h:132
int getZStripMax() const
Get last strip number for z plane.
Definition KLMHit2d.h:202
int getSection() const
Get section number.
Definition KLMHit2d.h:96
float getPositionZ() const
Get hit global position z coordinate.
Definition KLMHit2d.h:306
int getSector() const
Get sector number.
Definition KLMHit2d.h:114
float getPositionX() const
Get hit global position x coordinate.
Definition KLMHit2d.h:288
int getPhiStripMin() const
Get strip number for phi plane.
Definition KLMHit2d.h:218
int getZStripMin() const
Get strip number for z plane.
Definition KLMHit2d.h:194
int getPhiStripMax() const
Get last strip number for phi plane.
Definition KLMHit2d.h:226
float getPositionY() const
Get hit global position y coordinate.
Definition KLMHit2d.h:297
const Module * findModule(int section, int sector, int layer) const
Get the pointer to the definition of a module.
static GeometryPar * instance(void)
Static method to get a reference to the singleton GeometryPar instance.

◆ addCDCHit()

void addCDCHit ( const CDCHit * hit,
bool showTriggerHits = false )

show CDCHits directly.

Definition at line 1614 of file EVEVisualization.cc.

1615{
1616 static CDC::CDCGeometryPar& cdcgeo = CDC::CDCGeometryPar::Instance();
1617 const B2Vector3D& wire_pos_f = cdcgeo.wireForwardPosition(WireID(hit->getID()));
1618 const B2Vector3D& wire_pos_b = cdcgeo.wireBackwardPosition(WireID(hit->getID()));
1619 static CDC::RealisticTDCCountTranslator tdcTranslator;
1620 TEveGeoShape* cov_shape = new TEveGeoShape("cov_shape");
1621 //TODO: leftrightflag not set! (same for other parameters, unsure which ones should be set)
1622 double driftLength = tdcTranslator.getDriftLength(hit->getTDCCount(), WireID(hit->getID()));
1623 double driftLengthRes = tdcTranslator.getDriftLengthResolution(driftLength, WireID(hit->getID()));
1624 driftLengthRes = std::max(driftLengthRes, 0.005);
1625 const double lengthOfWireSection = 3.0;
1626
1627 //z in wire direction, x,y orthogonal
1628 const B2Vector3D zaxis = wire_pos_b - wire_pos_f;
1629 const B2Vector3D xaxis = zaxis.Orthogonal();
1630 const B2Vector3D yaxis = xaxis.Cross(zaxis);
1631
1632 // move to z=0
1633 const B2Vector3D midPoint = wire_pos_f - zaxis * (wire_pos_f.Z() / zaxis.Z());
1634
1635 cov_shape->SetShape(new TGeoTube(std::max(0., (double)(driftLength - driftLengthRes)), driftLength + driftLengthRes,
1636 lengthOfWireSection));
1637 fixGeoShapeRefCount(cov_shape);
1638
1639 TGeoRotation det_rot("det_rot",
1640 xaxis.Theta() * 180 / TMath::Pi(), xaxis.Phi() * 180 / TMath::Pi(),
1641 yaxis.Theta() * 180 / TMath::Pi(), yaxis.Phi() * 180 / TMath::Pi(),
1642 zaxis.Theta() * 180 / TMath::Pi(), zaxis.Phi() * 180 / TMath::Pi()
1643 );
1644
1645 TGeoCombiTrans det_trans(midPoint.X(), midPoint.Y(), midPoint.Z(), &det_rot);
1646 cov_shape->SetTransMatrix(det_trans);
1647
1648 // get relation to trigger track segments
1649 bool isPartOfTS = false;
1650 const auto segments = hit->getRelationsFrom<CDCTriggerSegmentHit>();
1651 if (showTriggerHits && segments.size() > 0) {
1652 isPartOfTS = true;
1653 }
1654
1655 if (hit->getISuperLayer() % 2 == 0) {
1656 if (isPartOfTS)
1657 cov_shape->SetMainColor(kCyan + 3);
1658 else
1659 cov_shape->SetMainColor(kCyan);
1660 } else {
1661 if (isPartOfTS)
1662 cov_shape->SetMainColor(kPink + 6);
1663 else
1664 cov_shape->SetMainColor(kPink + 7);
1665 }
1666
1667 cov_shape->SetMainTransparency(50);
1668 cov_shape->SetName(ObjectInfo::getIdentifier(hit));
1669 cov_shape->SetTitle(ObjectInfo::getInfo(hit) + TString::Format("\nWire ID: %d\nADC: %d\nTDC: %d",
1670 hit->getID(), hit->getADCCount(), hit->getTDCCount()));
1671
1672 addToGroup("CDCHits", cov_shape);
1673 addObject(hit, cov_shape);
1674 if (isPartOfTS) {
1675 addToGroup("CDCTriggerSegmentHits", cov_shape);
1676 for (auto rel : segments.relations()) {
1677 addObject(rel.object, cov_shape);
1678 }
1679 }
1680}
DataType Phi() const
The azimuth angle.
Definition B2Vector3.h:151
DataType Z() const
access variable Z (= .at(2) without boundary check)
Definition B2Vector3.h:435
DataType Theta() const
The polar angle.
Definition B2Vector3.h:153
B2Vector3< DataType > Cross(const B2Vector3< DataType > &p) const
Cross product.
Definition B2Vector3.h:296
DataType X() const
access variable X (= .at(0) without boundary check)
Definition B2Vector3.h:431
B2Vector3< DataType > Orthogonal() const
Vector orthogonal to this one.
Definition B2Vector3.h:277
DataType Y() const
access variable Y (= .at(1) without boundary check)
Definition B2Vector3.h:433
short getTDCCount() const
Getter for TDC count.
Definition CDCHit.h:219
unsigned short getID() const
Getter for encoded wire number.
Definition CDCHit.h:193
unsigned short getADCCount() const
Getter for integrated charge.
Definition CDCHit.h:230
unsigned short getISuperLayer() const
Getter for iSuperLayer.
Definition CDCHit.h:184
const B2Vector3D wireForwardPosition(uint layerId, int cellId, EWirePosition set=c_Base) const
Returns the forward position of the input sense wire.
const B2Vector3D wireBackwardPosition(uint layerId, int cellId, EWirePosition set=c_Base) const
Returns the backward position of the input sense wire.
static CDCGeometryPar & Instance(const CDCGeometry *=nullptr)
Static method to get a reference to the CDCGeometryPar instance.
double getDriftLength(unsigned short tdcCount, const WireID &wireID=WireID(), double timeOfFlightEstimator=0, bool leftRight=false, double z=0, double alpha=0, double theta=static_cast< double >(TMath::Pi()/2.), unsigned short adcCount=0) override
Get Drift length.
double getDriftLengthResolution(double driftLength, const WireID &wireID=WireID(), bool leftRight=false, double z=0, double alpha=0, double=static_cast< double >(TMath::Pi()/2.)) override
Get position resolution^2 corresponding to the drift length from getDriftLength of this class.
RelationVector< FROM > getRelationsFrom(const std::string &name="", const std::string &namedRelation="") const
Get the relations that point from another store array to this object.
B2Vector3< double > B2Vector3D
typedef for common usage with double
Definition B2Vector3.h:516
TString getIdentifier(const TObject *obj)
Where is this object in the datastore?
TString getInfo(const TObject *obj)
Get object info HTML (e.g.
Definition ObjectInfo.cc:55

◆ addCDCTriggerSegmentHit()

void addCDCTriggerSegmentHit ( const std::string & collectionName,
const CDCTriggerSegmentHit * hit )

show outline of track segments.

Definition at line 1682 of file EVEVisualization.cc.

1683{
1684 static CDC::CDCGeometryPar& cdcgeo = CDC::CDCGeometryPar::Instance();
1685 TEveStraightLineSet* shape = new TEveStraightLineSet();
1686
1687 // get center wire
1688 unsigned iL = WireID(hit->getID()).getICLayer();
1689 if (hit->getPriorityPosition() < 3) iL -= 1;
1690 unsigned nWires = cdcgeo.nWiresInLayer(iL);
1691 unsigned iCenter = hit->getIWire();
1692 if (hit->getPriorityPosition() == 1) iCenter += 1;
1693
1694 // a track segment consists of 11 wires (15 in SL0) in a special configuration
1695 // -> get the shift with respect to the center wire (*) for all wires
1696 // SL 1-8:
1697 // _ _ _
1698 // |_|_|_|
1699 // |_|_|
1700 // |*|
1701 // |_|_|
1702 // |_|_|_|
1703 std::vector<int> layershift = { -2, -1, 0, 1, 2};
1704 std::vector<std::vector<float>> cellshift = {
1705 { -1, 0, 1},
1706 { -0.5, 0.5},
1707 { 0},
1708 { -0.5, 0.5},
1709 { -1, 0, 1}
1710 };
1711 // SL 0:
1712 // _ _ _ _ _
1713 // |_|_|_|_|_|
1714 // |_|_|_|_|
1715 // |_|_|_|
1716 // |_|_|
1717 // |*|
1718 if (hit->getISuperLayer() == 0) {
1719 layershift = { 0, 1, 2, 3, 4};
1720 cellshift = {
1721 { 0},
1722 { -0.5, 0.5},
1723 { -1, 0, 1},
1724 { -1.5, -0.5, 0.5, 1.5},
1725 { -2, -1, 0, 1, 2}
1726 };
1727 }
1728
1729 // draw all cells in segment
1730 for (unsigned il = 0; il < layershift.size(); ++il) {
1731 for (unsigned ic = 0; ic < cellshift[il].size(); ++ic) {
1732 ROOT::Math::XYZVector corners[2][2];
1733 for (unsigned ir = 0; ir < 2; ++ir) {
1734 double r = cdcgeo.fieldWireR(iL + layershift[il] - ir);
1735 double fz = cdcgeo.fieldWireFZ(iL + layershift[il] - ir);
1736 double bz = cdcgeo.fieldWireBZ(iL + layershift[il] - ir);
1737 for (unsigned iphi = 0; iphi < 2; ++iphi) {
1738 double phib = (iCenter + cellshift[il][ic] + iphi - 0.5) * 2 * M_PI / nWires;
1739 double phif = phib + cdcgeo.nShifts(iL + layershift[il]) * M_PI / nWires;
1740
1741 ROOT::Math::XYZVector pos_f = ROOT::Math::XYZVector(cos(phif) * r, sin(phif) * r, fz);
1742 ROOT::Math::XYZVector pos_b = ROOT::Math::XYZVector(cos(phib) * r, sin(phib) * r, bz);
1743 ROOT::Math::XYZVector zaxis = pos_b - pos_f;
1744 corners[ir][iphi] = pos_f - zaxis * (pos_f.Z() / zaxis.Z());
1745 }
1746 }
1747
1748 shape->AddLine(corners[0][0].X(), corners[0][0].Y(), 0,
1749 corners[0][1].X(), corners[0][1].Y(), 0);
1750 shape->AddLine(corners[0][1].X(), corners[0][1].Y(), 0,
1751 corners[1][1].X(), corners[1][1].Y(), 0);
1752 shape->AddLine(corners[1][1].X(), corners[1][1].Y(), 0,
1753 corners[1][0].X(), corners[1][0].Y(), 0);
1754 shape->AddLine(corners[1][0].X(), corners[1][0].Y(), 0,
1755 corners[0][0].X(), corners[0][0].Y(), 0);
1756 }
1757 }
1758
1759 if (hit->getISuperLayer() % 2 == 0) {
1760 shape->SetMainColor(kCyan + 3);
1761 } else {
1762 shape->SetMainColor(kPink + 6);
1763 }
1764
1765 shape->SetName(ObjectInfo::getIdentifier(hit));
1766 shape->SetTitle(ObjectInfo::getTitle(hit) +
1767 TString::Format("\nPriority: %d\nLeft/Right: %d",
1768 hit->getPriorityPosition(), hit->getLeftRight()));
1769 addToGroup(collectionName, shape);
1770 addObject(hit, shape);
1771}
unsigned short getPriorityPosition() const
get position of the priority cell within the track segment (0: no hit, 3: 1st priority,...
unsigned short getIWire() const
get wire number of priority wire within layer.
unsigned short getID() const
get the encoded wire number of the priority wire.
unsigned short getISuperLayer() const
get super layer number.
unsigned short getLeftRight() const
get position of the priority cell relative to the track (0: no hit, 1: right, 2: left,...
int nShifts(int layerId) const
Returns number shift.
double fieldWireR(int layerId) const
Returns radius of field wire in each layer.
unsigned nWiresInLayer(int layerId) const
Returns wire numbers in a layer.
double fieldWireBZ(int layerId) const
Returns backward z position of field wire in each layer.
double fieldWireFZ(int layerId) const
Returns forward z position of field wire in each layer.
TString getTitle(const TObject *obj)
Get plain text for TEve object titles (shown on mouse-over).
Definition ObjectInfo.cc:68

◆ addCDCTriggerTrack()

void addCDCTriggerTrack ( const std::string & collectionName,
const CDCTriggerTrack & track )

Add a CDCTriggerTrack.

Definition at line 346 of file EVEVisualization.cc.

348{
349 const TString label = ObjectInfo::getIdentifier(&trgTrack);
350
351 B2Vector3D track_pos = B2Vector3D(0, 0, trgTrack.getZ0());
352 B2Vector3D track_mom = (trgTrack.getChargeSign() == 0) ?
353 trgTrack.getDirection() * 1000 :
354 trgTrack.getMomentum(1.5);
355
356 TEveRecTrack rectrack;
357 rectrack.fP.Set(track_mom);
358 rectrack.fV.Set(track_pos);
359
360 TEveTrack* track_lines = new TEveTrack(&rectrack, m_consttrackpropagator);
361 track_lines->SetName(label); //popup label set at end of function
362 track_lines->SetPropagator(m_consttrackpropagator);
363 track_lines->SetLineColor(kOrange + 2);
364 track_lines->SetLineWidth(1);
365 track_lines->SetTitle(ObjectInfo::getTitle(&trgTrack) +
366 TString::Format("\ncharge: %d, phi: %.2fdeg, pt: %.2fGeV, theta: %.2fdeg, z: %.2fcm",
367 trgTrack.getChargeSign(),
368 trgTrack.getPhi0() * 180 / M_PI,
369 trgTrack.getTransverseMomentum(1.5),
370 trgTrack.getDirection().Theta() * 180 / M_PI,
371 trgTrack.getZ0()));
372
373
374 track_lines->SetCharge(trgTrack.getChargeSign());
375
376 // show 2D tracks with dashed lines
377 if (trgTrack.getZ0() == 0 && trgTrack.getCotTheta() == 0)
378 track_lines->SetLineStyle(2);
379
380 addToGroup(collectionName, track_lines);
381 addObject(&trgTrack, track_lines);
382}

◆ addECLCluster()

void addECLCluster ( const ECLCluster * cluster)

Add a reconstructed cluster in the ECL.

Definition at line 1404 of file EVEVisualization.cc.

1405{
1406
1407 // only display c_nPhotons hypothesis clusters
1408 if (cluster->hasHypothesis(ECLCluster::EHypothesisBit::c_nPhotons)) {
1409
1410 const float phi = cluster->getPhi();
1411 float dPhi = cluster->getUncertaintyPhi();
1412 float dTheta = cluster->getUncertaintyTheta();
1413 if (dPhi >= M_PI / 4 or dTheta >= M_PI / 4 or cluster->getUncertaintyEnergy() == 1.0) {
1414 B2WARNING("Found ECL cluster with broken errors (unit matrix or too large). Using 0.05 as error in phi/theta. The 3x3 error matrix previously was:");
1415 cluster->getCovarianceMatrix3x3().Print();
1416 dPhi = dTheta = 0.05;
1417 }
1418
1419 if (!std::isfinite(dPhi) or !std::isfinite(dTheta)) {
1420 B2ERROR("ECLCluster phi or theta error is NaN or infinite, skipping cluster!");
1421 return;
1422 }
1423
1424 //convert theta +- dTheta into eta +- dEta
1425 ROOT::Math::XYZVector thetaLow;
1426 VectorUtil::setPtThetaPhi(thetaLow, 1.0, cluster->getTheta() - dTheta, phi);
1427 ROOT::Math::XYZVector thetaHigh;
1428 VectorUtil::setPtThetaPhi(thetaHigh, 1.0, cluster->getTheta() + dTheta, phi);
1429 float etaLow = thetaLow.Eta();
1430 float etaHigh = thetaHigh.Eta();
1431 if (etaLow > etaHigh) {
1432 std::swap(etaLow, etaHigh);
1433 }
1434
1435 int id = m_eclData->AddTower(etaLow, etaHigh, phi - dPhi, phi + dPhi);
1436 m_eclData->FillSlice(0, cluster->getEnergy(ECLCluster::EHypothesisBit::c_nPhotons));
1438 }
1439}
@ c_nPhotons
CR is split into n photons (N1)
Definition ECLCluster.h:41
void addCluster(const TObject *dataStoreObject, TEveCaloData *caloData, int towerID)
Selection inside TEveCalo* is complicated, use this to keep track of ECL clusters.
static VisualRepMap * getInstance()
get instance pointer.

◆ addEKLMHit2d()

void addEKLMHit2d ( const KLMHit2d * eklm2dhit)

Add a reconstructed 2d hit in the EKLM.

Definition at line 1537 of file EVEVisualization.cc.

1538{
1539 const double du = 2.0;
1540 const double dv = 2.0;
1541 ROOT::Math::XYZVector hitPosition = eklm2dhit->getPosition();
1542 ROOT::Math::XYZVector o(hitPosition.X(), hitPosition.Y(), hitPosition.Z());
1543 ROOT::Math::XYZVector u(1.0, 0.0, 0.0);
1544 ROOT::Math::XYZVector v(0.0, 1.0, 0.0);
1545 TEveBox* eklmbox = boxCreator(o, u, v, du, dv, 4.0);
1546 eklmbox->SetMainColor(kGreen);
1547 eklmbox->SetName("EKLMHit2d");
1548 addToGroup("EKLM2dHits", eklmbox);
1549 addObject(eklm2dhit, eklmbox);
1550}
ROOT::Math::XYZVector getPosition() const
Get hit global position.
Definition KLMHit2d.h:315

◆ addKLMCluster()

void addKLMCluster ( const KLMCluster * cluster)

Add a reconstructed cluster in the KLM.

Definition at line 1441 of file EVEVisualization.cc.

1442{
1443 const double layerThicknessCm = 3.16; //TDR, Fig 10.2
1444 const double layerDistanceCm = 9.1 - layerThicknessCm;
1445
1446 // Pposition of first RPC plane.
1447 ROOT::Math::XYZVector position = cluster->getClusterPosition();
1448 ROOT::Math::XYZVector startPos(position.X(), position.Y(), position.Z());
1449 ROOT::Math::XYZVector dir; //direction of cluster stack, Mag() == distance between planes
1450 ROOT::Math::XYZVector a, b; //defines RPC plane
1451 bool isBarrel = (startPos.Z() > -175.0 and startPos.Z() < 270.0);
1452 if (isBarrel) {
1453 //barrel
1454 b = ROOT::Math::XYZVector(0, 0, 1);
1455 a = startPos.Cross(b).Unit();
1456 double c = M_PI / 4.0;
1457 double offset = c / 2.0 + M_PI;
1458 VectorUtil::setPhi(a, int((a.Phi() + offset) / (c))*c - M_PI);
1459 ROOT::Math::XYZVector perp = b.Cross(a);
1460
1461 const double barrelRadiusCm = 204.0;
1462 VectorUtil::setMag(startPos, barrelRadiusCm / perp.Dot(startPos.Unit()));
1463
1464 dir = startPos.Unit();
1465 VectorUtil::setMag(dir, (layerDistanceCm + layerThicknessCm) / perp.Dot(dir));
1466 } else {
1467 //endcap
1468 b = ROOT::Math::XYZVector(startPos.X(), startPos.Y(), 0).Unit();
1469 a = startPos.Cross(b).Unit();
1470 double endcapStartZ = 284;
1471 if (startPos.Z() < 0)
1472 endcapStartZ = -189.5;
1473
1474 double scaleFac = endcapStartZ / startPos.Z();
1475 VectorUtil::setMag(startPos, startPos.R() * scaleFac);
1476
1477 dir = startPos.Unit();
1478 VectorUtil::setMag(dir, (layerDistanceCm + layerThicknessCm) / fabs(dir.Z()));
1479 }
1480
1481 for (int i = 0; i < cluster->getLayers(); i++) {
1482 ROOT::Math::XYZVector layerPos = startPos;
1483 layerPos += (cluster->getInnermostLayer() + i) * dir;
1484 auto* layer = boxCreator(layerPos, a, b, 20.0, 20.0, layerThicknessCm / 2);
1485 layer->SetMainColor(c_klmClusterColor);
1486 layer->SetMainTransparency(70);
1487 layer->SetName(ObjectInfo::getIdentifier(cluster));
1488 layer->SetTitle(ObjectInfo::getTitle(cluster));
1489
1490 addToGroup(std::string("KLMClusters/") + ObjectInfo::getIdentifier(cluster).Data(), layer);
1491 addObject(cluster, layer);
1492 }
1493}
static const int c_klmClusterColor
Color for KLMCluster objects.
bool isBarrel(int cellId)
Check whether the crystal is in barrel ECL.

◆ addMCParticle()

EVEVisualization::MCTrack * addMCParticle ( const MCParticle * particle)

Return MCTrack for given particle, add it if it doesn't exist yet.

If particle is NULL, a dummy MCTrack (with track=0) is created which can accept otherwise unassigned hits. Returns NULL if this particle and its hits shouldn't be shown.

Definition at line 1103 of file EVEVisualization.cc.

1104{
1105 if (!particle) {
1106 if (!m_mcparticleTracks[nullptr].simhits) {
1107 const TString pointsTitle("Unassigned SimHits");
1108 m_mcparticleTracks[nullptr].simhits = new TEvePointSet(pointsTitle);
1109 m_mcparticleTracks[nullptr].simhits->SetTitle(pointsTitle);
1110 m_mcparticleTracks[nullptr].simhits->SetMarkerStyle(6);
1111 m_mcparticleTracks[nullptr].simhits->SetMainColor(c_unassignedHitColor);
1112 //m_mcparticleTracks[nullptr].simhits->SetMainTransparency(50);
1113 m_mcparticleTracks[nullptr].track = NULL;
1114 }
1115 return &m_mcparticleTracks[nullptr];
1116 }
1117
1118 if (m_hideSecondaries and !particle->hasStatus(MCParticle::c_PrimaryParticle)) {
1119 return NULL;
1120 }
1121 if (m_assignToPrimaries) {
1122 while (!particle->hasStatus(MCParticle::c_PrimaryParticle) and particle->getMother())
1123 particle = particle->getMother();
1124 }
1125
1126 if (!m_mcparticleTracks[particle].track) {
1127 const ROOT::Math::XYZVector& p = particle->getMomentum();
1128 const ROOT::Math::XYZVector& vertex = particle->getProductionVertex();
1129 const int pdg = particle->getPDG();
1130 TParticle tparticle(pdg, particle->getStatus(),
1131 (particle->getMother() ? particle->getMother()->getIndex() : 0), 0, particle->getFirstDaughter(), particle->getLastDaughter(),
1132 p.X(), p.Y(), p.Z(), particle->getEnergy(),
1133 vertex.X(), vertex.Y(), vertex.Z(), particle->getProductionTime());
1134 TEveMCTrack mctrack;
1135 mctrack = tparticle;
1136 mctrack.fTDecay = particle->getDecayTime();
1137 mctrack.fVDecay.Set(B2Vector3D(particle->getDecayVertex()));
1138 mctrack.fDecayed = !std::isinf(mctrack.fTDecay);
1139 mctrack.fIndex = particle->getIndex();
1140 m_mcparticleTracks[particle].track = new TEveTrack(&mctrack, m_trackpropagator);
1141
1142 //Check if there is a trajectory stored for this particle
1143 const auto mcTrajectories = particle->getRelationsTo<MCParticleTrajectory>();
1144 bool hasTrajectory(false);
1145 for (auto rel : mcTrajectories.relations()) {
1146 //Trajectories with negative weight are from secondary daughters which
1147 //were ignored so we don't use them.
1148 if (rel.weight <= 0) continue;
1149 //Found one, let's add tose point as reference points to the TEveTrack.
1150 //This will force the track propagation to visit all points in order but
1151 //provide smooth helix interpolation between the points
1152 const MCParticleTrajectory& trajectory = dynamic_cast<const MCParticleTrajectory&>(*rel.object);
1153 for (const MCTrajectoryPoint& pt : trajectory) {
1154 m_mcparticleTracks[particle].track->AddPathMark(
1155 TEvePathMark(
1156 //Add the last trajectory point as decay point to prevent TEve to
1157 //propagate beyond the end of the track. So lets compare the address
1158 //to the address of last point and choose the pathmark accordingly
1159 (&pt == &trajectory.back()) ? TEvePathMark::kDecay : TEvePathMark::kReference,
1160 TEveVector(pt.x, pt.y, pt.z),
1161 TEveVector(pt.px, pt.py, pt.pz)
1162 ));
1163 }
1164 //"There can only be One" -> found a trajectory, stop the loop
1165 hasTrajectory = true;
1166 break;
1167 }
1168
1169 //If we have the full trajectory there is no need to add additional path marks
1170 if (!hasTrajectory) {
1171 //add daughter vertices - improves track rendering as lost momentum is taken into account
1172 for (int iDaughter = particle->getFirstDaughter(); iDaughter <= particle->getLastDaughter(); iDaughter++) {
1173 if (iDaughter == 0)
1174 continue; //no actual daughter
1175
1176 const MCParticle* daughter = StoreArray<MCParticle>()[iDaughter - 1];
1177
1178 TEvePathMarkD refMark(TEvePathMarkD::kDaughter);
1179 refMark.fV.Set(B2Vector3D(daughter->getProductionVertex()));
1180 refMark.fP.Set(B2Vector3D(daughter->getMomentum()));
1181 refMark.fTime = daughter->getProductionTime();
1182 m_mcparticleTracks[particle].track->AddPathMark(refMark);
1183 }
1184
1185 //neutrals and very short-lived particles should stop somewhere
1186 //(can result in wrong shapes for particles stopped in the detector, so not used there)
1187 if ((TMath::Nint(particle->getCharge()) == 0 or !particle->hasStatus(MCParticle::c_StoppedInDetector))
1188 and mctrack.fDecayed) {
1189 TEvePathMarkD decayMark(TEvePathMarkD::kDecay);
1190 decayMark.fV.Set(B2Vector3D(particle->getDecayVertex()));
1191 m_mcparticleTracks[particle].track->AddPathMark(decayMark);
1192 }
1193 }
1194 TString particle_name(mctrack.GetName());
1195
1196 //set track title (for popup)
1197 const MCParticle* mom = particle->getMother();
1198 if (mom) {
1199 m_mcparticleTracks[particle].parentParticle = mom;
1200 addMCParticle(mom);
1201 }
1202
1203 TString title = ObjectInfo::getTitle(particle);
1204 if (!hasTrajectory) {
1205 //Hijack the mother label to show that the track position is only
1206 //extrapolated, not known from simulation
1207 title += "\n(track estimated from initial momentum)";
1208 //Also, show those tracks with dashed lines
1209 m_mcparticleTracks[particle].track->SetLineStyle(2);
1210 }
1211
1212 m_mcparticleTracks[particle].track->SetTitle(title);
1213
1214 //add some color (avoid black & white)
1215 switch (abs(pdg)) {
1216 case 11:
1217 m_mcparticleTracks[particle].track->SetLineColor(kAzure);
1218 break;
1219 case 13:
1220 m_mcparticleTracks[particle].track->SetLineColor(kCyan + 1);
1221 break;
1222 case 22:
1223 m_mcparticleTracks[particle].track->SetLineColor(kSpring);
1224 break;
1225 case 211:
1226 m_mcparticleTracks[particle].track->SetLineColor(kGray + 1);
1227 break;
1228 case 321:
1229 m_mcparticleTracks[particle].track->SetLineColor(kRed + 1);
1230 break;
1231 case 2212:
1232 m_mcparticleTracks[particle].track->SetLineColor(kOrange - 2);
1233 break;
1234 default:
1235 m_mcparticleTracks[particle].track->SetLineColor(kMagenta);
1236 }
1237
1238 //create point set for hits
1239 const TString pointsTitle = "SimHits for " + ObjectInfo::getIdentifier(particle) + " - " + particle_name;
1240 m_mcparticleTracks[particle].simhits = new TEvePointSet(pointsTitle);
1241 m_mcparticleTracks[particle].simhits->SetTitle(pointsTitle);
1242 m_mcparticleTracks[particle].simhits->SetMarkerStyle(6);
1243 m_mcparticleTracks[particle].simhits->SetMainColor(m_mcparticleTracks[particle].track->GetLineColor());
1244 //m_mcparticleTracks[particle].simhits->SetMainTransparency(50);
1245 addObject(particle, m_mcparticleTracks[particle].track);
1246 }
1247 return &m_mcparticleTracks[particle];
1248}
std::map< const MCParticle *, MCTrack > m_mcparticleTracks
map MCParticles to MCTrack (so hits can be added to the correct track).
static const int c_unassignedHitColor
Color for unassigned (reco)hits.
bool m_hideSecondaries
If true, secondary MCParticles (and hits created by them) will not be shown.
MCTrack * addMCParticle(const MCParticle *particle)
Return MCTrack for given particle, add it if it doesn't exist yet.
@ c_PrimaryParticle
bit 0: Particle is primary particle.
Definition MCParticle.h:47
@ c_StoppedInDetector
bit 3: Particle was stopped in the detector (the simulation volume).
Definition MCParticle.h:53

◆ addObject()

void addObject ( const TObject * dataStoreObject,
TEveElement * visualRepresentation )
static

Generic function to keep track of which objects have which visual representation.

Should be called by functions adding TEveElements to the event scene (Hits are currently excluded).

Definition at line 1908 of file EVEVisualization.cc.

1909{
1910 VisualRepMap::getInstance()->add(dataStoreObject, visualRepresentation);
1911}
void add(const TObject *dataStoreObject, TEveElement *visualRepresentation)
Generic function to keep track of which objects have which visual representation.

◆ addRecoHit() [1/3]

void addRecoHit ( const CDCHit * hit,
TEveStraightLineSet * lines )
private

specialisation for CDCHit.

Definition at line 1603 of file EVEVisualization.cc.

1604{
1605 static CDC::CDCGeometryPar& cdcgeo = CDC::CDCGeometryPar::Instance();
1606 const ROOT::Math::XYZVector& wire_pos_f = cdcgeo.wireForwardPosition(WireID(hit->getID()));
1607 const ROOT::Math::XYZVector& wire_pos_b = cdcgeo.wireBackwardPosition(WireID(hit->getID()));
1608
1609 lines->AddLine(wire_pos_f.X(), wire_pos_f.Y(), wire_pos_f.Z(), wire_pos_b.X(), wire_pos_b.Y(), wire_pos_b.Z());
1610 m_shownRecohits.insert(hit);
1611
1612}
std::set< const TObject * > m_shownRecohits
List of shown recohits (PXDCluster, SVDCluster, CDCHit).

◆ addRecoHit() [2/3]

template<class SomeVXDHit>
void addRecoHit ( const SomeVXDHit * hit,
TEveStraightLineSet * lines )
inlineprivate

adds given VXD hit to lines.

Definition at line 288 of file EVEVisualization.h.

289 {
290 static VXD::GeoCache& geo = VXD::GeoCache::getInstance();
291
292 const ROOT::Math::XYZVector local_pos(hit->getU(), hit->getV(), 0.0); //z-component is height over the center of the detector plane
293 const VXD::SensorInfoBase& sensor = geo.getSensorInfo(hit->getSensorID());
294 const ROOT::Math::XYZVector global_pos = sensor.pointToGlobal(local_pos);
295 lines->AddMarker(global_pos.X(), global_pos.Y(), global_pos.Z());
296
297 m_shownRecohits.insert(hit);
298 }

◆ addRecoHit() [3/3]

void addRecoHit ( const SVDCluster * hit,
TEveStraightLineSet * lines )
private

specialisation for SVDCluster

Definition at line 1583 of file EVEVisualization.cc.

1584{
1585 static VXD::GeoCache& geo = VXD::GeoCache::getInstance();
1586 const VXD::SensorInfoBase& sensor = geo.getSensorInfo(hit->getSensorID());
1587
1588 ROOT::Math::XYZVector a, b;
1589 if (hit->isUCluster()) {
1590 const float u = hit->getPosition();
1591 a = sensor.pointToGlobal(ROOT::Math::XYZVector(sensor.getBackwardWidth() / sensor.getWidth(0) * u, -0.5 * sensor.getLength(), 0.0));
1592 b = sensor.pointToGlobal(ROOT::Math::XYZVector(sensor.getForwardWidth() / sensor.getWidth(0) * u, +0.5 * sensor.getLength(), 0.0));
1593 } else {
1594 const float v = hit->getPosition();
1595 a = sensor.pointToGlobal(ROOT::Math::XYZVector(-0.5 * sensor.getWidth(v), v, 0.0));
1596 b = sensor.pointToGlobal(ROOT::Math::XYZVector(+0.5 * sensor.getWidth(v), v, 0.0));
1597 }
1598
1599 lines->AddLine(a.X(), a.Y(), a.Z(), b.X(), b.Y(), b.Z());
1600 m_shownRecohits.insert(hit);
1601}
VxdID getSensorID() const
Get the sensor ID.
Definition SVDCluster.h:102
bool isUCluster() const
Get the direction of strips.
Definition SVDCluster.h:110
float getPosition(double v=0) const
Get the coordinate of reconstructed hit.
Definition SVDCluster.h:117
const SensorInfoBase & getSensorInfo(Belle2::VxdID id) const
Return a reference to the SensorInfo of a given SensorID.
Definition GeoCache.cc:67
static GeoCache & getInstance()
Return a reference to the singleton instance.
Definition GeoCache.cc:214

◆ addROI()

void addROI ( const ROIid * roi)

Add a Region Of Interest, computed by the PXDDataReduction module.

Definition at line 1552 of file EVEVisualization.cc.

1553{
1554 const VXD::GeoCache& aGeometry = VXD::GeoCache::getInstance();
1555
1556 VxdID sensorID = roi->getSensorID();
1557 const VXD::SensorInfoBase& aSensorInfo = aGeometry.getSensorInfo(sensorID);
1558
1559 double minU = aSensorInfo.getUCellPosition(roi->getMinUid(), roi->getMinVid());
1560 double minV = aSensorInfo.getVCellPosition(roi->getMinVid());
1561 double maxU = aSensorInfo.getUCellPosition(roi->getMaxUid(), roi->getMaxVid());
1562 double maxV = aSensorInfo.getVCellPosition(roi->getMaxVid());
1563
1564
1565 ROOT::Math::XYZVector localA(minU, minV, 0);
1566 ROOT::Math::XYZVector localB(minU, maxV, 0);
1567 ROOT::Math::XYZVector localC(maxU, minV, 0);
1568
1569 ROOT::Math::XYZVector globalA = aSensorInfo.pointToGlobal(localA);
1570 ROOT::Math::XYZVector globalB = aSensorInfo.pointToGlobal(localB);
1571 ROOT::Math::XYZVector globalC = aSensorInfo.pointToGlobal(localC);
1572
1573 TEveBox* ROIbox = boxCreator(globalB + globalC * 0.5, globalB - globalA, globalC - globalA, 1, 1, 0.01);
1574
1575 ROIbox->SetName(ObjectInfo::getIdentifier(roi));
1576 ROIbox->SetMainColor(kSpring - 9);
1577 ROIbox->SetMainTransparency(50);
1578
1579 addToGroup("ROIs", ROIbox);
1580 addObject(roi, ROIbox);
1581}
double getVCellPosition(int vID) const
Return the position of a specific strip/pixel in v direction.
double getUCellPosition(int uID, int vID=-1) const
Return the position of a specific strip/pixel in u direction.
ROOT::Math::XYZVector pointToGlobal(const ROOT::Math::XYZVector &local, bool reco=false) const
Convert a point from local to global coordinates.

◆ addSimHit() [1/5]

void addSimHit ( const CDCSimHit * hit,
const MCParticle * particle )

Add a CDCSimHit.

Definition at line 1074 of file EVEVisualization.cc.

1075{
1076 addSimHit(ROOT::Math::XYZVector(hit->getPosWire()), particle);
1077}
B2Vector3D getPosWire() const
The method to get position on wire.
Definition CDCSimHit.h:198
void addSimHit(const CDCSimHit *hit, const MCParticle *particle)
Add a CDCSimHit.

◆ addSimHit() [2/5]

void addSimHit ( const KLMSimHit * hit,
const MCParticle * particle )

Add a KLMSimHit.

Definition at line 1090 of file EVEVisualization.cc.

1091{
1092 const ROOT::Math::XYZVector& global_pos = hit->getPosition();
1093 addSimHit(global_pos, particle);
1094}
ROOT::Math::XYZVector getPosition() const
Get hit global position.
Definition KLMSimHit.h:414

◆ addSimHit() [3/5]

void addSimHit ( const PXDSimHit * hit,
const MCParticle * particle )

Add a PXDSimHit.

Definition at line 1078 of file EVEVisualization.cc.

1079{
1080 static VXD::GeoCache& geo = VXD::GeoCache::getInstance();
1081 const ROOT::Math::XYZVector& global_pos = geo.getSensorInfo(hit->getSensorID()).pointToGlobal(hit->getPosIn());
1082 addSimHit(global_pos, particle);
1083}
ROOT::Math::XYZVector getPosIn() const
Return the start point of the electron deposition in local coordinates.
Definition VXDSimHit.h:68
VxdID getSensorID() const
Return the sensorID of the sensor the electron was deposited in.
Definition VXDSimHit.h:64

◆ addSimHit() [4/5]

void addSimHit ( const ROOT::Math::XYZVector & v,
const MCParticle * particle )

Add simhit as a simple point.

Definition at line 1095 of file EVEVisualization.cc.

1096{
1097 MCTrack* track = addMCParticle(particle);
1098 if (!track)
1099 return; //hide hits from this particle
1100 track->simhits->SetNextPoint(v.X(), v.Y(), v.Z());
1101}
Hold MC tracks and associated visualisation objects.

◆ addSimHit() [5/5]

void addSimHit ( const SVDSimHit * hit,
const MCParticle * particle )

Add a SVDSimHit.

Definition at line 1084 of file EVEVisualization.cc.

1085{
1086 static VXD::GeoCache& geo = VXD::GeoCache::getInstance();
1087 const ROOT::Math::XYZVector& global_pos = geo.getSensorInfo(hit->getSensorID()).pointToGlobal(hit->getPosIn());
1088 addSimHit(global_pos, particle);
1089}

◆ addSimHits()

template<class T>
void addSimHits ( const StoreArray< T > & hits)
inline

Add all entries in the given 'hits' array (and the corresponding MCParticles) to the event scene.

Definition at line 135 of file EVEVisualization.h.

136 {
137 const int numHits = hits.getEntries();
138 for (int i = 0; i < numHits; i++) {
139 const RelationsObject* rel = hits[i];
140 const MCParticle* mcpart = rel->getRelatedFrom<MCParticle>();
141
142 addSimHit(hits[i], mcpart);
143 }
144 }
FROM * getRelatedFrom(const std::string &name="", const std::string &namedRelation="") const
Get the object from which this object has a relation.
RelationsInterface< TObject > RelationsObject
Provides interface for getting/adding relations to objects in StoreArrays.

◆ addToGroup()

void addToGroup ( const std::string & name,
TEveElement * elem )
private

Add 'elem' to the element group 'name' (created if necessary).

name can also be a path, e.g. MyOwnStuff/SpecialObject A, which will automatically create sub-groups.

slashes at beginning and end of name are ignored.

Definition at line 1913 of file EVEVisualization.cc.

1914{
1915 //slashes at beginning and end are ignored
1916 const std::string& groupName = boost::algorithm::trim_copy_if(name, boost::algorithm::is_any_of("/"));
1917
1918 TEveElementList* group = m_groups[groupName].group;
1919 if (!group) {
1920 group = new TEveElementList(groupName.c_str(), groupName.c_str());
1921 group->SetRnrState(m_groups[groupName].visible);
1922 m_groups[groupName].group = group;
1923
1924 //if groupName contains '/', remove last bit and add to parent group
1925 //e.g. if groupName=A/B/C, call addToGroup("A/B", groupC)
1926 auto lastSlash = boost::algorithm::find_last(groupName, "/");
1927 if (lastSlash) {
1928 const std::string parentGroup(groupName.begin(), lastSlash.begin());
1929 const std::string thisGroup(lastSlash.end(), groupName.end());
1930 group->SetElementName(thisGroup.c_str());
1931 addToGroup(parentGroup, group);
1932 } else {
1933 gEve->AddElement(group);
1934 }
1935 }
1936 group->AddElement(elem);
1937}
std::map< std::string, ElementGroup > m_groups
name -> grouping element.

◆ addTOPDigits()

void addTOPDigits ( const StoreArray< TOPDigit > & digits)

Add TOPDigits (shown aggregated per module).

Definition at line 1799 of file EVEVisualization.cc.

1800{
1801 /* TOP module ID -> #digits */
1802 std::map<int, int> m_topSummary;
1803 for (const TOPDigit& hit : digits) {
1804 int mod = hit.getModuleID();
1805 ++m_topSummary[mod];
1806 }
1807 int maxcount = 0;
1808 for (auto modCountPair : m_topSummary) {
1809 if (modCountPair.second > maxcount)
1810 maxcount = modCountPair.second;
1811 }
1812 for (auto modCountPair : m_topSummary) {
1813 const auto& topmod = TOP::TOPGeometryPar::Instance()->getGeometry()->getModule(modCountPair.first);
1814 double phi = topmod.getPhi();
1815 double r_center = topmod.getRadius();
1816 double z = topmod.getZc();
1817
1818 ROOT::Math::XYZVector centerPos3D;
1819 VectorUtil::setMagThetaPhi(centerPos3D, r_center, M_PI / 2, phi);
1820 centerPos3D.SetZ(z);
1821
1822 B2Vector3D channelX(1, 0, 0); channelX.RotateZ(phi);
1823 B2Vector3D channelY(0, 1, 0); channelY.RotateZ(phi);
1824
1825 //bar is a bit thicker so we can mouse over without getting the geometry
1826 auto* moduleBox = boxCreator(centerPos3D, channelX, channelY,
1827 3.0 * topmod.getBarThickness(), topmod.getBarWidth(), topmod.getBarLength());
1828 moduleBox->SetMainColor(kAzure + 10);
1829 double weight = double(modCountPair.second) / maxcount;
1830 moduleBox->SetMainTransparency(90 - weight * 50);
1831 moduleBox->SetName(("TOP module " + std::to_string(modCountPair.first)).c_str());
1832 moduleBox->SetTitle(TString::Format("#TOPDigits: %d ", modCountPair.second));
1833
1834 addToGroup("TOP Modules", moduleBox);
1835 //associate all TOPDigits with this module.
1836 for (const TOPDigit& hit : digits) {
1837 if (modCountPair.first == hit.getModuleID())
1838 addObject(&hit, moduleBox);
1839 }
1840 }
1841}
const TOPGeometry * getGeometry() const
Returns pointer to geometry object using basf2 units.
static TOPGeometryPar * Instance()
Static method to obtain the pointer to its instance.
const TOPGeoModule & getModule(int moduleID) const
Returns module.

◆ addTrack()

void addTrack ( const Belle2::Track * belle2Track)

Add this genfit::Track to event data.

Adapted from GenfitDisplay, originally written by Karl Bicker.

Definition at line 384 of file EVEVisualization.cc.

385{
386 // load the pion fit hypothesis or the hypothesis which is the closest in mass to a pion
387 // the tracking will not always successfully fit with a pion hypothesis
388 const TrackFitResult* fitResult = belle2Track->getTrackFitResultWithClosestMass(Const::pion);
389 if (!fitResult) {
390 B2ERROR("Track without TrackFitResult skipped.");
391 return;
392 }
393 const RecoTrack* track = belle2Track->getRelated<RecoTrack>();
394
395 const TString label = ObjectInfo::getIdentifier(belle2Track) + " (" + ObjectInfo::getIdentifier(fitResult) + ")";
396
397 // parse the option string ------------------------------------------------------------------------
398 bool drawDetectors = false;
399 bool drawHits = false;
400 bool drawPlanes = false;
401
402 if (m_options != "") {
403 for (size_t i = 0; i < m_options.length(); i++) {
404 if (m_options.at(i) == 'D') drawDetectors = true;
405 if (m_options.at(i) == 'H') drawHits = true;
406 if (m_options.at(i) == 'P') drawPlanes = true;
407 }
408 }
409 // finished parsing the option string -------------------------------------------------------------
410
411 // We loop over all points (scattering non-measurement points for GBL)
412 // and for Kalman we skip those with no measurements, which should
413 // not be there
414 bool isPruned = (track == nullptr);
415
416
417 TEveRecTrackD recTrack;
418 const B2Vector3D& poca = fitResult->getPosition();
419 recTrack.fV.Set(poca);
420
421 const B2Vector3D& poca_momentum = fitResult->getMomentum();
422 if (std::isfinite(poca_momentum.Mag()))
423 recTrack.fP.Set(poca_momentum);
424 else //use 1TeV momentum for tracks without curvature
425 recTrack.fP.Set(B2Vector3D(fitResult->getHelix().getDirection() * 1000));
426
427 recTrack.fSign = fitResult->getChargeSign();
428 TEveTrack* eveTrack = new TEveTrack(&recTrack, m_gftrackpropagator);
429 eveTrack->SetName(label);
430
431
432 if (track) {
433 const genfit::AbsTrackRep* representation;
434
435 if (m_drawCardinalRep) {
436 representation = track->getCardinalRepresentation();
437 B2DEBUG(100, "Draw cardinal rep");
438 } else {
439 const auto& representations = track->getRepresentations();
440 if (representations.empty()) {
441 B2ERROR("No representations found in the reco track!");
442 return;
443 }
444 B2DEBUG(100, "Draw representation number 0.");
445 representation = representations.front();
446 }
447
448 if (!track->hasTrackFitStatus(representation)) {
449 B2ERROR("RecoTrack without FitStatus: will be skipped!");
450 return;
451 }
452
453 const genfit::FitStatus* fitStatus = track->getTrackFitStatus(representation);
454
455 isPruned = fitStatus->isTrackPruned();
456
457 // GBL and Kalman cannot mix in a track.
458 // What is 0 after first loop will stay 0:
459 genfit::KalmanFitterInfo* fi = 0;
460 genfit::KalmanFitterInfo* prevFi = 0;
461 genfit::GblFitterInfo* gfi = 0;
462 genfit::GblFitterInfo* prevGFi = 0;
463 const genfit::MeasuredStateOnPlane* fittedState(NULL);
464 const genfit::MeasuredStateOnPlane* prevFittedState(NULL);
465
466
467 const auto& hitPoints = track->getHitPointsWithMeasurement();
468 const unsigned int numpoints = hitPoints.size();
469
470 int hitCounter = -1;
471 for (const genfit::TrackPoint* tp : hitPoints) { // loop over all points in the track
472 hitCounter++;
473
474 // get the fitter infos ------------------------------------------------------------------
475 if (! tp->hasFitterInfo(representation)) {
476 B2ERROR("trackPoint has no fitterInfo for rep");
477 continue;
478 }
479
480 genfit::AbsFitterInfo* fitterInfo = tp->getFitterInfo(representation);
481
482 fi = dynamic_cast<genfit::KalmanFitterInfo*>(fitterInfo);
483 gfi = dynamic_cast<genfit::GblFitterInfo*>(fitterInfo);
484
485 if (!gfi && !fi) {
486 B2ERROR("Can only display KalmanFitterInfo or GblFitterInfo");
487 continue;
488 }
489
490 if (gfi && fi)
491 B2FATAL("AbsFitterInfo dynamic-casted to both KalmanFitterInfo and GblFitterInfo!");
492
493
494 if (fi && ! tp->hasRawMeasurements()) {
495 B2ERROR("trackPoint has no raw measurements");
496 continue;
497 }
498
499 if (fi && ! fi->hasPredictionsAndUpdates()) {
500 B2ERROR("KalmanFitterInfo does not have all predictions and updates");
501 continue;
502 }
503
504 try {
505 if (gfi)
506 fittedState = &(gfi->getFittedState(true));
507 if (fi)
508 fittedState = &(fi->getFittedState(true));
509 } catch (genfit::Exception& e) {
510 B2ERROR(e.what() << " - can not get fitted state");
511 continue;
512 }
513
514 ROOT::Math::XYZVector track_pos = ROOT::Math::XYZVector(representation->getPos(*fittedState));
515
516 // draw track if corresponding option is set ------------------------------------------
517 if (prevFittedState != NULL) {
518
519 TEvePathMark::EType_e markType = TEvePathMark::kReference;
520 if (hitCounter + 1 == static_cast<int>(numpoints)) //track should stop here.
521 markType = TEvePathMark::kDecay;
522
523 // Kalman: non-null prevFi ensures that the previous fitter info was also KalmanFitterInfo
524 if (fi && prevFi) {
525 makeLines(eveTrack, prevFittedState, fittedState, representation, markType, m_drawErrors);
526 if (m_drawErrors) { // make sure to draw errors in both directions
527 makeLines(eveTrack, prevFittedState, fittedState, representation, markType, m_drawErrors, 0);
528 }
529 //these are currently disabled.
530 //TODO: if activated, I want to have a separate TEveStraightLineSet instead of eveTrack (different colors/options)
531 if (m_drawForward)
532 makeLines(eveTrack, prevFi->getForwardUpdate(), fi->getForwardPrediction(), representation, markType, m_drawErrors, 0);
533 if (m_drawBackward)
534 makeLines(eveTrack, prevFi->getBackwardPrediction(), fi->getBackwardUpdate(), representation, markType, m_drawErrors);
535 // draw reference track if corresponding option is set ------------------------------------------
536 if (m_drawRefTrack && fi->hasReferenceState() && prevFi->hasReferenceState())
537 makeLines(eveTrack, prevFi->getReferenceState(), fi->getReferenceState(), representation, markType, false);
538 }
539
540 // GBL: non-null prevGFi ensures that the previous fitter info was also GblFitterInfo
541 if (gfi && prevGFi) {
542 makeLines(eveTrack, prevFittedState, fittedState, representation, markType, m_drawErrors);
543 if (m_drawErrors) {
544 makeLines(eveTrack, prevFittedState, fittedState, representation, markType, m_drawErrors, 0);
545 }
546
547 if (m_drawRefTrack && gfi->hasReferenceState() && prevGFi->hasReferenceState()) {
548 genfit::StateOnPlane prevSop = prevGFi->getReferenceState();
549 genfit::StateOnPlane sop = gfi->getReferenceState();
550 makeLines(eveTrack, &prevSop, &sop, representation, markType, false);
551 }
552 }
553
554 }
555 prevFi = fi; // Kalman
556 prevGFi = gfi; // GBL
557 prevFittedState = fittedState;
558
559
560 //loop over all measurements for this point (e.g. u and v-type strips)
561 const int numMeasurements = tp->getNumRawMeasurements();
562 for (int iMeasurement = 0; iMeasurement < numMeasurements; iMeasurement++) {
563 const genfit::AbsMeasurement* m = tp->getRawMeasurement(iMeasurement);
564
565 TVectorT<double> hit_coords;
566 TMatrixTSym<double> hit_cov;
567
568 if (fi) {
569 // Kalman
570 genfit::MeasurementOnPlane* mop = fi->getMeasurementOnPlane(iMeasurement);
571 hit_coords.ResizeTo(mop->getState());
572 hit_cov.ResizeTo(mop->getCov());
573 hit_coords = mop->getState();
574 hit_cov = mop->getCov();
575 }
576 if (gfi) {
577 // GBL - has only one measurement (other should be already merged before the track is constructed)
578 // That means tp->getNumRawMeasurements() returns 1 for tracks fitted by GBL, because GBLfit Module
579 // merges all corresponding SVD strips to 2D combined clusters.
580 genfit::MeasurementOnPlane gblMeas = gfi->getMeasurement();
581 hit_coords.ResizeTo(gblMeas.getState());
582 hit_cov.ResizeTo(gblMeas.getCov());
583 hit_coords = gblMeas.getState();
584 hit_cov = gblMeas.getCov();
585 }
586
587 // finished getting the hit infos -----------------------------------------------------
588
589 // sort hit infos into variables ------------------------------------------------------
590 ROOT::Math::XYZVector o = ROOT::Math::XYZVector(fittedState->getPlane()->getO());
591 ROOT::Math::XYZVector u = ROOT::Math::XYZVector(fittedState->getPlane()->getU());
592 ROOT::Math::XYZVector v = ROOT::Math::XYZVector(fittedState->getPlane()->getV());
593
594 bool planar_hit = false;
595 bool planar_pixel_hit = false;
596 bool space_hit = false;
597 bool wire_hit = false;
598 bool wirepoint_hit = false;
599 double_t hit_u = 0;
600 double_t hit_v = 0;
601 double_t plane_size = 4;
602
603 if (dynamic_cast<const genfit::PlanarMeasurement*>(m) != NULL) {
604 int hit_coords_dim = m->getDim();
605 planar_hit = true;
606 if (hit_coords_dim == 1) {
607 hit_u = hit_coords(0);
608 } else if (hit_coords_dim == 2) {
609 planar_pixel_hit = true;
610 hit_u = hit_coords(0);
611 hit_v = hit_coords(1);
612 }
613 } else if (dynamic_cast<const genfit::SpacepointMeasurement*>(m) != NULL) {
614 space_hit = true;
615 } else if (dynamic_cast<const CDCRecoHit*>(m)
616 || dynamic_cast<const genfit::WireMeasurement*>(m)
617 || dynamic_cast<const genfit::WireMeasurementNew*>(m)) {
618 wire_hit = true;
619 hit_u = fabs(hit_coords(0));
620 hit_v = v.Dot(track_pos - o); // move the covariance tube so that the track goes through it
621 if (dynamic_cast<const genfit::WirePointMeasurement*>(m) != NULL) {
622 wirepoint_hit = true;
623 hit_v = hit_coords(1);
624 }
625 } else {
626 B2ERROR("Hit " << hitCounter << ", Measurement " << iMeasurement << ": Unknown measurement type: skipping hit!");
627 break;
628 }
629
630 // finished setting variables ---------------------------------------------------------
631
632 // draw planes if corresponding option is set -----------------------------------------
633 if (drawPlanes || (drawDetectors && planar_hit)) {
634 ROOT::Math::XYZVector move(0, 0, 0);
635 if (wire_hit) move = v * (v.Dot(track_pos - o)); // move the plane along the wire until the track goes through it
636 TEveBox* box = boxCreator(o + move, u, v, plane_size, plane_size, 0.01);
637 if (drawDetectors && planar_hit) {
638 box->SetMainColor(kCyan);
639 } else {
640 box->SetMainColor(kGray);
641 }
642 box->SetMainTransparency(50);
643 eveTrack->AddElement(box);
644 }
645 // finished drawing planes ------------------------------------------------------------
646
647 // draw detectors if option is set, only important for wire hits ----------------------
648 if (drawDetectors) {
649
650 if (wire_hit) {
651 TEveGeoShape* det_shape = new TEveGeoShape("det_shape");
652 det_shape->SetShape(new TGeoTube(std::max(0., (double)(hit_u - 0.0105 / 2.)), hit_u + 0.0105 / 2., plane_size));
653 fixGeoShapeRefCount(det_shape);
654
655 ROOT::Math::XYZVector norm = u.Cross(v);
656 TGeoRotation det_rot("det_rot", (u.Theta() * 180) / TMath::Pi(), (u.Phi() * 180) / TMath::Pi(),
657 (norm.Theta() * 180) / TMath::Pi(), (norm.Phi() * 180) / TMath::Pi(),
658 (v.Theta() * 180) / TMath::Pi(), (v.Phi() * 180) / TMath::Pi()); // move the tube to the right place and rotate it correctly
659 ROOT::Math::XYZVector move = v * (v.Dot(track_pos - o)); // move the tube along the wire until the track goes through it
660 TGeoCombiTrans det_trans(o.X() + move.X(),
661 o.Y() + move.Y(),
662 o.Z() + move.Z(),
663 &det_rot);
664 det_shape->SetTransMatrix(det_trans);
665 det_shape->SetMainColor(kCyan);
666 det_shape->SetMainTransparency(25);
667 if ((drawHits && (hit_u + 0.0105 / 2 > 0)) || !drawHits) {
668 eveTrack->AddElement(det_shape);
669 }
670 }
671
672 }
673 // finished drawing detectors ---------------------------------------------------------
674
675 if (drawHits) {
676
677 // draw planar hits, with distinction between strip and pixel hits ----------------
678 if (planar_hit) {
679 if (!planar_pixel_hit) {
680 //strip hit
681 static VXD::GeoCache& geo = VXD::GeoCache::getInstance();
682 const SVDRecoHit* recoHit = dynamic_cast<const SVDRecoHit*>(m);
683 if (!recoHit) {
684 B2WARNING("SVD recohit couldn't be converted... ");
685 continue;
686 }
687
688 const VXD::SensorInfoBase& sensor = geo.getSensorInfo(recoHit->getSensorID());
689 double du, dv;
690 ROOT::Math::XYZVector a = o; //defines position of sensor plane
691 double hit_res_u = hit_cov(0, 0);
692 if (recoHit->isU()) {
693 du = std::sqrt(hit_res_u);
694 dv = sensor.getLength();
695 a += hit_u * u;
696 } else {
697 du = sensor.getWidth();
698 dv = std::sqrt(hit_res_u);
699 a += hit_u * v;
700 }
701 double depth = sensor.getThickness();
702 TEveBox* hit_box = boxCreator(a, u, v, du, dv, depth);
703 hit_box->SetName("SVDRecoHit");
704 hit_box->SetMainColor(c_recoHitColor);
705 hit_box->SetMainTransparency(0);
706 eveTrack->AddElement(hit_box);
707 } else {
708 //pixel hit
709 // calculate eigenvalues to draw error-ellipse ----------------------------
710 TMatrixDSymEigen eigen_values(hit_cov);
711 TEveGeoShape* cov_shape = new TEveGeoShape("PXDRecoHit");
712 const TVectorD& ev = eigen_values.GetEigenValues();
713 const TMatrixD& eVec = eigen_values.GetEigenVectors();
714 double pseudo_res_0 = m_errorScale * std::sqrt(ev(0));
715 double pseudo_res_1 = m_errorScale * std::sqrt(ev(1));
716 // finished calculating, got the values -----------------------------------
717
718 // calculate the semiaxis of the error ellipse ----------------------------
719 cov_shape->SetShape(new TGeoEltu(pseudo_res_0, pseudo_res_1, 0.0105));
720 fixGeoShapeRefCount(cov_shape);
721 ROOT::Math::XYZVector pix_pos = o + hit_u * u + hit_v * v;
722 ROOT::Math::XYZVector u_semiaxis = (pix_pos + eVec(0, 0) * u + eVec(1, 0) * v) - pix_pos;
723 ROOT::Math::XYZVector v_semiaxis = (pix_pos + eVec(0, 1) * u + eVec(1, 1) * v) - pix_pos;
724 ROOT::Math::XYZVector norm = u.Cross(v);
725 // finished calculating ---------------------------------------------------
726
727 // rotate and translate everything correctly ------------------------------
728 TGeoRotation det_rot("det_rot", (u_semiaxis.Theta() * 180) / TMath::Pi(), (u_semiaxis.Phi() * 180) / TMath::Pi(),
729 (v_semiaxis.Theta() * 180) / TMath::Pi(), (v_semiaxis.Phi() * 180) / TMath::Pi(),
730 (norm.Theta() * 180) / TMath::Pi(), (norm.Phi() * 180) / TMath::Pi());
731 TGeoCombiTrans det_trans(pix_pos.X(), pix_pos.Y(), pix_pos.Z(), &det_rot);
732 cov_shape->SetTransMatrix(det_trans);
733 // finished rotating and translating --------------------------------------
734
735 cov_shape->SetMainColor(c_recoHitColor);
736 cov_shape->SetMainTransparency(0);
737 eveTrack->AddElement(cov_shape);
738 }
739 }
740 // finished drawing planar hits ---------------------------------------------------
741
742 // draw spacepoint hits -----------------------------------------------------------
743 if (space_hit) {
744
745 // get eigenvalues of covariance to know how to draw the ellipsoid ------------
746 TMatrixDSymEigen eigen_values(m->getRawHitCov());
747 TEveGeoShape* cov_shape = new TEveGeoShape("SpacePoint Hit");
748 cov_shape->SetShape(new TGeoSphere(0., 1.));
749 fixGeoShapeRefCount(cov_shape);
750 const TVectorD& ev = eigen_values.GetEigenValues();
751 const TMatrixD& eVec = eigen_values.GetEigenVectors();
752 ROOT::Math::XYZVector eVec1(eVec(0, 0), eVec(1, 0), eVec(2, 0));
753 ROOT::Math::XYZVector eVec2(eVec(0, 1), eVec(1, 1), eVec(2, 1));
754 ROOT::Math::XYZVector eVec3(eVec(0, 2), eVec(1, 2), eVec(2, 2));
755 // got everything we need -----------------------------------------------------
756
757
758 TGeoRotation det_rot("det_rot", (eVec1.Theta() * 180) / TMath::Pi(), (eVec1.Phi() * 180) / TMath::Pi(),
759 (eVec2.Theta() * 180) / TMath::Pi(), (eVec2.Phi() * 180) / TMath::Pi(),
760 (eVec3.Theta() * 180) / TMath::Pi(), (eVec3.Phi() * 180) / TMath::Pi()); // the rotation is already clear
761
762 // set the scaled eigenvalues -------------------------------------------------
763 double pseudo_res_0 = m_errorScale * std::sqrt(ev(0));
764 double pseudo_res_1 = m_errorScale * std::sqrt(ev(1));
765 double pseudo_res_2 = m_errorScale * std::sqrt(ev(2));
766 // finished scaling -----------------------------------------------------------
767
768 // rotate and translate -------------------------------------------------------
769 TGeoGenTrans det_trans(o.X(), o.Y(), o.Z(),
770 //std::sqrt(pseudo_res_0/pseudo_res_1/pseudo_res_2), std::sqrt(pseudo_res_1/pseudo_res_0/pseudo_res_2), std::sqrt(pseudo_res_2/pseudo_res_0/pseudo_res_1), // this workaround is necessary due to the "normalization" performed in TGeoGenTrans::SetScale
771 //1/(pseudo_res_0),1/(pseudo_res_1),1/(pseudo_res_2),
772 pseudo_res_0, pseudo_res_1, pseudo_res_2,
773 &det_rot);
774 cov_shape->SetTransMatrix(det_trans);
775 // finished rotating and translating ------------------------------------------
776
777 cov_shape->SetMainColor(c_recoHitColor);
778 cov_shape->SetMainTransparency(10);
779 eveTrack->AddElement(cov_shape);
780 }
781 // finished drawing spacepoint hits -----------------------------------------------
782
783 // draw wire hits -----------------------------------------------------------------
784 if (wire_hit) {
785 const double cdcErrorScale = 1.0;
786 TEveGeoShape* cov_shape = new TEveGeoShape("CDCRecoHit");
787 double pseudo_res_0 = cdcErrorScale * std::sqrt(hit_cov(0, 0));
788 double pseudo_res_1 = plane_size;
789 if (wirepoint_hit) pseudo_res_1 = cdcErrorScale * std::sqrt(hit_cov(1, 1));
790
791 cov_shape->SetShape(new TGeoTube(std::max(0., (double)(hit_u - pseudo_res_0)), hit_u + pseudo_res_0, pseudo_res_1));
792 fixGeoShapeRefCount(cov_shape);
793 ROOT::Math::XYZVector norm = u.Cross(v);
794
795 // rotate and translate -------------------------------------------------------
796 TGeoRotation det_rot("det_rot", (u.Theta() * 180) / TMath::Pi(), (u.Phi() * 180) / TMath::Pi(),
797 (norm.Theta() * 180) / TMath::Pi(), (norm.Phi() * 180) / TMath::Pi(),
798 (v.Theta() * 180) / TMath::Pi(), (v.Phi() * 180) / TMath::Pi());
799 TGeoCombiTrans det_trans(o.X() + hit_v * v.X(),
800 o.Y() + hit_v * v.Y(),
801 o.Z() + hit_v * v.Z(),
802 &det_rot);
803 cov_shape->SetTransMatrix(det_trans);
804 // finished rotating and translating ------------------------------------------
805
806 cov_shape->SetMainColor(c_recoHitColor);
807 cov_shape->SetMainTransparency(50);
808 eveTrack->AddElement(cov_shape);
809 }
810 // finished drawing wire hits -----------------------------------------------------
811 }
812
813 }
814 }
815
816 auto& firstref = eveTrack->RefPathMarks().front();
817 auto& lastref = eveTrack->RefPathMarks().back();
818 double f = firstref.fV.Distance(recTrack.fV);
819 double b = lastref.fV.Distance(recTrack.fV);
820 if (f > 100 and f > b) {
821 B2WARNING("Decay vertex is much closer to POCA than first vertex, reversing order of track points... (this is intended for cosmic tracks, if you see this message in other context it might indicate a problem)");
822 //last ref is better than first...
823 lastref.fType = TEvePathMarkD::kReference;
824 firstref.fType = TEvePathMarkD::kDecay;
825 std::reverse(eveTrack->RefPathMarks().begin(), eveTrack->RefPathMarks().end());
826 }
827 }
828 eveTrack->SetTitle(TString::Format("%s\n"
829 "pruned: %s\n"
830 "pT=%.3f, pZ=%.3f\n"
831 "pVal: %e",
832 label.Data(),
833 isPruned ? " yes" : "no",
834 poca_momentum.Pt(), poca_momentum.Pz(),
835 fitResult->getPValue()
836 ));
837 eveTrack->SetLineColor(c_trackColor);
838 eveTrack->SetLineStyle(1);
839 eveTrack->SetLineWidth(3.0);
840
841
842 addToGroup("Fitted Tracks", eveTrack);
843 if (track)
844 addObject(track, eveTrack);
845 addObject(belle2Track, eveTrack);
846}
DataType Pz() const
access variable Z (= .at(2) without boundary check)
Definition B2Vector3.h:441
DataType Mag() const
The magnitude (rho in spherical coordinate system).
Definition B2Vector3.h:159
DataType Pt() const
The transverse component (R in cylindrical coordinate system).
Definition B2Vector3.h:198
static const ChargedStable pion
charged pion particle
Definition Const.h:661
static const int c_recoHitColor
Color for reco hits.
bool m_drawRefTrack
Draw reference track in addTrack.
bool m_drawForward
draw forward in addTrack
bool m_drawCardinalRep
Draw cardinal representation in addTrack.
std::string m_options
Option string for genfit::Track visualisation.
static const int c_trackColor
Color for tracks.
bool m_drawBackward
draw backward in addTrack
bool m_drawErrors
Draw errors in addTrack.
double m_errorScale
Rescale PXD/SVD errors with this factor to ensure visibility.
static void makeLines(TEveTrack *eveTrack, const genfit::StateOnPlane *prevState, const genfit::StateOnPlane *state, const genfit::AbsTrackRep *rep, TEvePathMark::EType_e markType, bool drawErrors, int markerPos=1)
Create hit visualisation for the given options, and add them to 'eveTrack'.
T * getRelated(const std::string &name="", const std::string &namedRelation="") const
Get the object to or from which this object has a relation.
bool isU() const
Is the coordinate u or v?
Definition SVDRecoHit.h:91
VxdID getSensorID() const
Get the compact ID.
Definition SVDRecoHit.h:82
Helix getHelix() const
Conversion to framework Helix (without covariance).
short getChargeSign() const
Return track charge (1 or -1).
double getPValue() const
Getter for Chi2 Probability of the track fit.
ROOT::Math::XYZVector getMomentum() const
Getter for vector of momentum at closest approach of track in r/phi projection.
ROOT::Math::XYZVector getPosition() const
Getter for vector of position at closest approach of track in r/phi projection.
const TrackFitResult * getTrackFitResultWithClosestMass(const Const::ChargedStable &requestedType) const
Return the track fit for the fit hypothesis with the closest mass.
Definition Track.cc:104
ROOT::Math::XYZVector poca(ROOT::Math::XYZVector const &trackPos, ROOT::Math::XYZVector const &trackP, ROOT::Math::XYZVector const &vtxPos)
Returns the Point Of Closest Approach of a track to a vertex.

◆ addTrackCandidate()

void addTrackCandidate ( const std::string & collectionName,
const RecoTrack & recoTrack )

Add a RecoTrack, to evaluate track finding.

Definition at line 202 of file EVEVisualization.cc.

204{
205 const TString label = ObjectInfo::getIdentifier(&recoTrack);
206 // parse the option string ------------------------------------------------------------------------
207 bool drawHits = false;
208
209 if (m_options != "") {
210 for (size_t i = 0; i < m_options.length(); i++) {
211 if (m_options.at(i) == 'H') drawHits = true;
212 }
213 }
214 // finished parsing the option string -------------------------------------------------------------
215
216
217 //track seeds
218 const B2Vector3D& track_pos = recoTrack.getPositionSeed();
219 const B2Vector3D& track_mom = recoTrack.getMomentumSeed();
220
221 TEveStraightLineSet* lines = new TEveStraightLineSet("RecoHits for " + label);
222 lines->SetMainColor(c_recoTrackColor);
223 lines->SetMarkerColor(c_recoTrackColor);
224 lines->SetMarkerStyle(6);
225 lines->SetMainTransparency(60);
226
227 if (drawHits) {
228 // Loop over all hits in the track (three different types)
229 for (const RecoHitInformation::UsedPXDHit* pxdHit : recoTrack.getPXDHitList()) {
230 addRecoHit(pxdHit, lines);
231 }
232
233 for (const RecoHitInformation::UsedSVDHit* svdHit : recoTrack.getSVDHitList()) {
234 addRecoHit(svdHit, lines);
235 }
236
237 for (const RecoHitInformation::UsedCDCHit* cdcHit : recoTrack.getCDCHitList()) {
238 addRecoHit(cdcHit, lines);
239 }
240 }
241
242 TEveRecTrack rectrack;
243 rectrack.fP.Set(track_mom);
244 rectrack.fV.Set(track_pos);
245
246 TEveTrack* track_lines = new TEveTrack(&rectrack, m_gftrackpropagator);
247 track_lines->SetName(label); //popup label set at end of function
248 track_lines->SetPropagator(m_gftrackpropagator);
249 track_lines->SetLineColor(c_recoTrackColor);
250 track_lines->SetLineWidth(1);
251 track_lines->SetTitle(ObjectInfo::getTitle(&recoTrack));
252
253 track_lines->SetCharge((int)recoTrack.getChargeSeed());
254
255
256 track_lines->AddElement(lines);
257 addToGroup(collectionName, track_lines);
258 addObject(&recoTrack, track_lines);
259}
void addRecoHit(const SomeVXDHit *hit, TEveStraightLineSet *lines)
adds given VXD hit to lines.
static const int c_recoTrackColor
Color for TrackCandidates.
CDCHit UsedCDCHit
Define, use of CDC hits as CDC hits (for symmetry).
PXDCluster UsedPXDHit
Define, use of clusters or true hits for PXD.
SVDCluster UsedSVDHit
Define, use of clusters or true hits for SVD.
std::vector< Belle2::RecoTrack::UsedPXDHit * > getPXDHitList() const
Return an unsorted list of pxd hits.
Definition RecoTrack.h:449
std::vector< Belle2::RecoTrack::UsedSVDHit * > getSVDHitList() const
Return an unsorted list of svd hits.
Definition RecoTrack.h:452
std::vector< Belle2::RecoTrack::UsedCDCHit * > getCDCHitList() const
Return an unsorted list of cdc hits.
Definition RecoTrack.h:455
ROOT::Math::XYZVector getPositionSeed() const
Return the position seed stored in the reco track. ATTENTION: This is not the fitted position.
Definition RecoTrack.h:480
short int getChargeSeed() const
Return the charge seed stored in the reco track. ATTENTION: This is not the fitted charge.
Definition RecoTrack.h:508
ROOT::Math::XYZVector getMomentumSeed() const
Return the momentum seed stored in the reco track. ATTENTION: This is not the fitted momentum.
Definition RecoTrack.h:487

◆ addTrackCandidateImproved()

void addTrackCandidateImproved ( const std::string & collectionName,
const RecoTrack & recoTrack )

Add a RecoTrack, but use stored genfit track representation to make visualisation objects.

FIXME this is mostly just a workaround for monopoles.

Definition at line 261 of file EVEVisualization.cc.

263{
264 const TString label = ObjectInfo::getIdentifier(&recoTrack);
265 // parse the option string ------------------------------------------------------------------------
266 bool drawHits = false;
267
268 if (m_options != "") {
269 for (size_t i = 0; i < m_options.length(); i++) {
270 if (m_options.at(i) == 'H') drawHits = true;
271 }
272 }
273 // finished parsing the option string -------------------------------------------------------------
274
275 // Create a track as a polyline through reconstructed points
276 // FIXME this is snatched from PrimitivePlotter, need to add extrapolation out of CDC
277 TEveLine* track = new TEveLine(); // We are going to just add points with SetNextPoint
278 std::vector<ROOT::Math::XYZVector> posPoints; // But first we'll have to sort them as in RecoHits axial and stereo come in blocks
279 track->SetName(label); //popup label set at end of function
280 track->SetLineColor(c_recoTrackColor);
281 track->SetLineWidth(3);
282 track->SetTitle(ObjectInfo::getTitle(&recoTrack));
283 track->SetSmooth(true);
284
285 for (const auto recoHit : recoTrack.getRecoHitInformations()) {
286 // skip for reco hits which have not been used in the fit (and therefore have no fitted information on the plane
287 if (!recoHit->useInFit())
288 continue;
289
290 // Need TVector3 here for genfit interface below
291 TVector3 pos;
292 TVector3 mom;
293 TMatrixDSym cov;
294
295 try {
296 const auto* trackPoint = recoTrack.getCreatedTrackPoint(recoHit);
297 const auto* fittedResult = trackPoint->getFitterInfo();
298 if (not fittedResult) {
299 B2WARNING("Skipping unfitted track point");
300 continue;
301 }
302 const genfit::MeasuredStateOnPlane& state = fittedResult->getFittedState();
303 state.getPosMomCov(pos, mom, cov);
304 } catch (const genfit::Exception&) {
305 B2WARNING("Skipping state with strange pos, mom or cov");
306 continue;
307 }
308
309 posPoints.push_back(ROOT::Math::XYZVector(pos.X(), pos.Y(), pos.Z()));
310 }
311
312 sort(posPoints.begin(), posPoints.end(),
313 [](const ROOT::Math::XYZVector & a, const ROOT::Math::XYZVector & b) -> bool {
314 return a.X() * a.X() + a.Y() * a.Y() > b.X() * b.X() + b.Y() * b.Y();
315 });
316 for (auto vec : posPoints) {
317 track->SetNextPoint(vec.X(), vec.Y(), vec.Z());
318 }
319 // add corresponding hits ---------------------------------------------------------------------
320 TEveStraightLineSet* lines = new TEveStraightLineSet("RecoHits for " + label);
321 lines->SetMainColor(c_recoTrackColor);
322 lines->SetMarkerColor(c_recoTrackColor);
323 lines->SetMarkerStyle(6);
324 lines->SetMainTransparency(60);
325
326 if (drawHits) {
327 // Loop over all hits in the track (three different types)
328 for (const RecoHitInformation::UsedPXDHit* pxdHit : recoTrack.getPXDHitList()) {
329 addRecoHit(pxdHit, lines);
330 }
331
332 for (const RecoHitInformation::UsedSVDHit* svdHit : recoTrack.getSVDHitList()) {
333 addRecoHit(svdHit, lines);
334 }
335
336 for (const RecoHitInformation::UsedCDCHit* cdcHit : recoTrack.getCDCHitList()) {
337 addRecoHit(cdcHit, lines);
338 }
339 }
340
341 track->AddElement(lines);
342 addToGroup(collectionName, track);
343 addObject(&recoTrack, track);
344}
const genfit::TrackPoint * getCreatedTrackPoint(const RecoHitInformation *recoHitInformation) const
Get a pointer to the TrackPoint that was created from this hit.
Definition RecoTrack.cc:230
std::vector< RecoHitInformation * > getRecoHitInformations(bool getSorted=false) const
Return a list of all RecoHitInformations associated with the RecoTrack.
Definition RecoTrack.cc:557

◆ addUnassignedRecoHits()

template<class T>
void addUnassignedRecoHits ( const StoreArray< T > & hits)
inline

After adding recohits for tracks/candidates, this function adds the remaining hits in a global collection.

Definition at line 190 of file EVEVisualization.h.

191 {
192 if (hits.getEntries() == 0)
193 return;
194 if (!m_unassignedRecoHits) {
195 m_unassignedRecoHits = new TEveStraightLineSet("Unassigned RecoHits");
196 m_unassignedRecoHits->SetTitle("Unassigned RecoHits");
197 m_unassignedRecoHits->SetMainColor(c_unassignedHitColor);
198 m_unassignedRecoHits->SetMarkerColor(c_unassignedHitColor);
199 m_unassignedRecoHits->SetMarkerStyle(6);
200 //m_unassignedRecoHits->SetMainTransparency(60);
201 }
202 for (const T& hit : hits) {
203 if (m_shownRecohits.count(&hit) == 0) {
204 addRecoHit(&hit, m_unassignedRecoHits);
205 }
206 }
207 }

◆ addVertex()

void addVertex ( const genfit::GFRaveVertex * vertex)

Add a vertex point and its covariance matrix.

Definition at line 1352 of file EVEVisualization.cc.

1353{
1354 ROOT::Math::XYZVector v = ROOT::Math::XYZVector(vertex->getPos());
1355 TEvePointSet* vertexPoint = new TEvePointSet(ObjectInfo::getInfo(vertex));
1356 //sadly, setting a title for a TEveGeoShape doesn't result in a popup...
1357 vertexPoint->SetTitle(ObjectInfo::getTitle(vertex));
1358 vertexPoint->SetMainColor(c_recoHitColor);
1359 vertexPoint->SetNextPoint(v.X(), v.Y(), v.Z());
1360
1361 TMatrixDSymEigen eigen_values(vertex->getCov());
1362 TEveGeoShape* det_shape = new TEveGeoShape(ObjectInfo::getInfo(vertex) + " Error");
1363 det_shape->SetShape(new TGeoSphere(0., 1.)); //Initially created as a sphere, then "scaled" into an ellipsoid.
1364 fixGeoShapeRefCount(det_shape);
1365 const TVectorD& ev = eigen_values.GetEigenValues(); //Assigns the eigenvalues into the "ev" matrix.
1366 const TMatrixD& eVec = eigen_values.GetEigenVectors(); //Assigns the eigenvalues into the "eVec" matrix.
1367 //Define the 3 eigenvectors of the covariance matrix as objects of the ROOT::Math::XYZVector class using constructor.
1368 ROOT::Math::XYZVector eVec1(eVec(0, 0), eVec(1, 0), eVec(2, 0));
1369 //eVec(i,j) uses the method/overloaded operator ( . ) of the TMatrixT class to return the matrix entry.
1370 ROOT::Math::XYZVector eVec2(eVec(0, 1), eVec(1, 1), eVec(2, 1));
1371 ROOT::Math::XYZVector eVec3(eVec(0, 2), eVec(1, 2), eVec(2, 2));
1372 // got everything we need ----------------------------------------------------- //Eigenvalues(semi axis) of the covariance matrix acquired!
1373
1374
1375 TGeoRotation det_rot("det_rot", (eVec1.Theta() * 180) / TMath::Pi(), (eVec1.Phi() * 180) / TMath::Pi(),
1376 (eVec2.Theta() * 180) / TMath::Pi(), (eVec2.Phi() * 180) / TMath::Pi(),
1377 (eVec3.Theta() * 180) / TMath::Pi(), (eVec3.Phi() * 180) / TMath::Pi()); // the rotation is already clear
1378
1379 // set the scaled eigenvalues -------------------------------------------------
1380 //"Scaled" eigenvalues pseudo_res (lengths of the semi axis) are the sqrt of the eigenvalues.
1381 double pseudo_res_0 = std::sqrt(ev(0));
1382 double pseudo_res_1 = std::sqrt(ev(1));
1383 double pseudo_res_2 = std::sqrt(ev(2));
1384
1385 //B2INFO("The pseudo_res_0/1/2 are " << pseudo_res_0 << "," << pseudo_res_1 << "," << pseudo_res_2); //shows the scaled eigenvalues
1386
1387
1388
1389 // rotate and translate -------------------------------------------------------
1390 TGeoGenTrans det_trans(v.X(), v.Y(), v.Z(), pseudo_res_0, pseudo_res_1, pseudo_res_2,
1391 &det_rot); //Puts the ellipsoid at the position of the vertex, v(0)=v.X(), operator () overloaded.
1392 det_shape->SetTransMatrix(det_trans);
1393 // finished rotating and translating ------------------------------------------
1394
1395 det_shape->SetMainColor(kOrange); //The color of the error ellipsoid.
1396 det_shape->SetMainTransparency(0);
1397
1398 vertexPoint->AddElement(det_shape);
1399 addToGroup("Vertices", vertexPoint);
1400 addObject(vertex, vertexPoint);
1401}

◆ boxCreator()

TEveBox * boxCreator ( const ROOT::Math::XYZVector & o,
ROOT::Math::XYZVector u,
ROOT::Math::XYZVector v,
float ud,
float vd,
float depth )
staticprivate

Create a box around o, oriented along u and v with widths ud, vd and depth and return a pointer to the box object.

Definition at line 848 of file EVEVisualization.cc.

851{
852 //force minimum width of polygon to deal with Eve limits
853 float min = 0.04;
854 if (vd < min)
855 vd = min;
856 if (ud < min)
857 ud = min;
858 if (depth < min)
859 depth = min;
860
861 TEveBox* box = new TEveBox;
862 box->SetPickable(true);
863
864 ROOT::Math::XYZVector norm = u.Cross(v);
865 u *= (0.5 * ud);
866 v *= (0.5 * vd);
867 norm *= (0.5 * depth);
868
869
870 for (int k = 0; k < 8; ++k) {
871 // Coordinates for all eight corners of the box
872 // as two parallel rectangles, with vertices specified in clockwise direction
873 int signU = ((k + 1) & 2) ? -1 : 1;
874 int signV = (k & 4) ? -1 : 1;
875 int signN = (k & 2) ? -1 : 1;
876 float vertex[3];
877 // for (int i = 0; i < 3; ++i) {
878 // vertex[i] = o(i) + signU * u(i) + signV * v(i) + signN * norm(i);
879 // }
880 vertex[0] = o.X() + signU * u.X() + signV * v.X() + signN * norm.X();
881 vertex[1] = o.Y() + signU * u.Y() + signV * v.Y() + signN * norm.Y();
882 vertex[2] = o.Z() + signU * u.Z() + signV * v.Z() + signN * norm.Z();
883 box->SetVertex(k, vertex);
884 }
885
886 return box;
887}

◆ clearEvent()

void clearEvent ( )

clear event data.

Definition at line 1309 of file EVEVisualization.cc.

1310{
1311 if (!gEve)
1312 return;
1313
1315 for (auto& groupPair : m_groups) {
1316 //store visibility, invalidate pointers
1317 if (groupPair.second.group)
1318 groupPair.second.visible = groupPair.second.group->GetRnrState();
1319 groupPair.second.group = nullptr;
1320 }
1321
1322 m_mcparticleTracks.clear();
1323 m_shownRecohits.clear();
1324 m_tracklist->DestroyElements();
1325
1326 //remove ECL data from event
1327 m_calo3d->SetData(NULL);
1328 m_calo3d->DestroyElements();
1329
1330 //lower energy threshold for ECL
1331 float ecl_threshold = 0.01;
1332 if (m_eclData)
1333 ecl_threshold = m_eclData->GetSliceThreshold(0);
1334
1335 destroyEveElement(m_eclData);
1336 m_eclData = new TEveCaloDataVec(1); //#slices
1337 m_eclData->IncDenyDestroy();
1338 m_eclData->RefSliceInfo(0).Setup("ECL", ecl_threshold, kRed);
1339
1342 destroyEveElement(m_unassignedRecoHits);
1343
1344 gEve->GetSelection()->RemoveElements();
1345 gEve->GetHighlight()->RemoveElements();
1346 //other things are cleaned up by TEve...
1347}
bool m_unassignedRecoHitsVisibility
is m_unassignedRecoHits visible?
void clear()
Remove all contents in map.

◆ makeLines()

void makeLines ( TEveTrack * eveTrack,
const genfit::StateOnPlane * prevState,
const genfit::StateOnPlane * state,
const genfit::AbsTrackRep * rep,
TEvePathMark::EType_e markType,
bool drawErrors,
int markerPos = 1 )
staticprivate

Create hit visualisation for the given options, and add them to 'eveTrack'.

Definition at line 889 of file EVEVisualization.cc.

893{
894 using namespace genfit;
895
896 // Need TVector3 for genfit interface
897 TVector3 pos, dir, oldPos, oldDir;
898 rep->getPosDir(*state, pos, dir);
899 rep->getPosDir(*prevState, oldPos, oldDir);
900
901 double distA = (pos - oldPos).Mag();
902 double distB = distA;
903 if ((pos - oldPos)*oldDir < 0)
904 distA *= -1.;
905 if ((pos - oldPos)*dir < 0)
906 distB *= -1.;
907
908 TEvePathMark mark(
909 markType,
910 TEveVector(pos.X(), pos.Y(), pos.Z()),
911 TEveVector(dir.X(), dir.Y(), dir.Z())
912 );
913 eveTrack->AddPathMark(mark);
914
915
916 if (drawErrors) {
917 const MeasuredStateOnPlane* measuredState;
918 if (markerPos == 0)
919 measuredState = dynamic_cast<const MeasuredStateOnPlane*>(prevState);
920 else
921 measuredState = dynamic_cast<const MeasuredStateOnPlane*>(state);
922
923 if (measuredState != NULL) {
924
925 // step for evaluate at a distance from the original plane
926 ROOT::Math::XYZVector eval;
927 if (markerPos == 0)
928 eval = 0.2 * distA * oldDir;
929 else
930 eval = -0.2 * distB * dir;
931
932
933 // get cov at first plane
934 TMatrixDSym cov;
935 // Need TVector3 for genfit interface
936 TVector3 position, direction;
937 rep->getPosMomCov(*measuredState, position, direction, cov);
938
939 // get eigenvalues & -vectors
940 TMatrixDSymEigen eigen_values(cov.GetSub(0, 2, 0, 2));
941 const TVectorD& ev = eigen_values.GetEigenValues();
942 const TMatrixD& eVec = eigen_values.GetEigenVectors();
943 ROOT::Math::XYZVector eVec1, eVec2;
944 // limit
945 static const double maxErr = 1000.;
946 double ev0 = std::min(ev(0), maxErr);
947 double ev1 = std::min(ev(1), maxErr);
948 double ev2 = std::min(ev(2), maxErr);
949
950 // get two largest eigenvalues/-vectors
951 if (ev0 < ev1 && ev0 < ev2) {
952 eVec1.SetXYZ(eVec(0, 1), eVec(1, 1), eVec(2, 1));
953 eVec1 *= sqrt(ev1);
954 eVec2.SetXYZ(eVec(0, 2), eVec(1, 2), eVec(2, 2));
955 eVec2 *= sqrt(ev2);
956 } else if (ev1 < ev0 && ev1 < ev2) {
957 eVec1.SetXYZ(eVec(0, 0), eVec(1, 0), eVec(2, 0));
958 eVec1 *= sqrt(ev0);
959 eVec2.SetXYZ(eVec(0, 2), eVec(1, 2), eVec(2, 2));
960 eVec2 *= sqrt(ev2);
961 } else {
962 eVec1.SetXYZ(eVec(0, 0), eVec(1, 0), eVec(2, 0));
963 eVec1 *= sqrt(ev0);
964 eVec2.SetXYZ(eVec(0, 1), eVec(1, 1), eVec(2, 1));
965 eVec2 *= sqrt(ev1);
966 }
967
968 if (eVec1.Cross(eVec2).Dot(eval) < 0)
969 eVec2 *= -1;
970 //assert(eVec1.Cross(eVec2)*eval > 0);
971
972 ROOT::Math::XYZVector oldEVec1(eVec1);
973
974 const int nEdges = 24;
975 std::vector<ROOT::Math::XYZVector> vertices;
976
977 vertices.push_back(ROOT::Math::XYZVector(position));
978
979 // vertices at plane
980 for (int i = 0; i < nEdges; ++i) {
981 const double angle = 2 * TMath::Pi() / nEdges * i;
982 vertices.push_back(ROOT::Math::XYZVector(position) + cos(angle)*eVec1 + sin(angle)*eVec2);
983 }
984
985
986
987 SharedPlanePtr newPlane(new DetPlane(*(measuredState->getPlane())));
988 newPlane->setO(position + XYZToTVector(eval));
989
990 MeasuredStateOnPlane stateCopy(*measuredState);
991 try {
992 rep->extrapolateToPlane(stateCopy, newPlane);
993 } catch (Exception& e) {
994 B2ERROR(e.what());
995 return;
996 }
997
998 // get cov at 2nd plane
999 rep->getPosMomCov(stateCopy, position, direction, cov);
1000
1001 // get eigenvalues & -vectors
1002 {
1003 TMatrixDSymEigen eigen_values2(cov.GetSub(0, 2, 0, 2));
1004 const TVectorD& eVal = eigen_values2.GetEigenValues();
1005 const TMatrixD& eVect = eigen_values2.GetEigenVectors();
1006 // limit
1007 ev0 = std::min(eVal(0), maxErr);
1008 ev1 = std::min(eVal(1), maxErr);
1009 ev2 = std::min(eVal(2), maxErr);
1010
1011 // get two largest eigenvalues/-vectors
1012 if (ev0 < ev1 && ev0 < ev2) {
1013 eVec1.SetXYZ(eVect(0, 1), eVect(1, 1), eVect(2, 1));
1014 eVec1 *= sqrt(ev1);
1015 eVec2.SetXYZ(eVect(0, 2), eVect(1, 2), eVect(2, 2));
1016 eVec2 *= sqrt(ev2);
1017 } else if (ev1 < ev0 && ev1 < ev2) {
1018 eVec1.SetXYZ(eVect(0, 0), eVect(1, 0), eVect(2, 0));
1019 eVec1 *= sqrt(ev0);
1020 eVec2.SetXYZ(eVect(0, 2), eVect(1, 2), eVect(2, 2));
1021 eVec2 *= sqrt(ev2);
1022 } else {
1023 eVec1.SetXYZ(eVect(0, 0), eVect(1, 0), eVect(2, 0));
1024 eVec1 *= sqrt(ev0);
1025 eVec2.SetXYZ(eVect(0, 1), eVect(1, 1), eVect(2, 1));
1026 } eVec2 *= sqrt(ev1);
1027 }
1028
1029 if (eVec1.Cross(eVec2).Dot(eval) < 0)
1030 eVec2 *= -1;
1031 //assert(eVec1.Cross(eVec2)*eval > 0);
1032
1033 if (oldEVec1.Dot(eVec1) < 0) {
1034 eVec1 *= -1;
1035 eVec2 *= -1;
1036 }
1037
1038 // vertices at 2nd plane
1039 double angle0 = ROOT::Math::VectorUtil::Angle(eVec1, oldEVec1);
1040 if (eVec1.Dot(eval.Cross(oldEVec1)) < 0)
1041 angle0 *= -1;
1042 for (int i = 0; i < nEdges; ++i) {
1043 const double angle = 2 * TMath::Pi() / nEdges * i - angle0;
1044 vertices.push_back(ROOT::Math::XYZVector(position) + cos(angle)*eVec1 + sin(angle)*eVec2);
1045 }
1046
1047 vertices.push_back(ROOT::Math::XYZVector(position));
1048
1049
1050 TEveTriangleSet* error_shape = new TEveTriangleSet(vertices.size(), nEdges * 2);
1051 for (unsigned int k = 0; k < vertices.size(); ++k) {
1052 error_shape->SetVertex(k, vertices[k].X(), vertices[k].Y(), vertices[k].Z());
1053 }
1054
1055 assert(vertices.size() == 2 * nEdges + 2);
1056
1057 int iTri(0);
1058 for (int i = 0; i < nEdges; ++i) {
1059 //error_shape->SetTriangle(iTri++, 0, i+1, (i+1)%nEdges+1);
1060 error_shape->SetTriangle(iTri++, i + 1, i + 1 + nEdges, (i + 1) % nEdges + 1);
1061 error_shape->SetTriangle(iTri++, (i + 1) % nEdges + 1, i + 1 + nEdges, (i + 1) % nEdges + 1 + nEdges);
1062 //error_shape->SetTriangle(iTri++, 2*nEdges+1, i+1+nEdges, (i+1)%nEdges+1+nEdges);
1063 }
1064
1065 //assert(iTri == nEdges*4);
1066
1067 error_shape->SetMainColor(c_trackColor);
1068 error_shape->SetMainTransparency(25);
1069 eveTrack->AddElement(error_shape);
1070 }
1071 }
1072}
static constexpr auto XYZToTVector
Helper function to convert XYZVector to TVector3.
Definition VectorUtil.h:26
double sqrt(double a)
sqrt for double
Definition beamHelpers.h:28
double eval(const std::vector< double > &spl, const std::vector< double > &vals, double x)
Evaluate spline (zero order or first order) in point x.
Definition tools.h:115

◆ makeTracks()

void makeTracks ( )

Create visual representation of all tracks.

Needs to be called after all hits / tracks are added.

Definition at line 1250 of file EVEVisualization.cc.

1251{
1252 for (auto& mcTrackPair : m_mcparticleTracks) {
1253 MCTrack& mcTrack = mcTrackPair.second;
1254 if (mcTrack.track) {
1255 if (mcTrack.simhits->Size() > 0) {
1256 mcTrack.track->AddElement(mcTrack.simhits);
1257 } else {
1258 //if we don't add it, remove empty collection
1259 destroyEveElement(mcTrack.simhits);
1260 }
1261
1262 TEveElement* parent = m_tracklist;
1263 if (mcTrack.parentParticle) {
1264 const auto& parentIt = m_mcparticleTracks.find(mcTrack.parentParticle);
1265 if (parentIt != m_mcparticleTracks.end()) {
1266 parent = parentIt->second.track;
1267 }
1268 }
1269 parent->AddElement(mcTrack.track);
1270 } else { //add simhits directly
1271 gEve->AddElement(mcTrack.simhits);
1272 }
1273 }
1274 gEve->AddElement(m_tracklist);
1275 m_tracklist->MakeTracks();
1276 m_tracklist->SelectByP(c_minPCut, FLT_MAX); //don't show too many particles by default...
1277
1278 for (size_t i = 0; i < m_options.length(); i++) {
1279 if (m_options.at(i) == 'M') {
1280 m_gftrackpropagator->SetRnrDaughters(true);
1281 m_gftrackpropagator->SetRnrReferences(true);
1282 //m_gftrackpropagator->SetRnrFV(true); //TODO: this crashes?
1283 TMarker m;
1284 m.SetMarkerColor(c_trackMarkerColor);
1285 m.SetMarkerStyle(1); //a single pixel
1286 m.SetMarkerSize(1); //ignored.
1287 m_gftrackpropagator->RefPMAtt() = m;
1288 m_gftrackpropagator->RefFVAtt() = m;
1289 }
1290 }
1291
1292 m_consttrackpropagator->SetMagField(0, 0, -1.5);
1293
1294 m_eclData->DataChanged(); //update limits (Empty() won't work otherwise)
1295 if (!m_eclData->Empty()) {
1296 m_eclData->SetAxisFromBins(0.0,
1297 0.0); //epsilon_x/y = 0 so we don't merge neighboring bins. This avoids some rendering issues with projections of small clusters.
1298 m_calo3d->SetData(m_eclData);
1299 }
1300 gEve->AddElement(m_calo3d);
1301
1304 gEve->AddElement(m_unassignedRecoHits);
1305 }
1306
1307}
static const int c_trackMarkerColor
Color for track markers.

◆ setAssignToPrimaries()

void setAssignToPrimaries ( bool on)
inline

If true, hits created by secondary particles (e.g.

delta electrons) will be assigned to the original primary particle.

Definition at line 266 of file EVEVisualization.h.

266{ m_assignToPrimaries = on; }

◆ setErrScale()

void setErrScale ( double errScale = 1.)

Set the scaling factor for the visualization of track hit errors.

(only affects PXD and SpacePoint hits, which are somewhat small

Definition at line 184 of file EVEVisualization.cc.

184{ m_errorScale = errScale; }

◆ setHideSecondaries()

void setHideSecondaries ( bool on)
inline

If true, secondary MCParticles (and hits created by them) will not be shown.

Definition at line 269 of file EVEVisualization.h.

269{ m_hideSecondaries = on; }

◆ setOptions()

void setOptions ( const std::string & opts)

Set the display options.

The option string lets you steer the way the events are displayed. The following options are available:

'D': Draw detectors. This causes a simple representation for all detectors to be drawn. For planar detectors, this is a plane with the same position and orientation of the real detector plane, but with different size. For wires, this is a tube whose diameter is equal to the value measured by the wire. Spacepoint hits are not affected by this option.

'H': Draw hits. This causes the hits to be visualized. Normally, the size of the hit representation is connected to the covariance matrix of the hit, scaled by the value set in setErrScale which is normally 1. Normally used in connection with 'D'.

'M': Draw track markers. Draw the intersection points between the track and the virtual (and/or real) detector planes. 'P': Draw detector planes. Draws the virtual (and/or real) detector planes.

Definition at line 182 of file EVEVisualization.cc.

182{ m_options = opts; }

◆ showUserData()

void showUserData ( const DisplayData & displayData)

Add user-defined data (labels, points, etc.)

Definition at line 1844 of file EVEVisualization.cc.

1845{
1846 for (const auto& labelPair : displayData.m_labels) {
1847 TEveText* text = new TEveText(labelPair.first.c_str());
1848 text->SetName(labelPair.first.c_str());
1849 text->SetTitle(labelPair.first.c_str());
1850 text->SetMainColor(kGray + 1);
1851 const ROOT::Math::XYZVector& p = labelPair.second;
1852 text->PtrMainTrans()->SetPos(p.X(), p.Y(), p.Z());
1853 addToGroup("DisplayData", text);
1854 }
1855
1856 for (const auto& pointPair : displayData.m_pointSets) {
1857 TEvePointSet* points = new TEvePointSet(pointPair.first.c_str());
1858 points->SetTitle(pointPair.first.c_str());
1859 points->SetMarkerStyle(7);
1860 points->SetMainColor(kGreen);
1861 for (const auto& p : pointPair.second) {
1862 points->SetNextPoint(p.X(), p.Y(), p.Z());
1863 }
1864 addToGroup("DisplayData", points);
1865 }
1866
1867 int randomColor = 2; //primary colours, changing rapidly with index
1868 for (const auto& arrow : displayData.m_arrows) {
1869 const ROOT::Math::XYZVector pos = arrow.start;
1870 const ROOT::Math::XYZVector dir = arrow.end - pos;
1871 TEveArrow* eveArrow = new TEveArrow(dir.X(), dir.Y(), dir.Z(), pos.X(), pos.Y(), pos.Z());
1872 eveArrow->SetName(arrow.name.c_str());
1873 eveArrow->SetTitle(arrow.name.c_str());
1874 int arrowColor = arrow.color;
1875 if (arrowColor == -1) {
1876 arrowColor = randomColor;
1877 randomColor++;
1878 }
1879 eveArrow->SetMainColor(arrowColor);
1880
1881 //add label
1882 TEveText* text = new TEveText(arrow.name.c_str());
1883 text->SetMainColor(arrowColor);
1884 //place label in middle of arrow, with some slight offset
1885 // orthogonal direction is arbitrary, set smallest component zero
1886 ROOT::Math::XYZVector orthogonalDir;
1887 if (std::abs(dir.X()) < std::abs(dir.Y())) {
1888 if (std::abs(dir.X()) < std::abs(dir.Z())) {
1889 orthogonalDir.SetCoordinates(0, dir.Z(), -dir.Y());
1890 } else {
1891 orthogonalDir.SetCoordinates(dir.Y(), -dir.X(), 0);
1892 }
1893 } else {
1894 if (std::abs(dir.Y()) < std::abs(dir.Z())) {
1895 orthogonalDir.SetCoordinates(-dir.Z(), 0, dir.X());
1896 } else {
1897 orthogonalDir.SetCoordinates(dir.Y(), -dir.X(), 0);
1898 }
1899 }
1900 const ROOT::Math::XYZVector& labelPos = pos + 0.5 * dir + 0.1 * orthogonalDir;
1901 text->PtrMainTrans()->SetPos(labelPos.X(), labelPos.Y(), labelPos.Z());
1902 eveArrow->AddElement(text);
1903 addToGroup("DisplayData", eveArrow);
1904 }
1905
1906}
std::vector< Arrow > m_arrows
List of arrows.
std::map< std::string, std::vector< ROOT::Math::XYZVector > > m_pointSets
name -> points map
std::vector< std::pair< std::string, ROOT::Math::XYZVector > > m_labels
text labels (to be shown at a given position).

Member Data Documentation

◆ c_klmClusterColor

const int c_klmClusterColor = getTColorID("Chameleon", 1)
staticprivate

Color for KLMCluster objects.

Definition at line 99 of file EVEVisualization.h.

◆ c_minPCut

double c_minPCut = 0.00
staticconstexprprivate

don't show MCParticles with momentum below this cutoff.

Definition at line 362 of file EVEVisualization.h.

◆ c_recoHitColor

const int c_recoHitColor = getTColorID("Orange", 1)
staticprivate

Color for reco hits.

Definition at line 89 of file EVEVisualization.h.

◆ c_recoTrackColor

const int c_recoTrackColor = getTColorID("Sky Blue", 1)
staticprivate

Color for TrackCandidates.

Definition at line 91 of file EVEVisualization.h.

◆ c_trackColor

const int c_trackColor = getTColorID("Sky Blue", 2)
staticprivate

Color for tracks.

Definition at line 93 of file EVEVisualization.h.

◆ c_trackMarkerColor

const int c_trackMarkerColor = getTColorID("Chameleon", 3)
staticprivate

Color for track markers.

Definition at line 95 of file EVEVisualization.h.

◆ c_unassignedHitColor

const int c_unassignedHitColor = getTColorID("Plum", 1)
staticprivate

Color for unassigned (reco)hits.

Definition at line 97 of file EVEVisualization.h.

◆ m_assignToPrimaries

bool m_assignToPrimaries
private

If true, hits created by secondary particles (e.g.

delta electrons) will be assigned to the original primary particle.

Definition at line 324 of file EVEVisualization.h.

◆ m_bfield

EveVisBField* m_bfield
private

The global magnetic field.

Definition at line 351 of file EVEVisualization.h.

◆ m_calo3d

TEveCalo3D* m_calo3d
private

Object for the energy bar visualisation.

Definition at line 274 of file EVEVisualization.h.

◆ m_consttrackpropagator

TEveTrackPropagator* m_consttrackpropagator
private

Track propagator for CDCTriggerTracks (uses constant B field)

Definition at line 345 of file EVEVisualization.h.

◆ m_drawBackward

bool m_drawBackward = false
private

draw backward in addTrack

Definition at line 377 of file EVEVisualization.h.

◆ m_drawCardinalRep

bool m_drawCardinalRep = true
private

Draw cardinal representation in addTrack.

Definition at line 365 of file EVEVisualization.h.

◆ m_drawErrors

bool m_drawErrors = false
private

Draw errors in addTrack.

Definition at line 368 of file EVEVisualization.h.

◆ m_drawForward

bool m_drawForward = false
private

draw forward in addTrack

Definition at line 374 of file EVEVisualization.h.

◆ m_drawRefTrack

bool m_drawRefTrack = false
private

Draw reference track in addTrack.

Definition at line 371 of file EVEVisualization.h.

◆ m_eclData

TEveCaloDataVec* m_eclData
private

ECL cluster data.

Definition at line 348 of file EVEVisualization.h.

◆ m_errorScale

double m_errorScale
private

Rescale PXD/SVD errors with this factor to ensure visibility.

Definition at line 315 of file EVEVisualization.h.

◆ m_gftrackpropagator

TEveTrackPropagator* m_gftrackpropagator
private

Track propagator for genfit::Tracks (different mainly because of drawing options)

Definition at line 342 of file EVEVisualization.h.

◆ m_groups

std::map<std::string, ElementGroup> m_groups
private

name -> grouping element.

Definition at line 333 of file EVEVisualization.h.

◆ m_hideSecondaries

bool m_hideSecondaries {false}
private

If true, secondary MCParticles (and hits created by them) will not be shown.

Definition at line 327 of file EVEVisualization.h.

327{false};

◆ m_mcparticleTracks

std::map<const MCParticle*, MCTrack> m_mcparticleTracks
private

map MCParticles to MCTrack (so hits can be added to the correct track).

Definition at line 330 of file EVEVisualization.h.

◆ m_options

std::string m_options
private

Option string for genfit::Track visualisation.

See also
setOptions

Definition at line 321 of file EVEVisualization.h.

◆ m_shownRecohits

std::set<const TObject*> m_shownRecohits
private

List of shown recohits (PXDCluster, SVDCluster, CDCHit).

Definition at line 354 of file EVEVisualization.h.

◆ m_tracklist

TEveTrackList* m_tracklist
private

parent object for MC tracks.

Definition at line 336 of file EVEVisualization.h.

◆ m_trackpropagator

TEveTrackPropagator* m_trackpropagator
private

Track propagator for MCParticles.

Definition at line 339 of file EVEVisualization.h.

◆ m_unassignedRecoHits

TEveStraightLineSet* m_unassignedRecoHits = nullptr
private

Unassigned recohits.

Definition at line 357 of file EVEVisualization.h.

◆ m_unassignedRecoHitsVisibility

bool m_unassignedRecoHitsVisibility = true
private

is m_unassignedRecoHits visible?

Definition at line 359 of file EVEVisualization.h.


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