TOMO FH: added tomo cube as breakpoint.

9theWorld9 · Jan 18, 2023 · c569027 · c569027
1 parent 6b3547d
commit c569027
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/biopal/tomo/processing_TOMO.py b/biopal/tomo/processing_TOMO.py
@@ -229,7 +229,7 @@ class opt_str:
     power_peak = out_str.peak
     canopy_height = medfilt2d(canopy_height.astype("float64"), kernel_size=proc_conf.median_factor)
 
-    return canopy_height, power_peak, rg_vec_subs, az_vec_subs, subs_F_r, subs_F_a
+    return canopy_height, power_peak, rg_vec_subs, az_vec_subs, subs_F_r, subs_F_a, tomo_cube
 
 
 def tomo_cube_generator(z, I, model_sign, kz_stack):

diff --git a/biopal/tomo_fh/main_TOMO_FH.py b/biopal/tomo_fh/main_TOMO_FH.py
@@ -440,7 +440,7 @@ def _run(self, input_file):
                     conf_params_obj.estimate_tomo_fh.vertical_range.sampling,
                 )
 
-                (estimated_height, power_peak, rg_vec_subs, az_vec_subs, subs_F_r, subs_F_a,) = BiomassForestHeightSKPD(
+                (estimated_height, power_peak, rg_vec_subs, az_vec_subs, subs_F_r, subs_F_a,tomo_cube,) = BiomassForestHeightSKPD(
                     data_SLC,
                     cov_est_window_size,
                     raster_info.pixel_spacing_slant_rg,
@@ -520,9 +520,9 @@ def _run(self, input_file):
                 logging.info("TOMO FH: saving mail results (in slant range geometry) on " + breakpoints_output_folder)
                 post_string = "_SR_" + unique_stack_id
 
-                breakpoint_names = ["estimated_height" + post_string]
+                breakpoint_names = ["estimated_height" + post_string, "tomo_cube", "vertical_vector"]
 
-                save_breakpoints(breakpoints_output_folder, breakpoint_names, [estimated_height])
+                save_breakpoints(breakpoints_output_folder, breakpoint_names, [estimated_height, tomo_cube, vertical_vector])
                 logging.info("...done.\n")
             del estimated_height