From 06a4cbc6cde3d08cfd72289d4d1935cb7dabb4a5 Mon Sep 17 00:00:00 2001
From: jupfi <git@jupfi.me>
Date: Fri, 26 Apr 2024 14:42:20 +0200
Subject: [PATCH] Implemented apodization.

---
 src/nqrduck_spectrometer/measurement.py | 34 +++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/nqrduck_spectrometer/measurement.py b/src/nqrduck_spectrometer/measurement.py
index 2415678..4648706 100644
--- a/src/nqrduck_spectrometer/measurement.py
+++ b/src/nqrduck_spectrometer/measurement.py
@@ -2,7 +2,9 @@
 
 import logging
 import numpy as np
+from decimal import Decimal
 from nqrduck.helpers.signalprocessing import SignalProcessing as sp
+from nqrduck.helpers.functions import Function
 
 logger = logging.getLogger(__name__)
 
@@ -39,12 +41,44 @@ class Measurement:
         IF_frequency: float = 0,
     ) -> None:
         """Initializes the measurement."""
+        # Convert to decimal 
         self.tdx = tdx
         self.tdy = tdy
         self.target_frequency = target_frequency
         self.fdx, self.fdy = sp.fft(tdx, tdy, frequency_shift)
         self.IF_frequency = IF_frequency
 
+    def apodization(self, function : Function):
+        """
+        Applies apodization to the measurement data.
+
+        Args:
+            function (Function): Apodization function.
+
+        returns:
+            Measurement : The apodized measurement.
+        """
+        # Get the y data weights from the function
+        duration = (self.tdx[-1] - self.tdx[0]) * 1e-6
+
+        resolution = duration / len(self.tdx)
+
+        logger.debug("Resolution: %s", resolution)
+
+        y_weight = function.get_pulse_amplitude(duration, resolution)
+
+        tdy_measurement = self.tdy * y_weight
+
+        apodized_measurement = Measurement(
+            self.tdx,
+            tdy_measurement,
+            target_frequency=self.target_frequency,
+            IF_frequency=self.IF_frequency,
+        )
+
+        return apodized_measurement
+
+
     # Data saving and loading
 
     def to_json(self):