Builder.h

Go to the documentation of this file.

//===- Builder.h - Builder 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
//
//===----------------------------------------------------------------------===//
//
// Builder wrapper to simplify using TableGen Record for building
// operations/types/etc.
//
//===----------------------------------------------------------------------===//
 
#ifndef MLIR_TABLEGEN_BUILDER_H_
#define MLIR_TABLEGEN_BUILDER_H_
 
#include "mlir/Support/LLVM.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringRef.h"
 
namespace llvm {
class Init;
class Record;
class SMLoc;
} // namespace llvm
 
namespace mlir {
namespace tblgen {
 
/// Wrapper class with helper methods for accessing Builders defined in
/// TableGen.
class Builder {
public:
  /// This class represents a single parameter to a builder method.
  class Parameter {
  public:
    /// Return a string containing the C++ type of this parameter.
    StringRef getCppType() const;
 
    /// Return an optional string containing the name of this parameter. If
    /// std::nullopt, no name was specified for this parameter by the user.
    std::optional<StringRef> getName() const { return name; }
 
    /// Return an optional string containing the default value to use for this
    /// parameter.
    std::optional<StringRef> getDefaultValue() const;
 
  private:
    Parameter(std::optional<StringRef> name, const llvm::Init *def)
        : name(name), def(def) {}
 
    /// The optional name of the parameter.
    std::optional<StringRef> name;
 
    /// The tablegen definition of the parameter. This is either a StringInit,
    /// or a CArg DefInit.
    const llvm::Init *def;
 
    // Allow access to the constructor.
    friend Builder;
  };
 
  /// Construct a builder from the given Record instance.
  Builder(const llvm::Record *record, ArrayRef<SMLoc> loc);
 
  /// Return a list of parameters used in this build method.
  ArrayRef<Parameter> getParameters() const { return parameters; }
 
  /// Return an optional string containing the body of the builder.
  std::optional<StringRef> getBody() const;
 
  /// Return the deprecation message of the builder.
  /// Empty optional if the builder is not deprecated.
  std::optional<StringRef> getDeprecatedMessage() const;
 
protected:
  /// The TableGen definition of this builder.
  const llvm::Record *def;
 
private:
  /// A collection of parameters to the builder.
  SmallVector<Parameter> parameters;
};
 
} // namespace tblgen
} // namespace mlir
 
#endif // MLIR_TABLEGEN_BUILDER_H_