GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-07-27 23:32:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve a bit AutoDiffVector, but it still not working

Browse Source
This commit is contained in:
Gael Guennebaud 2009-11-06 11:34:58 +01:00
parent 6647a58847
commit 5dc02fe5e9
1 changed files with 74 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

84
unsupported/Eigen/src/AutoDiff/AutoDiffVector.h
View File

@ -48,7 +48,11 @@ template<typename ValueType, typename JacobianType>
class AutoDiffVector class AutoDiffVector
{ {
public: public:
typedef typename ei_traits<ValueType>::Scalar Scalar; //typedef typename ei_traits<ValueType>::Scalar Scalar;
typedef typename ei_traits<ValueType>::Scalar BaseScalar;
typedef AutoDiffScalar<Matrix<BaseScalar,JacobianType::RowsAtCompileTime,1> > ActiveScalar;
typedef ActiveScalar Scalar;
typedef AutoDiffScalar<typename JacobianType::ColXpr> CoeffType;
inline AutoDiffVector() {} inline AutoDiffVector() {}
@ -58,6 +62,22 @@ class AutoDiffVector
m_jacobian.setZero(); m_jacobian.setZero();
} }
CoeffType operator[] (int i) { return CoeffType(m_values[i], m_jacobian.col(i)); }
const CoeffType operator[] (int i) const { return CoeffType(m_values[i], m_jacobian.col(i)); }
CoeffType operator() (int i) { return CoeffType(m_values[i], m_jacobian.col(i)); }
const CoeffType operator() (int i) const { return CoeffType(m_values[i], m_jacobian.col(i)); }
CoeffType coeffRef(int i) { return CoeffType(m_values[i], m_jacobian.col(i)); }
const CoeffType coeffRef(int i) const { return CoeffType(m_values[i], m_jacobian.col(i)); }
int size() const { return m_values.size(); }
// FIXME here we could return an expression of the sum
Scalar sum() const { /*std::cerr << "sum \n\n";*/ /*std::cerr << m_jacobian.rowwise().sum() << "\n\n";*/ return Scalar(m_values.sum(), m_jacobian.rowwise().sum()); }
inline AutoDiffVector(const ValueType& values, const JacobianType& jac) inline AutoDiffVector(const ValueType& values, const JacobianType& jac)
: m_values(values), m_jacobian(jac) : m_values(values), m_jacobian(jac)
{} {}
@ -72,7 +92,7 @@ class AutoDiffVector
{} {}
template<typename OtherValueType, typename OtherJacobianType> template<typename OtherValueType, typename OtherJacobianType>
inline AutoDiffScalar& operator=(const AutoDiffVector<OtherValueType, OtherJacobianType>& other) inline AutoDiffVector& operator=(const AutoDiffVector<OtherValueType, OtherJacobianType>& other)
{ {
m_values = other.values(); m_values = other.values();
m_jacobian = other.jacobian(); m_jacobian = other.jacobian();
@ -94,20 +114,20 @@ class AutoDiffVector
template<typename OtherValueType,typename OtherJacobianType> template<typename OtherValueType,typename OtherJacobianType>
inline const AutoDiffVector< inline const AutoDiffVector<
CwiseBinaryOp<ei_scalar_sum_op<Scalar>,ValueType,OtherValueType> > typename MakeCwiseBinaryOp<ei_scalar_sum_op<BaseScalar>,ValueType,OtherValueType>::Type,
CwiseBinaryOp<ei_scalar_sum_op<Scalar>,JacobianType,OtherJacobianType> > typename MakeCwiseBinaryOp<ei_scalar_sum_op<BaseScalar>,JacobianType,OtherJacobianType>::Type >
operator+(const AutoDiffScalar<OtherDerType>& other) const operator+(const AutoDiffVector<OtherValueType,OtherJacobianType>& other) const
{ {
return AutoDiffVector< return AutoDiffVector<
CwiseBinaryOp<ei_scalar_sum_op<Scalar>,ValueType,OtherValueType> > typename MakeCwiseBinaryOp<ei_scalar_sum_op<BaseScalar>,ValueType,OtherValueType>::Type,
CwiseBinaryOp<ei_scalar_sum_op<Scalar>,JacobianType,OtherJacobianType> >( typename MakeCwiseBinaryOp<ei_scalar_sum_op<BaseScalar>,JacobianType,OtherJacobianType>::Type >(
m_values + other.values(), m_values + other.values(),
m_jacobian + other.jacobian()); m_jacobian + other.jacobian());
} }
template<typename OtherValueType, typename OtherJacobianType> template<typename OtherValueType, typename OtherJacobianType>
inline AutoDiffVector& inline AutoDiffVector&
operator+=(const AutoDiffVector<OtherValueType,OtherDerType>& other) operator+=(const AutoDiffVector<OtherValueType,OtherJacobianType>& other)
{ {
m_values += other.values(); m_values += other.values();
m_jacobian += other.jacobian(); m_jacobian += other.jacobian();
@ -116,20 +136,20 @@ class AutoDiffVector
template<typename OtherValueType,typename OtherJacobianType> template<typename OtherValueType,typename OtherJacobianType>
inline const AutoDiffVector< inline const AutoDiffVector<
CwiseBinaryOp<ei_scalar_difference_op<Scalar>,ValueType,OtherValueType> > typename MakeCwiseBinaryOp<ei_scalar_difference_op<Scalar>,ValueType,OtherValueType>::Type,
CwiseBinaryOp<ei_scalar_difference_op<Scalar>,JacobianType,OtherJacobianType> > typename MakeCwiseBinaryOp<ei_scalar_difference_op<Scalar>,JacobianType,OtherJacobianType>::Type >
operator-(const AutoDiffScalar<OtherDerType>& other) const operator-(const AutoDiffVector<OtherValueType,OtherJacobianType>& other) const
{ {
return AutoDiffVector< return AutoDiffVector<
CwiseBinaryOp<ei_scalar_difference_op<Scalar>,ValueType,OtherValueType> > typename MakeCwiseBinaryOp<ei_scalar_difference_op<Scalar>,ValueType,OtherValueType>::Type,
CwiseBinaryOp<ei_scalar_difference_op<Scalar>,JacobianType,OtherJacobianType> >( typename MakeCwiseBinaryOp<ei_scalar_difference_op<Scalar>,JacobianType,OtherJacobianType>::Type >(
m_values - other.values(), m_values - other.values(),
m_jacobian - other.jacobian()); m_jacobian - other.jacobian());
} }
template<typename OtherValueType, typename OtherJacobianType> template<typename OtherValueType, typename OtherJacobianType>
inline AutoDiffVector& inline AutoDiffVector&
operator-=(const AutoDiffVector<OtherValueType,OtherDerType>& other) operator-=(const AutoDiffVector<OtherValueType,OtherJacobianType>& other)
{ {
m_values -= other.values(); m_values -= other.values();
m_jacobian -= other.jacobian(); m_jacobian -= other.jacobian();
@ -137,37 +157,37 @@ class AutoDiffVector
} }
inline const AutoDiffVector< inline const AutoDiffVector<
CwiseUnaryOp<ei_scalar_opposite_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_opposite_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_opposite_op<Scalar>, JacobianType> > typename MakeCwiseUnaryOp<ei_scalar_opposite_op<Scalar>, JacobianType>::Type >
operator-() const operator-() const
{ {
return AutoDiffVector< return AutoDiffVector<
CwiseUnaryOp<ei_scalar_opposite_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_opposite_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_opposite_op<Scalar>, JacobianType> >( typename MakeCwiseUnaryOp<ei_scalar_opposite_op<Scalar>, JacobianType>::Type >(
-m_values, -m_values,
-m_jacobian); -m_jacobian);
} }
inline const AutoDiffVector< inline const AutoDiffVector<
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType> > typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType>::Type>
operator*(const Scalar& other) const operator*(const BaseScalar& other) const
{ {
return AutoDiffVector< return AutoDiffVector<
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType> >( typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType>::Type >(
m_values * other, m_values * other,
(m_jacobian * other)); m_jacobian * other);
} }
friend inline const AutoDiffVector< friend inline const AutoDiffVector<
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType> > typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType>::Type >
operator*(const Scalar& other, const AutoDiffVector& v) operator*(const Scalar& other, const AutoDiffVector& v)
{ {
return AutoDiffVector< return AutoDiffVector<
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType> typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, ValueType>::Type,
CwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType> >( typename MakeCwiseUnaryOp<ei_scalar_multiple_op<Scalar>, JacobianType>::Type >(
v.values() * other, v.values() * other,
v.jacobian() * other); v.jacobian() * other);
} }