Merge pull request #71 from LIBRA-project/rem/spectrum-fit

Spectrum fitting for activation foils

Author: RemDelaporteMathurin

example.ipynb

@@ -1,19 +1 @@
-from datetime import datetime
-from . import calibration
-
-from dataclasses import dataclass
-
-
-@dataclass
-class CheckSource:
-    nuclide: calibration.Nuclide
-    activity_date: datetime.date
-    activity: float
-
-
-class ActivationFoil:
-    nuclide: calibration.Nuclide
-    mass: float
-
-
 from . import explicit, settings, calculations

libra_toolbox/neutron_detection/activation_foils/__init__.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
 from typing import List
+import datetime
+import numpy as np
 
 
 @dataclass
@@ -55,3 +57,36 @@ class Nuclide:
     intensity=[0.99976],
     half_life=312.20 * 24 * 3600,
 )
+
+
+@dataclass
+class CheckSource:
+    nuclide: Nuclide
+    activity_date: datetime.date
+    activity: float
+
+    def get_expected_activity(self, date: datetime.date) -> float:
+
+        decay_constant = np.log(2) / self.nuclide.half_life
+
+        # Convert date to datetime if needed
+        if isinstance(self.activity_date, datetime.date) and not isinstance(
+            self.activity_date, datetime.datetime
+        ):
+
+            activity_datetime = datetime.datetime.combine(
+                self.activity_date, datetime.datetime.min.time()
+            )
+            # add a timezone
+            activity_datetime = activity_datetime.replace(tzinfo=date.tzinfo)
+        else:
+            activity_datetime = self.activity_date
+
+        time = (date - activity_datetime).total_seconds()
+        act_expec = self.activity * np.exp(-decay_constant * time)
+        return act_expec
+
+
+class ActivationFoil:
+    nuclide: Nuclide
+    mass: float