From 29e754796a4ec83cae2f2bfb44e697250f7f222f Mon Sep 17 00:00:00 2001
From: Andrej Babic <andrej.babic@psi.ch>
Date: Mon, 29 Jun 2020 14:50:40 +0200
Subject: [PATCH] Reshape and change the data before returning

---
 scripts/BinaryBufferReader.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/BinaryBufferReader.py b/scripts/BinaryBufferReader.py
index 9e57535..d728829 100644
--- a/scripts/BinaryBufferReader.py
+++ b/scripts/BinaryBufferReader.py
@@ -108,7 +108,8 @@ class BinaryBufferReader(object):
         if not metadata_init:
             metadata["is_good_frame"] = False
 
-        return metadata, data
+        data_shape = [self.n_modules * MODULE_Y_SIZE, MODULE_X_SIZE]
+        return metadata, data.astype(dtype="uint16").reshape(shape=data_shape)
 
 
 def get_file_frame_index(pulse_id):