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   const static auto fieldName = "description";
48   auto *val = def.getValue(fieldName);
49   if (!val)
50     return "";
51 
52   return def.getValueAsString(fieldName);
53 }
54 
55 tblgen::TypeConstraint::TypeConstraint(const llvm::DefInit &init)
56     : TypeConstraint(*init.getDef()) {}
57 
58 tblgen::Type::Type(const llvm::Record &record) : TypeConstraint(record) {
59   assert(def.isSubClassOf("Type") &&
60          "must be subclass of TableGen 'Type' class");
61 }
62 
63 tblgen::Type::Type(const llvm::DefInit *init) : Type(*init->getDef()) {}
64 
65 StringRef tblgen::Type::getTableGenDefName() const { return def.getName(); }
66