1 //===- Type.cpp - Type class ------------------------------------*- C++ -*-===//
2 //
3 // Copyright 2019 The MLIR Authors.
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 //   http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 // =============================================================================
17 //
18 // Type wrapper to simplify using TableGen Record defining a MLIR Type.
19 //
20 //===----------------------------------------------------------------------===//
21 
22 #include "mlir/TableGen/Type.h"
23 #include "llvm/TableGen/Record.h"
24 
25 using namespace mlir;
26 
27 tblgen::TypeConstraint::TypeConstraint(const llvm::Record &record)
28     : def(&record) {
29   assert(def->isSubClassOf("TypeConstraint") &&
30          "must be subclass of TableGen 'TypeConstraint' class");
31 }
32 
33 tblgen::Pred tblgen::TypeConstraint::getPredicate() const {
34   auto *val = def->getValue("predicate");
35   assert(val &&
36          "TableGen 'TypeConstraint' class should have 'predicate' field");
37 
38   const auto *pred = dyn_cast<llvm::DefInit>(val->getValue());
39   return Pred(pred);
40 }
41 
42 std::string tblgen::TypeConstraint::getConditionTemplate() const {
43   return getPredicate().getCondition();
44 }
45 
46 llvm::StringRef tblgen::TypeConstraint::getDescription() const {
47   auto doc = def->getValueAsString("description");
48   if (doc.empty())
49     return def->getName();
50   return doc;
51 }
52 
53 tblgen::TypeConstraint::TypeConstraint(const llvm::DefInit &init)
54     : TypeConstraint(*init.getDef()) {}
55 
56 tblgen::Type::Type(const llvm::Record &record) : TypeConstraint(record) {
57   assert(def->isSubClassOf("Type") &&
58          "must be subclass of TableGen 'Type' class");
59 }
60 
61 tblgen::Type::Type(const llvm::DefInit *init) : Type(*init->getDef()) {}
62 
63 StringRef tblgen::Type::getTableGenDefName() const { return def->getName(); }
64 
65 bool tblgen::Type::isVariadic() const { return def->isSubClassOf("Variadic"); }
66 
67 tblgen::Type tblgen::Type::getVariadicBaseType() const {
68   assert(isVariadic() && "must be variadic type");
69   return Type(def->getValueAsDef("baseType"));
70 }
71