75 lines
2.9 KiB
C++
75 lines
2.9 KiB
C++
//===- Builder.cpp - Builder 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Builder.h"
|
|
#include "llvm/TableGen/Error.h"
|
|
#include "llvm/TableGen/Record.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::tblgen;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Builder::Parameter
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Return a string containing the C++ type of this parameter.
|
|
StringRef Builder::Parameter::getCppType() const {
|
|
if (const auto *stringInit = dyn_cast<llvm::StringInit>(def))
|
|
return stringInit->getValue();
|
|
const llvm::Record *record = cast<llvm::DefInit>(def)->getDef();
|
|
return record->getValueAsString("type");
|
|
}
|
|
|
|
/// Return an optional string containing the default value to use for this
|
|
/// parameter.
|
|
Optional<StringRef> Builder::Parameter::getDefaultValue() const {
|
|
if (isa<llvm::StringInit>(def))
|
|
return llvm::None;
|
|
const llvm::Record *record = cast<llvm::DefInit>(def)->getDef();
|
|
Optional<StringRef> value = record->getValueAsOptionalString("defaultValue");
|
|
return value && !value->empty() ? value : llvm::None;
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Builder
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
Builder::Builder(const llvm::Record *record, ArrayRef<llvm::SMLoc> loc)
|
|
: def(record) {
|
|
// Initialize the parameters of the builder.
|
|
const llvm::DagInit *dag = def->getValueAsDag("dagParams");
|
|
auto *defInit = dyn_cast<llvm::DefInit>(dag->getOperator());
|
|
if (!defInit || !defInit->getDef()->getName().equals("ins"))
|
|
PrintFatalError(def->getLoc(), "expected 'ins' in builders");
|
|
|
|
bool seenDefaultValue = false;
|
|
for (unsigned i = 0, e = dag->getNumArgs(); i < e; ++i) {
|
|
const llvm::StringInit *paramName = dag->getArgName(i);
|
|
const llvm::Init *paramValue = dag->getArg(i);
|
|
Parameter param(paramName ? paramName->getValue() : Optional<StringRef>(),
|
|
paramValue);
|
|
|
|
// Similarly to C++, once an argument with a default value is detected, the
|
|
// following arguments must have default values as well.
|
|
if (param.getDefaultValue()) {
|
|
seenDefaultValue = true;
|
|
} else if (seenDefaultValue) {
|
|
PrintFatalError(loc,
|
|
"expected an argument with default value after other "
|
|
"arguments with default values");
|
|
}
|
|
parameters.emplace_back(param);
|
|
}
|
|
}
|
|
|
|
/// Return an optional string containing the body of the builder.
|
|
Optional<StringRef> Builder::getBody() const {
|
|
Optional<StringRef> body = def->getValueAsOptionalString("body");
|
|
return body && !body->empty() ? body : llvm::None;
|
|
}
|