From 080367c98a6a590f7460947026cf52e9ba85de3d Mon Sep 17 00:00:00 2001
From: Jonathan Bloedow <jhb4@cornell.edu>
Date: Fri, 2 Aug 2024 12:47:47 -0700
Subject: [PATCH] Added ctypes call to ri.py.

---
 nnmm/ri.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/nnmm/ri.py b/nnmm/ri.py
index 9c54dac..8f40743 100644
--- a/nnmm/ri.py
+++ b/nnmm/ri.py
@@ -1,5 +1,19 @@
 import numpy as np
 import numba as nb
+import ctypes
+
+lib = ctypes.CDLL('./libri.so')
+
+# Define the argument types for the C function
+lib.update_susceptibility_based_on_ri_timer.argtypes = [
+    ctypes.c_uint32,                  # count
+    np.ctypeslib.ndpointer(dtype=np.uint16, ndim=1, flags='C_CONTIGUOUS'),  # ri_timer
+    np.ctypeslib.ndpointer(dtype=np.uint8, ndim=1, flags='C_CONTIGUOUS'),   # susceptibility
+    np.ctypeslib.ndpointer(dtype=np.uint16, ndim=1, flags='C_CONTIGUOUS'),  # age_at_vax
+    np.ctypeslib.ndpointer(dtype=np.int32, ndim=1, flags='C_CONTIGUOUS'),   # dob
+    ctypes.c_int64                    # tick
+]
+
 
 def add(model, count_births, istart, iend):
     # Randomly set ri_timer for coverage fraction of agents to a value between 8.5*30.5 and 9.5*30.5 days
@@ -55,6 +69,10 @@ def add_with_ips(model, count_births, istart, iend):
 
     return
 
+def _update_susceptibility_based_on_ri_timer(count, ri_timer, susceptibility, age_at_vax, dob, tick):
+    lib.update_susceptibility_based_on_ri_timer(count, ri_timer, susceptibility, age_at_vax, dob, tick)
+
+"""
 # Define the function to decrement ri_timer and update susceptibility
 @nb.njit((nb.uint32, nb.uint16[:], nb.uint8[:], nb.uint16[:], nb.int32[:], nb.int64 ), parallel=True)
 def _update_susceptibility_based_on_ri_timer(count, ri_timer, susceptibility, age_at_vax, dob, tick):
@@ -67,3 +85,5 @@ def _update_susceptibility_based_on_ri_timer(count, ri_timer, susceptibility, ag
                 susceptibility[i] = 0
                 #age_at_vax[i] = tick-dob[i] # optional for reporting
 
+
+"""