tableHit.h

/*
 * Copyright 2017 VMware, Inc.
 * SPDX-FileCopyrightText: 2017 VMware, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef MIDEND_TABLEHIT_H_
#define MIDEND_TABLEHIT_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/typeChecking/typeChecker.h"
#include "ir/ir.h"
 
namespace P4 {

class DoTableHit : public Transform, public ResolutionContext {
    TypeMap *typeMap;
    enum op_t { None, And, Or, Xor };
 
    const IR::Node *process(IR::BaseAssignmentStatement *statement, op_t op);
 
 public:
    const IR::Node *postorder(IR::BaseAssignmentStatement *statement) override {
        return process(statement, None);
    }
    const IR::Node *postorder(IR::OpAssignmentStatement *statement) override { return statement; }
    const IR::Node *postorder(IR::BAndAssign *statement) override {
        return process(statement, And);
    }
    const IR::Node *postorder(IR::BOrAssign *statement) override { return process(statement, Or); }
    const IR::Node *postorder(IR::BXorAssign *statement) override {
        return process(statement, Xor);
    }
 
    explicit DoTableHit(TypeMap *typeMap) : typeMap(typeMap) {
        CHECK_NULL(typeMap);
        setName("DoTableHit");
    }
};
 
class TableHit : public PassManager {
 public:
    explicit TableHit(TypeMap *typeMap, TypeChecking *typeChecking = nullptr) {
        if (!typeChecking) typeChecking = new TypeChecking(nullptr, typeMap);
        passes.push_back(typeChecking);
        passes.push_back(new DoTableHit(typeMap));
        setName("TableHit");
    }
};
 
}  // namespace P4
 
#endif /* MIDEND_TABLEHIT_H_ */