Add image_int_ptr and coeff_int_ptr

pierrepaleo · Dec 11, 2019 · 3d7da10 · 3d7da10
1 parent 08e1bd6
commit 3d7da10
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/src/wt.cu b/src/wt.cu
@@ -656,15 +656,14 @@ int Wavelets::add_wavelet(Wavelets W, DTYPE alpha) {
     return 0;
 }
 
+/// Return the memory pointer (unsigned int64) of the device image
+__intptr_t Wavelets::image_int_ptr(void) {
+    return (__intptr_t) d_image;
+}
 
-
-
-
-
-
-
-
-
-
+/// Return the memory pointer (unsigned int64) of the device coefficient.
+__intptr_t Wavelets::coeff_int_ptr(int num) {
+    return (__intptr_t) d_coeffs[num];
+}
 
 
diff --git a/src/wt.h b/src/wt.h
@@ -71,6 +71,8 @@ class Wavelets {
     int set_filters_inverse(DTYPE* filter1, DTYPE* filter2, DTYPE* filter3 = NULL, DTYPE* filter4 = NULL);
 
     int add_wavelet(Wavelets W, DTYPE alpha=1.0f);
+    __intptr_t image_int_ptr(void);
+    __intptr_t coeff_int_ptr(int num);
 };