mlir/lib/CAPI/IR/IntegerSet.cpp

   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
  19 MlirContext mlirIntegerSetGetContext(MlirIntegerSet set) {
  20   return wrap(unwrap(set).getContext());
  21 }
  22
  23 bool mlirIntegerSetEqual(MlirIntegerSet s1, MlirIntegerSet s2) {
  24   return unwrap(s1) == unwrap(s2);
  25 }
  26
  27 void mlirIntegerSetPrint(MlirIntegerSet set, MlirStringCallback callback,
  28                          void *userData) {
  29   mlir::detail::CallbackOstream stream(callback, userData);
  30   unwrap(set).print(stream);
  31 }
  32
  33 void mlirIntegerSetDump(MlirIntegerSet set) { unwrap(set).dump(); }
  34
  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
  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
  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
  69 bool mlirIntegerSetIsCanonicalEmpty(MlirIntegerSet set) {
  70   return unwrap(set).isEmptyIntegerSet();
  71 }
  72
  73 intptr_t mlirIntegerSetGetNumDims(MlirIntegerSet set) {
  74   return static_cast<intptr_t>(unwrap(set).getNumDims());
  75 }
  76
  77 intptr_t mlirIntegerSetGetNumSymbols(MlirIntegerSet set) {
  78   return static_cast<intptr_t>(unwrap(set).getNumSymbols());
  79 }
  80
  81 intptr_t mlirIntegerSetGetNumInputs(MlirIntegerSet set) {
  82   return static_cast<intptr_t>(unwrap(set).getNumInputs());
  83 }
  84
  85 intptr_t mlirIntegerSetGetNumConstraints(MlirIntegerSet set) {
  86   return static_cast<intptr_t>(unwrap(set).getNumConstraints());
  87 }
  88
  89 intptr_t mlirIntegerSetGetNumEqualities(MlirIntegerSet set) {
  90   return static_cast<intptr_t>(unwrap(set).getNumEqualities());
  91 }
  92
  93 intptr_t mlirIntegerSetGetNumInequalities(MlirIntegerSet set) {
  94   return static_cast<intptr_t>(unwrap(set).getNumInequalities());
  95 }
  96
  97 MlirAffineExpr mlirIntegerSetGetConstraint(MlirIntegerSet set, intptr_t pos) {
  98   return wrap(unwrap(set).getConstraint(static_cast<unsigned>(pos)));
  99 }
 100
 101 bool mlirIntegerSetIsConstraintEq(MlirIntegerSet set, intptr_t pos) {
 102   return unwrap(set).isEq(pos);
 103 }