dpdkMetadata.h

/*
 * Copyright 2022 Intel Corp.
 * SPDX-FileCopyrightText: 2022 Intel Corp.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_DPDK_DPDKMETADATA_H_
#define BACKENDS_DPDK_DPDKMETADATA_H_
 
#include "dpdkUtils.h"
#include "frontends/common/resolveReferences/referenceMap.h"
#include "ir/ir.h"
 
namespace P4::DPDK {
 
class IsDirectionMetadataUsed : public Inspector {
    bool &is_direction_used;
 
 public:
    explicit IsDirectionMetadataUsed(bool &is_direction_used)
        : is_direction_used(is_direction_used) {}
    bool preorder(const IR::Member *m) override {
        if (!is_direction_used && isDirection(m)) {
            is_direction_used = true;
        }
        return false;
    }
};

class DirectionToRegRead : public Transform {
    ordered_map<cstring, cstring> dirToDirMapping;
    ordered_map<cstring, cstring> inputToInputPortMapping;
    ordered_set<cstring> usedNames;
    cstring registerInstanceName;
 
 public:
    DirectionToRegRead() {
        dirToDirMapping.insert(std::make_pair(cstring("pna_main_input_metadata_direction"),
                                              cstring("pna_main_input_metadata_direction")));
        dirToDirMapping.insert(std::make_pair(cstring("pna_pre_input_metadata_direction"),
                                              cstring("pna_main_input_metadata_direction")));
        dirToDirMapping.insert(std::make_pair(cstring("pna_main_parser_input_metadata_direction"),
                                              cstring("pna_main_input_metadata_direction")));
        inputToInputPortMapping.insert(
            std::make_pair(cstring("pna_main_input_metadata_input_port"),
                           cstring("pna_main_input_metadata_input_port")));
        inputToInputPortMapping.insert(
            std::make_pair(cstring("pna_pre_input_metadata_direction"),
                           cstring("pna_main_input_metadata_input_port")));
        inputToInputPortMapping.insert(
            std::make_pair(cstring("pna_main_parser_input_metadata_direction"),
                           cstring("pna_main_input_metadata_input_port")));
    }
    const IR::Node *preorder(IR::DpdkAsmProgram *p) override;
    const IR::Node *preorder(IR::Member *m) override;
 
    void uniqueNames(IR::DpdkAsmProgram *p);
    IR::DpdkExternDeclaration *addRegDeclInstance(cstring instanceName);
    IR::DpdkListStatement *replaceDirection(IR::DpdkListStatement *l);
    IR::IndexedVector<IR::DpdkAsmStatement> addRegReadStmtForDirection(
        IR::IndexedVector<IR::DpdkAsmStatement> stmts);
};

class PrependPassRecircId : public Transform {
    IR::IndexedVector<IR::DpdkAsmStatement> newStmts;
 
 public:
    PrependPassRecircId() {}
    bool isPass(const IR::Member *m);
    const IR::Node *postorder(IR::DpdkAction *a) override;
    const IR::Node *postorder(IR::DpdkListStatement *l) override;
    IR::IndexedVector<IR::DpdkAsmStatement> prependPassWithRecircid(
        IR::IndexedVector<IR::DpdkAsmStatement> stmts);
};
 
}  // namespace P4::DPDK
#endif  // BACKENDS_DPDK_DPDKMETADATA_H_