Próbuję napisać nowe wyrażenie Eigen po najnowszej dokumentacji https://eigen.tuxfamily.org/dox-devel/TopicNewExpressionType.html. Zasadniczo to, czego chcę, jest częścią funkcji przekształcania, której wciąż nie ma w Eigen. Tak więc chop_expr (tutaj wyrażenie wektorów własnych) powinno przekształcić wektor wejściowy na macierz n razy.Pisanie nowego wyrażenia Eigen
Niestety, to, co zaimplementowałem, nie działa z wyrażeniami przydzielonymi do sterty, na przykład poniższy kod nie działa, ale po zmianie MAXV na 10, wszystko idzie idealnie.
Inną kwestią jest to o
enum {Flags = EvalBeforeNestingBit}
znalazłem, że muszę go inaczej, kiedy posiekać mnożenia macierzy, Eigen nie tworzy temporaries, ale myślę, że w ten sposób wymusić chop_expr stworzyć tymczasowy dla każdego inne wyrażenia również. Pytanie brzmi: jak mam to zrobić właściwie?
namespace Eigen {
template <int chunk, typename Derived> struct ChoppedExpression;
namespace internal {
template <int chunk, typename Derived>
struct traits<ChoppedExpression<chunk, Derived>> : traits<Derived> {
enum {Flags = EvalBeforeNestingBit};
enum {IsRowMajor = 0};
enum { RowsAtCompileTime = chunk};
enum {MaxRowsAtCompileTime = chunk};
enum {ColsAtCompileTime = (Derived::RowsAtCompileTime == Eigen::Dynamic
? Eigen::Dynamic : Derived::RowsAtCompileTime/chunk)};
enum {MaxColsAtCompileTime = (Derived::MaxRowsAtCompileTime == Eigen::Dynamic
? Eigen::Dynamic : (Derived::MaxRowsAtCompileTime + chunk - 1)/chunk)};
};
} // namespace internal
template <int chunk, class Derived> struct ChoppedExpression
: public MatrixBase<ChoppedExpression<chunk, Derived>> {
ChoppedExpression(const Derived& arg) : m_arg(arg) {
EIGEN_STATIC_ASSERT(Derived::ColsAtCompileTime == 1,
YOU_TRIED_CALLING_A_VECTOR_METHOD_ON_A_MATRIX);
EIGEN_STATIC_ASSERT(Derived::RowsAtCompileTime % chunk == 0
|| Derived::RowsAtCompileTime == Eigen::Dynamic,
VECTOR_SHOULD_HAVE_INTEGER_NUMBER_OF_CHUNKS_FOR_CHOPPING);
}
typedef Index Index;
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index rows() const { return chunk; }
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index cols() const { return m_arg.size()/chunk; }
typedef typename internal::ref_selector<ChoppedExpression>::type Nested;
typedef typename internal::ref_selector<Derived>::type DerivedTypeNested;
DerivedTypeNested m_arg;
};
namespace internal {
template<int chunk, typename Derived>
struct evaluator<ChoppedExpression<chunk, Derived>>
: public evaluator_base<ChoppedExpression<chunk, Derived>> {
typedef ChoppedExpression<chunk, Derived> XprType;
typedef typename nested_eval<Derived, XprType::ColsAtCompileTime>::type DerivedNested;
typedef typename remove_all<DerivedNested>::type DerivedNestedCleaned;
typedef typename XprType::CoeffReturnType CoeffReturnType;
enum {
CoeffReadCost = evaluator<DerivedNestedCleaned>::CoeffReadCost,
Flags = traits<XprType>::Flags, IsRowMajor = 0
};
evaluator(const XprType& xpr) : m_argImpl(xpr.m_arg) {}
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType coeff(Index row, Index col) const
{ return m_argImpl.coeff(col * chunk + row); }
evaluator<DerivedNestedCleaned> m_argImpl;
};
} // namespace internal
} // namespace Eigen
template<int chunk, typename Derived> EIGEN_ALWAYS_INLINE
EIGEN_DEVICE_FUNC static Eigen::ChoppedExpression<chunk, Derived>
chop_expr(const Eigen::MatrixBase<Derived> &expr)
{ return Eigen::ChoppedExpression<chunk, Derived>(expr.derived()); }
#define MAXV -1
Eigen::Matrix<double, -1, 1, 0, std::max(3*MAXV, -1)> _blendshapes(2, 1);
int main() {
for (int i = 0; i < 2; ++i) _blendshapes[i] = double(i + 10);
std::cout << chop_expr<2>(_blendshapes + Eigen::Matrix<double, 2, 1>(1, 1)) << std::endl;
}
Aktualizacja
Wreszcie znalazłem sposób, aby to działało. Rozwiązaniem jest usunięcie DeredNested i DerivedNestedCleaned typedefs (oczywiście nie potrzebuję ich w wyrażeniu, ponieważ jest to tylko zmiana kształtu, ale nie mogę wyjaśnić, dlaczego powoduje to nieprawidłowe wyniki). Pozostaje tylko pytanie, co powinienem zrobić z EvalBeforeNestingBit?
Wydaje się, że cechy, dla przynajmniej niektórych wyrażeń (np Eigen :: :: wewnętrzne cechy :: Flagi) są zerowe, więc użyłem oceniający :: Zamiast tego flagi. Ale zdecydowanie wskazałeś na problem. –