From 4c4fa8e732609b17c8af344867e77b4c82f466db Mon Sep 17 00:00:00 2001
From: Johannes Stelzer <johannes.stelzer@tuebingen.mpg.de>
Date: Mon, 11 Mar 2019 13:54:35 +0100
Subject: [PATCH] 4D fix

---
 vviewer/VistaLoad.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/vviewer/VistaLoad.py b/vviewer/VistaLoad.py
index e6c1370..6cc1929 100644
--- a/vviewer/VistaLoad.py
+++ b/vviewer/VistaLoad.py
@@ -184,7 +184,7 @@ def load_vista(fp_input):
             
         img3D = np.transpose(np.reshape(img1D, (zdim,ydim,xdim)), (2,1,0))
         data = img3D
-        dim = "4D"
+        dim = "3D"
     else:
         list_images = []
         for i in range(len(idx_images)):
@@ -202,11 +202,13 @@ def load_vista(fp_input):
             img1D[i*xdim*ydim*tdim:(i+1)*xdim*ydim*tdim] = list_images[i]
         img4D = np.transpose(np.reshape(img1D, (zdim,tdim,ydim,xdim)), (3,2,0,1))
         data = img4D
-        dim = "3D"
+        dim = "4D"
         
         
     #%% re-parse the header to get the complete header information
     mm = get_property_str(header_img, "voxel", "list")
+    if dim == "4D":
+        mm.append(tdim)
     
     #get s-form code
     sform_code = get_property_str(header, "sform_code", "int")
@@ -240,6 +242,7 @@ def load_vista(fp_input):
     
     #%% convert to nii object... fill in data and header
     nii_loaded = nib.Nifti1Image(data, affine=np.eye(4))
+    # import pdb; pdb.set_trace()
     nii_loaded.header.set_zooms(mm)
     nii_loaded.header['pixdim'][4] = tr
     nii_loaded.set_sform(sform2D)
-- 
GitLab