updated to build constant boundary file

unit_tests/gen_tools.py

@@ -449,4 +449,48 @@ def plot_grids(lat_in,lon_in,new_lat,new_lon,off_lat,off_lon,src_lat,src_lon):
     # tweak margins of subplots as tight layout doesn't work
     plt.subplots_adjust(left=0.01, right=1, top=0.9, bottom=0.05,wspace=0.01)
 
-    plt.show()
\ No newline at end of file
+    plt.show()
+
+def write_parameter(fileout, grid_h,grid_z,params):
+    '''
+    Writes out a
+
+    Args:
+
+    Returns:
+    '''
+
+    #TODO: implement multiple parameters using dicts.
+
+    # Open pointer to netcdf file
+    dataset = Dataset(fileout, 'w', format='NETCDF4_CLASSIC')
+
+    # Get input size and create appropriate dimensions
+    # TODO: add some sort of error handling
+    nx, ny, nz = np.shape(grid_z['e3t'])
+    dataset.createDimension('x', nx)
+    dataset.createDimension('y', ny)
+    dataset.createDimension('z', nz)
+
+    # Create Variables
+    longitude = dataset.createVariable('longitude', np.float32, ('y', 'x'))
+    latitude = dataset.createVariable('latitude', np.float32, ('y', 'x'))
+    depth = dataset.createVariable('depth', np.float32, 'z')
+    longitude.units, longitude.long_name = 'km', 'X'
+    latitude.units, latitude.long_name = 'km', 'Y'
+    depth.units, depth.long_name = 'm', 'Z'
+    # Populate file with input data
+    longitude[:, :] = grid_h['lont'].T
+    latitude[:, :] = grid_h['latt'].T
+    depth[:] = grid_z['dept_1d']
+
+    parameter = dataset.createVariable(str(params['name']), np.float64, ('z', 'y', 'x'))
+    parameter.units, parameter.long_name = str(params['units']), str(params['longname'])
+    value_fill = np.ones(np.shape(grid_z['e3t']))
+    value_fill = value_fill*params['const_value']
+    parameter[:, :, :] = value_fill.T
+
+    # Close off pointer
+    dataset.close()
+
+    return 0
\ No newline at end of file

unit_tests/test_gen.py

@@ -101,6 +101,11 @@ def _main():
     # close data files
     #ds.close()
     #src.close()
+    out_fname = 'unit_tests/test_data/output_boundary.nc'
+    params = {'name':'thetao','const_value':15.0,'longname':'temperature','units':'degreesC'}
+    boundary = gt.write_parameter(out_fname,grid_h1,grid_z1,params)
+    if boundary == 0:
+        print('Success!')
 
 if __name__ == '__main__':
     _main()
\ No newline at end of file