calibration.hpp

#ifndef _CHRONOPTICS_TOF_CALIBRATION_HPP_
#define _CHRONOPTICS_TOF_CALIBRATION_HPP_
 
#include <chronoptics/tof/calibration.h>
 
#include <chronoptics/tof/base.hpp>
 
namespace chronoptics {
namespace tof {
 
/** This class contains all the calibration information
*/
class Calibration : public detail::Base<tof_calibration, tof_calibration_delete> {
 public:
  /** Construct from pointer */
  Calibration(tof_calibration_t ptr = nullptr) {
    this->ptr_ = ptr;
  }
 
  /** Read calibration from disk
   * @param file_location Location of calibration file
   */
  Calibration(StringView file_location) {
    this->ptr_ = tof_calibration_new_from_disk(file_location, TOF_ERROR_HANDLER{});
  }
 
  /** Write calibration to disk
   * @param file_location Location to save the calibration file to
   */
  void write(StringView file_location) const {
    return tof_calibration_write(this->ptr_, file_location, TOF_ERROR_HANDLER{});
  }
 
  /** The the calibrated frequencies in the calibration file
   * @return The calibrated modulation frequencies
   */
  std::vector<float> get_calibrated_frequencies() const {
    size_t size = tof_calibration_get_calibrated_frequencies(this->ptr_, nullptr, 0, TOF_ERROR_HANDLER{});
    std::vector<float> vec(size);
    size = tof_calibration_get_calibrated_frequencies(this->ptr_, vec.data(), vec.size(), TOF_ERROR_HANDLER{});
    return vec;
  }
 
  /** Get focal length x
   * @return Focal length
   */
  double get_focal_length_x() const {
    return tof_calibration_get_focal_length_x(this->ptr_, TOF_ERROR_HANDLER{});
  }
 
  /** Get focal length y
   * @return Focal length
   */
  double get_focal_length_y() const {
    return tof_calibration_get_focal_length_y(this->ptr_, TOF_ERROR_HANDLER{});
  }
 
  /** Get principal point x
   * @return Principal point
   */
  double get_principal_point_x() const {
    return tof_calibration_get_principal_point_x(this->ptr_, TOF_ERROR_HANDLER{});
  }
 
  /** Get principal point y
   * @return Principal point
   */
  double get_principal_point_y() const {
    return tof_calibration_get_principal_point_y(this->ptr_, TOF_ERROR_HANDLER{});
  }
 
  /** Get the camera matrix of the depth sensor and lens
   * @return Camera matrix, 3x3 matrix
   */
  std::array<double, 9> get_depth_camera_matrix() {
    std::array<double, 9> array;
    auto data = tof_calibration_get_depth_camera_matrix(this->ptr_, TOF_ERROR_HANDLER{});
    std::copy(data, data + array.size(), array.data());
    return array;
  }
 
  /** Get the distortion coefficients of the depth lens
   * @return Distortion coefficients, 5x1 matrix
   */
  std::array<double, 5> get_depth_distortion_coefficients() {
    std::array<double, 5> array;
    auto data = tof_calibration_get_depth_distortion_coefficients(this->ptr_, TOF_ERROR_HANDLER{});
    std::copy(data, data + array.size(), array.data());
    return array;
  }
 
  /** Get the camera matrix of the rgb sensor and lens
   * @return Camera matrix, 3x3 matrix
   */
  std::array<double, 9> get_rgb_camera_matrix() {
    std::array<double, 9> array;
    auto data = tof_calibration_get_rgb_camera_matrix(this->ptr_, TOF_ERROR_HANDLER{});
    std::copy(data, data + array.size(), array.data());
    return array;
  }
 
  /** Get the distortion coefficients of the rgb lens
   * @return Distortion coefficients, 5x1 matrix
   */
  std::array<double, 5> get_rgb_distortion_coefficients() {
    std::array<double, 5> array;
    auto data = tof_calibration_get_rgb_distortion_coefficients(this->ptr_, TOF_ERROR_HANDLER{});
    std::copy(data, data + array.size(), array.data());
    return array;
  }
 
};
 
} // tof
} // chronoptics
 
#endif