Skip to content

Commit

Permalink
preserve curves before filters
Browse files Browse the repository at this point in the history
  • Loading branch information
marcosmc committed Jan 27, 2025
1 parent a61853c commit f240a7e
Showing 1 changed file with 18 additions and 2 deletions.
20 changes: 18 additions & 2 deletions src/dgcv/curves/manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ def __init__(
self._working_dir = parameters.get_working_dir()
self._producer = parameters.get_producer()
self._pcs_name = pcs_name
self._before_filters_curves = {"calculated": pd.DataFrame(), "reference": pd.DataFrame()}
self._curves = {"calculated": pd.DataFrame(), "reference": pd.DataFrame()}
self._windows = {"calculated": dict(), "reference": dict()}

Expand All @@ -34,6 +35,19 @@ def __init__(
)
self._reference_curves_generator = curves_factory.get_reference(parameters)

def __copy_curves_to_before_filters(self):
self._before_filters_curves["calculated"] = self._curves["calculated"].copy()
self._before_filters_curves["reference"] = self._curves["reference"].copy()

def __get_before_filters_curves(self, curve: str) -> pd.DataFrame:
if curve not in self._before_filters_curves:
return pd.DataFrame()

if self._before_filters_curves[curve].empty:
return pd.DataFrame()

return self._before_filters_curves[curve]

def __get_producer_curves_generator(self):
return self._producer_curves_generator

Expand Down Expand Up @@ -82,6 +96,8 @@ def __obtain_curve(
reference_event_start_time,
)

self.__copy_curves_to_before_filters()

return (
working_oc_dir,
jobs_output_dir,
Expand Down Expand Up @@ -259,8 +275,8 @@ def apply_signal_processing(
"""
# TODO: refactor this function so that it really adheres to the Method described above.

csv_calculated_curves = self.get_curves("calculated")
csv_reference_curves = self.get_curves("reference")
csv_calculated_curves = self.__get_before_filters_curves("calculated")
csv_reference_curves = self.__get_before_filters_curves("reference")

# Activate this code to use the curve calculated as a reference curve,
# only for debug cases without reference curves.
Expand Down

0 comments on commit f240a7e

Please sign in to comment.