TrackFiducialVolume.cpp

Go to the documentation of this file.

#include "NSL/Tracker/TrackFiducialVolume.h"

// NAIA headers
#include <Containers/Utils.hpp>

namespace NSL {
namespace Selections {
namespace Track {

// Beware, for L9 the check is X only, not R
constexpr std::array<float, 9> fid_R = {62, 62, 46, 46, 46, 46, 46, 46, 43};
constexpr std::array<float, 9> fid_Y = {47, 40, 44, 44, 36, 36, 44, 44, 29};

bool IsInsideFiducial(unsigned int ilayer, NAIA::TrTrack::Fit fitType, NAIA::TrTrack::Span spanType,
                      NAIA::Event &event) {

  assert(ilayer > 0 && ilayer < 10);
  auto layer = static_cast<NAIA::TrTrack::FitPositionHeight>(ilayer-1);

  float x = event.trTrackBase->TrTrackFitPos[layer][fitType][spanType][NAIA::TrTrack::Side::X];
  float y = event.trTrackBase->TrTrackFitPos[layer][fitType][spanType][NAIA::TrTrack::Side::Y];

  if (layer == NAIA::TrTrack::FitPositionHeight::Layer9) {
    if (std::fabs(x) > fid_R[layer] || std::fabs(y) > fid_Y[layer])
      return false;
  } else {
    if (std::sqrt(x * x + y * y) > fid_R[layer] || std::fabs(y) > fid_Y[layer])
      return false;
  }

  return true;
}

InnerFiducialVolume::InnerFiducialVolume(NAIA::TrTrack::Fit fitType, NAIA::TrTrack::Span spanType) {
  m_matcher = std::make_shared<boolMatcher>([=](Event &event) {
    if (!event.trTrackBase->FitIDExists(fitType, spanType))
      return false;

    unsigned int nhits = 0;

    for (unsigned int layer = 2; layer < 9; ++layer)
      if (IsInsideFiducial(layer, fitType, spanType, event))
        nhits++;
    if (nhits < 5)
      return false;

    if (!IsInsideFiducial(2, fitType, spanType, event))
      return false;
    if (!(IsInsideFiducial(3, fitType, spanType, event) || IsInsideFiducial(4, fitType, spanType, event)))
      return false;
    if (!(IsInsideFiducial(5, fitType, spanType, event) || IsInsideFiducial(6, fitType, spanType, event)))
      return false;
    if (!(IsInsideFiducial(7, fitType, spanType, event) || IsInsideFiducial(8, fitType, spanType, event)))
      return false;

    return true;
  });
}

L1FiducialVolume::L1FiducialVolume(NAIA::TrTrack::Fit fitType, NAIA::TrTrack::Span spanType) {
  m_matcher = std::make_shared<boolMatcher>([=](Event &event) {
    if (!event.trTrackBase->FitIDExists(fitType, spanType))
      return false;

    return IsInsideFiducial(1, fitType, spanType, event);
  });
}

L9FiducialVolume::L9FiducialVolume(NAIA::TrTrack::Fit fitType, NAIA::TrTrack::Span spanType) {
  m_matcher = std::make_shared<boolMatcher>([=](Event &event) {
    if (!event.trTrackBase->FitIDExists(fitType, spanType))
      return false;

    return IsInsideFiducial(9, fitType, spanType, event);
  });
}

} // namespace Track
} // namespace Selections
} // namespace NSL