annotate_with_in_hash.h

#ifndef BACKENDS_TOFINO_BF_P4C_MIDEND_ANNOTATE_WITH_IN_HASH_H_
#define BACKENDS_TOFINO_BF_P4C_MIDEND_ANNOTATE_WITH_IN_HASH_H_
 
#include "backends/tofino/bf-p4c/midend/type_checker.h"
#include "ir/ir.h"
 
namespace BFN {

class DoAnnotateWithInHash : public Transform {
    P4::TypeMap *typeMap;
 
    bool checkKeyDefaultAction(const IR::P4Control &control, const IR::P4Action &action) const;
    bool checkAssignmentStructure(const IR::AssignmentStatement &assignment,
                                  const IR::Expression **op, const IR::Expression **opConcat);
    bool checkAluSuitability(const IR::Expression &op) const;
    bool checkHeaderMetadataReference(const IR::Expression &op) const;
 
 public:
    explicit DoAnnotateWithInHash(P4::TypeMap *typeMap) : typeMap(typeMap) {}
    const IR::Node *preorder(IR::BlockStatement *b) override;
    const IR::Node *preorder(IR::Expression *e) override;
    const IR::Node *preorder(IR::AssignmentStatement *assignment) override;
};

 
class AnnotateWithInHash : public PassManager {
 public:
    AnnotateWithInHash(P4::ReferenceMap *refMap, P4::TypeMap *typeMap,
                       BFN::TypeChecking *typeChecking = nullptr) {
        if (!typeChecking) typeChecking = new BFN::TypeChecking(refMap, typeMap);
        passes.push_back(typeChecking);
        passes.push_back(new DoAnnotateWithInHash(typeMap));
    }
};
 
}  // namespace BFN
 
#endif  // BACKENDS_TOFINO_BF_P4C_MIDEND_ANNOTATE_WITH_IN_HASH_H_