handling uint16 data in fits

thepetabyteproject · Oct 15, 2020 · 1c70986 · 1c70986
1 parent b4dda7f
commit 1c70986
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/your/formats/psrfits.py b/your/formats/psrfits.py
@@ -253,7 +253,13 @@ def read_subint(
                     data = data + ((sdata[:, 0, :] - sdata[:, 1, :]) / 2).squeeze()
                 else:
                     raise ValueError(f"pol={pol} value not supported.")
-
+            elif len(shp) == 4 and shp[-1] == 2 and self.poln_order == 'IQUV':
+                logger.warning("Data is packed as two uint8 arrays. Concatenating them to get uint16.")
+                logger.warning("Polarization is IQUV. Just using Stokes I.")
+                data = np.zeros((self.nsamp_per_subint, self.nchan), dtype=np.float32)
+                data1 = sdata[:, 0, :, 0].astype(np.uint16)
+                data2 = sdata[:, 0, :, 1].astype(np.uint16)
+                data += np.left_shift(data2, 8) + data1
             else:
                 data = np.asarray(sdata)
         data = data.reshape((self.nsamp_per_subint, self.nchan)).astype(np.float32)

diff --git a/your/your.py b/your/your.py
@@ -338,7 +338,7 @@ def __init__(self, your):
         if self.nbits <= 8:
             self.dtype = np.uint8
         elif self.nbits == 16:
-            if self.format == "fits":
+            if self.format == "fits" and int(your.fits['SUBINT'].header['TDIM17'][1]) == 1:
                 self.dtype = np.int16
             else:
                 self.dtype = np.uint16