funcarg.hpp

Go to the documentation of this file.

#include "teq/itensor.hpp"
#include "teq/coord.hpp"

#ifndef TEQ_FUNCARG_HPP
#define TEQ_FUNCARG_HPP

namespace teq
{

Shape apply_shaper (const CoordptrT& shaper, Shape inshape);

struct FuncArg final
{
    FuncArg (TensptrT tensor, CoordptrT shaper) :
        tensor_(tensor), shaper_(shaper)
    {
        if (tensor_ == nullptr)
        {
            logs::fatal("cannot map a null tensor");
        }
        map_io_ = tensor_->shape().n_elems() > shape().n_elems();
        if (shaper == identity || map_io_)
        {
            coorder_ = shaper;
        }
        else
        {
            coorder_ = CoordptrT(shaper->reverse());
        }
    }

    FuncArg (TensptrT tensor, CoordptrT shaper,
        bool map_io, CoordptrT coorder) :
        tensor_(tensor), shaper_(shaper),
        map_io_(map_io), coorder_(coorder)
    {
        if (tensor_ == nullptr)
        {
            logs::fatal("cannot map a null tensor");
        }
    }

    Shape shape (void) const
    {
        return apply_shaper(shaper_, tensor_->shape());
    }

    TensptrT get_tensor (void) const
    {
        return tensor_;
    }

    CoordptrT get_shaper (void) const
    {
        return shaper_;
    }

    bool map_io (void) const
    {
        return map_io_;
    }

    CoordptrT get_coorder (void) const
    {
        return coorder_;
    }

private:
    TensptrT tensor_;

    CoordptrT shaper_;

    bool map_io_;

    CoordptrT coorder_;
};

using ArgsT = std::vector<FuncArg>;

FuncArg identity_map (TensptrT tensor);

FuncArg reduce_1d_map (TensptrT tensor, RankT rank);

FuncArg reduce_map (TensptrT tensor,
    RankT rank, std::vector<DimT> red);

FuncArg extend_map (TensptrT tensor,
    RankT rank, std::vector<DimT> ext);

FuncArg permute_map (TensptrT tensor, std::vector<RankT> order);

FuncArg flip_map (TensptrT tensor, RankT dim);

ArgsT to_args (TensptrsT tens);

}

#endif // TEQ_FUNCARG_HPP