bindings: Use verilator_lib in Python

2 files changed, 19 insertions, 1 deletions

diff --git a/bindings/galearn_pdm.cpp b/bindings/galearn_pdm.cpp
index 9274a01..5b7e4f5 100644
--- a/bindings/galearn_pdm.cpp
+++ b/bindings/galearn_pdm.cpp
@@ -1,6 +1,8 @@
 #include <pybind11/pybind11.h>
 #include <pybind11/numpy.h>
 
+int sim(void);
+
 namespace py = pybind11;
 
 void process(py::array_t<uint8_t> arr1, py::array_t<int16_t> arr2) {
@@ -16,12 +18,19 @@ void process(py::array_t<uint8_t> arr1, py::array_t<int16_t> arr2) {
         throw std::runtime_error("Input 1 must be same or larger than input 2");
     }
 
+#if 0
     // Example: access data
     uint8_t* in = static_cast<uint8_t*>(buf1.ptr);
     int16_t* out = static_cast<int16_t*>(buf2.ptr);
     for (int i=0; i<buf2.size; i++) {
         out[i] = in[i] + 1;
     }
+#else
+
+    sim();
+
+#endif
+
 }
 
 PYBIND11_MODULE(galearn_pdm, m) {
diff --git a/bindings/setup.py b/bindings/setup.py
index 487949a..c2f93f7 100644
--- a/bindings/setup.py
+++ b/bindings/setup.py
@@ -1,14 +1,23 @@
 from setuptools import setup, Extension
 import pybind11
 import numpy
+import os.path
+
+verilated_build_dir = '../verilator_lib/obj_dir/'
 
 ext_modules = [
     Extension(
         'galearn_pdm',
-        ['galearn_pdm.cpp'],
+        sources=[
+            'galearn_pdm.cpp',
+        ],
         include_dirs=[
             pybind11.get_include(),
             numpy.get_include(),
+            verilated_build_dir,
+        ],
+        extra_objects=[
+            os.path.join(verilated_build_dir, 'libsim.so')
         ],
         language='c++',
     ),