"""Create mesh and time boundary for the Okushiri island validation """ #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! # Assume that the root AnuGA dir is included in your pythonpath #!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! from Numeric import array, zeros, Float, allclose from anuga.pmesh.mesh import * from anuga.coordinate_transforms.geo_reference import Geo_reference import project def prepare_timeboundary(filename): """Converting benchmark 2 time series to NetCDF tms file. This is a 'throw-away' code taylor made for files like 'Benchmark_2_input.txt' from the LWRU2 benchmark """ print 'Preparing time boundary from %s' %filename from Numeric import array fid = open(filename) #Skip first line line = fid.readline() #Read remaining lines lines = fid.readlines() fid.close() N = len(lines) T = zeros(N, Float) #Time Q = zeros(N, Float) #Values for i, line in enumerate(lines): fields = line.split() T[i] = float(fields[0]) Q[i] = float(fields[1]) #Create tms file from Scientific.IO.NetCDF import NetCDFFile outfile = filename[:-4] + '.tms' print 'Writing to', outfile fid = NetCDFFile(outfile, 'w') fid.institution = 'Geoscience Australia' fid.description = 'Input wave for Benchmark 2' fid.starttime = 0.0 fid.createDimension('number_of_timesteps', len(T)) fid.createVariable('time', Float, ('number_of_timesteps',)) fid.variables['time'][:] = T fid.createVariable('stage', Float, ('number_of_timesteps',)) fid.variables['stage'][:] = Q[:] fid.createVariable('xmomentum', Float, ('number_of_timesteps',)) fid.variables['xmomentum'][:] = 0.0 fid.createVariable('ymomentum', Float, ('number_of_timesteps',)) fid.variables['ymomentum'][:] = 0.0 fid.close() #------------------------------------------------------------- if __name__ == "__main__": #Preparing points #(Only when using original Benchmark_2_Bathymetry.txt file) #from anuga.abstract_2d_finite_volumes.data_manager import xya2pts #xya2pts(project.bathymetry_filename, verbose = True, # z_func = lambda z: -z) # Prepare time boundary prepare_timeboundary(project.boundary_filename) # Create Mesh #Basic geometry xleft = 0 xright = 5.448 ybottom = 0 ytop = 3.402 #Outline point_sw = [xleft, ybottom] point_se = [xright, ybottom] point_nw = [xleft, ytop] point_ne = [xright, ytop] #Midway points (left) point_mtop = [xleft + (xright-xleft)/3+1, ytop] point_mbottom = [xleft + (xright-xleft)/3+1, ybottom] geo = Geo_reference(xllcorner = xleft, yllcorner = ybottom) print "***********************" print "geo ref", geo print "***********************" m = Mesh(geo_reference=geo) #Boundary dict = {} dict['points'] = [point_se, #se point_ne, point_mtop, point_nw, point_sw, point_mbottom] dict['segments'] = [[0,1], [1,2], [2,3], [3,4], [4,5], [5,0], #The outer border [2,5]] #Separator dict['segment_tags'] = ['wall', 'wall', 'wall', 'wave', 'wall', 'wall', ''] #Interior m.addVertsSegs(dict) #Localised refined area for gulleys dict = {} xl = 4.8 xr = 5.3 yb = 1.6 yt = 2.3 p0 = [xl, yb] p1 = [xr, yb] p2 = [xr, yt] p3 = [xl, yt] dict['points'] = [p0, p1, p2, p3] dict['segments'] = [[0,1], [1,2], [2,3], [3,0]] dict['segment_tags'] = ['', '', '', ''] m.addVertsSegs(dict) #Island area island_0 = [xleft + 2*(xright-xleft)/3+1.2, ytop-0.5] island_1 = [xleft + 2*(xright-xleft)/3+0.5, ybottom + 2*(ytop-ybottom)/3] island_2 = [xleft + (xright-xleft)/2+0.3, ybottom + 2*(ytop-ybottom)/3-0.3] island_3 = [xleft + (xright-xleft)/2+0.3, ybottom + (ytop-ybottom)/3+0.3] island_4 = [xleft + 2*(xright-xleft)/3+0.4, ybottom + (ytop-ybottom)/3-0.3] island_5 = [xleft + 2*(xright-xleft)/3+1.2, ybottom+0.2] island_6 = [xl-.01, yb] #OK island_7 = [xl-.01, yt] #OK dict['points'] = [island_0, island_1, island_2, island_3, island_4, island_5, #p0, p3] island_6, island_7] dict['segments'] = [[0,1], [1,2], [2,3], [3,4], [4,5], [5,6], [6,7], [7,0]] dict['segment_tags'] = ['', '', '', '', '', '', '', ''] m.addVertsSegs(dict) # base_resolution = 1 ocean = m.addRegionEN(xleft+.1, ybottom+.1) ocean.setMaxArea(0.1*base_resolution) mid = m.addRegionEN(point_mbottom[0]+.1, ybottom+.1) mid.setMaxArea(0.0005*base_resolution) inner = m.addRegionEN(island_3[0]+.1, island_3[1]+.1) inner.setMaxArea(0.0003*base_resolution) gulleys = m.addRegionEN(p0[0]+0.1, p0[1]+0.1) gulleys.setMaxArea(0.00002*base_resolution) # From r 1709 11 August 2005 #base_resolution = 100 # #ocean = m.addRegionEN(xleft+.1, ybottom+.1) #ocean.setMaxArea(0.01*base_resolution) # #mid = m.addRegionEN(point_mbottom[0]+.1, ybottom+.1) #mid.setMaxArea(0.001*base_resolution) #inner = m.addRegionEN(island_3[0]+.1, island_3[1]+.1) #inner.setMaxArea(0.0001*base_resolution) #gulleys = m.addRegionEN(p0[0]+0.1, p0[1]+0.1) #gulleys.setMaxArea(0.00001*base_resolution) m.generateMesh('pzq28.0za1000000a') import project m.export_mesh_file(project.mesh_filename)