1 | from anuga.utilities.numerical_tools import ensure_numeric |
---|
2 | from Scientific.IO.NetCDF import NetCDFFile |
---|
3 | from Numeric import asarray,transpose,sqrt,argmax,argmin,arange,Float,\ |
---|
4 | compress,zeros,fabs,allclose,ones |
---|
5 | from anuga.utilities.polygon import inside_polygon,read_polygon |
---|
6 | from os import sep |
---|
7 | from time import localtime, strftime, gmtime |
---|
8 | from anuga.shallow_water.data_manager import urs2sts,create_sts_boundary |
---|
9 | import os |
---|
10 | import project |
---|
11 | |
---|
12 | |
---|
13 | ###################################### |
---|
14 | # Create urs boundary from mux2files # |
---|
15 | ###################################### |
---|
16 | |
---|
17 | #print project.muxhome |
---|
18 | dir=os.path.join(project.muxhome,'mux') |
---|
19 | #print dir |
---|
20 | prefix=os.path.join(dir,'Java-00') |
---|
21 | suffix='-z.grd' |
---|
22 | |
---|
23 | |
---|
24 | |
---|
25 | |
---|
26 | urs_filenames={} |
---|
27 | for filename in os.listdir(dir): |
---|
28 | basename = filename[:-7] |
---|
29 | |
---|
30 | urs_filenames[basename] = True |
---|
31 | |
---|
32 | |
---|
33 | # Make list of basenames |
---|
34 | urs_filenames = [os.path.join(dir, basename) for basename in urs_filenames.keys()] |
---|
35 | |
---|
36 | # Reduce this list slightly |
---|
37 | urs_filenames = urs_filenames[:30] |
---|
38 | for name in urs_filenames: |
---|
39 | print name |
---|
40 | |
---|
41 | weights=10.4544*ones(len(urs_filenames),Float) |
---|
42 | |
---|
43 | base_name=project.scenario_name |
---|
44 | order_filename=os.path.join(project.boundaries_dir, 'boundary_ordering.txt') |
---|
45 | |
---|
46 | print 'reading', order_filename |
---|
47 | # Create ordered sts file |
---|
48 | print 'creating sts file' |
---|
49 | urs2sts(urs_filenames,basename_out=base_name, |
---|
50 | ordering_filename=order_filename, |
---|
51 | weights=weights, |
---|
52 | mean_stage=project.tide, |
---|
53 | verbose=True) |
---|
54 | |
---|
55 | # Read in boundary from ordered sts file |
---|
56 | urs_boundary=create_sts_boundary(base_name) |
---|
57 | |
---|