compute_lowered_parser_ir.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_COMPUTE_LOWERED_PARSER_IR_H_
#define BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_COMPUTE_LOWERED_PARSER_IR_H_
 
#include "backends/tofino/bf-p4c/parde/allocate_parser_checksum.h"
#include "backends/tofino/bf-p4c/parde/clot/clot_info.h"
#include "backends/tofino/bf-p4c/parde/count_strided_header_refs.h"
#include "backends/tofino/bf-p4c/parde/parde_visitor.h"
 
namespace Parde::Lowered {
 
struct ComputeLoweredParserIR : public ParserInspector {
    ComputeLoweredParserIR(
        const PhvInfo &phv, ClotInfo &clotInfo, const AllocateParserChecksums &checksumAlloc,
        std::map<gress_t, std::set<PHV::Container>> &origParserZeroInitContainers)
        : phv(phv),
          clotInfo(clotInfo),
          checksumAlloc(checksumAlloc),
          origParserZeroInitContainers(origParserZeroInitContainers) {
        // Initialize the map from high-level parser states to low-level parser
        // states so that null, which represents the end of the parser program
        // in the high-level IR, is mapped to null, which conveniently enough
        // means the same thing in the lowered IR.
        loweredStates[nullptr] = nullptr;
    }
 
    std::map<const IR::BFN::ParserState *, const IR::BFN::LoweredParserState *> loweredStates;
    std::set<const IR::BFN::LoweredParserState *> dontMergeStates;
 
    const IR::BFN::ContainerRef *igParserError = nullptr;
    const IR::BFN::ContainerRef *egParserError = nullptr;
    unsigned egressMetaOpt = 0;
    unsigned egressMetaSize = 0;  // in bytes
 
 private:
    profile_t init_apply(const IR::Node *node) override;
 
    bool preorder(const IR::Type_Header *type) override;
 
    IR::Vector<IR::BFN::LoweredParserChecksum> lowerParserChecksums(
        const IR::BFN::Parser *parser, const IR::BFN::ParserState *state,
        const std::vector<const IR::BFN::ParserChecksumPrimitive *> &checksums);
 
    unsigned int rangeToInt(const IR::BFN::PacketRVal *range);
 
    IR::BFN::LoweredParserChecksum *lowerParserChecksum(
        const IR::BFN::Parser *parser, const IR::BFN::ParserState *state, cstring name,
        std::vector<const IR::BFN::ParserChecksumPrimitive *> &checksums);
 
    unsigned getOffsetIncAmt(const IR::BFN::ParserState *state);
 
    void postorder(const IR::BFN::ParserState *state) override;
 
    void end_apply() override;
 
    const PhvInfo &phv;
    ClotInfo &clotInfo;
    const AllocateParserChecksums &checksumAlloc;
    std::map<gress_t, std::set<PHV::Container>> &origParserZeroInitContainers;
    // Maps clot tag to checksum unit whose output will be deposited in that CLOT for each gress
    std::map<gress_t, std::map<unsigned, unsigned>> clotTagToCsumUnit;
};
 
}  // namespace Parde::Lowered
 
#endif /* BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_COMPUTE_LOWERED_PARSER_IR_H_ */