Merge pull request #518 from RocketPy-Team/enh/save-filtered-data

ENH : Add of a line for saving the filtered dataset

rocketpy/mathutils/function.py

@@ -1086,7 +1086,7 @@ def to_frequency_domain(self, lower, upper, sampling_frequency, remove_dc=True):
             extrapolation="zero",
         )
 
-    def low_pass_filter(self, alpha):
+    def low_pass_filter(self, alpha, file_path=None):
         """Implements a low pass filter with a moving average filter
 
         Parameters
@@ -1097,6 +1097,9 @@ def low_pass_filter(self, alpha):
             filtered function returned will match the function the smaller
             alpha is, the smoother the filtered function returned will be
             (but with a phase shift)
+        file_path : string
+            File path or file name of the CSV to save. Don't save any CSV if
+            if no argument is passed. Initiated to None.
 
         Returns
         -------
@@ -1112,6 +1115,10 @@ def low_pass_filter(self, alpha):
                 alpha * self.source[i] + (1 - alpha) * filtered_signal[i - 1]
             )
 
+        # Save the new csv file with filtered data
+        if isinstance(file_path, str):
+            np.savetxt(file_path, filtered_signal, delimiter=",")
+
         return Function(
             source=filtered_signal,
             interpolation=self.__interpolation__,