MeasurementCalibrator.h

#ifndef LDMXMEASUREMENTCALIBRATOR_H_
#define LDMXMEASUREMENTCALIBRATOR_H_
 
#include <vector>
 
#include "Acts/Definitions/Algebra.hpp"
#include "Acts/Definitions/TrackParametrization.hpp"
#include "Acts/EventData/SourceLink.hpp"
#include "Acts/Utilities/CalibrationContext.hpp"
#include "Tracking/Event/Measurement.h"
#include "Tracking/Sim/IndexSourceLink.h"
 
 
namespace tracking {
namespace sim {
 
class LdmxMeasurementCalibrator {
 public:
  LdmxMeasurementCalibrator() = default;
 
  // The calibrator needs to access the sim hit container
  LdmxMeasurementCalibrator(
      const std::vector<ldmx::Measurement>& measurements) {
    m_measurements_ = &measurements;
  }
 
  template <typename traj_t>
  void calibrate(const Acts::GeometryContext& /*gctx*/,
                 const Acts::CalibrationContext& /*cctx*/,
                 const Acts::SourceLink& genericSourceLink /*sourceLink*/,
                 typename traj_t::TrackStateProxy trackState) const {
    acts_examples::IndexSourceLink source_link{
        genericSourceLink.get<acts_examples::IndexSourceLink>()};
    assert(m_measurements_ and
           "Undefined measurement container in LdmxMeasurementCalibrator");
    assert((source_link.index() < m_measurements_->size()) and
           "Source link index is outside the container bounds in "
           "LdmxMeasurementCalibrator");
 
    auto meas = m_measurements_->at(source_link.index());
    Acts::Vector2 local_pos{meas.getLocalPosition()[0],
                            meas.getLocalPosition()[1]};
    trackState.allocateCalibrated(2);
    auto ts_cal{trackState.template calibrated<2>()};
    auto ts_cal_cov{trackState.template calibratedCovariance<2>()};
    ts_cal.setZero();
    ts_cal.template head<2>() = local_pos;
    Acts::SquareMatrix2 local_cov;
    local_cov.setZero();
    local_cov(0, 0) = meas.getLocalCovariance()[0];
    local_cov(1, 1) = meas.getLocalCovariance()[1];
    ts_cal_cov.setZero();
    // make tsCalCov 2x2 block the local_cov we just set
    ts_cal_cov.block(0, 0, 2, 2) = local_cov;
 
    trackState.setProjectorSubspaceIndices(
        std::array<Acts::BoundIndices, 2>{Acts::eBoundLoc0, Acts::eBoundLoc1});
    trackState.setUncalibratedSourceLink(Acts::SourceLink{genericSourceLink});
  }
 
  template <typename traj_t>
  void calibrate1d(const Acts::GeometryContext& /*gctx*/,
                   const Acts::CalibrationContext& /*cctx*/,
                   const Acts::SourceLink& genericSourceLink /*sourceLink*/,
                   typename traj_t::TrackStateProxy trackState) const {
    acts_examples::IndexSourceLink source_link{
        genericSourceLink.get<acts_examples::IndexSourceLink>()};
 
    assert(m_measurements_ and
           "Undefined measurement container in LdmxMeasurementCalibrator");
    assert((source_link.index() < m_measurements_->size()) and
           "Source link index is outside the container bounds in "
           "LdmxMeasurementCalibrator");
 
    auto meas = m_measurements_->at(source_link.index());
 
    trackState.allocateCalibrated(1);
    auto ts_cal{trackState.template calibrated<1>()};
    auto ts_cal_cov{trackState.template calibratedCovariance<1>()};
 
    ts_cal.setZero();
    ts_cal(0) = (meas.getLocalPosition())[0];
    ts_cal_cov.setZero();
    ts_cal_cov(0, 0) = (meas.getLocalCovariance())[0];
 
    trackState.setProjectorSubspaceIndices(
        std::array<Acts::BoundIndices, 1>{Acts::eBoundLoc0});
    trackState.setUncalibratedSourceLink(Acts::SourceLink{genericSourceLink});
  }
 
  // Function to test the measurement calibrator
  // It takes an user defined source link and returns the information of the
  // linked measurement
  void test(const Acts::GeometryContext& /*gctx*/,
            const acts_examples::IndexSourceLink& sourceLink) const {
    auto meas = m_measurements_->at(sourceLink.index());
 
    Acts::Vector3 global_pos{meas.getGlobalPosition()[0],
                             meas.getGlobalPosition()[1],
                             meas.getGlobalPosition()[2]};
    std::cout << "Measurement global_position::\n" << global_pos << std::endl;
 
    Acts::Vector2 local_pos{meas.getLocalPosition()[0],
                            meas.getLocalPosition()[1]};
    std::cout << "Measurement local_position::\n" << local_pos << std::endl;
  }
 
 private:
  // use pointer so the calibrator is copyable and default constructible.
  const std::vector<ldmx::Measurement>* m_measurements_ = nullptr;
};
 
}  // namespace sim
}  // namespace tracking
 
#endif