StorageEigen.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_UTILS_STORAGE_EIGEN_HEADER_FILE
#define AI_TOOLBOX_UTILS_STORAGE_EIGEN_HEADER_FILE
 
#include <AIToolbox/Types.hpp>
 
namespace AIToolbox {
    class StorageVector {
        private:
            Vector storage_;
 
        public:
            Eigen::Ref<Vector> vector;
 
            StorageVector(size_t startSize);
 
            StorageVector(Vector vector);
 
            void pop_back(size_t num = 1);
 
            void push_back(double val);
 
            void resize(size_t size);
 
            void reserve(size_t size);
    };
 
    class StorageMatrix2D {
        private:
            Matrix2D storage_;
 
        public:
            Eigen::Ref<Matrix2D> matrix;
 
            StorageMatrix2D(size_t startRows, size_t cols);
 
            StorageMatrix2D(Matrix2D matrix);
 
            void pop_back(size_t num = 1);
 
            void push_back();
 
            template <typename V>
            void push_back(const Eigen::MatrixBase<V> & row);
 
            void resize(size_t rows);
 
            void reserve(size_t rows);
    };
 
    template <typename V>
    void StorageMatrix2D::push_back(const Eigen::MatrixBase<V> & row) {
        const auto currRows = matrix.rows();
 
        if (storage_.rows() == currRows)
            storage_.conservativeResize(currRows * 2, Eigen::NoChange);
 
        storage_.row(currRows) = row;
 
        new (&matrix) Eigen::Ref<Matrix2D>(storage_.topRows(currRows + 1));
    }
}
 
#endif