bmv2/pna_nic/pnaProgramStructure.h

/*
 * Copyright 2024 Marvell Technology, Inc.
 * SPDX-FileCopyrightText: 2024 Marvell Technology, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_BMV2_PNA_NIC_PNAPROGRAMSTRUCTURE_H_
#define BACKENDS_BMV2_PNA_NIC_PNAPROGRAMSTRUCTURE_H_
 
#include "backends/common/portableProgramStructure.h"

namespace P4::BMV2 {
 
enum pna_block_t {
    MAIN_PARSER,
    MAIN_CONTROL,
    MAIN_DEPARSER,
};
 
class PnaProgramStructure : public P4::PortableProgramStructure {
 public:
    ordered_map<const IR::Node *, pna_block_t> block_type;
 
 public:
    PnaProgramStructure(P4::ReferenceMap *refMap, P4::TypeMap *typeMap)
        : PortableProgramStructure(refMap, typeMap) {}

    static bool isCounterMetadata(cstring ptName) { return ptName == "PNA_CounterType_t"; }

    static bool isStandardMetadata(cstring ptName) {
        return ptName == "pna_main_parser_input_metadata_t" ||
               ptName == "pna_main_input_metadata_t" || ptName == "pna_main_output_metadata_t";
    }
};
 
class ParsePnaArchitecture : public P4::ParsePortableArchitecture {
    PnaProgramStructure *structure;
 
 public:
    explicit ParsePnaArchitecture(PnaProgramStructure *structure) : structure(structure) {
        CHECK_NULL(structure);
    }
 
    void modelError(const char *format, const IR::INode *node) {
        ::P4::error(ErrorType::ERR_MODEL,
                    (cstring(format) + "\nAre you using an up-to-date 'pna.p4'?").c_str(),
                    node->getNode());
    }
 
    bool preorder(const IR::PackageBlock *block) override;
    bool preorder(const IR::ExternBlock *block) override;
 
    profile_t init_apply(const IR::Node *root) override {
        structure->block_type.clear();
        structure->globals.clear();
        return Inspector::init_apply(root);
    }
};
 
class InspectPnaProgram : public P4::InspectPortableProgram {
    PnaProgramStructure *pinfo;
 
 public:
    InspectPnaProgram(P4::ReferenceMap *refMap, P4::TypeMap *typeMap, PnaProgramStructure *pinfo)
        : InspectPortableProgram(refMap, typeMap), pinfo(pinfo) {
        CHECK_NULL(pinfo);
        setName("InspectPnaProgram");
    }
 
    void postorder(const IR::P4Parser *p) override;
    void postorder(const IR::P4Control *c) override;
    void postorder(const IR::Declaration_Instance *di) override;
 
    void addTypesAndInstances(const IR::Type_StructLike *type, bool meta);
    void addHeaderType(const IR::Type_StructLike *st);
    void addHeaderInstance(const IR::Type_StructLike *st, cstring name);
    bool preorder(const IR::Declaration_Variable *dv) override;
    bool preorder(const IR::Parameter *parameter) override;
};
 
}  // namespace P4::BMV2
 
#endif /* BACKENDS_BMV2_PNA_NIC_PNAPROGRAMSTRUCTURE_H_ */