/* ----------------------------------------------------------------------------

 * GTSAM Copyright 2010, Georgia Tech Research Corporation,
 * Atlanta, Georgia 30332-0415
 * All Rights Reserved
 * Authors: Frank Dellaert, et al. (see THANKS for the full author list)

 * See LICENSE for the license information

 * -------------------------------------------------------------------------- */

/**
 *  @file  BetweenFactor.h
 *  @author Frank Dellaert, Viorela Ila
 **/
#pragma once

#include <ostream>

#include <gtsam/nonlinear/NonlinearFactor.h>
#include <gtsam/geometry/concepts.h>
#include <gtsam/base/Testable.h>

namespace gtsam {

  /**
   * A class for a measurement predicted by "between(config[key1],config[key2])"
   * @tparam POSE the Pose type
   * @ingroup slam
   */
  template<class POSE>
  class PoseBetweenFactor: public NoiseModelFactorN<POSE, POSE> {

  private:

    typedef PoseBetweenFactor<POSE> This;
    typedef NoiseModelFactorN<POSE, POSE> Base;

    POSE measured_; /** The measurement */
    std::optional<POSE> body_P_sensor_; ///< The pose of the sensor in the body frame

    /** concept check by type */
    GTSAM_CONCEPT_TESTABLE_TYPE(POSE)
    GTSAM_CONCEPT_POSE_TYPE(POSE)
  public:

    // Provide access to the Matrix& version of evaluateError:
    using Base::evaluateError;

    // shorthand for a smart pointer to a factor
    typedef typename std::shared_ptr<PoseBetweenFactor> shared_ptr;

    /** default constructor - only use for serialization */
    PoseBetweenFactor() {}

    /** Constructor */
    PoseBetweenFactor(Key key1, Key key2, const POSE& measured,
        const SharedNoiseModel& model, std::optional<POSE> body_P_sensor = {}) :
      Base(model, key1, key2), measured_(measured), body_P_sensor_(body_P_sensor) {
    }

    ~PoseBetweenFactor() override {}

    /// @return a deep copy of this factor
    gtsam::NonlinearFactor::shared_ptr clone() const override {
      return std::static_pointer_cast<gtsam::NonlinearFactor>(
          gtsam::NonlinearFactor::shared_ptr(new This(*this))); }

    /** implement functions needed for Testable */

    /** print */
    void print(const std::string& s, const KeyFormatter& keyFormatter = DefaultKeyFormatter) const override {
      std::cout << s << "BetweenFactor("
          << keyFormatter(this->key1()) << ","
          << keyFormatter(this->key2()) << ")\n";
      measured_.print("  measured: ");
      if(this->body_P_sensor_)
        this->body_P_sensor_->print("  sensor pose in body frame: ");
      this->noiseModel_->print("  noise model: ");
    }

    /** equals */
    bool equals(const NonlinearFactor& expected, double tol=1e-9) const override {
      const This *e =  dynamic_cast<const This*> (&expected);
      return e != nullptr
          && Base::equals(*e, tol)
          && this->measured_.equals(e->measured_, tol)
          && ((!body_P_sensor_ && !e->body_P_sensor_) || (body_P_sensor_ && e->body_P_sensor_ && body_P_sensor_->equals(*e->body_P_sensor_)));
    }

    /** implement functions needed to derive from Factor */

    /** vector of errors */
    Vector evaluateError(const POSE& p1, const POSE& p2,
        OptionalMatrixType H1, OptionalMatrixType H2) const override {
      if(body_P_sensor_) {
        POSE hx;
        if(H1 || H2) {
          Matrix H3;
          hx = p1.compose(*body_P_sensor_,H3).between(p2.compose(*body_P_sensor_), H1, H2); // h(x)
          if(H1) (*H1) *= H3;
          if(H2) (*H2) *= H3;
        } else {
          hx = p1.compose(*body_P_sensor_).between(p2.compose(*body_P_sensor_)); // h(x)
        }
        // manifold equivalent of h(x)-z -> log(z,h(x))
        return measured_.localCoordinates(hx);
      } else {
        POSE hx = p1.between(p2, H1, H2); // h(x)
        // manifold equivalent of h(x)-z -> log(z,h(x))
        return measured_.localCoordinates(hx);
      }
    }

    /** return the measured */
    const POSE& measured() const {
      return measured_;
    }

  private:

#if GTSAM_ENABLE_BOOST_SERIALIZATION
    /** Serialization function */
    friend class boost::serialization::access;
    template<class ARCHIVE>
    void serialize(ARCHIVE & ar, const unsigned int /*version*/) {
      ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(Base);
      ar & BOOST_SERIALIZATION_NVP(measured_);

    }
#endif
  }; // \class PoseBetweenFactor

} /// namespace gtsam