sot · javierggt · Jan 15, 2025 · Apr 29, 2025 · Apr 29, 2025 · Apr 29, 2025
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,3 @@
+.eggs
+.vscode
+__pycache__
diff --git a/pyrightconfig.json b/pyrightconfig.json
diff --git a/setup.py b/setup.py
@@ -5,6 +5,7 @@
     "console_scripts": [
         "ska_trend_wrong_box_updates=ska_trend.wrong_box_anom.wrong_box_anom:main",
         "ska_trend_bad_periscope=ska_trend.bad_periscope_gradient.periscope_update:main",
+        "ska_trend_periscope_drift=ska_trend.periscope_drift.scripts.periscope_drift_reports:main",
     ]
 }
 
@@ -26,12 +27,18 @@
         "ska_trend",
         "ska_trend.wrong_box_anom",
         "ska_trend.bad_periscope_gradient",
+        "ska_trend.periscope_drift",
+        "ska_trend.periscope_drift.scripts",
     ],
     package_data={
         "ska_trend": [
             "wrong_box_anom/index_template.html",
             "wrong_box_anom/task_schedule.cfg",
             "bad_periscope_gradient/task_schedule.cfg",
-        ]
+        ],
+        "ska_trend.periscope_drift": [
+            "task_schedule.cfg",
+            "templates/periscope_drift/index.html",
+        ],
     },
 )
diff --git a/ska_trend/periscope_drift/__init__.py b/ska_trend/periscope_drift/__init__.py
diff --git a/ska_trend/periscope_drift/correction.py b/ska_trend/periscope_drift/correction.py
@@ -0,0 +1,59 @@
+"""
+Periscope Drift Correction
+
+This module contains the function and constants used to correct the periscope drift.
+"""
+
+import numpy as np
+import scipy
+
+__all__ = [
+    "GRADIENTS",
+    "get_expected_correction",
+]
+
+GRADIENTS = {
+    "OOBAGRD3": {
+        "yag": 6.98145650e-04,
+        "zag": 9.51578351e-05,
+    },
+    "OOBAGRD6": {
+        "yag": -1.67009240e-03,
+        "zag": -2.79084775e-03,
+    },
+}
+
+
+def get_expected_correction(telem):
+    """
+    Calculate the expected periscope drift correction.
+
+    Parameters
+    ----------
+    telem : dict
+        Dictionary containing the telemetry data.
+        Required keys are "OOBAGRD3" and "OOBAGRD6".
+    """
+    corr = {
+        "ang_y_corr": np.zeros_like(telem["OOBAGRD6"]),
+        "ang_z_corr": np.zeros_like(telem["OOBAGRD6"]),
+    }
+
+    msids = [key for key in telem if key != "time"]
+    for msid in msids:
+        if msid not in GRADIENTS:
+            continue
+        mean_gradient = np.mean(telem[msid])
+        corr["ang_y_corr"] -= (telem[msid] - mean_gradient) * GRADIENTS[msid]["yag"]
+        corr["ang_z_corr"] -= (telem[msid] - mean_gradient) * GRADIENTS[msid]["zag"]
+
+    corr["ang_y_corr"] *= 3600
+    corr["ang_z_corr"] *= 3600
+
+    covariance = np.cov([telem["OOBAGRD3"], telem["OOBAGRD6"]])
+    eig_vals, eig_vec = scipy.linalg.eig(covariance)
+    vec = eig_vec[:, np.argmax(eig_vals)]
+    corr_angle = np.arctan2(vec[1], vec[0])
+    corr["OOBAGRD_corr_angle"] = corr_angle
+
+    return corr