HWTrackingDetectorEventDisplay.hpp

Go to the documentation of this file.

#include "herd/analysis/algorithms/display/HWTrackingDetectorEventDisplay.h"
 
namespace Herd {
 
template <class TrackingGeoParams>
void HWTrackingDetectorEventDisplay::ImportClusters(const ClustersColl &clColl,
                                                    const SidesArray<std::vector<TrackingGeoParams>> &geoParamsColl) {
 
  if (TrkHits3D)
    TrkHits3D = new TEveStraightLineSet("Clusters");
 
  TrkHitsXY->Reset();
  TrkHitsXZ->Reset();
  TrkHitsYZ->Reset();
 
  TEvePointSet *TrkHits = nullptr;
 
  for (auto side : RefFrame::Sides) {
    auto &sideColl = clColl.at(side);
 
    for (size_t iLayer = 0; iLayer < sideColl.size(); iLayer++) {
      auto geoParams = geoParamsColl.at(side).at(iLayer);
 
      if (RefFrame::ToAxis(geoParams.NormalDirection()) != RefFrame::Axis::Y &&
          RefFrame::ToAxis(geoParams.SegmentationDirections().first) != RefFrame::Axis::Y) {
        TrkHits = TrkHitsXZ;
      } else if (RefFrame::ToAxis(geoParams.NormalDirection()) != RefFrame::Axis::X &&
                 RefFrame::ToAxis(geoParams.SegmentationDirections().first) != RefFrame::Axis::X) {
        TrkHits = TrkHitsYZ;
      } else if (RefFrame::ToAxis(geoParams.NormalDirection()) != RefFrame::Axis::Z &&
                 RefFrame::ToAxis(geoParams.SegmentationDirections().first) != RefFrame::Axis::Z) {
        TrkHits = TrkHitsXY;
      }
 
      for (auto &cluster : sideColl.at(iLayer)) {
        auto longitudinalRange = LineRange(cluster, geoParams);
        auto clusterPos = cluster.COGPosition(geoParams);
        TrkHits->SetNextPoint(clusterPos[RefFrame::Coo::X], clusterPos[RefFrame::Coo::Y], clusterPos[RefFrame::Coo::Z]);
 
        auto lineStart = clusterPos, lineEnd = clusterPos;
        lineStart[RefFrame::ToCoo(geoParams.SegmentationDirections().second)] = longitudinalRange.first;
        lineEnd[RefFrame::ToCoo(geoParams.SegmentationDirections().second)] = longitudinalRange.second;
        TrkHits3D->AddLine(lineStart[RefFrame::Coo::X], lineStart[RefFrame::Coo::Y], lineStart[RefFrame::Coo::Z],
                           lineEnd[RefFrame::Coo::X], lineEnd[RefFrame::Coo::Y], lineEnd[RefFrame::Coo::Z]);
      }
    }
  }
 
  return;
}
 
template <>
inline std::pair<float, float> HWTrackingDetectorEventDisplay::LineRange(const Cluster &cluster,
                                                                         const ScdGeoParams &geoParams) {
  RefFrame::Axis axis = RefFrame::ToAxis(geoParams.SegmentationDirections().second);
 
  // get ladder size and position
  float elementSize =
      geoParams.WaferSize()[axis] *
      (geoParams.LadderBonding() == ScdGeoParams::BondingType::NONE ? 1 : geoParams.NWafersPerBondedLadder()[axis]);
  float elementPos = geoParams.LadderPosition(
      geoParams.LadderIDFromStripID(cluster.Hits()[cluster.SeedHitIndex()].VolumeID()))[RefFrame::ToCoo(axis)];
 
  return {elementPos - 0.5 * elementSize, elementPos + 0.5 * elementSize};
}
 
template <>
inline std::pair<float, float> HWTrackingDetectorEventDisplay::LineRange(const Cluster &cluster,
                                                                         const FitGeoParams &geoParams) {
  RefFrame::Axis axis = RefFrame::ToAxis(geoParams.SegmentationDirections().second);
 
  // get ladder size and position
  float elementSize = geoParams.MatSize()[axis];
  float elementPos = geoParams.MatPosition(
      geoParams.MatIDFromChannelID(cluster.Hits()[cluster.SeedHitIndex()].VolumeID()))[RefFrame::ToCoo(axis)];
 
  return {elementPos - 0.5 * elementSize, elementPos + 0.5 * elementSize};
}
 
} // namespace Herd