1 //===- Predicate.cpp - Predicate 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 // Wrapper around predicates defined in TableGen.
19 //
20 //===----------------------------------------------------------------------===//
21 
22 #include "mlir/TableGen/Predicate.h"
23 #include "llvm/ADT/SetVector.h"
24 #include "llvm/ADT/StringExtras.h"
25 #include "llvm/Support/FormatVariadic.h"
26 #include "llvm/TableGen/Error.h"
27 #include "llvm/TableGen/Record.h"
28 
29 using namespace mlir;
30 
31 // Construct a Predicate from a record.
32 tblgen::Pred::Pred(const llvm::Record *def) : def(def) {
33   assert(def->isSubClassOf("Pred") &&
34          "must be a subclass of TableGen 'Pred' class");
35 }
36 
37 // Construct a Predicate from an initializer.
38 tblgen::Pred::Pred(const llvm::Init *init) : def(nullptr) {
39   if (const auto *defInit = dyn_cast_or_null<llvm::DefInit>(init))
40     def = defInit->getDef();
41 }
42 
43 // Get condition of the Predicate.
44 StringRef tblgen::Pred::getCondition() const {
45   assert(!isNull() && "null predicate does not have a condition");
46   return def->getValueAsString("predCall");
47 }
48