From 162baae448770b19e4de828a24e2226da5ccd2ee Mon Sep 17 00:00:00 2001
From: Harry Carey <harry.carey95@gmail.com>
Date: Sat, 16 Nov 2024 14:40:27 +0100
Subject: [PATCH] allow angle change to take into consideration bad_sections
---
DeepSlice/coord_post_processing/angle_methods.py | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/DeepSlice/coord_post_processing/angle_methods.py b/DeepSlice/coord_post_processing/angle_methods.py
index 3c31ee9..0ca898d 100644
--- a/DeepSlice/coord_post_processing/angle_methods.py
+++ b/DeepSlice/coord_post_processing/angle_methods.py
@@ -105,11 +105,16 @@ def propagate_angles(df, method, species):
:rtype: pandas.DataFrame
"""
# get the angles for each section in the dataset
- DV_angle_list, ML_angle_list = calculate_angles(df)
- if method == "weighted_mean":
- depths = calculate_brain_center_depths(
- df[["ox", "oy", "oz", "ux", "uy", "uz", "vx", "vy", "vz"]]
- )
+ temp = df.copy()
+ if "bad_section" in temp:
+ temp = temp[~temp["bad_section"]]
+ DV_angle_list, ML_angle_list = calculate_angles(temp)
+
+
+ depths = calculate_brain_center_depths(
+ temp[["ox", "oy", "oz", "ux", "uy", "uz", "vx", "vy", "vz"]]
+ )
+
DV_angle, ML_angle = get_mean_angle(
DV_angle_list, ML_angle_list, method, depths, species
)