coord.hpp

Go to the documentation of this file.

#include <functional>

#include "teq/matops.hpp"

#ifndef TEQ_COORD_HPP
#define TEQ_COORD_HPP

namespace teq
{

struct iCoordMap
{
    virtual ~iCoordMap (void) = default;

    virtual iCoordMap* connect (const iCoordMap& rhs) const = 0;

    virtual void forward (CoordT::iterator out,
        CoordT::const_iterator in) const = 0;

    virtual iCoordMap* reverse (void) const = 0;

    virtual std::string to_string (void) const = 0;

    virtual void access (std::function<void(const MatrixT&)> cb) const = 0;

    virtual bool is_bijective (void) const = 0;
};

struct CoordMap final : public iCoordMap
{
    CoordMap (std::function<void(MatrixT)> init)
    {
        std::memset(fwd_, 0, mat_size);
        fwd_[rank_cap][rank_cap] = 1;
        init(fwd_);
    }

    iCoordMap* connect (const iCoordMap& rhs) const override
    {
        return new CoordMap([&](MatrixT out)
        {
            rhs.access([&](const MatrixT& in)
            {
                matmul(out, fwd_, in);
            });
        });
    }

    void forward (CoordT::iterator out,
        CoordT::const_iterator in) const override;

    iCoordMap* reverse (void) const override
    {
        return new CoordMap([this](MatrixT m)
        {
            inverse(m, this->fwd_);
        });
    }

    std::string to_string (void) const override
    {
        return teq::to_string(fwd_);
    }

    void access (std::function<void(const MatrixT&)> cb) const override
    {
        cb(fwd_);
    }

    bool is_bijective (void) const override
    {
        return (int) determinant(fwd_) != 0;
    }

private:
    MatrixT fwd_;
};

using CoordptrT = std::shared_ptr<iCoordMap>;

extern CoordptrT identity;

bool is_identity (iCoordMap* coorder);

CoordptrT reduce (RankT rank, std::vector<DimT> red);

CoordptrT extend (RankT rank, std::vector<DimT> ext);

CoordptrT permute (std::vector<RankT> order);

CoordptrT flip (RankT dim);

}

#endif // TEQ_COORD_HPP