Merge pull request #11 from Cervest/fix/missing-features

fix/missing-features
Cervest · Jan 12, 2023 · 2b75e98 · 2b75e98
2 parents feea92b + 7cc48c0
commit 2b75e98
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 10 deletions.
diff --git a/src/jlrichdem/common/jlcommon.cpp b/src/jlrichdem/common/jlcommon.cpp
@@ -26,8 +26,8 @@ namespace jlrichdem
             wrapped.method("noData", &WrappedT::noData);
             wrapped.method("get_projection", [](const WrappedT &mat)
                            { return mat.projection; });
-            wrapped.method("isNoData", [](WrappedT &mat, xyT x, xyT y)
-                           { return mat.isNoData(x, y); });
+            wrapped.method("isNoData", [](WrappedT &mat, uint32_t i)
+                           { return mat.isNoData(i); });
             wrapped.method("setNoData", [](WrappedT &mat, const ScalarT &ndval)
                            { return mat.setNoData(ndval); });
             wrapped.method("resize", [](WrappedT &mat, const xyT width, const xyT height, const ScalarT &val)

diff --git a/src/jlrichdem/depressions/jldepressions.cpp b/src/jlrichdem/depressions/jldepressions.cpp
@@ -16,15 +16,33 @@ JLCXX_MODULE define_depressions_module(jlcxx::Module &mod)
     jlcxx::TypeWrapper<rd::dephier::DepressionHierarchy<double>> depression_hierarchy_double =
         mod.add_type<rd::dephier::DepressionHierarchy<double>>("DepressionHierarchyDouble");
 
+    jlcxx::TypeWrapper<rd::dephier::Depression<float>> depression_float =
+        mod.add_type<rd::dephier::Depression<float>>("DepressionFloat");
+
+    jlcxx::TypeWrapper<rd::dephier::DepressionHierarchy<float>> depression_hierarchy_float =
+        mod.add_type<rd::dephier::DepressionHierarchy<float>>("DepressionHierarchyFloat");
+
     mod.method("GetDepressionHierarchyDoubleD8", [](const rd::Array2D<double> &topo,
                                                     rd::Array2D<rd::dephier::dh_label_t> &label,
                                                     rd::Array2D<int8_t> &flowdirs)
                { return rd::dephier::GetDepressionHierarchy<double, rd::Topology::D8>(topo, label, flowdirs); });
 
-    mod.method("FillSpillMerge", [](const rd::Array2D<double> &topo,
-                                    const rd::Array2D<rd::dephier::dh_label_t> &label,
-                                    const rd::Array2D<rd::flowdir_t> &flowdirs,
-                                    rd::dephier::DepressionHierarchy<double> &deps,
-                                    rd::Array2D<double> &wtd)
+    mod.method("FillSpillMergeDouble", [](const rd::Array2D<double> &topo,
+                                          const rd::Array2D<rd::dephier::dh_label_t> &label,
+                                          const rd::Array2D<rd::flowdir_t> &flowdirs,
+                                          rd::dephier::DepressionHierarchy<double> &deps,
+                                          rd::Array2D<double> &wtd)
+               { return rd::dephier::FillSpillMerge(topo, label, flowdirs, deps, wtd); });
+
+    mod.method("GetDepressionHierarchyFloatD8", [](const rd::Array2D<float> &topo,
+                                                   rd::Array2D<rd::dephier::dh_label_t> &label,
+                                                   rd::Array2D<int8_t> &flowdirs)
+               { return rd::dephier::GetDepressionHierarchy<float, rd::Topology::D8>(topo, label, flowdirs); });
+
+    mod.method("FillSpillMergeFloat", [](const rd::Array2D<float> &topo,
+                                         const rd::Array2D<rd::dephier::dh_label_t> &label,
+                                         const rd::Array2D<rd::flowdir_t> &flowdirs,
+                                         rd::dephier::DepressionHierarchy<float> &deps,
+                                         rd::Array2D<double> &wtd)
                { return rd::dephier::FillSpillMerge(topo, label, flowdirs, deps, wtd); });
 }
diff --git a/src/jlrichdem/misc/jlmisc.cpp b/src/jlrichdem/misc/jlmisc.cpp
@@ -11,8 +11,13 @@ namespace rd = richdem;
 
 JLCXX_MODULE define_misc_module(jlcxx::Module &mod)
 {
-    mod.method("BucketFillFromEdgesD8", [](const rd::Array2D<double> &check_raster,
-                                           rd::Array2D<rd::dephier::dh_label_t> &set_raster, 
-                                           double check_value, rd::dephier::dh_label_t set_value)
+    mod.method("BucketFillFromEdgesD8Double", [](const rd::Array2D<double> &check_raster,
+                                                 rd::Array2D<rd::dephier::dh_label_t> &set_raster,
+                                                 double check_value, rd::dephier::dh_label_t set_value)
+               { return rd::BucketFillFromEdges<rd::Topology::D8>(check_raster, set_raster, check_value, set_value); });
+
+    mod.method("BucketFillFromEdgesD8Float", [](const rd::Array2D<float> &check_raster,
+                                                rd::Array2D<rd::dephier::dh_label_t> &set_raster,
+                                                float check_value, rd::dephier::dh_label_t set_value)
                { return rd::BucketFillFromEdges<rd::Topology::D8>(check_raster, set_raster, check_value, set_value); });
 }