MeasurementCalibrator.h

#ifndef LDMXMEASUREMENTCALIBRATOR_H_
#define LDMXMEASUREMENTCALIBRATOR_H_
 
#include <vector>
 
#include "Acts/Definitions/Algebra.hpp"
#include "Acts/EventData/MultiTrajectory.hpp"
#include "Acts/EventData/SourceLink.hpp"
#include "Acts/EventData/VectorMultiTrajectory.hpp"
#include "Acts/Utilities/CalibrationContext.hpp"
#include "Tracking/Event/Measurement.h"
#include "Tracking/Sim/IndexSourceLink.h"
#include "Tracking/Sim/LdmxSpacePoint.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 {
    ActsExamples::IndexSourceLink sourceLink{
        genericSourceLink.get<ActsExamples::IndexSourceLink>()};
    assert(m_measurements and
           "Undefined measurement container in LdmxMeasurementCalibrator");
    assert((sourceLink.index() < m_measurements->size()) and
           "Source link index is outside the container bounds in "
           "LdmxMeasurementCalibrator");
 
    auto meas = m_measurements->at(sourceLink.index());
    Acts::Vector2 local_pos{meas.getLocalPosition()[0],
                            meas.getLocalPosition()[1]};
    auto tsCal{trackState.template calibrated<2>()};
    auto tsCalCov{trackState.template calibratedCovariance<2>()};
    tsCal.setZero();
    tsCal.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];
    tsCalCov.setZero();
    // make tsCalCov 2x2 block the local_cov we just set
    tsCalCov.block(0, 0, 2, 2) = local_cov;
 
    Acts::ActsMatrix<2, 6> projector;
    projector.setZero();
    projector(0, 0) = 1.;
    projector(1, 1) = 1.;
 
    trackState.setProjector(projector);
  }
 
  template <typename traj_t>
  void calibrate_1d(const Acts::GeometryContext& /*gctx*/,
                    const Acts::CalibrationContext& /*cctx*/,
                    const Acts::SourceLink& genericSourceLink /*sourceLink*/,
                    typename traj_t::TrackStateProxy trackState) const {
    ActsExamples::IndexSourceLink sourceLink{
        genericSourceLink.get<ActsExamples::IndexSourceLink>()};
 
    assert(m_measurements and
           "Undefined measurement container in LdmxMeasurementCalibrator");
    assert((sourceLink.index() < m_measurements->size()) and
           "Source link index is outside the container bounds in "
           "LdmxMeasurementCalibrator");
 
    auto meas = m_measurements->at(sourceLink.index());
 
    trackState.allocateCalibrated(1);
    auto tsCal{trackState.template calibrated<1>()};
    auto tsCalCov{trackState.template calibratedCovariance<1>()};
 
    tsCal.setZero();
    tsCal(0) = (meas.getLocalPosition())[0];
    tsCalCov.setZero();
    tsCalCov(0, 0) = (meas.getLocalCovariance())[0];
 
    Acts::ActsMatrix<2, 6> projector;
    projector.setZero();
    projector(0, 0) = 1.;
    projector(1, 1) = 1.;
    trackState.setProjector(projector.row(0));
    trackState.setUncalibratedSourceLink(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 ActsExamples::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