From f5cc56d8a7a1a245d43acd7cbfe0bf6a6c72b62a Mon Sep 17 00:00:00 2001
From: jupfi <github@jupfi.me>
Date: Sat, 29 Jul 2023 18:07:46 +0200
Subject: [PATCH] Changed to decimals.

---
 src/nqrduck_spectrometer/pulseparameters.py | 56 ++++++++++++++++++---
 1 file changed, 48 insertions(+), 8 deletions(-)

diff --git a/src/nqrduck_spectrometer/pulseparameters.py b/src/nqrduck_spectrometer/pulseparameters.py
index b48829a..d6fdad7 100644
--- a/src/nqrduck_spectrometer/pulseparameters.py
+++ b/src/nqrduck_spectrometer/pulseparameters.py
@@ -1,6 +1,7 @@
 import logging
 import numpy as np
 import sympy
+from decimal import Decimal
 from pathlib import Path
 from PyQt6.QtGui import QPixmap
 from nqrduck.contrib.mplwidget import MplWidget
@@ -14,25 +15,25 @@ class Function:
     name: str
     parameters: list
     expression: str | sympy.Expr
-    resolution: float
+    resolution: Decimal
     start_x: float
     end_x: float
 
     def __init__(self, expr) -> None:
         self.parameters = []
         self.expr = expr
-        self.resolution = 1/30.72e6
+        self.resolution = Decimal(1/30.72e6)
         self.start_x = -1
         self.end_x = 1
 
-    def get_time_points(self, pulse_length: float) -> np.ndarray:
+    def get_time_points(self, pulse_length: Decimal) -> np.ndarray:
         """Returns the time domain points for the function with the given pulse length."""
         # Get the time domain points
         n = int(pulse_length / self.resolution)
-        t = np.linspace(0, pulse_length, n)
+        t = np.linspace(0, float(pulse_length), n)
         return t
 
-    def evaluate(self, pulse_length: float) -> np.ndarray:
+    def evaluate(self, pulse_length: Decimal) -> np.ndarray:
         """Evaluates the function for the given pulse length."""
         n = int(pulse_length / self.resolution)
         t = np.linspace(self.start_x, self.end_x, n)
@@ -52,7 +53,7 @@ class Function:
 
         return f(t)
 
-    def frequency_domain_plot(self, pulse_length: float) -> MplWidget:
+    def frequency_domain_plot(self, pulse_length: Decimal) -> MplWidget:
         mpl_widget = MplWidget()
         td = self.get_time_points(pulse_length)
         yd = self.evaluate(pulse_length)
@@ -63,7 +64,7 @@ class Function:
         mpl_widget.canvas.ax.grid(True)
         return mpl_widget
 
-    def time_domain_plot(self, pulse_length: float) -> MplWidget:
+    def time_domain_plot(self, pulse_length: Decimal) -> MplWidget:
         mpl_widget = MplWidget()
         td = self.get_time_points(pulse_length)
         mpl_widget.canvas.ax.plot(td, self.evaluate(pulse_length))
@@ -72,7 +73,7 @@ class Function:
         mpl_widget.canvas.ax.grid(True)
         return mpl_widget
     
-    def get_pulse_amplitude(self, pulse_length: float) -> np.array:
+    def get_pulse_amplitude(self, pulse_length: Decimal) -> np.array:
         """Returns the pulse amplitude in the time domain."""
         return self.evaluate(pulse_length)
 
@@ -124,6 +125,45 @@ class Function:
         elif isinstance(expr, sympy.Expr):
             self._expr = expr
 
+    @property
+    def resolution(self):
+        """ The resolution of the function in seconds."""
+        return self._resolution
+    
+    @resolution.setter
+    def resolution(self, resolution):
+        try:
+            self._resolution = Decimal(resolution)
+        except:
+            logger.error("Could not convert %s to a decimal", resolution)
+            raise SyntaxError("Could not convert %s to a decimal" % resolution)
+        
+    @property
+    def start_x(self):
+        """ The x value where the evalution of the function starts."""
+        return self._start_x
+    
+    @start_x.setter
+    def start_x(self, start_x):
+        try:
+            self._start_x = float(start_x)
+        except:
+            logger.error("Could not convert %s to a float", start_x)
+            raise SyntaxError("Could not convert %s to a float" % start_x)
+        
+    @property
+    def end_x(self):
+        """ The x value where the evalution of the function ends."""
+        return self._end_x
+    
+    @end_x.setter
+    def end_x(self, end_x):
+        try:
+            self._end_x = float(end_x)
+        except:
+            logger.error("Could not convert %s to a float", end_x)
+            raise SyntaxError("Could not convert %s to a float" % end_x)
+
 
     class Parameter:
         def __init__(self, name: str, symbol: str, value: float) -> None: