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 && "TableGen 'Type' class should have 'predicate' field");
36 
37   const auto *pred = dyn_cast<llvm::DefInit>(val->getValue());
38   return Pred(pred);
39 }
40 
41 llvm::StringRef tblgen::TypeConstraint::getConditionTemplate() const {
42   return getPredicate().getCondition();
43 }
44 
45 llvm::StringRef tblgen::TypeConstraint::getDescription() const {
46   const static auto fieldName = "description";
47   auto *val = def.getValue(fieldName);
48   if (!val)
49     return "";
50 
51   return def.getValueAsString(fieldName);
52 }
53 
54 tblgen::TypeConstraint::TypeConstraint(const llvm::DefInit &init)
55     : def(*init.getDef()) {}
56 
57 tblgen::Type::Type(const llvm::Record &record) : TypeConstraint(record) {
58   assert(def.isSubClassOf("Type") &&
59          "must be subclass of TableGen 'Type' class");
60 }
61 
62 tblgen::Type::Type(const llvm::DefInit *init) : Type(*init->getDef()) {}
63 
64 StringRef tblgen::Type::getTableGenDefName() const { return def.getName(); }
65 
66 StringRef tblgen::Type::getBuilderCall() const {
67   const auto *val = def.getValue("builderCall");
68   assert(val && "TableGen 'Type' class should have 'builderCall' field");
69 
70   if (const auto *builder = dyn_cast<llvm::CodeInit>(val->getValue()))
71     return builder->getValue();
72   return {};
73 }
74 
75