parser_loops_info.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PARDE_PARSER_LOOPS_INFO_H_
#define BACKENDS_TOFINO_BF_P4C_PARDE_PARSER_LOOPS_INFO_H_
 
#include "frontends/common/resolveReferences/referenceMap.h"
#include "ir/ir.h"
#include "lib/cstring.h"
 
namespace BFN {
 
struct ParserPragmas : public Inspector {
    static bool checkNumArgs(cstring pragma, const IR::Vector<IR::Expression> &exprs,
                             unsigned expected);
 
    static bool checkGress(cstring pragma, const IR::StringLiteral *gress);
 
    bool preorder(const IR::Annotation *annot) override;
 
    std::set<const IR::ParserState *> terminate_parsing;
    std::map<const IR::ParserState *, unsigned> force_shift;
    std::map<const IR::ParserState *, unsigned> max_loop_depth;
 
    std::set<cstring> dont_unroll;
};

struct ParserLoopsInfo {
    struct GetMaxLoopDepth;
 
    ParserLoopsInfo(P4::ReferenceMap *refMap, const IR::BFN::TnaParser *parser,
                    const ParserPragmas &pm);
 
    const ParserPragmas &parserPragmas;
 
    std::set<std::set<cstring>> loops;
    std::map<cstring, int> max_loop_depth;
    std::set<cstring> has_next;  // states that have stack "next" references
 
    const std::set<cstring> *find_loop(cstring state) const;

    bool has_next_on_loop(cstring state) const;
 
    bool dont_unroll(cstring state) const;

    bool need_strided_allocation(cstring state) const;
};
 
}  // namespace BFN
 
#endif /* BACKENDS_TOFINO_BF_P4C_PARDE_PARSER_LOOPS_INFO_H_ */