From 2f2e37ce901eb2e287c7c3cf2bcbad58ca2a276f Mon Sep 17 00:00:00 2001
From: Jan David Mol <mol@astron.nl>
Date: Sat, 26 Aug 2023 15:16:58 +0200
Subject: [PATCH] Fix reshaping XST attributes if we have less than 192 inputs

---
 .../tangostationcontrol/devices/sdp/xst.py                | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tangostationcontrol/tangostationcontrol/devices/sdp/xst.py b/tangostationcontrol/tangostationcontrol/devices/sdp/xst.py
index f2ccb6172..36e4e8baa 100644
--- a/tangostationcontrol/tangostationcontrol/devices/sdp/xst.py
+++ b/tangostationcontrol/tangostationcontrol/devices/sdp/xst.py
@@ -445,22 +445,22 @@ class XST(Statistics):
 
     def read_xst_real_R(self):
         return numpy.real(self.statistics_client.collector.xst_values()).reshape(
-            MAX_PARALLEL_SUBBANDS, MAX_INPUTS * MAX_INPUTS
+            MAX_PARALLEL_SUBBANDS, self.statistics_collector.nr_signal_inputs**2
         )
 
     def read_xst_imag_R(self):
         return numpy.imag(self.statistics_client.collector.xst_values()).reshape(
-            MAX_PARALLEL_SUBBANDS, MAX_INPUTS * MAX_INPUTS
+            MAX_PARALLEL_SUBBANDS, self.statistics_collector.nr_signal_inputs**2
         )
 
     def read_xst_power_R(self):
         return numpy.abs(self.statistics_client.collector.xst_values()).reshape(
-            MAX_PARALLEL_SUBBANDS, MAX_INPUTS * MAX_INPUTS
+            MAX_PARALLEL_SUBBANDS, self.statistics_collector.nr_signal_inputs**2
         )
 
     def read_xst_phase_R(self):
         return numpy.angle(self.statistics_client.collector.xst_values()).reshape(
-            MAX_PARALLEL_SUBBANDS, MAX_INPUTS * MAX_INPUTS
+            MAX_PARALLEL_SUBBANDS, self.statistics_collector.nr_signal_inputs**2
         )
 
     # xst_R, but as a matrix of input x input, for each specific subband index
-- 
GitLab