rewrite_parser_locals.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PARDE_REWRITE_PARSER_LOCALS_H_
#define BACKENDS_TOFINO_BF_P4C_PARDE_REWRITE_PARSER_LOCALS_H_
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
#include "backends/tofino/bf-p4c/phv/phv_parde_mau_use.h"
#include "collect_parser_usedef.h"
 
class GetSelectFieldInfo : public Inspector {
    const PhvInfo &phv;
    bool preorder(const IR::BFN::Select *select) override;
 
 public:
    ordered_map<const PHV::Field *, ordered_set<const IR::BFN::ParserState *>>
        select_field_to_state;
    explicit GetSelectFieldInfo(const PhvInfo &phv) : phv(phv) {}
    Visitor::profile_t init_apply(const IR::Node *root) override {
        select_field_to_state.clear();
        return Inspector::init_apply(root);
    }
};
 
class AddParserMatchDefs : public Modifier {
    const PhvInfo &phv;
    CollectParserInfo &parserInfo;
    const GetSelectFieldInfo &getSelectField;
    const PhvUse &uses;
    bool preorder(IR::BFN::Extract *) override;
 
 public:
    AddParserMatchDefs(const PhvInfo &phv, CollectParserInfo &parserInfo,
                       const GetSelectFieldInfo &getSelectField, const PhvUse &uses)
        : phv(phv), parserInfo(parserInfo), getSelectField(getSelectField), uses(uses) {}
};
 
class RewriteParserMatchDefs : public PassManager {
 public:
    explicit RewriteParserMatchDefs(const PhvInfo &phv);
};
 
#endif /*BACKENDS_TOFINO_BF_P4C_PARDE_REWRITE_PARSER_LOCALS_H_*/