resources_parser.h

 
#ifndef _BACKENDS_TOFINO_BF_P4C_LOGGING_RESOURCES_PARSER_H_
#define _BACKENDS_TOFINO_BF_P4C_LOGGING_RESOURCES_PARSER_H_
 
/* clang-format off */
#include "ir/ir.h"
#include "backends/tofino/bf-p4c/parde/parde_visitor.h"
#include "resources_schema.h"
/* clang-format on */
 
using Logging::Resources_Schema_Logger;
 
namespace BFN {
 
class ParserResourcesLogging : public ParserInspector {
 public:
    using ElementUsage = Resources_Schema_Logger::ElementUsage;
    using ParserResources = Resources_Schema_Logger::ParserResources;
    using ParserResourceUsage = Resources_Schema_Logger::ParserResourceUsage;
    using Phase0ResourceUsage = Resources_Schema_Logger::Phase0ResourceUsage;
 
    using ParserStateTransition = Resources_Schema_Logger::ParserStateTransitionResourceUsage;
 
    using ClotExtracts = ParserStateTransition::Clot_Extracts;
    using StateExtracts = ParserStateTransition::Extracts;
    using StateMatchesOn = ParserStateTransition::MatchesOn;
    using StateSavesTo = ParserStateTransition::SavesTo;
 
 private:
    struct ParserLogData {
        ParserResourceUsage *usage = nullptr;
        Phase0ResourceUsage *phase0 = nullptr;
    };
 
    bool collected = false;
    ParserResources *parserLogger = nullptr;
 
    std::map<cstring, ParserLogData> parsers;
 
    std::map<cstring, int> stateIds;
    std::map<const IR::BFN::LoweredParserMatch *, int> tcamIds[2];
    int nextTcamId[2] = {0};  // Next TCAM ID value array
 
    bool preorder(const IR::BFN::LoweredParserState *state) override;
    bool preorder(const IR::BFN::LoweredParser *parser) override;
    void end_apply() override { collected = true; }
 
    std::vector<ParserStateTransition *> logStateTransitionsByMatch(
        const std::string &nextStateName, const IR::BFN::LoweredParserState *prevState,
        const IR::BFN::LoweredParserMatch *match);
 
    int getTcamId(const IR::BFN::LoweredParserMatch *match, gress_t gress);
 
    int getStateId(const std::string &state);
 
    void logStateExtracts(const IR::BFN::LoweredParserMatch *match,
                          std::vector<StateExtracts *> &result);
 
    void logStateMatches(const IR::BFN::LoweredParserState *prevState,
                         const IR::BFN::LoweredParserMatch *match,
                         std::vector<StateMatchesOn *> &result);
 
    void logStateSaves(const IR::BFN::LoweredParserMatch *match,
                       std::vector<StateSavesTo *> &result);
 
 public:
    ParserResourcesLogging() {
        const int tcamRowsInit = Device::pardeSpec().numTcamRows() - 1;
        nextTcamId[0] = tcamRowsInit;
        nextTcamId[1] = tcamRowsInit;
    }
 
    const ParserResources *getLogger();
};
 
}  // namespace BFN
 
#endif /* _BACKENDS_TOFINO_BF_P4C_LOGGING_RESOURCES_PARSER_H_ */