diff --git a/Common/CUDA/PICCS.cu b/Common/CUDA/PICCS.cu index 481ede08..0ddae170 100644 --- a/Common/CUDA/PICCS.cu +++ b/Common/CUDA/PICCS.cu @@ -54,7 +54,7 @@ Codes : https://github.com/CERN/TIGRE #define MAXTHREADS 1024 #include "PICCS.hpp" - +#include "gpuUtils.hpp" @@ -114,17 +114,21 @@ do { \ unsigned long size2d = rows*cols; unsigned long long idx = z * size2d + y * cols + x; - float uidx = u[idx]; + // float uidx = u[idx]; + float uidx = 0; if ( z - 1 >= 0 && z= 0 && y= 0 && y= 0 && x= 0 && x src.shape[0] + size_img[1]= src.shape[1] + size_img[2]= src.shape[2] + + cdef float* c_imgout = malloc(size_img[0] *size_img[1] *size_img[2]* sizeof(float)) + + cdef long imgsize[3] + imgsize[0] = size_img[2] + imgsize[1] = size_img[1] + imgsize[2] = size_img[0] + + src = np.ascontiguousarray(src) + prior = np.ascontiguousarray(prior) + + cdef float* c_src = src.data + cdef float* c_prior = prior.data + cdef np.npy_intp c_maxiter = maxiter + cuda_raise_errors(piccs_tv(c_src, c_prior, c_imgout, alpha, ratio, imgsize, c_maxiter, c_gpuids[0])) + imgout = np.PyArray_SimpleNewFromData(3, size_img, np.NPY_FLOAT32, c_imgout) + PyArray_ENABLEFLAGS(imgout, np.NPY_OWNDATA) + + return imgout