tofino/bf-p4c/arch/arch_spec.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_ARCH_ARCH_SPEC_H_
#define BACKENDS_TOFINO_BF_P4C_ARCH_ARCH_SPEC_H_
 
#include "bf-p4c/ir/gress.h"
#include "ir/ir.h"
 
struct IntrinsicField {
 public:
    std::string name;
    std::string asm_name;
    explicit IntrinsicField(std::string n) : name(n), asm_name(n) {}
    IntrinsicField(std::string n, std::string an) : name(n), asm_name(an) {}
};
 
struct IntrinsicParam {
 public:
    std::string type;
    std::string name;
    IntrinsicParam(std::string t, std::string n) : type(t), name(n) {}
};
 
class ArchSpec {
 protected:
    std::vector<IntrinsicField> intrinsic_metadata[GRESS_T_COUNT];
    std::vector<IntrinsicField> intrinsic_metadata_for_tm[GRESS_T_COUNT];
    std::vector<IntrinsicField> intrinsic_metadata_from_parser[GRESS_T_COUNT];
    std::vector<IntrinsicField> intrinsic_metadata_for_deparser[GRESS_T_COUNT];
    std::vector<IntrinsicField> intrinsic_metadata_for_output_port[GRESS_T_COUNT];
 
    std::vector<IntrinsicParam> parser_intrinsic_types[GRESS_T_COUNT];
    std::vector<IntrinsicParam> mauppu_intrinsic_types[GRESS_T_COUNT];
    std::vector<IntrinsicParam> deparser_intrinsic_types[GRESS_T_COUNT];
 
    int deparser_intrinsic_metadata_for_deparser_param_index = -1;
 
 public:
    enum ArchType_t { TNA, T2NA };
 
    ArchSpec();
 
    std::vector<IntrinsicField> getIngressIntrinsicMetadata() const {
        return intrinsic_metadata[INGRESS];
    }
 
    std::vector<IntrinsicField> getIngressInstrinicMetadataFromParser() const {
        return intrinsic_metadata_from_parser[INGRESS];
    }
 
    std::vector<IntrinsicField> getIngressInstrinicMetadataForTM() const {
        return intrinsic_metadata_for_tm[INGRESS];
    }
 
    std::vector<IntrinsicField> getIngressInstrinicMetadataForDeparser() const {
        return intrinsic_metadata_for_deparser[INGRESS];
    }
 
    std::vector<IntrinsicField> getEgressIntrinsicMetadata() const {
        return intrinsic_metadata[EGRESS];
    }
 
    std::vector<IntrinsicField> getEgressIntrinsicMetadataFromParser() const {
        return intrinsic_metadata_from_parser[EGRESS];
    }
 
    std::vector<IntrinsicField> getEgressIntrinsicMetadataForTM() const {
        return intrinsic_metadata_for_tm[EGRESS];
    }
 
    std::vector<IntrinsicField> getEgressIntrinsicMetadataForDeparser() const {
        return intrinsic_metadata_for_deparser[EGRESS];
    }
 
    std::vector<IntrinsicField> getEgressIntrinsicMetadataForOutputPort() const {
        return intrinsic_metadata_for_output_port[EGRESS];
    }
 
    void add_md(gress_t gress, IntrinsicField f) { intrinsic_metadata[gress].push_back(f); }
 
    void add_tm_md(gress_t gress, IntrinsicField f) {
        intrinsic_metadata_for_tm[gress].push_back(f);
    }
 
    void add_prsr_md(gress_t gress, IntrinsicField f) {
        intrinsic_metadata_from_parser[gress].push_back(f);
    }
 
    void add_dprsr_md(gress_t gress, IntrinsicField f) {
        intrinsic_metadata_for_deparser[gress].push_back(f);
    }
 
    void add_outport_md(gress_t gress, IntrinsicField f) {
        intrinsic_metadata_for_output_port[gress].push_back(f);
    }
 
    std::vector<IntrinsicParam> getParserIntrinsicTypes(gress_t g) const {
        return parser_intrinsic_types[g];
    }
 
    std::vector<IntrinsicParam> getMAUIntrinsicTypes(gress_t g) const {
        return mauppu_intrinsic_types[g];
    }
 
    std::vector<IntrinsicParam> getPPUIntrinsicTypes(gress_t g) const {
        return mauppu_intrinsic_types[g];
    }
 
    std::vector<IntrinsicParam> getDeparserIntrinsicTypes(gress_t g) const {
        return deparser_intrinsic_types[g];
    }
 
    int getDeparserIntrinsicMetadataForDeparserParamIndex() const {
        return deparser_intrinsic_metadata_for_deparser_param_index;
    }
 
    void setTofinoIntrinsicTypes();
};
 
class TofinoArchSpec : public ArchSpec {
 public:
    TofinoArchSpec();
};
 
class JBayArchSpec : public ArchSpec {
 public:
    JBayArchSpec();
};
 
#endif /* BACKENDS_TOFINO_BF_P4C_ARCH_ARCH_SPEC_H_ */