From f80c6578496f342703a871f45c70749cde4f5a71 Mon Sep 17 00:00:00 2001
From: rudy <rudy.sicard@gmail.com>
Date: Mon, 24 Oct 2022 10:45:29 +0200
Subject: [PATCH] feat(cpp): access to round operator

---
 concrete-optimizer-cpp/src/concrete-optimizer.rs   | 14 ++++++++++++++
 .../src/cpp/concrete-optimizer.cpp                 |  7 +++++++
 .../src/cpp/concrete-optimizer.hpp                 |  1 +
 3 files changed, 22 insertions(+)

diff --git a/concrete-optimizer-cpp/src/concrete-optimizer.rs b/concrete-optimizer-cpp/src/concrete-optimizer.rs
index cf768481e..e90334620 100644
--- a/concrete-optimizer-cpp/src/concrete-optimizer.rs
+++ b/concrete-optimizer-cpp/src/concrete-optimizer.rs
@@ -208,6 +208,14 @@ impl OperationDag {
             .into()
     }
 
+    fn add_round_op(
+        &mut self,
+        input: ffi::OperatorIndex,
+        rounded_precision: Precision,
+    ) -> ffi::OperatorIndex {
+        self.0.add_round_op(input.into(), rounded_precision).into()
+    }
+
     fn optimize_v0(&self, options: ffi::Options) -> ffi::Solution {
         let processing_unit = processing_unit(options);
 
@@ -345,6 +353,12 @@ mod ffi {
             comment: &str,
         ) -> OperatorIndex;
 
+        fn add_round_op(
+            self: &mut OperationDag,
+            input: OperatorIndex,
+            rounded_precision: u8,
+        ) -> OperatorIndex;
+
         fn optimize_v0(self: &OperationDag, options: Options) -> Solution;
 
         fn optimize(self: &OperationDag, options: Options) -> DagSolution;
diff --git a/concrete-optimizer-cpp/src/cpp/concrete-optimizer.cpp b/concrete-optimizer-cpp/src/cpp/concrete-optimizer.cpp
index a98abce1c..02543d865 100644
--- a/concrete-optimizer-cpp/src/cpp/concrete-optimizer.cpp
+++ b/concrete-optimizer-cpp/src/cpp/concrete-optimizer.cpp
@@ -960,6 +960,7 @@ struct OperationDag final : public ::rust::Opaque {
   ::concrete_optimizer::dag::OperatorIndex add_lut(::concrete_optimizer::dag::OperatorIndex input, ::rust::Slice<::std::uint64_t const> table, ::std::uint8_t out_precision) noexcept;
   ::concrete_optimizer::dag::OperatorIndex add_dot(::rust::Slice<::concrete_optimizer::dag::OperatorIndex const> inputs, ::rust::Box<::concrete_optimizer::Weights> weights) noexcept;
   ::concrete_optimizer::dag::OperatorIndex add_levelled_op(::rust::Slice<::concrete_optimizer::dag::OperatorIndex const> inputs, double lwe_dim_cost_factor, double fixed_cost, double manp, ::rust::Slice<::std::uint64_t const> out_shape, ::rust::Str comment) noexcept;
+  ::concrete_optimizer::dag::OperatorIndex add_round_op(::concrete_optimizer::dag::OperatorIndex input, ::std::uint8_t rounded_precision) noexcept;
   ::concrete_optimizer::v0::Solution optimize_v0(::concrete_optimizer::Options options) const noexcept;
   ::concrete_optimizer::dag::DagSolution optimize(::concrete_optimizer::Options options) const noexcept;
   ::rust::String dump() const noexcept;
@@ -1100,6 +1101,8 @@ extern "C" {
 
 ::concrete_optimizer::dag::OperatorIndex concrete_optimizer$cxxbridge1$OperationDag$add_levelled_op(::concrete_optimizer::OperationDag &self, ::rust::Slice<::concrete_optimizer::dag::OperatorIndex const> inputs, double lwe_dim_cost_factor, double fixed_cost, double manp, ::rust::Slice<::std::uint64_t const> out_shape, ::rust::Str comment) noexcept;
 
+::concrete_optimizer::dag::OperatorIndex concrete_optimizer$cxxbridge1$OperationDag$add_round_op(::concrete_optimizer::OperationDag &self, ::concrete_optimizer::dag::OperatorIndex input, ::std::uint8_t rounded_precision) noexcept;
+
 ::concrete_optimizer::v0::Solution concrete_optimizer$cxxbridge1$OperationDag$optimize_v0(::concrete_optimizer::OperationDag const &self, ::concrete_optimizer::Options options) noexcept;
 
 void concrete_optimizer$cxxbridge1$OperationDag$optimize(::concrete_optimizer::OperationDag const &self, ::concrete_optimizer::Options options, ::concrete_optimizer::dag::DagSolution *return$) noexcept;
@@ -1159,6 +1162,10 @@ namespace dag {
   return concrete_optimizer$cxxbridge1$OperationDag$add_levelled_op(*this, inputs, lwe_dim_cost_factor, fixed_cost, manp, out_shape, comment);
 }
 
+::concrete_optimizer::dag::OperatorIndex OperationDag::add_round_op(::concrete_optimizer::dag::OperatorIndex input, ::std::uint8_t rounded_precision) noexcept {
+  return concrete_optimizer$cxxbridge1$OperationDag$add_round_op(*this, input, rounded_precision);
+}
+
 ::concrete_optimizer::v0::Solution OperationDag::optimize_v0(::concrete_optimizer::Options options) const noexcept {
   return concrete_optimizer$cxxbridge1$OperationDag$optimize_v0(*this, options);
 }
diff --git a/concrete-optimizer-cpp/src/cpp/concrete-optimizer.hpp b/concrete-optimizer-cpp/src/cpp/concrete-optimizer.hpp
index 7c5a7fa1b..1f86726ae 100644
--- a/concrete-optimizer-cpp/src/cpp/concrete-optimizer.hpp
+++ b/concrete-optimizer-cpp/src/cpp/concrete-optimizer.hpp
@@ -941,6 +941,7 @@ struct OperationDag final : public ::rust::Opaque {
   ::concrete_optimizer::dag::OperatorIndex add_lut(::concrete_optimizer::dag::OperatorIndex input, ::rust::Slice<::std::uint64_t const> table, ::std::uint8_t out_precision) noexcept;
   ::concrete_optimizer::dag::OperatorIndex add_dot(::rust::Slice<::concrete_optimizer::dag::OperatorIndex const> inputs, ::rust::Box<::concrete_optimizer::Weights> weights) noexcept;
   ::concrete_optimizer::dag::OperatorIndex add_levelled_op(::rust::Slice<::concrete_optimizer::dag::OperatorIndex const> inputs, double lwe_dim_cost_factor, double fixed_cost, double manp, ::rust::Slice<::std::uint64_t const> out_shape, ::rust::Str comment) noexcept;
+  ::concrete_optimizer::dag::OperatorIndex add_round_op(::concrete_optimizer::dag::OperatorIndex input, ::std::uint8_t rounded_precision) noexcept;
   ::concrete_optimizer::v0::Solution optimize_v0(::concrete_optimizer::Options options) const noexcept;
   ::concrete_optimizer::dag::DagSolution optimize(::concrete_optimizer::Options options) const noexcept;
   ::rust::String dump() const noexcept;