import slim3d from slimPre import make_directory output_directory = 'output' make_directory(output_directory) domain = slim3d.Domain('data/mesh3d.msh') equations = slim3d.Slim3d_equations(domain, salinity=True) equations.set_coriolis(('data/coriolis.nc', 'coriolis')) equations.set_initial_temperature('netcdf', temperature=('data/initialCondition.nc', 'temperature')) equations.set_initial_salinity('netcdf', salinity=('data/initialCondition.nc', 'salinity')) #equations.set_initial_elevation(('data/initialCondition.nc', 'elevation')) equations.set_boundary_coast('coast') #equations.set_boundary_coast('yellow_river') #equations.set_boundary_coast('yellow_sea_north') #equations.set_boundary_coast('yellow_sea_center') #equations.set_boundary_coast('yellow_sea_south') equations.set_boundary_open('yellow_river', flux=('data/river_discharge.nc', 'river_discharge'), temperature=('data/initialCondition.nc', 'temperature'), salinity=('data/river_salinity.nc', 'river_salinity')) equations.set_boundary_open('yellow_sea_north', eta=('data/yellow_sea_north.nc', 'h'), u=('data/yellow_sea_north.nc', 'u'), v=('data/yellow_sea_north.nc', 'v'), temperature=('data/initialCondition.nc', 'temperature'), salinity=('data/initialCondition.nc', 'salinity'), transport=True) equations.set_boundary_open('yellow_sea_center', eta=('data/yellow_sea_center.nc', 'h'), u=('data/yellow_sea_center.nc', 'u'), v=('data/yellow_sea_center.nc', 'v'), temperature=('data/initialCondition.nc', 'temperature'), salinity=('data/initialCondition.nc', 'salinity'), transport=True) equations.set_boundary_open('yellow_sea_south', eta=('data/yellow_sea_south.nc', 'h'), u=('data/yellow_sea_south.nc', 'u'), v=('data/yellow_sea_south.nc', 'v'), temperature=('data/initialCondition.nc', 'temperature'), salinity=('data/initialCondition.nc', 'salinity'), transport=True) initial_time = '2016-01-01 00:00:00' final_time = '2016-01-05 00:00:00' time_loop = slim3d.Loop(equations, 1, 50, 300, initial_time, final_time, output_directory) time_loop.export_elevation() time_loop.export_salinity() time_loop.export_uv() time_loop.export_uv2d() time_loop.setup() time_loop.loop()