functor.hpp

Go to the documentation of this file.

#include "teq/ifunctor.hpp"

#ifndef TEQ_FUNCTOR_HPP
#define TEQ_FUNCTOR_HPP

namespace teq
{

struct Functor final : public iFunctor
{
    static Functor* get (Opcode opcode, ArgsT args)
    {
        if (0 == args.size())
        {
            logs::fatalf("cannot perform `%s` with no arguments",
                opcode.name_.c_str());
        }

        Shape shape = args[0].shape();
        for (size_t i = 1, n = args.size(); i < n; ++i)
        {
            Shape ishape = args[i].shape();
            if (false == ishape.compatible_after(shape, 0))
            {
                logs::fatalf("cannot perform `%s` with incompatible shapes %s "
                    "and %s", opcode.name_.c_str(), shape.to_string().c_str(),
                    ishape.to_string().c_str());
            }
        }
        return new Functor(opcode, shape, args);
    }

    static Functor* get (const Functor& other)
    {
        return new Functor(other);
    }

    static Functor* get (Functor&& other)
    {
        return new Functor(std::move(other));
    }

    Functor& operator = (const Functor& other) = delete;

    Functor& operator = (Functor&& other) = delete;

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

    std::string to_string (void) const override
    {
        return opcode_.name_;
    }

    Opcode get_opcode (void) const override
    {
        return opcode_;
    }

    const ArgsT& get_children (void) const override
    {
        return args_;
    }

    void update_child (FuncArg arg, size_t index) override
    {
        logs::warn("teq::Functor does not allow editing of children");
    }

private:
    Functor (Opcode opcode, Shape shape, ArgsT args) :
        opcode_(opcode), shape_(shape), args_(args) {}

    Functor (const Functor& other) = default;

    Functor (Functor&& other) = default;

    Opcode opcode_;

    Shape shape_;

    ArgsT args_;
};

}

#endif // TEQ_FUNCTOR_HPP