compile_context.h

/*
 * SPDX-FileCopyrightText: 2013 Barefoot Networks, Inc.
 * Copyright 2013-present Barefoot Networks, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef LIB_COMPILE_CONTEXT_H_
#define LIB_COMPILE_CONTEXT_H_
 
#include <memory>
#include <typeinfo>
#include <vector>
 
#include "lib/cstring.h"
#include "lib/error_reporter.h"
 
namespace P4 {

class ICompileContext {
 public:
    virtual ~ICompileContext() = 0;
};

struct CompileContextStack final {
    CompileContextStack() = delete;

    template <typename CompileContextType>
    static CompileContextType &top() {
        auto &stack = getStack();
        if (stack.empty()) reportNoContext();
        auto *current = dynamic_cast<CompileContextType *>(stack.back());
        if (!current) reportContextMismatch(typeid(CompileContextType).name());
        return *current;
    }
 
    static bool isEmpty() { return getStack().empty(); }
 
 private:
    friend struct AutoCompileContext;
 
    using StackType = std::vector<ICompileContext *>;

    static void reportNoContext();
    static void reportContextMismatch(const char *desiredContextType);

    static void push(ICompileContext *context);
    static void pop();
    static StackType &getStack();
};

struct AutoCompileContext {
 private:
    std::unique_ptr<ICompileContext> ownedContext;
 
 public:
    explicit AutoCompileContext(ICompileContext *context);
    explicit AutoCompileContext(std::unique_ptr<ICompileContext> context);
    AutoCompileContext(const AutoCompileContext &) = delete;
    AutoCompileContext &operator=(const AutoCompileContext &) = delete;
    ~AutoCompileContext();
};

class BaseCompileContext : public ICompileContext {
 protected:
    BaseCompileContext() = default;
    BaseCompileContext(const BaseCompileContext &other) = default;
    BaseCompileContext &operator=(const BaseCompileContext &other) = default;
 
 public:
    static BaseCompileContext &get();

    virtual ErrorReporter &errorReporter();

    virtual DiagnosticAction getDefaultInfoDiagnosticAction();

    virtual DiagnosticAction getDefaultWarningDiagnosticAction();

    virtual DiagnosticAction getDefaultErrorDiagnosticAction();
 
 private:
    ErrorReporter errorReporterInstance;
};
 
}  // namespace P4
 
#endif /* LIB_COMPILE_CONTEXT_H_ */