Changeset 3677

Timestamp:

Sep 29, 2006, 1:43:41 PM (18 years ago)

Author:

sexton

Message:

(i) make thinbathy into a function (ii) make work for Hobart test (iii) needs a unit test (esp if other uses end up using MOST?)

Location:

anuga_core/source/anuga/extras

Files:

• __init__.py (added)
• thinbathy.py (modified) (1 diff)

anuga_core/source/anuga/extras/thinbathy.py

@@ -3 +3 @@
 """
 
-input_file = 'e5_bathy_corr5.cdf'
-output_file = 'SU-AU_e.nc'
+#input_file = 'e5_bathy_corr5.cdf'
+#output_file = 'SU-AU_e.nc'
 
-lat_thin = lon_thin = 4
+#This should have a test
 
-from Scientific.IO.NetCDF import NetCDFFile
+def thinbathy(input_file, output_file):
 
-fid = NetCDFFile(input_file, 'r')
+    lat_thin = lon_thin = 4 
 
-latitudes = fid.variables['LAT'][::lat_thin]
-longitudes = fid.variables['LON'][::lon_thin]
-elevations = fid.variables['ELEVATION'][::lat_thin,::lon_thin]
+    from Scientific.IO.NetCDF import NetCDFFile
 
+    fid = NetCDFFile(input_file, 'r')
 
-outfile = NetCDFFile(output_file, 'w')
-outfile.createDimension('LON', fid.dimensions['LON']/lon_thin + 1)
-outfile.createDimension('LAT', fid.dimensions['LAT']/lat_thin + 1)
+    latitudes = fid.variables['LAT'][::lat_thin]
+    longitudes = fid.variables['LON'][::lon_thin]
+    elevations = fid.variables['ELEVATION'][::lat_thin,::lon_thin]
 
+    outfile = NetCDFFile(output_file, 'w')
+    outfile.createDimension('LON', fid.dimensions['LON']/lon_thin) # + 1)
+    outfile.createDimension('LAT', fid.dimensions['LAT']/lat_thin) # + 1)
 
-import Numeric
-outfile.createVariable('LON', Numeric.Float64, ('LON',))
-outfile.createVariable('LAT', Numeric.Float64, ('LAT',))
-outfile.createVariable('ELEVATION', Numeric.Float32, ('LAT', 'LON'))
+    import Numeric
+    outfile.createVariable('LON', Numeric.Float64, ('LON',))
+    outfile.createVariable('LAT', Numeric.Float64, ('LAT',))
+    outfile.createVariable('ELEVATION', Numeric.Float32, ('LAT', 'LON'))
 
-#Assign to NC file
-#Precision used by MOST for lat/lon is 4 or 5 decimals
-outfile.variables['LAT'][:] = Numeric.around(latitudes, 5)
-outfile.variables['LON'][:] = Numeric.around(longitudes, 5)
-outfile.variables['ELEVATION'][:] = elevations
+    #Assign to NC file
+    #Precision used by MOST for lat/lon is 4 or 5 decimals
+    outfile.variables['LAT'][:] = Numeric.around(latitudes, 5)
+    outfile.variables['LON'][:] = Numeric.around(longitudes, 5)
+    outfile.variables['ELEVATION'][:] = elevations
 
-fid.close()
-outfile.close()
+    fid.close()
+    outfile.close()