get_partitions now returns python list

Author: Felipe Garrido R.

include/spatialize/abstract_esi.hpp

@@ -404,20 +404,32 @@ namespace sptlz{
 				std::vector<std::vector<std::vector<float>>> partitions;
 				int n = mondrian_forest.size();
 
+				long count = 0;
+				std::cout << "UNO" << std::endl;
 				for(int i=0; i<n; i++){
+
 					std::vector<std::vector<float>> this_partition;
 					// get leaves for i-th tree
 					for(auto leaf: mondrian_forest.at(i)->leaves){
+						std::cout << "aaaa" << std::endl;
 						std::vector<float> this_leaf;
+						std::cout << "bbbb" << std::endl;
 						this_leaf.push_back(leaf->leaf_id);
+						std::cout << "cccc" << std::endl;
 						for(auto ax_lims: leaf->bbox){
+							std::cout << "dddd" << std::endl;
 							this_leaf.push_back(ax_lims[0]);
+							std::cout << "eeee" << std::endl;
 							this_leaf.push_back(ax_lims[1]);
 						}
+						count += this_leaf.size();
+						std::cout << "ffff" << std::endl;
 						this_partition.push_back(this_leaf);
 					}
+					std::cout << "gggg" << std::endl;
 					partitions.push_back(this_partition);
 				}
+				std::cout << "hhhh " << (8*count/1024.0) << std::endl;
 				return(partitions);
 			}
 

src/c++/libspatialize.cpp

@@ -14,7 +14,7 @@
 
 namespace py = pybind11;
 
-py::array_t<float> get_partitions_using_esi(py::array_t<float> samples, int forest_size, float alpha, std::optional<py::function> visitor, int seed){
+std::vector<std::vector<std::vector<float>>> get_partitions_using_esi(py::array_t<float> samples, int forest_size, float alpha, std::optional<py::function> visitor, int seed){
     py::buffer_info smp_info = samples.request();
 
     if (smp_info.ndim != 2)
@@ -41,7 +41,8 @@ py::array_t<float> get_partitions_using_esi(py::array_t<float> samples, int fore
 
     delete esi;
 
-    return(sptlz::vector_3d_to_ndarray(&r));
+//    return(sptlz::vector_3d_to_ndarray(&r));
+    return(r);
 }
 
 py::array_t<float> get_leaf_for_samples_using_esi(py::array_t<float> samples, int forest_size, float alpha, std::optional<py::function> visitor, int seed){
@@ -1267,8 +1268,8 @@ PYBIND11_MODULE(libspatialize, m) {
       "get leaves bbox for every partition"
     );
     m.def(
-      "get_leaves_by_samples_using_esi", 
-      &get_leaves_by_samples_using_esi, 
+      "get_leaf_for_samples_using_esi", 
+      &get_leaf_for_samples_using_esi, 
       "get several partitions using MondrianTree"
     );
     /* plain NN IDW*/