From 3cc20874098867aab11313b7fb1d1f65c400fa0c Mon Sep 17 00:00:00 2001
From: Edoardo Pasca <edo.paskino@gmail.com>
Date: Mon, 22 Jan 2018 16:15:20 +0000
Subject: fixed setup.py, bugfixes

---
 Wrappers/Python/fista_module.cpp | 18 ++++++++++++++----
 Wrappers/Python/setup.py         |  9 +++++----
 2 files changed, 19 insertions(+), 8 deletions(-)

(limited to 'Wrappers/Python')

diff --git a/Wrappers/Python/fista_module.cpp b/Wrappers/Python/fista_module.cpp
index f3add76..bb43b84 100644
--- a/Wrappers/Python/fista_module.cpp
+++ b/Wrappers/Python/fista_module.cpp
@@ -592,6 +592,8 @@ bp::list FGP_TV(np::ndarray input, double d_mu, int iter, double d_epsil, int me
 bp::list LLT_model(np::ndarray input, double d_lambda, double d_tau, int iter, double d_epsil, int switcher) {
 	// the result is in the following list
 	bp::list result;
+	
+	std::cout << "Calling LLT_model" << std::endl;
 
 	int number_of_dims, dimX, dimY, dimZ, ll, j, count;
 	//const int  *dim_array;
@@ -644,7 +646,8 @@ bp::list LLT_model(np::ndarray input, double d_lambda, double d_tau, int iter, d
 		np::ndarray npD1 = np::zeros(shape, dtype);
 		np::ndarray npD2 = np::zeros(shape, dtype);
 		
-
+        //result.append<np::ndarray>(npU);
+		
 		U = reinterpret_cast<float *>(npU.get_data());
 		U_old = reinterpret_cast<float *>(npU_old.get_data());
 		D1 = reinterpret_cast<float *>(npD1.get_data());
@@ -657,7 +660,7 @@ bp::list LLT_model(np::ndarray input, double d_lambda, double d_tau, int iter, d
 		re_old = 0.0f;
 
 		for (ll = 0; ll < iter; ll++) {
-
+			std::cout << "iteration " << ll << " of " << iter << " count " << count <<std::endl;
 			copyIm(U, U_old, dimX, dimY, dimZ);
 
 			/*estimate inner derrivatives */
@@ -683,9 +686,11 @@ bp::list LLT_model(np::ndarray input, double d_lambda, double d_tau, int iter, d
 			re_old = re;
 
 		} /*end of iterations*/
-		  //printf("HO iterations stopped at iteration: %i\n", ll);
-
+		  printf("HO iterations stopped at iteration: %i\n", ll);
 		result.append<np::ndarray>(npU);
+		std::cout << "npU shape " << bp::extract<char const *>(bp::str(shape)) <<std::endl;
+		std::cout << "npU  " << bp::extract<char const *>(bp::str(npU)) <<std::endl;
+		
 	}
 	else if (number_of_dims == 3) {
 		/*3D case*/
@@ -766,6 +771,11 @@ bp::list LLT_model(np::ndarray input, double d_lambda, double d_tau, int iter, d
 		if (switcher != 0) result.append<np::ndarray>(npMap);
 
 	}
+	std::cout << "Call to LLT_model ended" << std::endl;
+	std::cout << "result length " << bp::len(result) << std::endl;
+	//std::cout << "npU shape " << bp::extract<char const *>(bp::str(shape)) <<std::endl;
+	std::cout << "npU  " << bp::extract<char const *>(bp::str(result[0])) <<std::endl;
+			
 	return result;
 }
 
diff --git a/Wrappers/Python/setup.py b/Wrappers/Python/setup.py
index ef20a27..d2129b0 100644
--- a/Wrappers/Python/setup.py
+++ b/Wrappers/Python/setup.py
@@ -32,8 +32,9 @@ extra_compile_args = []
 extra_link_args = []
 extra_libraries = ['cilreg']
 
-extra_include_dirs += [os.path.join(".." , ".." , "Core",  "regularizers_CPU"),
-                       	   os.path.join(".." , ".." , "Core",  "regularizers_GPU") , 
+extra_include_dirs += [os.path.join(".." , ".." , "Core"),
+					   os.path.join(".." , ".." , "Core",  "regularizers_CPU"),
+                       os.path.join(".." , ".." , "Core",  "regularizers_GPU") , 
 						   "."]
 
 if platform.system() == 'Windows':
@@ -57,7 +58,7 @@ setup(
 	description='CCPi Core Imaging Library - Image Regularizers',
 	version=cil_version,
     cmdclass = {'build_ext': build_ext},
-    ext_modules = [Extension("ccpi.imaging.cpu_regularizers",
+    ext_modules = [Extension("ccpi.filters.cpu_regularizers",
                              sources=[os.path.join("." , "fista_module.cpp" ),
                                      # os.path.join("@CMAKE_SOURCE_DIR@" , "main_func" ,  "regularizers_CPU", "FGP_TV_core.c"),
 									 # os.path.join("@CMAKE_SOURCE_DIR@" , "main_func" ,  "regularizers_CPU", "SplitBregman_TV_core.c"),
@@ -73,7 +74,7 @@ setup(
     
     ],
 	zip_safe = False,	
-	packages = {'ccpi','ccpi.imaging'},
+	packages = {'ccpi','ccpi.filters'},
 )
 
 
-- 
cgit v1.2.3