Dialect.h

Go to the documentation of this file.

//===- Dialect.h - PDLL ODS Dialect -----------------------------*- 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_TOOLS_PDLL_ODS_DIALECT_H_
#define MLIR_TOOLS_PDLL_ODS_DIALECT_H_
 
#include <string>
 
#include "mlir/Support/LLVM.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/StringMap.h"
 
namespace mlir {
namespace pdll {
namespace ods {
class Operation;
 
/// This class represents an ODS dialect, and contains information on the
/// constructs held within the dialect.
class Dialect {
public:
  ~Dialect();
 
  /// Return the name of this dialect.
  StringRef getName() const { return name; }
 
  /// Insert a new operation with the dialect. Returns the inserted operation,
  /// and a boolean indicating if the operation newly inserted (false if the
  /// operation already existed).
  std::pair<Operation *, bool>
  insertOperation(StringRef name, StringRef summary, StringRef desc,
                  StringRef nativeClassName, bool supportsResultTypeInferrence,
                  SMLoc loc);
 
  /// Lookup an operation registered with the given name, or null if no
  /// operation with that name is registered.
  Operation *lookupOperation(StringRef name) const;
 
  /// Return a map of all of the operations registered to this dialect.
  const llvm::StringMap<std::unique_ptr<Operation>> &getOperations() const {
    return operations;
  }
 
private:
  explicit Dialect(StringRef name);
 
  /// The name of the dialect.
  std::string name;
 
  /// The operations defined by the dialect.
  llvm::StringMap<std::unique_ptr<Operation>> operations;
 
  /// Allow access to the constructor.
  friend class Context;
};
} // namespace ods
} // namespace pdll
} // namespace mlir
 
#endif // MLIR_TOOLS_PDLL_ODS_DIALECT_H_