Changeset 26358 for issm/trunk-jpl/src/m/solve/loadresultfromdisk.py

Timestamp:

07/24/21 17:33:34 (4 years ago)

Author:

jdquinn

Message:

CHG: Completed MATLAB -> Python updates for SE; archive updates now that GMSH can be used on macOS and Linux; various minor bug fixes; formatting; cleanup

File:

• issm/trunk-jpl/src/m/solve/loadresultfromdisk.py (modified) (2 diffs)

issm/trunk-jpl/src/m/solve/loadresultfromdisk.py

@@ -4 +4 @@
 
 def loadresultfromdisk(filename, step, name, *args):  # {{{
-    """
-    LOADRESULTFROMDISK - load specific result of solution sequence from disk
+    """LOADRESULTFROMDISK - load specific result of solution sequence from disk
     file "filename"
 
-        Usage:
-            variable = loadresultsfromdisk(filename, step, name)
+    Usage:
+        variable = loadresultsfromdisk(filename, step, name)
 
-        TODO:
-        - Test this module against output of src/m/solve/loadresultsfromdisk.m
+    TODO:
+    - Test this module against output of src/m/solve/loadresultsfromdisk.m
     """
-    print('Got hwrwefew')
-    exit()
+
     # Open file
     try:
@@ -59 +57 @@
                 for i in range(M):
                     field[i, :] = struct.unpack('{}i'.format(N), fid.read(N * struct.calcsize('i')))
+            elif datatype == 5:
+                # TODO:
+                # - Check that the following results in the same output as 
+                # MATLAB
+                #
+                N = struct.unpack('i', fid.read(struct.calcsize('i')))[0]
+                fieldr = np.zeros(shape=(M, N), dtype=float)
+                fieldi = np.zeros(shape=(M, N), dtype=float)
+                for i in range(M):
+                    fieldr[i, :] = struct.unpack('{}d'.format(N), fid.read(N * struct.calcsize('d')))
+                    fieldi[i, :] = struct.unpack('{}d'.format(N), fid.read(N * struct.calcsize('d')))
+                field = np.vectorize(complex)(fieldr, fieldi)
+                print(field)
             else:
                 raise TypeError("cannot read data of type {}".format(datatype))