p4tools/modules/smith/targets/tofino/target.h

/*
 * SPDX-FileCopyrightText: 2024 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_SMITH_TARGETS_TOFINO_TARGET_H_
#define BACKENDS_P4TOOLS_MODULES_SMITH_TARGETS_TOFINO_TARGET_H_
 
#include <ostream>
#include <string>
 
#include "backends/p4tools/modules/smith/common/declarations.h"
#include "backends/p4tools/modules/smith/common/expressions.h"
#include "backends/p4tools/modules/smith/common/parser.h"
#include "backends/p4tools/modules/smith/common/statements.h"
#include "backends/p4tools/modules/smith/common/table.h"
#include "backends/p4tools/modules/smith/core/target.h"
#include "ir/ir.h"
 
namespace P4::P4Tools::P4Smith::Tofino {
 
class AbstractTofinoSmithTarget : public SmithTarget {
 protected:
    explicit AbstractTofinoSmithTarget(const std::string &deviceName, const std::string &archName);
};
 
class TofinoTnaSmithTarget : public AbstractTofinoSmithTarget {
 private:
    DeclarationGenerator *_declarationGenerator = new DeclarationGenerator(*this);
    ExpressionGenerator *_expressionGenerator = new ExpressionGenerator(*this);
    StatementGenerator *_statementGenerator = new StatementGenerator(*this);
    ParserGenerator *_parserGenerator = new ParserGenerator(*this);
    TableGenerator *_tableGenerator = new TableGenerator(*this);
 
    [[nodiscard]] IR::P4Parser *generateIngressParserBlock() const;
    [[nodiscard]] IR::P4Control *generateIngressBlock() const;
    [[nodiscard]] IR::P4Control *generateIngressDeparserBlock() const;
    [[nodiscard]] IR::P4Parser *generateEgressParserBlock() const;
    [[nodiscard]] IR::P4Control *generateEgressBlock() const;
    [[nodiscard]] IR::P4Control *generateEgressDeparserBlock() const;
 
 public:
    static void make();
 
    [[nodiscard]] int writeTargetPreamble(std::ostream *ostream) const override;
 
    [[nodiscard]] const IR::P4Program *generateP4Program() const override;
 
    [[nodiscard]] DeclarationGenerator &declarationGenerator() const override {
        return *_declarationGenerator;
    }
 
    [[nodiscard]] ExpressionGenerator &expressionGenerator() const override {
        return *_expressionGenerator;
    }
 
    [[nodiscard]] StatementGenerator &statementGenerator() const override {
        return *_statementGenerator;
    }
 
    [[nodiscard]] ParserGenerator &parserGenerator() const override { return *_parserGenerator; }
 
    [[nodiscard]] TableGenerator &tableGenerator() const override { return *_tableGenerator; }
 
 private:
    TofinoTnaSmithTarget();
};
 
}  // namespace P4::P4Tools::P4Smith::Tofino
 
#endif /* BACKENDS_P4TOOLS_MODULES_SMITH_TARGETS_TOFINO_TARGET_H_ */