Changeset 2644

Timestamp:

Mar 31, 2006, 4:20:22 PM (17 years ago)

Author:

nick

Message:

added functionality to data_manager.py can now read MOST and Ferret netCDF files

File:

• inundation/pyvolution/data_manager.py (modified) (4 diffs)

inundation/pyvolution/data_manager.py

@@ -2119 +2119 @@
                                   #for the Wollongong tsunami
                                   #scenario but is very hacky
-    """Convert 'Ferret' NetCDF format for wave propagation to
+    """Convert MOST and 'Ferret' NetCDF format for wave propagation to
     sww format native to pyvolution.
 
@@ -2175 +2175 @@
         swwname = basename_out + '.sww'
 
-    times = file_h.variables['TIME']
-    latitudes = file_h.variables['LAT']
-    longitudes = file_h.variables['LON']
-
+#    Get dimensions of file_h
+    for dimension in file_h.dimensions.keys():
+        if dimension[:3] == 'LON':
+            dim_h_longitude = dimension
+        if dimension[:3] == 'LAT':
+            dim_h_latitude = dimension
+        if dimension[:4] == 'TIME':
+            dim_h_time = dimension
+
+#    print 'long:', dim_h_longitude
+#    print 'lats:', dim_h_latitude
+#    print 'times:', dim_h_time
+
+    times = file_h.variables[dim_h_time]
+    latitudes = file_h.variables[dim_h_latitude]
+    longitudes = file_h.variables[dim_h_longitude]
+    
+# get dimensions for file_e
+    for dimension in file_e.dimensions.keys():
+        if dimension[:3] == 'LON':
+            dim_e_longitude = dimension
+        if dimension[:3] == 'LAT':
+            dim_e_latitude = dimension
+        if dimension[:4] == 'TIME':
+            dim_e_time = dimension
+
+# get dimensions for file_u
+    for dimension in file_u.dimensions.keys():
+        if dimension[:3] == 'LON':
+            dim_u_longitude = dimension
+        if dimension[:3] == 'LAT':
+            dim_u_latitude = dimension
+        if dimension[:4] == 'TIME':
+            dim_u_time = dimension
+            
+# get dimensions for file_v
+    for dimension in file_v.dimensions.keys():
+        if dimension[:3] == 'LON':
+            dim_v_longitude = dimension
+        if dimension[:3] == 'LAT':
+            dim_v_latitude = dimension
+        if dimension[:4] == 'TIME':
+            dim_v_time = dimension
 
 
     #Precision used by most for lat/lon is 4 or 5 decimals
-    e_lat = around(file_e.variables['LAT'][:], 5)
-    e_lon = around(file_e.variables['LON'][:], 5)
+    e_lat = around(file_e.variables[dim_e_latitude][:], 5)
+    e_lon = around(file_e.variables[dim_e_longitude][:], 5)
 
     #Check that files are compatible
-    assert allclose(latitudes, file_u.variables['LAT'])
-    assert allclose(latitudes, file_v.variables['LAT'])
+    assert allclose(latitudes, file_u.variables[dim_u_latitude])
+    assert allclose(latitudes, file_v.variables[dim_v_latitude])
     assert allclose(latitudes, e_lat)
 
-    assert allclose(longitudes, file_u.variables['LON'])
-    assert allclose(longitudes, file_v.variables['LON'])
+    assert allclose(longitudes, file_u.variables[dim_u_longitude])
+    assert allclose(longitudes, file_v.variables[dim_v_longitude])
     assert allclose(longitudes, e_lon)
-
-
 
     if mint == None:
@@ -2226 +2263 @@
         lmax = searchsorted(longitudes, maxlon)
 
-
-
+#    print' j', jmin, jmax
     times = times[jmin:jmax]
     latitudes = latitudes[kmin:kmax]
@@ -2235 +2271 @@
     if verbose: print 'cropping'
     zname = 'ELEVATION'
-
 
     amplitudes = file_h.variables['HA'][jmin:jmax, kmin:kmax, lmin:lmax]