From 8836fa83480fe0781eb1af3d83274a00748a822f Mon Sep 17 00:00:00 2001
From: "Michael Bradley, Jr" <michaelsbradleyjr@gmail.com>
Date: Tue, 1 Mar 2022 15:14:56 -0600
Subject: [PATCH] [wip] leo_encode, leo_decode, etc. are succeeding at basic
 level

need to gen random data and do random deletions similar to what's done in the
c++ benchmark program being roughly ported in tests/dagger/testleopard.nim
---
 dagger/leopard.nim           | 15 +++---
 tests/dagger/testleopard.nim | 97 ++++++++++++++++++++++++++++++++++--
 tests/testAll.nim            |  1 -
 3 files changed, 102 insertions(+), 11 deletions(-)

diff --git a/dagger/leopard.nim b/dagger/leopard.nim
index 2fa791de..3bae67b5 100644
--- a/dagger/leopard.nim
+++ b/dagger/leopard.nim
@@ -12,21 +12,22 @@ type
 const
   header = "leopard.h"
 
-{.pragma: leo, cdecl, header: header, importCpp.}
+{.pragma: leo, cdecl, header: header.}
 
-proc leo_init*(): cint {.leo.}
+proc leo_init*(): cint {.leo, importCpp.}
 
-func leo_result_string*(res: LeopardResult): cstring {.leo.}
+func leo_result_string*(res: LeopardResult): cstring {.leo, importc.}
 
 func leo_encode_work_count*(original_count, recovery_count: cuint): cuint
-  {.leo.}
+  {.leo, importc.}
 
 proc leo_encode*(buffer_bytes: uint64, original_count, recovery_count,
-  work_count: cuint, original_data, work_data: pointer): LeopardResult {.leo.}
+  work_count: cuint, original_data, work_data: pointer): LeopardResult
+  {.leo, importc.}
 
 func leo_decode_work_count*(original_count, recovery_count: cuint): cuint
-  {.leo.}
+  {.leo, importc.}
 
 proc leo_decode*(buffer_bytes: uint64, original_count, recovery_count,
   work_count: cuint, original_data, recovery_data, work_data: pointer):
-  LeopardResult {.leo.}
+  LeopardResult {.leo, importc.}
diff --git a/tests/dagger/testleopard.nim b/tests/dagger/testleopard.nim
index 9d5f4f12..7314add9 100644
--- a/tests/dagger/testleopard.nim
+++ b/tests/dagger/testleopard.nim
@@ -1,6 +1,97 @@
+# import std/os
+# import std/random
+import std/sequtils
+import std/strformat
+
 import pkg/dagger/leopard
 
-let lin = leo_init()
+type
+  TestParameters = object
+    original_count: cuint
+    recovery_count: cuint
+    buffer_bytes  : cuint
+    loss_count    : cuint
+    seed          : cuint
 
-echo ""
-echo "leo_init() return code: " & $lin
+proc init(
+    T: type TestParameters,
+    original_count: cuint = 100,
+    recovery_count: cuint = 10,
+    buffer_bytes  : cuint = 64000,
+    loss_count    : cuint = 32768,
+    seed          : cuint = 2): T =
+  T(original_count: original_count,
+    recovery_count: recovery_count,
+    buffer_bytes  : buffer_bytes,
+    loss_count    : loss_count,
+    seed          : seed)
+
+proc benchmark(params: TestParameters) =
+  var
+    original_data = newSeqWith(params.original_count.int,
+      newSeq[byte](params.buffer_bytes))
+
+  let
+    encode_work_count = leo_encode_work_count(params.original_count,
+      params.recovery_count)
+    decode_work_count = leo_decode_work_count(params.original_count,
+      params.recovery_count)
+
+  debugEcho "encode_work_count: " & $encode_work_count
+  debugEcho "decode_work_count: " & $decode_work_count
+
+  let
+    total_bytes = (params.buffer_bytes * params.original_count).uint64
+
+  debugEcho "total_bytes: " & $total_bytes
+
+  var
+    encode_work_data = newSeqWith(encode_work_count.int,
+      newSeq[byte](params.buffer_bytes))
+    decode_work_data = newSeqWith(decode_work_count.int,
+      newSeq[byte](params.buffer_bytes))
+
+  let
+    encodeResult = leo_encode(
+      params.buffer_bytes,
+      params.original_count,
+      params.recovery_count,
+      encode_work_count,
+      addr original_data[0],
+      addr encode_work_data[0]
+    )
+
+  debugEcho "encodeResult: " & $leo_result_string(encodeResult)
+
+  let
+    decodeResult = leo_decode(
+      params.buffer_bytes,
+      params.original_count,
+      params.recovery_count,
+      decode_work_count,
+      addr original_data[0],
+      addr encode_work_data[0],
+      addr decode_work_data[0]
+    )
+
+  debugEcho "decodeResult: " & $leo_result_string(decodeResult)
+
+proc main() =
+  if leo_init() != 0: raise (ref Defect)(msg: "Leopard failed to initialize")
+
+  var params = TestParameters.init
+
+  debugEcho fmt(
+    "Parameters: "                              &
+    "[original count={params.original_count}] " &
+    "[recovery count={params.recovery_count}] " &
+    "[buffer bytes={params.buffer_bytes}] "     &
+    "[loss count={params.loss_count}] "         &
+    "[random seed={params.seed}]"
+  )
+
+  benchmark params
+
+when true: # isMainModule:
+  # randomize()
+  main()
diff --git a/tests/testAll.nim b/tests/testAll.nim
index 232b633b..5ad7d0ca 100644
--- a/tests/testAll.nim
+++ b/tests/testAll.nim
@@ -6,5 +6,4 @@ import ./dagger/testmanifest
 import ./dagger/testnode
 import ./dagger/testleopard
 
-
 {.warning[UnusedImport]: off.}