//===- Interfaces.h - Interface wrapper classes -----------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef MLIR_TABLEGEN_INTERFACES_H_
#define MLIR_TABLEGEN_INTERFACES_H_

#include "mlir/Support/LLVM.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringMap.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/iterator.h"

namespace llvm {
class Init;
class Record;
} // namespace llvm

namespace mlir {
namespace tblgen {

// Wrapper class with helper methods for accessing InterfaceMethod defined
// in TableGen.
class InterfaceMethod {
public:
  // This struct represents a single method argument.
  struct Argument {
    StringRef type;
    StringRef name;
  };

  explicit InterfaceMethod(const llvm::Record *def, std::string uniqueName);

  // Return the return type of this method.
  StringRef getReturnType() const;

  // Return the name of this method.
  StringRef getName() const;

  // Return the dedup name of this method.
  StringRef getUniqueName() const;

  // Return if this method is static.
  bool isStatic() const;

  // Return if the method is a pure virtual one.
  bool isPureVirtual() const;

  // Return if the method is only a declaration.
  bool isDeclaration() const;

  // Return the body for this method if it has one.
  std::optional<StringRef> getBody() const;

  // Return the default implementation for this method if it has one.
  std::optional<StringRef> getDefaultImplementation() const;

  // Return the description of this method if it has one.
  std::optional<StringRef> getDescription() const;

  // Arguments.
  ArrayRef<Argument> getArguments() const;
  bool arg_empty() const;

private:
  // The TableGen definition of this method.
  const llvm::Record *def;

  // The arguments of this method.
  SmallVector<Argument, 2> arguments;

  // The unique name of this method, to distinguish it from other methods with
  // the same name (overloaded methods)
  std::string uniqueName;
};

//===----------------------------------------------------------------------===//
// Interface
//===----------------------------------------------------------------------===//

// Wrapper class with helper methods for accessing Interfaces defined in
// TableGen.
class Interface {
public:
  explicit Interface(const llvm::Record *def);
  Interface(const Interface &rhs) : def(rhs.def), methods(rhs.methods) {
    for (auto &base : rhs.baseInterfaces)
      baseInterfaces.push_back(std::make_unique<Interface>(*base));
  }

  // Return the name of this interface.
  StringRef getName() const;

  // Returns this interface's name prefixed with namespaces.
  std::string getFullyQualifiedName() const;

  // Return the C++ namespace of this interface.
  StringRef getCppNamespace() const;

  // Return the methods of this interface.
  ArrayRef<InterfaceMethod> getMethods() const;

  // Return the description of this method if it has one.
  std::optional<StringRef> getDescription() const;

  // Return the interfaces extra class declaration code.
  std::optional<StringRef> getExtraClassDeclaration() const;

  // Return the traits extra class declaration code.
  std::optional<StringRef> getExtraTraitClassDeclaration() const;

  // Return the extra class declaration code shared between the interface and
  // trait classes.
  std::optional<StringRef> getExtraSharedClassDeclaration() const;

  // Return the extra classof method code.
  std::optional<StringRef> getExtraClassOf() const;

  // Return the verify method body if it has one.
  std::optional<StringRef> getVerify() const;

  // Return the base interfaces of this interface.
  auto getBaseInterfaces() const {
    return llvm::make_pointee_range(baseInterfaces);
  }

  // If there's a verify method, return if it needs to access the ops in the
  // regions.
  bool verifyWithRegions() const;

  // Returns the Tablegen definition this interface was constructed from.
  const llvm::Record &getDef() const { return *def; }

private:
  // The TableGen definition of this interface.
  const llvm::Record *def;

  // The methods of this interface.
  SmallVector<InterfaceMethod, 8> methods;

  // The base interfaces of this interface.
  SmallVector<std::unique_ptr<Interface>> baseInterfaces;
};

// An interface that is registered to an Attribute.
struct AttrInterface : public Interface {
  using Interface::Interface;

  static bool classof(const Interface *interface);
};
// An interface that is registered to an Operation.
struct OpInterface : public Interface {
  using Interface::Interface;

  static bool classof(const Interface *interface);
};
// An interface that is registered to a Type.
struct TypeInterface : public Interface {
  using Interface::Interface;

  static bool classof(const Interface *interface);
};
// An interface that is registered to a Dialect.
struct DialectInterface : public Interface {
  using Interface::Interface;

  static bool classof(const Interface *interface);
};

} // namespace tblgen
} // namespace mlir

#endif // MLIR_TABLEGEN_INTERFACES_H_