HWTrackingDetectorEventDisplay.hpp

Go to the documentation of this file.

#include "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 = GetLineRange(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::GetLineRange(const Cluster &cluster,
                                                                            const SiliconDetectorGeoParams &geoParams) {
  RefFrame::Axis axis = RefFrame::ToAxis(geoParams.SegmentationDirections().second);

  // get ladder size and position
  float elementSize = geoParams.WaferSize()[axis] * geoParams.NWafersPerLadder()[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::GetLineRange(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