Changeset 1122

Timestamp:

Mar 22, 2005, 4:19:10 PM (20 years ago)

Author:

prow

Message:

 

Location:

inundation/ga/storm_surge/pyvolution

Files:

• most2nc.py (modified) (1 diff)
• test_most2nc.py (added)

inundation/ga/storm_surge/pyvolution/most2nc.py

@@ -1 @@
-input_file = 'small.txt'
-output_file = 'small_e.nc'
+def most2nc(input_file=None,output_file=None,inverted_bathymetry = False):
+    #input_file = 'small.txt'
+    #output_file = 'small_e.nc'
 
-long_name = 'LON'
-lat_name = 'LAT'
+    long_name = 'LON'
+    lat_name = 'LAT'
+    if inverted_bathymetry:
+        up = +1.
+    else:
+        up = -1.
+    
+    from Scientific.IO.NetCDF import NetCDFFile
+    import sys
+    
+    try:
+        if input_file is None:
+            input_file = sys.argv[1]
+        if output_file is None:
+            output_file = sys.argv[2]
+    except:
+        raise 'usage is: most2nc input output'
+    
+    in_file = open(input_file,'r')
+    print 'reading header'
+    nx_ny_str = in_file.readline()
+    nx_str,ny_str = nx_ny_str.split()
+    nx = int(nx_str)
+    ny = int(ny_str)
+    h1_list=[]
+    for i in range(nx):
+        h1_list.append(float(in_file.readline()))
 
-up = +1.
+    h2_list=[]
+    for j in range(ny):
+        h2_list.append(float(in_file.readline()))
 
-from Scientific.IO.NetCDF import NetCDFFile
+    h2_list.reverse()
 
-in_file = open(input_file,'r')
-print 'reading header'
-nx_ny_str = in_file.readline()
-nx_str,ny_str = nx_ny_str.split()
-nx = int(nx_str)
-ny = int(ny_str)
-h1_list=[]
-for i in range(nx):
-    h1_list.append(float(in_file.readline()))
+    print 'reading depths'
+    in_depth_list = in_file.readlines()
+    in_file.close()
 
-h2_list=[]
-for j in range(ny):
-    h2_list.append(float(in_file.readline()))
+    out_depth_list = [[]]
 
-h2_list.reverse()
+    print 'processing depths'
+    k=1
+    for in_line in in_depth_list:
+        for string in in_line.split():
+            #j = k/nx
+            out_depth_list[(k-1)/nx].append(float(string)*up)
+            #print k,len(out_depth_list),(k-1)/nx,out_depth_list[(k-1)/nx][-1],len(out_depth_list[(k-1)/nx])
+            if k==nx*ny:
+                break
+            if k-(k/nx)*nx ==0:
+                out_depth_list.append([])
+            k+=1
 
-print 'reading depths'
-in_depth_list = in_file.readlines()
-in_file.close()
+    in_file.close()
+    out_depth_list.reverse()
+    depth_list = out_depth_list
+    print out_depth_list
 
-out_depth_list = [[]]
+    print 'writing results'
+    out_file = NetCDFFile(output_file,'w')
 
-print 'processing depths'
-k=1
-for in_line in in_depth_list:
-    for string in in_line.split():
-        #j = k/nx
-        out_depth_list[(k-1)/nx].append(float(string)*up)
-        #print k,len(out_depth_list),(k-1)/nx,out_depth_list[(k-1)/nx][-1],len(out_depth_list[(k-1)/nx])
-        if k==nx*ny:
-            break
-        if k-(k/nx)*nx ==0:
-            out_depth_list.append([])
-        k+=1
+    out_file.createDimension(long_name,nx)
+    out_file.createVariable(long_name,'d',(long_name,))
+    out_file.variables[long_name].point_spacing='uneven'
+    out_file.variables[long_name].units='degrees_east'
+    out_file.variables[long_name].assignValue(h1_list)
 
-in_file.close()
-out_depth_list.reverse()
-depth_list = out_depth_list
-print out_depth_list
+    out_file.createDimension(lat_name,ny)
+    out_file.createVariable(lat_name,'d',(lat_name,))
+    out_file.variables[lat_name].point_spacing='uneven'
+    out_file.variables[lat_name].units='degrees_north'
+    out_file.variables[lat_name].assignValue(h2_list)
 
-print 'writing results'
-out_file = NetCDFFile(output_file,'w')
+    out_file.createVariable('ELEVATION','d',(lat_name,long_name))
+    out_file.variables['ELEVATION'].point_spacing='uneven'
+    out_file.variables['ELEVATION'].units='meters'
+    out_file.variables['ELEVATION'].assignValue(depth_list)
 
-out_file.createDimension(long_name,nx)
-out_file.createVariable('LON','d',(long_name,))
-out_file.variables['LON'].point_spacing='uneven'
-out_file.variables['LON'].units='degrees_east'
-out_file.variables['LON'].assignValue(h1_list)
-
-out_file.createDimension(lat_name,ny)
-out_file.createVariable('LAT','d',(lat_name,))
-out_file.variables['LAT'].point_spacing='uneven'
-out_file.variables['LAT'].units='degrees_north'
-out_file.variables['LAT'].assignValue(h2_list)
-
-out_file.createVariable('ELEVATION','d',(lat_name,long_name))
-out_file.variables['ELEVATION'].point_spacing='uneven'
-out_file.variables['ELEVATION'].units='meters'
-out_file.variables['ELEVATION'].assignValue(depth_list)
-
-out_file.close()
+    out_file.close()