1 //===- Type.cpp - Type class ----------------------------------------------===//
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     : Constraint(Constraint::CK_Type, record) {
29   assert(def->isSubClassOf("TypeConstraint") &&
30          "must be subclass of TableGen 'TypeConstraint' class");
31 }
32 
33 tblgen::TypeConstraint::TypeConstraint(const llvm::DefInit *init)
34     : TypeConstraint(init->getDef()) {}
35 
36 bool tblgen::TypeConstraint::isVariadic() const {
37   return def->isSubClassOf("Variadic");
38 }
39 
40 tblgen::Type::Type(const llvm::Record *record) : TypeConstraint(record) {
41   assert(def->isSubClassOf("Type") &&
42          "must be subclass of TableGen 'Type' class");
43 }
44 
45 tblgen::Type::Type(const llvm::DefInit *init) : Type(init->getDef()) {}
46 
47 StringRef tblgen::Type::getTableGenDefName() const { return def->getName(); }
48 
49 tblgen::Type tblgen::Type::getVariadicBaseType() const {
50   assert(isVariadic() && "must be variadic type constraint");
51   return Type(def->getValueAsDef("baseType"));
52 }
53