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::Type::Type(const llvm::Record &def) : def(def) {
28   assert(def.isSubClassOf("Type") &&
29          "must be subclass of TableGen 'Type' class");
30 }
31 
32 tblgen::Type::Type(const llvm::DefInit *init) : Type(*init->getDef()) {}
33 
34 StringRef tblgen::Type::getTableGenDefName() const { return def.getName(); }
35 
36 StringRef tblgen::Type::getBuilderCall() const {
37   const auto *val = def.getValue("builderCall");
38   assert(val && "TableGen 'Type' class should have 'builderCall' field");
39 
40   if (const auto *builder = dyn_cast<llvm::CodeInit>(val->getValue()))
41     return builder->getValue();
42   return {};
43 }
44 
45 tblgen::PredCNF tblgen::Type::getPredicate() const {
46   auto *val = def.getValue("predicate");
47   assert(val && "TableGen 'Type' class should have 'predicate' field");
48 
49   const auto *pred = dyn_cast<llvm::DefInit>(val->getValue());
50   return PredCNF(pred);
51 }
52