Support.h

Go to the documentation of this file.

//===-- Support.h - C API general utilities ---------------------*- C++ -*-===//
//
// Part of the LLZK Project, under the Apache License v2.0.
// See LICENSE.txt for license information.
// Copyright 2025 Veridise Inc.
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//
//
// This header declares utility macros for working with the C API from the
// C++ side.
//
//===----------------------------------------------------------------------===//
 
#pragma once
 
#include "llzk-c/Support.h"
 
#include "llzk/Dialect/LLZK/IR/AttributeHelper.h"
 
#include <mlir-c/IR.h>
 
#include <mlir/CAPI/IR.h>
#include <mlir/IR/OperationSupport.h>
#include <mlir/IR/ValueRange.h>
 
#define LLZK_DEFINE_SUFFIX_OP_BUILD_METHOD(dialect, op, suffix, ...)                               \
  MlirOperation llzk##dialect##_##op##Build##suffix(                                               \
      MlirOpBuilder builder, MlirLocation location, __VA_ARGS__                                    \
  )
#define LLZK_DEFINE_OP_BUILD_METHOD(dialect, op, ...)                                              \
  LLZK_DEFINE_SUFFIX_OP_BUILD_METHOD(dialect, op, , __VA_ARGS__)
 
//===----------------------------------------------------------------------===//
// Custom wrap/unwrap functions for C API types
//===----------------------------------------------------------------------===//
 
static inline int64_t wrap(const llvm::APInt &cpp) { return llzk::fromAPInt(cpp); }
 
static inline llvm::APInt unwrap(int64_t c) { return llzk::toAPInt(c); }
 
static inline MlirNamedAttribute wrap(mlir::NamedAttribute attr) {
  return MlirNamedAttribute {wrap(attr.getName()), wrap(attr.getValue())};
}
 
static inline mlir::NamedAttribute unwrap(MlirNamedAttribute attr) {
  return mlir::NamedAttribute(unwrap(attr.name), unwrap(attr.attribute));
}
 
namespace mlir {
template <typename To> inline auto unwrap_cast(auto &from) { return cast<To>(unwrap(from)); }
} // namespace mlir
 
//===----------------------------------------------------------------------===//
// Other helpers
//===----------------------------------------------------------------------===//
 
constexpr int DEFAULT_ELTS = 5;

template <int N = DEFAULT_ELTS> class MapOperandsHelper {
public:
  MapOperandsHelper(intptr_t nMapOperands, MlirValueRange const *mapOperands) {
    // resize allocates elements (needed for operator[] access on line 44)
    storage.resize(nMapOperands);
    ranges.reserve(nMapOperands);
 
    for (intptr_t i = 0; i < nMapOperands; i++) {
      mlir::SmallVector<mlir::Value, N> &sto = storage[i];
      MlirValueRange ops = mapOperands[i];
      ranges.push_back(mlir::ValueRange(unwrapList(ops.size, ops.values, sto)));
    }
  }
 
  mlir::ArrayRef<mlir::ValueRange> operator*() const { return ranges; }
 
private:
  mlir::SmallVector<mlir::SmallVector<mlir::Value, N>, N> storage;
  mlir::SmallVector<mlir::ValueRange, N> ranges;
};