OMAttributes.cpp

Go to the documentation of this file.

//===- OMAttributes.cpp - Object Model attribute definitions --------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// This file contains the Object Model attribute definitions.
//
//===----------------------------------------------------------------------===//
 
#include "circt/Dialect/OM/OMAttributes.h"
#include "circt/Dialect/OM/OMDialect.h"
#include "circt/Dialect/OM/OMTypes.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/DialectImplementation.h"
#include "llvm/ADT/TypeSwitch.h"
 
using namespace mlir;
using namespace circt::om;
 
#define GET_ATTRDEF_CLASSES
#include "circt/Dialect/OM/OMAttributes.cpp.inc"
 
Type circt::om::ReferenceAttr::getType() {
  return ReferenceType::get(getContext());
}
 
Type circt::om::SymbolRefAttr::getType() {
  return SymbolRefType::get(getContext());
}
 
Type circt::om::ListAttr::getType() {
  return ListType::get(getContext(), getElementType());
}
 
Type circt::om::MapAttr::getType() {
  return MapType::get(getContext(), StringType::get(getContext()),
                      getValueType());
}
 
circt::om::SymbolRefAttr circt::om::SymbolRefAttr::get(mlir::Operation *op) {
  return om::SymbolRefAttr::get(op->getContext(),
                                mlir::FlatSymbolRefAttr::get(op));
}
 
circt::om::SymbolRefAttr
circt::om::SymbolRefAttr::get(mlir::StringAttr symName) {
  return om::SymbolRefAttr::get(symName.getContext(),
                                mlir::FlatSymbolRefAttr::get(symName));
}
 
LogicalResult
circt::om::ListAttr::verify(function_ref<InFlightDiagnostic()> emitError,
                            mlir::Type elementType, mlir::ArrayAttr elements) {
  return success(llvm::all_of(elements, [&](mlir::Attribute attr) {
    auto typedAttr = llvm::dyn_cast<mlir::TypedAttr>(attr);
    if (!typedAttr) {
      emitError()
          << "an element of a list attribute must be a typed attr but got "
          << attr;
      return false;
    }
    if (typedAttr.getType() != elementType) {
      emitError() << "an element of a list attribute must have a type "
                  << elementType << " but got " << typedAttr.getType();
      return false;
    }
 
    return true;
  }));
}
 
LogicalResult
circt::om::MapAttr::verify(function_ref<InFlightDiagnostic()> emitError,
                           mlir::Type valueType,
                           mlir::DictionaryAttr elements) {
  for (auto attr : elements) {
    auto typedAttr = llvm::dyn_cast<mlir::TypedAttr>(attr.getValue());
    if (!typedAttr)
      return emitError()
             << "a value of a map attribute must be a typed attr but got "
             << attr.getValue();
    if (typedAttr.getType() != valueType)
      return emitError() << "a value of a map attribute must have a type "
                         << valueType << " but field " << attr.getName()
                         << " has " << typedAttr.getType();
  }
  return success();
}
 
void PathAttr::print(AsmPrinter &odsPrinter) const {
  odsPrinter << '[';
  llvm::interleaveComma(getPath(), odsPrinter, [&](PathElement element) {
    odsPrinter.printKeywordOrString(element.module);
    odsPrinter << ':';
    odsPrinter.printKeywordOrString(element.instance);
  });
  odsPrinter << ']';
}
 
Attribute PathAttr::parse(AsmParser &odsParser, Type odsType) {
  auto *context = odsParser.getContext();
  SmallVector<PathElement> path;
  if (odsParser.parseCommaSeparatedList(
          OpAsmParser::Delimiter::Square, [&]() -> ParseResult {
            std::string module;
            std::string instance;
            if (odsParser.parseKeywordOrString(&module) ||
                odsParser.parseColon() ||
                odsParser.parseKeywordOrString(&instance))
              return failure();
            path.emplace_back(StringAttr::get(context, module),
                              StringAttr::get(context, instance));
            return success();
          }))
    return nullptr;
  return PathAttr::get(context, path);
}
 
LogicalResult PathAttr::verify(function_ref<mlir::InFlightDiagnostic()>,
                               ArrayRef<PathElement> path) {
  return success();
}
 
Type circt::om::IntegerAttr::getType() {
  return OMIntegerType::get(getContext());
}
 
void circt::om::OMDialect::registerAttributes() {
  addAttributes<
#define GET_ATTRDEF_LIST
#include "circt/Dialect/OM/OMAttributes.cpp.inc"
      >();
}