def parse_args(args):
if len(args) == 1:
print('Usage is: ws2ModEM <input data> <output data>')
return False
elif len(args) == 2:
in_file = sys.argv[1]
print('No output specified. Using "ModEM_input.data".\n')
out_file = 'ModEM_input.data'
elif len(args) == 3:
in_file = sys.argv[1]
out_file = sys.argv[2]
else:
print('Too many inputs.\n')
print('Usage is: ws2ModEM <input data> <output data>\n')
return
if not utils.check_file(in_file):
print('{} not found.\n')
return False
if utils.check_file(out_file):
print('{} already exists.\n'.format(out_file))
user_response = WSIO.verify_input('Overwrite?',
expected='yn',
default='y')
if user_response.lower() == 'n':
out_file = WSIO.verify_input('Please specify another output file.',
expected='write')
return in_file, out_file
def confirm_details(best_model, models):
print('Best model is {} with an RMS of {}'.format(best_model[0], best_model[1]))
resp = WSIO.verify_input('This is the correct model?', expected='yn', default='y')
if resp == 'n':
print('Options are:')
for model in models:
print('\t{} RMS: {}'.format(model[0], model[1]))
model = WSIO.verify_input(message='New selection', expected='read', default=best_model[0])
else:
model = best_model[0]
listfile = WSIO.verify_input(message='Enter list file:', expected='read', default=None)
outfile = WSIO.verify_input(message='Output file name?', expected='write', default='Report.txt')
return model, listfile, outfile
def pick_periods(sorted_periods, period_set, interactive):
sorted_ratios = [period_set[p] * 100 for p in sorted_periods]
log_per = [utils.truncate(np.log10(x)) for x in sorted_periods]
num = list(range(1, len(sorted_periods) + 1))
for ii, p in enumerate(sorted_periods):
if p < 1:
p = -1 / p
sorted_periods[ii] = utils.truncate(p)
data = np.transpose(np.array([num, sorted_periods, log_per,
sorted_ratios]))
if interactive and len(num) <= 45:
col_labels = ("Number", "Period", "Log10(Period)",
"Percentage of sites")
fig, ax = plt.subplots(1, 1)
ax.axis('off')
# Make this table clickable for true interactive period selection.
ax.table(cellText=data, colLabels=col_labels, loc='center')
plt.tight_layout(rect=[0.05, 0.15, 0.95, 0.9])
plt.ion()
plt.show()
else:
print('Number\tPeriod\tLog10(Period)\t%-age of Sites\n')
for idx in range(data.shape[0]):
print('{:<8d}{:7.2f}{:13.2f}{:14.2f}'.format(
int(data[idx, 0]), data[idx, 1], data[idx, 2], data[idx, 3]))
chosen = []
print(
'Select periods by entering the corresponding integers. Enter 0 when done.'
)
while True:
chosen.append(
WSIO.verify_input('Enter period number', expected=int, default=0))
if chosen[-1] == 0:
del chosen[-1]
chosen = [idx - 1 for idx in chosen]
print('Selected Periods:')
for idx in chosen:
print('{}\n'.format(sorted_periods[idx]))
resp = WSIO.verify_input('Continue with these?',
default='y',
expected='yn')
if resp == 'y':
break
else:
chosen = []
print('Starting over...')
ret = list(sorted([sorted_periods[ii] for ii in chosen]))
for ii, p in enumerate(ret):
if p < 0:
ret[ii] = -1 / p
return list(sorted([sorted_periods[ii] for ii in chosen]))
def main():
best_model, models = scan_models()
startup = WSIO.read_startup()
model, listfile, outfile = confirm_details(best_model, models)
response = model.replace('model', 'resp')
dataset = WSDS.Dataset(modelfile=model, datafile=startup['datafile'],
responsefile=response, listfile=listfile)
write_template(dataset, startup, outfile)
generate_misfit_plots(dataset)
def MTU_data_types(data, dType=''):
MTU_5a = [
-10400., -8800., -7200., -6000., -5200., -4400., -3600., -3000, -2600.,
-2200., -1800., -1500., -1300., -1100., -900., -780., -640., -530.,
-460., -390., -360., -320., -312., -265., -229., -216., -194., -180.,
-159., -132., -115., -108., -97., -90., -79., -66., -57., -54., -49.,
-45., -40., -39., -33., -27.5, -27., -22.50, -19.50, -18.80, -16.50,
-16.20, -13.70, -13.50, -11.20, -9.700, -9.40, -8.300, -8.100, -6.900,
-6.800, -5.600, -4.900, -4.700, -4.100, -3.400, -2.810, -2.440, -2.340,
-2.060, -1.720, -1.690, -1.410, -1.220, -1.170, -1.030, -1.020, 1.163,
1.190, 1.429, 1.695, 1.961, 2.326, 2.857, 3.413, 3.937, 4.651, 5.682,
6.849, 7.874, 9.346, 11.36, 13.70, 15.87, 18.52, 22.73, 27.03, 31.25,
37.17, 45.45, 54.64, 62.89, 74.63, 90.91, 108.7, 126.6, 149.3, 181.8,
217.4, 250.0, 294.1, 363.6, 436.7, 505.1, 595.2, 729.9, 877.2, 1010.,
1190., 1449., 1754., 2000., 2381., 2941.
]
MTU_5 = [
-320.0, -265.0, -229.0, -194.0, -159.0, -132.0, -115.0, -97.00, -79.00,
-66.00, -57.00, -49.00, -40.00, -33.00, -27.50, -22.50, -18.80, -16.20,
-13.70, -11.20, -9.400, -8.100, -6.900, -5.600, -4.700, -4.100, -3.400,
-2.810, -2.340, -2.030, -1.720, -1.410, -1.170, -1.020, 1.163, 1.429,
1.695, 1.961, 2.326, 2.857, 3.413, 3.937, 4.651, 5.682, 6.849, 7.874,
9.346, 11.36, 13.70, 15.87, 18.52, 22.73, 27.03, 31.25, 37.17, 45.45,
54.64, 62.89, 74.63, 90.91, 108.7, 126.6, 149.3, 181.8, 217.4, 250.0,
294.1, 363.6, 436.7, 505.1, 595.2, 729.9, 877.2, 1010., 1190., 1449.,
1754., 2000., 2381., 2941.
]
if dType.lower() == '5':
retval = MTU_5
elif dType.lower() == 'a':
retval = MTU_5a
elif dType.lower() == 'b':
retval = list(set(MTU_5).union(set(MTU_5a)))
elif dType.lower() == 'o':
retval = WSIO.read_freqset()
elif dType.lower() == 'm':
retval = data.master_periods.keys()
elif dType.lower() == 'n':
retval = data.narrow_periods.keys()
else:
print('Input {} not recognized, try again.'.format(dType))
return False
for ii, val in enumerate(retval):
if val < 0:
retval[ii] = -1 / val
return retval
def test_readDirect_raw_data_nonexistant_data(self):
sites = IO.read_sites(badlist)
with assert_raises(WSFileError):
IO.read_raw_data(site_names=sites)
def test_readDirect_data_nonexistant_data(self):
with assert_raises(WSFileError):
IO.read_data(datafile='this_is_not_a_file.lst')
def test_readDirect_sites_nonexistant_list(self):
with assert_raises(WSFileError):
IO.read_sites('this_is_not_a_file.lst')
def main_data(args):
defaults = {
'azimuth': 0.0,
'flag_outliers': 'y',
'outlier_errmap': 10,
'lowTol': 2.0,
'highTol': 10.0,
'use_TF': 'y',
'period_choice': 'n',
'cTol': 0.5,
'hfreq_errmap': 20,
'XXYY_errmap': 10,
'no_period_errmap': 50
}
if set(('dbg', 'debug', '-dbg',
'-debug')).intersection(set([x.lower() for x in args])):
DEBUG = True
WSDS.DEBUG = True
if set(('i', '-i')).intersection(set([x.lower() for x in args])):
interactive = True
else:
interactive = False
# Get list file
list_file = WSIO.verify_input('Enter list file name:', expected='read')
try:
site_names = WSIO.read_sites(list_file)
except WSFileError as e: # Error is raised if something is wrong with the file.
print(e.message)
print('Exiting...')
return
dataset = WSDS.Dataset(listfile=list_file)
# Get desired azimuth
azimuth = WSIO.verify_input('Desired azimuth (deg. from True North)',
expected=float,
default=defaults['azimuth'])
flag_outliers = WSIO.verify_input('Adjust error map for outliers?',
default=defaults['flag_outliers'],
expected='yn')
if flag_outliers == 'y':
dataset.data.OUTLIER_MAP = WSIO.verify_input(
'Set outlier error map',
default=defaults['outlier_errmap'],
expected=int)
else:
dataset.data.OUTLIER_MAP = 1
dataset.data.HIGHFREQ_MAP = WSIO.verify_input(
'Error map on high frequencies (>1000Hz)',
default=defaults['hfreq_errmap'],
expected=int)
dataset.data.XXYY_MAP = WSIO.verify_input('XXYY error map',
default=defaults['XXYY_errmap'],
expected=int)
dataset.data.NO_PERIOD_MAP = WSIO.verify_input(
'Missing period error map',
default=defaults['no_period_errmap'],
expected=int)
lowTol = WSIO.verify_input(
'High frequency (>1Hz) matching tolerance %-age',
default=defaults['lowTol'],
expected=float)
lowTol /= 100
highTol = WSIO.verify_input(
'High frequency (>=10s) matching tolerance %-age',
default=defaults['highTol'],
expected=float)
highTol /= 100
raw_data = dataset.raw_data
num_TF = 0
for site in raw_data.sites.values():
if 'TZXR' in site.components:
num_TF += 1
print('{} sites out of {} have tipper data'.format(num_TF,
len(site_names)))
use_TF = WSIO.verify_input('Would you like to include TF data?',
default=defaults['use_TF'],
expected='yn')
if use_TF:
defaults.update({'inv_type': 5})
else:
defaults.update({'inv_type': 1})
dType = WSIO.verify_input(
'Which periods would you like to choose from?\n'
'Options are MTU-5, MTU-A, both, freqset, all, or program selected '
'(5/A/b/o/m/n)',
default=defaults['period_choice'],
expected='5aboamn')
# if dType == 'n':
cTol = WSIO.verify_input(
'Required fraction of sites containing each period?',
default=defaults['cTol'],
expected=float)
if cTol > 1:
cTol /= 100
period_set = raw_data.narrow_period_list(periods=MTU_data_types(
data=raw_data, dType=dType),
high_tol=highTol,
low_tol=lowTol,
count_tol=cTol)
sorted_periods = sorted(period_set.keys())
# Can set this up later to only run if -i is set (or set that as the default)
# Want user to be able to decide between interactive (plotting) mode, and just
# straight command line input.
chosen = pick_periods(sorted_periods, period_set, interactive)
for ii, p in enumerate(chosen):
if p < 0:
chosen[ii] = -1 / p
else:
chosen[ii] = p
inv_type = int(
WSIO.verify_input('Inversion Type?',
default=defaults['inv_type'],
expected='12345'))
# Still have to make sure this method also applies the error map properly.
dataset.get_data_from_raw(
lTol=lowTol,
hTol=highTol,
periods=chosen,
components=WSIO.get_components(invType=inv_type)[0])
dataset.data.set_error_map()
if azimuth != 0:
dataset.rotate_sites(azimuth)
while True:
outdata = WSIO.verify_input('Output file name?',
default='.data',
expected='write')
if outdata:
if outdata[-5:] != '.data':
outdata = ''.join([outdata, '.data'])
dataset.write_data(outfile=outdata, overwrite=True)
break
return dataset
def main_mesh(args, data=None):
if data is None:
datafile = WSIO.verify_input('Data file to use?', expected='read')
data = WSDS.Data(datafile=datafile)
avg_rho = np.mean([
utils.compute_rho(site, calc_comp='det', errtype='none')
for site in data.sites.values()
])
print('Average determinant apparent resistivity is {}'.format(avg_rho))
bg_resistivity = WSIO.verify_input('Background resistivity?',
expected=float,
default=avg_rho)
xmesh = utils.generate_lateral_mesh(site_locs=data.locations[:, 0])
ymesh = utils.generate_lateral_mesh(site_locs=data.locations[:, 1])
max_depth = min([500, utils.skin_depth(bg_resistivity, data.periods[-1])])
min_depth = min([1, utils.skin_depth(bg_resistivity, data.periods[0])])
print(
'Note: Default min and max depths are based on skin depth of lowest and highest periods.'
)
min_depth = WSIO.verify_input('Depth of first layer?',
expected=float,
default=min_depth)
max_depth = WSIO.verify_input('Depth of last layer?',
expected=float,
default=max_depth)
NZ = WSIO.verify_input('Total # of layers or # of layers per decade?',
expected='numtuple',
default=60)
zmesh = utils.generate_zmesh(min_depth=min_depth,
max_depth=max_depth,
NZ=NZ)
model = WSDS.Model()
model.dx = xmesh
model.dy = ymesh
model.dz = zmesh
model.vals = bg_resistivity * np.ones((model.nx, model.ny, model.nz))
if '-i' in args:
viewer = meshview(model, data)
viewer.show()
else:
nxpads = WSIO.verify_input('Number of pads in X direction',
expected=int,
default=5)
nypads = WSIO.verify_input('Number of pads in y direction',
expected=int,
default=5)
xpads = [model.xCS[0]]
ypads = [model.yCS[0]]
for ii in range(nxpads):
xpads.append(
WSIO.verify_input('Pad size',
expected=float,
default=xpads[ii] * 1.2))
xpads = xpads[1:]
for ii in range(nypads):
ypads.append(
WSIO.verify_input('Pad size',
expected=float,
default=ypads[ii] * 1.2))
ypads = ypads[:1]
model.dx = xpads[-1::-1] + model.dx + xpads
model.dy = ypads[-1::-1] + model.dy + ypads
modelname = WSIO.verify_input('Model name',
expected='write',
default='mod')
if '.model' not in modelname:
modelname = ''.join([modelname, '.model'])
model.write(modelname)
return