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/StringExtras.h"
24 #include "llvm/Support/FormatVariadic.h"
25 #include "llvm/TableGen/Error.h"
26 #include "llvm/TableGen/Record.h"
27 
28 using namespace mlir;
29 
30 std::string mlir::PredCNF::createTypeMatcherTemplate() const {
31   std::string outString;
32   llvm::raw_string_ostream ss(outString);
33   auto conjunctiveList = conditions;
34   bool firstDisjunctive = true;
35   for (auto disjunctiveInit : *conjunctiveList) {
36     ss << (firstDisjunctive ? "(" : " && (");
37     firstDisjunctive = false;
38     bool firstConjunctive = true;
39     for (auto atom : *cast<llvm::ListInit>(disjunctiveInit)) {
40       auto predAtom = cast<llvm::DefInit>(atom)->getDef();
41       ss << (firstConjunctive ? "" : " || ")
42          << (predAtom->getValueAsBit("negated") ? "!" : "")
43          << predAtom->getValueAsString("predCall");
44       firstConjunctive = false;
45     }
46     ss << ")";
47   }
48   ss.flush();
49   return outString;
50 }
51