From 469e1c0ab06c8fa58372f6691670cc3e3f286a07 Mon Sep 17 00:00:00 2001
From: Artur Glavic <artur.glavic@psi.ch>
Date: Fri, 10 Oct 2025 08:50:57 +0200
Subject: [PATCH] Add theta projection

---
 eos/projection.py    | 52 ++++++++++++++++++++++++++++++++++++++++++++
 eos/reduction_e2h.py |  7 +++---
 2 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/eos/projection.py b/eos/projection.py
index 530dd2a..06b7b1f 100644
--- a/eos/projection.py
+++ b/eos/projection.py
@@ -709,6 +709,58 @@ class LProjection(ProjectionInterface):
         """
         self._graph[0].set_ydata(self.data.I.T)
 
+class TProjection(ProjectionInterface):
+    theta: np.ndarray
+    z: np.ndarray
+
+    data: np.recarray
+    _dtype = np.dtype([
+            ('cts', np.float64),
+            ('I', np.float64),
+            ('err', np.float64),
+            ])
+
+    def __init__(self, tthh):
+        self.z = np.arange(Detector.nBlades*Detector.nWires+1)-0.5
+        dd = Detector.delta_z[1]-Detector.delta_z[0]
+        delta = np.hstack([Detector.delta_z, Detector.delta_z[-1]+dd])-dd/2.
+        self.theta = tthh+delta
+        self.data = np.zeros(self.theta.shape[0]-1, dtype=self._dtype).view(np.recarray)
+        self.monitor = 0.
+
+    def project(self, dataset: EventDatasetProtocol, monitor: float):
+        detYi, detZi, detX, delta = Detector.pixelLookUp[dataset.data.events.pixelID-1].T
+
+        cts , *_ = np.histogram(detZi, bins=self.z)
+        self.data.cts += cts
+        self.monitor += monitor
+
+        self.data.I = self.data.cts / self.monitor
+        self.data.err = np.sqrt(self.data.cts) / self.monitor
+
+    def clear(self):
+        self.data[:] = 0
+        self.monitor = 0.
+
+    def plot(self, **kwargs):
+        from matplotlib import pyplot as plt
+        for key in ONLY_MAP:
+            if key in kwargs: del(kwargs[key])
+
+
+        self._graph = plt.plot(self.theta[:-1], self.data.I, **kwargs)
+
+        plt.xlabel('Reflection Angle / °')
+        plt.ylabel('I / cpm')
+        plt.xlim(self.theta[0], self.theta[-1])
+        plt.title('Theta')
+
+    def update_plot(self):
+        """
+        Inline update of previous plot by just updating the data.
+        """
+        self._graph[0].set_ydata(self.data.I.T)
+
 
 class CombinedProjection(ProjectionInterface):
     """
diff --git a/eos/reduction_e2h.py b/eos/reduction_e2h.py
index 6b34a97..c93ba67 100644
--- a/eos/reduction_e2h.py
+++ b/eos/reduction_e2h.py
@@ -20,9 +20,7 @@ from .options import E2HConfig, E2HPlotArguments, IncidentAngle, MonitorType, E2
 from . import event_handling as eh
 from .path_handling import PathResolver
 from .projection import CombinedProjection, LProjection, LZProjection, ProjectionInterface, ReflectivityProjector, \
-    TofProjection, \
-    TofZProjection, \
-    YTProjection, YZProjection
+    TofProjection,  TofZProjection,  TProjection, YTProjection, YZProjection
 
 NEEDS_LAMDA = (E2HPlotSelection.All, E2HPlotSelection.LT, E2HPlotSelection.Q, E2HPlotSelection.L)
 
@@ -170,6 +168,9 @@ class E2HReduction:
         if self.config.reduction.plot==E2HPlotSelection.YT:
             self.projection = YTProjection(tthh)
 
+        if self.config.reduction.plot==E2HPlotSelection.T:
+            self.projection = TProjection(tthh)
+
         if self.config.reduction.plot==E2HPlotSelection.L:
             self.projection = LProjection()