ileaf.hpp

Go to the documentation of this file.

#include "teq/ileaf.hpp"

#include "eteq/eigen.hpp"

#ifndef ETEQ_ILEAF_HPP
#define ETEQ_ILEAF_HPP

namespace eteq
{

template <typename T>
struct iLeaf : public teq::iLeaf
{
    virtual ~iLeaf (void) = default;

    const teq::Shape& shape (void) const override
    {
        return shape_;
    }

    void* data (void) override
    {
        return data_.data();
    }

    const void* data (void) const override
    {
        return data_.data();
    }

    size_t type_code (void) const override
    {
        return egen::get_type<T>();
    }

    std::string type_label (void) const override
    {
        return egen::name_type(egen::get_type<T>());
    }

    size_t nbytes (void) const override
    {
        return sizeof(T) * shape_.n_elems();
    }

protected:
    iLeaf (T* data, teq::Shape shape) :
        data_(make_tensmap(data, shape)),
        shape_(shape) {}

    TensorT<T> data_;

    teq::Shape shape_;
};

}

#endif // ETEQ_ILEAF_HPP