| 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 | # Create urs boundary from mux2files # |
|---|
| 14 | ###################################### |
|---|
| 15 | |
|---|
| 16 | dir=os.path.join(project.muxhome,'mux') |
|---|
| 17 | |
|---|
| 18 | # taken from David's event list |
|---|
| 19 | urs_filenames = [ |
|---|
| 20 | os.path.join(dir,'Java-0000-z.grd'), |
|---|
| 21 | os.path.join(dir,'Java-0001-z.grd'), |
|---|
| 22 | os.path.join(dir,'Java-0002-z.grd'), |
|---|
| 23 | os.path.join(dir,'Java-0003-z.grd'), |
|---|
| 24 | os.path.join(dir,'Java-0005-z.grd'), |
|---|
| 25 | os.path.join(dir,'Java-0006-z.grd'), |
|---|
| 26 | os.path.join(dir,'Java-0007-z.grd'), |
|---|
| 27 | os.path.join(dir,'Java-0008-z.grd'), |
|---|
| 28 | os.path.join(dir,'Java-0010-z.grd'), |
|---|
| 29 | os.path.join(dir,'Java-0011-z.grd'), |
|---|
| 30 | os.path.join(dir,'Java-0012-z.grd'), |
|---|
| 31 | os.path.join(dir,'Java-0013-z.grd'), |
|---|
| 32 | os.path.join(dir,'Java-0015-z.grd'), |
|---|
| 33 | os.path.join(dir,'Java-0016-z.grd'), |
|---|
| 34 | os.path.join(dir,'Java-0017-z.grd'), |
|---|
| 35 | os.path.join(dir,'Java-0018-z.grd'), |
|---|
| 36 | os.path.join(dir,'Java-0020-z.grd'), |
|---|
| 37 | os.path.join(dir,'Java-0021-z.grd'), |
|---|
| 38 | os.path.join(dir,'Java-0022-z.grd'), |
|---|
| 39 | os.path.join(dir,'Java-0023-z.grd'), |
|---|
| 40 | os.path.join(dir,'Java-0025-z.grd'), |
|---|
| 41 | os.path.join(dir,'Java-0026-z.grd'), |
|---|
| 42 | os.path.join(dir,'Java-0027-z.grd'), |
|---|
| 43 | os.path.join(dir,'Java-0028-z.grd'), |
|---|
| 44 | os.path.join(dir,'Java-0029-z.grd'), |
|---|
| 45 | os.path.join(dir,'Java-0030-z.grd'), |
|---|
| 46 | os.path.join(dir,'Java-0031-z.grd'), |
|---|
| 47 | os.path.join(dir,'Java-0032-z.grd'), |
|---|
| 48 | os.path.join(dir,'Java-0033-z.grd'), |
|---|
| 49 | os.path.join(dir,'Java-0034-z.grd'), |
|---|
| 50 | os.path.join(dir,'Java-0035-z.grd'), |
|---|
| 51 | os.path.join(dir,'Java-0036-z.grd'), |
|---|
| 52 | os.path.join(dir,'Java-0037-z.grd'), |
|---|
| 53 | os.path.join(dir,'Java-0038-z.grd'), |
|---|
| 54 | os.path.join(dir,'Java-0039-z.grd'), |
|---|
| 55 | os.path.join(dir,'Java-0040-z.grd'), |
|---|
| 56 | os.path.join(dir,'Java-0041-z.grd'), |
|---|
| 57 | os.path.join(dir,'Java-0042-z.grd'), |
|---|
| 58 | os.path.join(dir,'Java-0043-z.grd'), |
|---|
| 59 | os.path.join(dir,'Java-0044-z.grd'), |
|---|
| 60 | os.path.join(dir,'Java-0045-z.grd'), |
|---|
| 61 | os.path.join(dir,'Java-0046-z.grd'), |
|---|
| 62 | os.path.join(dir,'Java-0047-z.grd'), |
|---|
| 63 | os.path.join(dir,'Java-0048-z.grd')] |
|---|
| 64 | |
|---|
| 65 | print 'number of sources', len(urs_filenames) |
|---|
| 66 | |
|---|
| 67 | #from os import sys |
|---|
| 68 | #sys.exit() |
|---|
| 69 | # for all mux files |
|---|
| 70 | #urs_filenames={} |
|---|
| 71 | ##for filename in os.listdir(dir) |
|---|
| 72 | ## basename = filename[:-7] |
|---|
| 73 | ## |
|---|
| 74 | ## urs_filenames[basename] = True |
|---|
| 75 | |
|---|
| 76 | # Make list of basenames |
|---|
| 77 | #urs_filenames = [os.path.join(dir, basename) for basename in urs_filenames.keys()] |
|---|
| 78 | |
|---|
| 79 | # AS per David Burbidge email on Friday 4th July the mag 9.3 event |
|---|
| 80 | # has 1m worth of slip on each sub fault therefore mutliple each unit |
|---|
| 81 | # source by the slip (10.4544) and sum the 44 time series together to |
|---|
| 82 | # get the time series for this event at the points on your boundary. |
|---|
| 83 | |
|---|
| 84 | weight_factor = 10.4544 |
|---|
| 85 | weights=weight_factor*ones(len(urs_filenames),Float) |
|---|
| 86 | |
|---|
| 87 | scenario_name=project.scenario_name |
|---|
| 88 | order_filename=os.path.join(project.boundaries_dir, 'thinned_boundary_ordering.txt') |
|---|
| 89 | |
|---|
| 90 | print 'reading', order_filename |
|---|
| 91 | # Create ordered sts file |
|---|
| 92 | print 'creating sts file' |
|---|
| 93 | |
|---|
| 94 | urs2sts(urs_filenames, |
|---|
| 95 | basename_out=os.path.join(project.boundaries_dir,scenario_name), |
|---|
| 96 | ordering_filename=order_filename, |
|---|
| 97 | zone=50, #enforcing zone 50 |
|---|
| 98 | weights=weights, |
|---|
| 99 | mean_stage=project.tide, |
|---|
| 100 | verbose=True) |
|---|
