candidate.hpp

Go to the documentation of this file.

#include <string>
#include <unordered_map>

#include <boost/functional/hash.hpp>

#include "teq/itensor.hpp"

#ifndef OPT_CAND_HPP
#define OPT_CAND_HPP

namespace opt
{

using CtxValT = std::set<teq::TensptrT>;

using ContexT = std::map<std::string,CtxValT>;

using CtxsT = std::unordered_set<ContexT,boost::hash<ContexT>>;

enum CAND_TYPE
{
    SCALAR = 0,
    CONST,
    INTERM,
    CONVRT,
};

struct Symbol final
{
    CAND_TYPE type_;

    // type_=SCALAR: scalar label
    // type_=INTERM: intermediate id
    // type_=CONVRT: conversion ref
    std::string reference_;
};

struct SymbolHash final
{
    size_t operator() (const Symbol& sym) const
    {
        size_t seed = 0;
        boost::hash_combine(seed, sym.type_);
        boost::hash_combine(seed, sym.reference_);
        return seed;
    }
};

inline bool operator == (const Symbol& lhs, const Symbol& rhs)
{
    return lhs.type_ == rhs.type_ && lhs.reference_ == rhs.reference_;
}

using CandsT = std::unordered_map<Symbol,CtxsT,SymbolHash>;

struct CandArg
{
    teq::TensptrT tensor_;

    CandsT candidates_;

    teq::CoordptrT shaper_;

    teq::CoordptrT coorder_;
};

using CandArgsT = std::vector<CandArg>;

}

#endif // OPT_CAND_HPP