parse_pb_expr.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_PARSE_PB_EXPR_H
#define FREE_TENSOR_PARSE_PB_EXPR_H
 
#include <iostream>
 
#include <expr.h>
#include <math/presburger.h>
 
namespace freetensor {
 
struct SimplePBFuncAST {
    std::vector<std::string> args_;
    std::vector<Expr> values_;
    Expr cond_; // Maybe null
};
 
std::ostream &operator<<(std::ostream &os, const SimplePBFuncAST &ast);
 
typedef std::vector<SimplePBFuncAST> PBFuncAST;
 
PBFuncAST parsePBFunc(const PBFunc::Serialized &f);
PBFuncAST parsePBFunc(const PBSingleFunc::Serialized &f);
PBFuncAST parsePBFuncReconstructMinMax(const PBSet &set);
PBFuncAST parsePBFuncReconstructMinMax(const PBMap &map);
inline SimplePBFuncAST parseSimplePBFunc(const auto &f) {
    auto ret = parsePBFunc(f);
    if (ret.size() != 1) {
        throw ParserError(FT_MSG << f << " is not a simple PBFunc");
    }
    return ret.front();
}
inline SimplePBFuncAST parseSimplePBFuncReconstructMinMax(const auto &f) {
    auto ret = parsePBFuncReconstructMinMax(f);
    if (ret.size() != 1) {
        throw ParserError(FT_MSG << f << " is not a simple PBFunc");
    }
    return ret.front();
}
} // namespace freetensor
 
#endif // FREE_TENSOR_PARSE_PB_EXPR_H