//
// Copyright (c) 2015-2020 CNRS INRIA
// Copyright (c) 2016 Wandercraft, 86 rue de Paris 91400 Orsay, France.
//

#pragma once

// IWYU pragma: private, include "pinocchio/multibody/joint-motion-subspace.hpp"

#ifdef PINOCCHIO_LSP
  #undef PINOCCHIO_LSP
  #include "pinocchio/multibody/joint-motion-subspace.hpp"
#endif // PINOCCHIO_LSP

// S   : v   \in M^6              -> v_J \in lie(Q) ~= R^nv
// S^T : f_J \in lie(Q)^* ~= R^nv -> f    \in F^6

#define PINOCCHIO_CONSTRAINT_TYPEDEF_GENERIC(DERIVED, TYPENAME)                                    \
  typedef TYPENAME traits<DERIVED>::Scalar Scalar;                                                 \
  typedef TYPENAME traits<DERIVED>::JointMotion JointMotion;                                       \
  typedef TYPENAME traits<DERIVED>::JointForce JointForce;                                         \
  typedef TYPENAME traits<DERIVED>::DenseBase DenseBase;                                           \
  typedef TYPENAME traits<DERIVED>::MatrixReturnType MatrixReturnType;                             \
  typedef TYPENAME traits<DERIVED>::ConstMatrixReturnType ConstMatrixReturnType;                   \
  static constexpr int LINEAR = traits<DERIVED>::LINEAR;                                           \
  static constexpr int ANGULAR = traits<DERIVED>::ANGULAR;                                         \
  static constexpr int Options = traits<DERIVED>::Options;

#define PINOCCHIO_CONSTRAINT_TYPEDEF_TPL(DERIVED)                                                  \
  PINOCCHIO_CONSTRAINT_TYPEDEF_GENERIC(DERIVED, typename)
#define PINOCCHIO_CONSTRAINT_TYPEDEF(DERIVED)                                                      \
  PINOCCHIO_CONSTRAINT_TYPEDEF_GENERIC(DERIVED, PINOCCHIO_EMPTY_ARG)

namespace pinocchio
{

  /// \brief Return type of the Constraint::Transpose * Force operation
  template<class ConstraintDerived, typename Force>
  struct ConstraintForceOp
  {
    typedef UndefinedReturnType ReturnType;
  };

  /// \brief Return type of the Constraint::Transpose * ForceSet operation
  template<class ConstraintDerived, typename ForceSet>
  struct ConstraintForceSetOp
  {
    typedef UndefinedReturnType ReturnType;
  };

  template<class Derived>
  class JointMotionSubspaceBase : public NumericalBase<Derived>
  {
  protected:
    PINOCCHIO_CONSTRAINT_TYPEDEF_TPL(Derived)

  public:
    Derived & derived()
    {
      return *static_cast<Derived *>(this);
    }
    const Derived & derived() const
    {
      return *static_cast<const Derived *>(this);
    }

    template<typename VectorLike>
    JointMotion operator*(const Eigen::MatrixBase<VectorLike> & vj) const
    {
      return derived().__mult__(vj);
    }

    MatrixReturnType matrix()
    {
      return derived().matrix_impl();
    }
    ConstMatrixReturnType matrix() const
    {
      return derived().matrix_impl();
    }

    int nv() const
    {
      return derived().nv_impl();
    }

    static int rows()
    {
      return 6;
    }
    int cols() const
    {
      return nv();
    }

    template<class OtherDerived>
    bool isApprox(
      const JointMotionSubspaceBase<OtherDerived> & other,
      const Scalar & prec = Eigen::NumTraits<Scalar>::dummy_precision()) const
    {
      return matrix().isApprox(other.matrix(), prec);
    }

    void disp(std::ostream & os) const
    {
      derived().disp_impl(os);
    }
    friend std::ostream & operator<<(std::ostream & os, const JointMotionSubspaceBase<Derived> & X)
    {
      X.disp(os);
      return os;
    }

    typename SE3GroupAction<Derived>::ReturnType se3Action(const SE3Tpl<Scalar, Options> & m) const
    {
      return derived().se3Action(m);
    }

    typename SE3GroupAction<Derived>::ReturnType
    se3ActionInverse(const SE3Tpl<Scalar, Options> & m) const
    {
      return derived().se3ActionInverse(m);
    }

    template<typename MotionDerived>
    typename MotionAlgebraAction<Derived, MotionDerived>::ReturnType
    motionAction(const MotionDense<MotionDerived> & v) const
    {
      return derived().motionAction(v);
    }

    bool operator==(const JointMotionSubspaceBase<Derived> & other) const
    {
      return derived().isEqual(other.derived());
    }

  }; // class JointMotionSubspaceBase

  ///  \brief Operation Y * S used in the CRBA algorithm for instance
  template<typename Scalar, int Options, typename ConstraintDerived>
  typename MultiplicationOp<InertiaTpl<Scalar, Options>, ConstraintDerived>::ReturnType operator*(
    const InertiaTpl<Scalar, Options> & Y,
    const JointMotionSubspaceBase<ConstraintDerived> & constraint)
  {
    return impl::LhsMultiplicationOp<InertiaTpl<Scalar, Options>, ConstraintDerived>::run(
      Y, constraint.derived());
  }

  ///  \brief Operation Y_matrix * S used in the ABA algorithm for instance
  template<typename MatrixDerived, typename ConstraintDerived>
  typename MultiplicationOp<Eigen::MatrixBase<MatrixDerived>, ConstraintDerived>::ReturnType
  operator*(
    const Eigen::MatrixBase<MatrixDerived> & Y,
    const JointMotionSubspaceBase<ConstraintDerived> & constraint)
  {
    return impl::LhsMultiplicationOp<Eigen::MatrixBase<MatrixDerived>, ConstraintDerived>::run(
      Y.derived(), constraint.derived());
  }

  namespace details
  {
    template<typename Constraint>
    struct StDiagonalMatrixSOperation
    {
      typedef typename traits<Constraint>::StDiagonalMatrixSOperationReturnType ReturnType;
      typedef typename traits<Constraint>::ReducedSquaredMatrix ReducedSquaredMatrix;

      static ReturnType run(const JointMotionSubspaceBase<Constraint> & /*constraint*/)
      {
        return ReducedSquaredMatrix::Identity(Constraint::NV, Constraint::NV);
      }
    };
  } // namespace details

  template<class ConstraintDerived>
  struct JointMotionSubspaceTransposeBase
  {
    typedef typename traits<ConstraintDerived>::StDiagonalMatrixSOperationReturnType
      StDiagonalMatrixSOperationReturnType;
  };

  template<class ConstraintDerived>
  typename JointMotionSubspaceTransposeBase<ConstraintDerived>::StDiagonalMatrixSOperationReturnType
  operator*(
    const JointMotionSubspaceTransposeBase<ConstraintDerived> & /*S_transpose*/,
    const JointMotionSubspaceBase<ConstraintDerived> & S)
  {
    return details::StDiagonalMatrixSOperation<ConstraintDerived>::run(S.derived());
  }

} // namespace pinocchio
