from tools.source_shapefile_builder import get_preferred_catalogue, \
get_completeness_model, get_aus_shmax_vectors, \
get_rate_adjust_factor, build_source_shape, \
get_ul_seismo_depths
except:
print 'Add PYTHONPATH to NSHA18 root directory'
###############################################################################
# parse Mcomp shp
###############################################################################
#arupshp = 'ARUP_source_model.shp'
mcshp = 'completeness_zones.shp'
# get preferred catalogues
prefCat = get_preferred_catalogue(mcshp)
print 'Reading source shapefile...'
sf = shapefile.Reader(mcshp)
shapes = sf.shapes()
polygons = []
for poly in shapes:
polygons.append(Polygon(poly.points))
# get src name
src_names = get_field_data(sf, 'NAME', 'str')
src_codes = src_names
###############################################################################
# get neotectonic domains class and Mmax from zone centroid
###############################################################################
dep_b.append(10.)
dep_u.append(5.0)
dep_l.append(15.)
# don't care about depth logic tree outside Australia
else:
dep_b.append(hd[i])
dep_u.append(-999)
dep_l.append(-999)
###############################################################################
# get preferred catalogues
###############################################################################
# get preferred catalogues for each zone
prefCat = get_preferred_catalogue(domshp)
# fix catalogue for source zones
prefCat[55] = 'NSHA18CAT_V0.1_hmtk_declustered.csv'
prefCat[56] = 'NSHA18CAT_V0.1_hmtk_declustered.csv'
###############################################################################
# load 2018 completeness models
###############################################################################
single_mc = 0
ycomp, mcomp, min_rmag = get_completeness_model(src_codes, shapes, domains,
single_mc)
# use manual modification
for i in range(0, len(trt_new)):
if trt_new[i] == 'Active':
# don't care about depth logic tree outside Australia
else:
dep_b.append(hd[i])
dep_u.append(-999)
dep_l.append(-999)
# fix preferred upper/lower seismo depths from Domains
usd, lsd = get_ul_seismo_depths(src_codes, usd, lsd)
'''
###############################################################################
# get preferred catalogues
###############################################################################
# get preferred catalogues for each zone
prefCat = get_preferred_catalogue(outshp)
'''
# fix catalogue for source zones
prefCat[44] = 'NSHA18CAT_V0.1_hmtk_declustered.csv'
prefCat[2] = 'NSHA18CAT_V0.1_hmtk_declustered.csv'
'''
###############################################################################
# load 2018 completeness models
###############################################################################
single_mc = 0
ycomp, mcomp, min_rmag_ignore = get_completeness_model(src_codes, shapes,
neo_domains, single_mc)
#ycomp, mcomp, min_rmag_ignore = get_completeness_model_point(src_codes, shapes, single_mc)
# use values from Domains model instead
min_rmag = neo_min_rmag
