1 //===- IntegerSet.cpp - C API for MLIR Integer Sets -----------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "mlir-c/IntegerSet.h"
10 #include "mlir-c/AffineExpr.h"
11 #include "mlir/CAPI/AffineExpr.h"
12 #include "mlir/CAPI/IR.h"
13 #include "mlir/CAPI/IntegerSet.h"
14 #include "mlir/CAPI/Utils.h"
15 #include "mlir/IR/IntegerSet.h"
16
17 using namespace mlir;
18
mlirIntegerSetGetContext(MlirIntegerSet set)19 MlirContext mlirIntegerSetGetContext(MlirIntegerSet set) {
20 return wrap(unwrap(set).getContext());
21 }
22
mlirIntegerSetEqual(MlirIntegerSet s1,MlirIntegerSet s2)23 bool mlirIntegerSetEqual(MlirIntegerSet s1, MlirIntegerSet s2) {
24 return unwrap(s1) == unwrap(s2);
25 }
26
mlirIntegerSetPrint(MlirIntegerSet set,MlirStringCallback callback,void * userData)27 void mlirIntegerSetPrint(MlirIntegerSet set, MlirStringCallback callback,
28 void *userData) {
29 mlir::detail::CallbackOstream stream(callback, userData);
30 unwrap(set).print(stream);
31 }
32
mlirIntegerSetDump(MlirIntegerSet set)33 void mlirIntegerSetDump(MlirIntegerSet set) { unwrap(set).dump(); }
34
mlirIntegerSetEmptyGet(MlirContext context,intptr_t numDims,intptr_t numSymbols)35 MlirIntegerSet mlirIntegerSetEmptyGet(MlirContext context, intptr_t numDims,
36 intptr_t numSymbols) {
37 return wrap(IntegerSet::getEmptySet(static_cast<unsigned>(numDims),
38 static_cast<unsigned>(numSymbols),
39 unwrap(context)));
40 }
41
mlirIntegerSetGet(MlirContext context,intptr_t numDims,intptr_t numSymbols,intptr_t numConstraints,const MlirAffineExpr * constraints,const bool * eqFlags)42 MlirIntegerSet mlirIntegerSetGet(MlirContext context, intptr_t numDims,
43 intptr_t numSymbols, intptr_t numConstraints,
44 const MlirAffineExpr *constraints,
45 const bool *eqFlags) {
46 SmallVector<AffineExpr> mlirConstraints;
47 (void)unwrapList(static_cast<size_t>(numConstraints), constraints,
48 mlirConstraints);
49 return wrap(IntegerSet::get(
50 static_cast<unsigned>(numDims), static_cast<unsigned>(numSymbols),
51 mlirConstraints,
52 llvm::ArrayRef(eqFlags, static_cast<size_t>(numConstraints))));
53 }
54
55 MlirIntegerSet
mlirIntegerSetReplaceGet(MlirIntegerSet set,const MlirAffineExpr * dimReplacements,const MlirAffineExpr * symbolReplacements,intptr_t numResultDims,intptr_t numResultSymbols)56 mlirIntegerSetReplaceGet(MlirIntegerSet set,
57 const MlirAffineExpr *dimReplacements,
58 const MlirAffineExpr *symbolReplacements,
59 intptr_t numResultDims, intptr_t numResultSymbols) {
60 SmallVector<AffineExpr> mlirDims, mlirSymbols;
61 (void)unwrapList(unwrap(set).getNumDims(), dimReplacements, mlirDims);
62 (void)unwrapList(unwrap(set).getNumSymbols(), symbolReplacements,
63 mlirSymbols);
64 return wrap(unwrap(set).replaceDimsAndSymbols(
65 mlirDims, mlirSymbols, static_cast<unsigned>(numResultDims),
66 static_cast<unsigned>(numResultSymbols)));
67 }
68
mlirIntegerSetIsCanonicalEmpty(MlirIntegerSet set)69 bool mlirIntegerSetIsCanonicalEmpty(MlirIntegerSet set) {
70 return unwrap(set).isEmptyIntegerSet();
71 }
72
mlirIntegerSetGetNumDims(MlirIntegerSet set)73 intptr_t mlirIntegerSetGetNumDims(MlirIntegerSet set) {
74 return static_cast<intptr_t>(unwrap(set).getNumDims());
75 }
76
mlirIntegerSetGetNumSymbols(MlirIntegerSet set)77 intptr_t mlirIntegerSetGetNumSymbols(MlirIntegerSet set) {
78 return static_cast<intptr_t>(unwrap(set).getNumSymbols());
79 }
80
mlirIntegerSetGetNumInputs(MlirIntegerSet set)81 intptr_t mlirIntegerSetGetNumInputs(MlirIntegerSet set) {
82 return static_cast<intptr_t>(unwrap(set).getNumInputs());
83 }
84
mlirIntegerSetGetNumConstraints(MlirIntegerSet set)85 intptr_t mlirIntegerSetGetNumConstraints(MlirIntegerSet set) {
86 return static_cast<intptr_t>(unwrap(set).getNumConstraints());
87 }
88
mlirIntegerSetGetNumEqualities(MlirIntegerSet set)89 intptr_t mlirIntegerSetGetNumEqualities(MlirIntegerSet set) {
90 return static_cast<intptr_t>(unwrap(set).getNumEqualities());
91 }
92
mlirIntegerSetGetNumInequalities(MlirIntegerSet set)93 intptr_t mlirIntegerSetGetNumInequalities(MlirIntegerSet set) {
94 return static_cast<intptr_t>(unwrap(set).getNumInequalities());
95 }
96
mlirIntegerSetGetConstraint(MlirIntegerSet set,intptr_t pos)97 MlirAffineExpr mlirIntegerSetGetConstraint(MlirIntegerSet set, intptr_t pos) {
98 return wrap(unwrap(set).getConstraint(static_cast<unsigned>(pos)));
99 }
100
mlirIntegerSetIsConstraintEq(MlirIntegerSet set,intptr_t pos)101 bool mlirIntegerSetIsConstraintEq(MlirIntegerSet set, intptr_t pos) {
102 return unwrap(set).isEq(pos);
103 }
104