From 7e7e8fc07a1f82b14a9918fe9834333b819d3d7f Mon Sep 17 00:00:00 2001
From: Ben <ben.curtis@zama.ai>
Date: Tue, 10 Aug 2021 09:31:39 +0100
Subject: [PATCH] testing

---
 cpp/gen_cpp.py | 169 +++++++++++++++++++++++++++++++++++++++++++++++++
 cpp/v0curves   |   1 +
 gen_cpp.py     |   1 -
 3 files changed, 170 insertions(+), 1 deletion(-)
 create mode 100644 cpp/gen_cpp.py
 create mode 100644 cpp/v0curves
 delete mode 100644 gen_cpp.py

diff --git a/cpp/gen_cpp.py b/cpp/gen_cpp.py
new file mode 100644
index 000000000..7d5819800
--- /dev/null
+++ b/cpp/gen_cpp.py
@@ -0,0 +1,169 @@
+from V0Parameters.compile_test import check_codegen
+from V0Parameters.tabulation import main_optimization_v0
+from V0Parameters.compile_test import check_codegen
+from V0Parameters.misc import NORM2_MAX, P_MAX, v0_parameters_path
+import numpy as np
+
+import_string = f"""
+
+#include <iostream>
+using namespace std;"""
+
+
+constant_string = f"""
+const int num_sec_levels = {num_sec_levels};"""
+
+struct_string = """
+typedef struct V0Parameter
+{
+    int k;
+    int polynomialSize;
+    int nSmall;
+    int brLevel;
+    int brLogBase;
+    int ksLevel;
+    int ksLogBase;
+
+    V0Parameter(int k_,
+                int polynomialSize_,
+                int nSmall_,
+                int brLevel_,
+                int brLogBase_,
+                int ksLevel_,
+                int ksLogBase_)
+    {
+        k = k_;
+        polynomialSize = polynomialSize_;
+        nSmall = nSmall_;
+        brLevel = brLevel_;
+        brLogBase = brLogBase_;
+        ksLevel = ksLevel_;
+        ksLogBase = ksLogBase_;
+    }
+
+} V0Parameter;"""
+
+
+table_string = """
+V0Parameter parameters[NORM2_MAX][P_MAX] = """
+
+get_string = """
+extern "C" V0Parameter *get(int norm, int p)
+{
+    // - 1 is an offset as norm and p are in [1, ...] and not [0, ...]
+    return &parameters[norm - 1][p - 1];
+}"""
+
+
+def constructor(k, polynomialSize, nSmall, brLevel, brLogBase, ksLevel, ksLogBase):
+    return f"V0Parameter({k}, {polynomialSize}, {nSmall}, {brLevel}, {brLogBase}, {ksLevel}, {ksLogBase}),"
+
+
+def fill_parameters(
+    # Return a string with parameters for the c++ array initialization
+    polynomial_size_results,
+    rlwe_dimension_results,
+):
+    parameters = "{}{{".format(table_string)
+    for norm in range(1, NORM2_MAX + 1):
+        line = "{"
+        for p in range(1, P_MAX + 1):
+            try:
+                line += constructor(
+                    1,
+                    int(polynomial_size_results[(norm, p)]),
+                    int(lwe_dimension_results[(norm, p)]),
+                    int(br_level_results[(norm, p)]),
+                    int(br_logbase_results[(norm, p)]),
+                    int(ks_level_results[(norm, p)]),
+                    int(ks_logbase_results[(norm, p)]),
+                )
+            except ValueError:
+                line += constructor(0, 0, 0, 0, 0, 0, 0)
+        line = line[:-1]
+        line += "},"
+        parameters += line
+    parameters = parameters[:-1]
+    parameters += "} ;"
+    return parameters
+
+
+def codegen(
+    polynomial_size_results,
+    lwe_dimension_results,
+    ks_base_results,
+    ks_level_results,
+    br_base_results,
+    br_level_results,
+):
+    # Generate the C++ file as a string
+    code = f"""
+    {import_string}
+    {constant_string}
+    {struct_string} 
+    {fill_parameters(
+        polynomial_size_results,
+        lwe_dimension_results,
+        ks_base_results,
+        ks_level_results,
+        br_base_results,
+        br_level_results,
+    )}
+    {get_string}
+    """
+    return code
+
+
+def write_codegen(
+    polynomial_size_results,
+    lwe_dimension_results,
+    ks_base_results,
+    ks_level_results,
+    br_base_results,
+    br_level_results,
+):
+    # Create the c++ source
+    code = codegen(
+        polynomial_size_results,
+        lwe_dimension_results,
+        ks_base_results,
+        ks_level_results,
+        br_base_results,
+        br_level_results,
+    )
+
+    with open(f"{v0_parameters_path}/parameters.cpp", "w") as f:
+        f.write(code)
+    print("> Successfully wrote C++ source to disk")
+
+
+def main_codegen():
+    # finding parameters for V0
+    (
+        polynomial_size_results,
+        lwe_dimension_results,
+        ks_base_results,
+        ks_level_results,
+        br_base_results,
+        br_level_results,
+    ) = main_optimization_v0()
+
+    # code generation
+    write_codegen(
+        polynomial_size_results,
+        lwe_dimension_results,
+        ks_base_results,
+        ks_level_results,
+        br_base_results,
+        br_level_results,
+    )
+
+    # check code gen worked
+    check_codegen(
+        polynomial_size_results,
+        lwe_dimension_results,
+        ks_base_results,
+        ks_level_results,
+        br_base_results,
+        br_level_results,
+    )
diff --git a/cpp/v0curves b/cpp/v0curves
new file mode 100644
index 000000000..a82d1b41e
--- /dev/null
+++ b/cpp/v0curves
@@ -0,0 +1 @@
+#placeholder
diff --git a/gen_cpp.py b/gen_cpp.py
deleted file mode 100644
index fdffa2a0f..000000000
--- a/gen_cpp.py
+++ /dev/null
@@ -1 +0,0 @@
-# placeholder