Changeset 8780 for trunk/anuga_core/source/anuga/file_conversion/sww2pts.py

Timestamp:

Mar 27, 2013, 8:04:00 PM (11 years ago)

Author:

steve

Message:

Some changes to allow netcdf4 use

File:

• trunk/anuga_core/source/anuga/file_conversion/sww2pts.py (modified) (2 diffs)

trunk/anuga_core/source/anuga/file_conversion/sww2pts.py

@@ -56 +56 @@
     # Read sww file
     if verbose: log.critical('Reading from %s' % name_in)
-    from Scientific.IO.NetCDF import NetCDFFile
+    from anuga.file.netcdf import NetCDFFile
     fid = NetCDFFile(name_in)
 
@@ -64 +64 @@
     volumes = fid.variables['volumes'][:]
 
-    number_of_timesteps = fid.dimensions['number_of_timesteps']
-    number_of_points = fid.dimensions['number_of_points']
+
+    try: # works with netcdf4
+        number_of_timesteps = len(fid.dimensions['number_of_timesteps'])
+        number_of_points = len(fid.dimensions['number_of_points'])
+    except: #works with scientific.io.netcdf
+        number_of_timesteps = fid.dimensions['number_of_timesteps']
+        number_of_points = fid.dimensions['number_of_points']
+
+        
     if origin is None:
         # Get geo_reference