Context Navigation

← Previous Revision
Next Revision →
Normal
Revision Log

EventSelection.py

Last change on this file was 7214, checked in by rwilson, 16 years ago
Back- (and forward)-merge, Numeric to numpy.
File size: 38.5 KB

Rev	Line
[6251]	1	#!/usr/bin/env python
	2	################################################################################
	3	# Earthquake Event Selection.
	4	################################################################################
	5
[6330]	6
[6251]	7	import os
	8	import os.path
	9	import sys
	10	import re
	11	import shutil
	12	import time
[6330]	13
[6251]	14	import tempfile
	15	try:
[7200]	16	import cpickle as pickle
[6251]	17	except:
	18	import pickle
	19
	20	try:
	21	import wx
	22	except ImportError:
[6333]	23	import tkinter_error
[6330]	24	msg = ('Sorry, you have to install WxPython.\n'
	25	'Get it from either:\n'
[6364]	26	r' N:\georisk\downloads\wxpython, or' '\n'
[6330]	27	' [http://www.wxpython.org/download.php#binaries].')
[6333]	28	tkinter_error.tkinter_error(msg)
[6330]	29	print msg
[6251]	30	sys.exit(1)
[6330]	31
	32
[6251]	33	Imported_PyEmbeddedImage = True
	34	try:
	35	from wx.lib.embeddedimage import PyEmbeddedImage
	36	except ImportError:
	37	Imported_PyEmbeddedImage = False
	38
	39
	40	######
	41	# Various constants
	42	######
	43
	44	# where the data lives
	45	if sys.platform == 'win32':
	46	BaseDirectory = r'N:\georisk_models\tsunami\models\PTHA'
	47	else:
	48	BaseDirectory = '/nas/gemd/georisk_models/tsunami/models/PTHA'
	49
	50	# name of the fault name file (in multimux directory)
	51	FaultNameFilename = 'fault_list.txt'
	52
	53	# program name and version
	54	APP_NAME = 'EventSelection'
[7214]	55	APP_VERSION = '1.1'
[6251]	56
	57	# name of the configuration filename
	58	# GUI values are saved in this file for next time
	59	ConfigFilename = 'EventSelection.cfg'
	60
	61	# name of the 'fault XY' output file
	62	FaultXYFilename = 'fault.xy'
	63
	64	# name of the quake probability output file
	65	QuakeProbFilename = 'quake_prob.txt'
	66
	67	# name of the output URS data module
	68	# note: the %d is where the event number goes
	69	URSDataFilename = 'urs_data_%6.6d.py'
	70
	71	# name of the output bash params file
	72	# note: the %d is where the event number goes
	73	GridBashScriptName = 'faults_%6.6d.params'
	74
	75	# map display name to directory name
	76	Regions = [ ('Australia', 'Australia'),
	77	('India', 'Indian'),
	78	('SW Pacific', 'SW_Pacific')
	79	]
	80
	81	# pattern string used to split multimax data
	82	SpacesPatternString = ' +'
	83
	84	# generate 're' pattern for 'any number of spaces'
	85	SpacesPattern = re.compile(SpacesPatternString)
	86
	87	# Global holding current results sub-directory
	88	ResultSubdir = ''
	89
	90	# a small value to perturb the user limits
	91	Epsilon = 1.0E-6
	92
	93	# GUI definitions
	94	BOX_WIDTH = 400
	95	BOX_HEIGHT = 360
	96	BOX2_HEIGHT = 285
	97
	98
	99	FORM_WIDTH = BOX_WIDTH + 15
	100	FORM_HEIGHT = 755
	101
	102	START_YOFFSET = 7
	103	OUT_TEXT_XOFFSET = 8
	104	OUT_TEXT_YOFFSET = 11
	105	OUTPUT_BOX_HEIGHT = 45
	106
	107	GEN_YOFFSET = 15
	108	GEN_DELTAY = 25
	109	GEN_LABELXOFFSET = 5
	110	GEN_TEXTXOFFSET = BOX_WIDTH/2 + GEN_LABELXOFFSET + 10
	111
	112	TEXTLIST_WIDTH = BOX_WIDTH - 10
	113	TEXTLIST_HEIGHT = 200
	114
	115	CTL_WIDTH = BOX_WIDTH / 2
	116	CTL_HEIGHT = 22
	117
	118	BUTTON_WIDTH = 100
	119	BUTTON_HEIGHT = 30
	120
	121	DOUBLE_BUTTON_OFFSET = 8
	122
	123	COMBO_FUDGE = 2
	124
	125	# Copies of various parameters
	126	Region = ''
	127	GaugeNumber = ''
	128	MinimumHeight = ''
	129	MaximumHeight = ''
	130
	131
	132	################################################################################
	133	# This code was generated by img2py.py
	134	# Embed the ICON image here, so we don't need an external *.ico file.
	135	# Icon image inspired by Hokusai's print "The Great Wave off Kanagawa"
	136	# (Kanagawa okinami ura) from the "Thirty-Six Views of Mount Fuji" series.
	137	# http://commons.wikimedia.org/wiki/Image:The_Great_Wave_off_Kanagawa.jpg
	138	################################################################################
	139
	140	if Imported_PyEmbeddedImage:
	141	def getIcon():
	142	return PyEmbeddedImage(
	143	"iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAABHNCSVQICAgIfAhkiAAABDBJ"
	144	"REFUWIXtlmtsFFUUx393dnbbbh+hlPQVKEEBKSLyEEEhqFQSNYAQNahAIJJUidEEtUQIRgk+"
	145	"GkSMJBJjUEADCVghtipSjeWhWIoh5VXoBqgU2aWvLbt0H52dmesH6La1U7OVbvCD59PcmTPn"
	146	"/M69/3vuFYDkFppyK5P/D/CfAFD//kIgcGbkkpk/GYczjat/umiqrUIaBgA2RwKG1h71lxZP"
	147	"/xJAkJiazn3LNnD7g8+gqI5o0Iaawxza+CJ3zlnGiIKF+C6f5+iWVTicaaRkDSXY4uHCwa/Q"
	148	"20N9BhGAdA7IIjVnGNNf3Uz60NEghYWr7PJLz3GguZ6Tuz/C8/teQoFrDBoxEefAbPyeOtzV"
	149	"FZhGxBpg9voDMnvMVIRi6xN57xVJJF0LkDScreSHVbNob/P28Fdyxk7vt+TX0wmuz46k/si3"
	150	"uMq3kTVqCmPmvWTp30OE/WWu8q1UbV5J2N9C7rgZvS5BXACklFTveI9BIycweNKj1FeWcWrP"
	151	"RojOTqfdVB8QvSheILEnp6EF/OgBPwNvG8v8LS6e3lrLhAWrUZTOukVhuYzLYSRNo4u2ZLfi"
	152	"qza/zvFd7yOR8euEXZNfOFjCzsV3sOu5fLwXTzP84UXRuYt7K24+d4yK4oUEvW600DXqDpTQ"
	153	"1ngxulHjtgs67OLhUgxDx56cxl1zX0ZKk0MbCqPf46aBDpPSIBJqI3y1Gb/7HJ4T+6nd+xlB"
	154	"X1NXAAlYtV/LkN18TSNCqLUR58DsmBtaqNXD9ysfwVt3skMDsSa/7msaEa6c/pWg14PeHmTn"
	155	"khGUvTKda1cuWAI3nj3CH7/soaHmNwy9naT0bKYUrgdE7yIUwmplJH73OUqXT+OnNU9wfNc6"
	156	"bPZEFNVO45lKKtYtBiRhfzOGFqZj39WUbuLHtU9SunwquwvHEWhxkzF8AopN7R1AWpyIquZn"
	157	"3xuz8F2qZdKSNeTdM5MzO9/BnpBE7riHULU2VAWqd7zL1y+Mp7W+BhDcPX8FAsHo2cuItAcJ"
	158	"eT00uaqQht43Edbu+5wDHyzt8T5j2BimLV2Lt+EyWijAsS/ewjQN0vPymfvxUYRQ2P7sEOyJ"
	159	"yeTPep6w30tN2Sa0gK9v2/D8z9u7z4iqous6LXWn+Gb1PBwOB5qmkZiaQcGKbZS/+Tju6goG"
	160	"T5yJPSkV32UXlZ8WdYsRcyMSSPRQoHMsBKZpdvPRNA2AlMwhaG1eMkdNJiUrj5CviUDTJcu4"
	161	"MQNIIDlraOdYyh4AHdZSd4JIOMhjxfsAyf7iRTdEaVFYrBqwKZLqkg+p/OQ1ZAz3PiEENjUB"
	162	"U9cwpTUogBLrJVJKGD3jKewJSTH6S/RI+B+T3wCIrQmZUmAbMJh7F6xAtfXfEdLH01Awcl4R"
	163	"eeMfwOFw3AoAUBOc3F/0JZnDx5OUFNty9CsAgDM9h4K3vyNvyhxSUlJvCuAvPDCzruPdn6UA"
	164	"AAAASUVORK5CYII=")
	165
	166
	167	################################################################################
	168	# An object that behaves like a dictionary but is pickled/unpickled from a file.
	169	################################################################################
	170
	171	class Config(object):
[6362]	172	"""An object that behaves like a dictionary but is persistent
[6251]	173
	174	cfg = Config('filename')
	175	cfg['save'] = 'A saved value'
	176	cfg[123] = 'value 123'
	177	print cfg['save']
	178
	179	The values stored in the file 'filename' will be available to any
	180	application using that config file.
	181	"""
	182
	183	def __init__(self, configfile=None):
[6362]	184	"""__init__(self, String filename=None) -> Config"""
[6251]	185
	186	self.delconf = False
	187	self.cfgdict = {}
	188	self.changed = False
	189	if not configfile:
	190	self.delconf = True
	191	configfile =tempfile.mktemp()
	192	self.configfile = os.path.abspath(configfile)
	193	if os.path.exists(self.configfile):
	194	try:
	195	f = open(configfile, "r")
	196	u = pickle.Unpickler(f)
	197	self.cfgdict = u.load()
	198	except pickle.UnpicklingError, e:
	199	print e
	200	except:
	201	pass
	202	else:
	203	f.close()
	204
	205	def __setitem__(self, key, value):
[6362]	206	"""Override to allow: cfg[<key>] = <value>"""
	207
[6251]	208	self.cfgdict[key] = value
	209	self.changed = True
	210
	211	def __getitem__(self, key):
[6362]	212	"""Override to allow: <var> = cfg[<key>]"""
	213
[6251]	214	return self.cfgdict.get(key, None)
	215
	216	def __str__(self):
[6362]	217	"""__str__(self) -> String"""
	218
[6251]	219	return "<config object at %s>" % hex(id(self))
	220
	221	def getfilename(self):
[6362]	222	"""getfilename(self) -> String filename"""
	223
[6251]	224	return self.configfile
	225
	226	def setdeleted(self):
[6362]	227	"""setdeleted(self)"""
	228
[6251]	229	self.delconf = True
	230
	231	def save(self):
[6362]	232	"""save(self) -> save data to a file"""
	233
[6251]	234	try:
	235	f = open(self.configfile, "w")
	236	p = pickle.Pickler(f)
	237	p.dump(self.cfgdict)
	238	except pickle.PicklingError, e:
	239	print e
	240	else:
	241	f.close()
	242	self.changed = False
	243
	244	def close(self):
[6362]	245	"""close(self) - close the save file"""
[6251]	246	if self.changed:
	247	self.save()
	248	if self.delconf:
	249	if os.path.exists(self.configfile):
	250	os.remove(self.configfile)
	251
	252	def __del__(self):
[6362]	253	"""__del__(self) - ensure save file is closed"""
	254
[6251]	255	self.close()
	256
	257	################################################################################
	258	# This code implements the logic of David Burbidge's 'list_quakes' program.
	259	# Except the two output files are now in CSV format.
	260	#################################################################################
	261
	262	##
	263	# @brief Function to do all the work - list the quakes selected.
	264	def list_quakes(event_num, min_height, max_height, InvallFilename,
	265	TStarFilename, FaultXYFilename, QuakeProbFilename):
	266	##
	267	# @brief Class to hold i_invall data
	268	class Inval(object):
	269	def __init__(self, lon, lat):
	270	self.lon = lon
	271	self.lat = lat
	272
	273	def __str__(self):
	274	return '.lon=%g, .lat=%g' % (self.lon, self.lat)
	275
	276
	277	##
	278	# @brief Class to hold T-**** data
	279	class TStar(object):
	280	def __init__(self, ipt, zquake, zprob, mag, slip, ng, ng_data):
	281	self.ipt = ipt
	282	self.zquake = zquake
	283	self.zprob = zprob
	284	self.mag = mag
	285	self.slip = slip
	286	self.ng_data = [int(i) for i in SpacesPattern.split(ng_data)]
	287	if len(self.ng_data) != ng:
	288	raise RuntimeError(1, "Error parsing T-***** data: %s" % msg)
	289
	290	def __str__(self):
	291	return ('.ipt=%d, .zquake=%g, .zprob=%g, .mag=%g, .slip=%g, .ng_data=%s' %
	292	(self.ipt, self.zquake, self.zprob, self.mag,
	293	self.slip, str(self.ng_data)))
	294
	295
	296	# read i_invall file
	297	try:
	298	fd = open(InvallFilename, "r")
	299	invall_lines = fd.readlines()
	300	fd.close()
	301	except IOError, msg:
	302	raise RuntimeError(1, "Error reading file '%s': %s" %
	303	(InvallFilename, msg))
	304
	305	# trash the first three lines
	306	invall_lines = invall_lines[3:]
	307
	308	# split i_invall data into fields
	309	invall_data = []
	310	for line in invall_lines:
	311	l = line.strip()
	312	(lon, lat, _) = SpacesPattern.split(l, maxsplit=2)
	313	invall_data.append(Inval(float(lon), float(lat)))
	314
	315	del invall_lines
	316
	317	# now read T-**** file, creating filename from event #
	318	try:
	319	fd = open(TStarFilename, "r")
	320	tstar_lines = fd.readlines()
	321	fd.close()
	322	except IOError, msg:
	323	raise RuntimeError(1, "Error reading file: %s" % msg)
	324
	325	# trash the first line of T-**** data
	326	tstar_lines = tstar_lines[1:]
	327
	328	# get the data from tstar_lines
	329	tstar_data = []
	330	min_wave = min_height - Epsilon
	331	max_wave = max_height + Epsilon
	332	for (ipt, line) in enumerate(tstar_lines):
	333	l = line.strip()
	334	(zquake, zprob, mag, slip, ng, ng_data) = SpacesPattern.split(l, maxsplit=5)
	335	zquake = float(zquake)
	336	zprob = float(zprob)
	337	mag = float(mag)
	338	slip = float(slip)
	339	ng = int(ng)
	340	ng_data = ng_data.strip()
	341	# only remember the data if zquake in wave height range and zprob > 0.0
	342	if zprob > 0.0 and (min_wave <= zquake <= max_wave):
	343	tstar_data.append(TStar(ipt+1, zquake, zprob, mag, slip, ng, ng_data))
	344	del tstar_lines
	345
	346	# write out lines joining centroids
	347	try:
	348	outfd = open(FaultXYFilename, "w")
	349	except IOError, msg:
	350	raise RuntimeError(1, "Error opening output file: %s" % msg)
	351
	352	outfd.write('Lon,Lat,Quake_ID,Subfault_ID\n')
	353	for t in tstar_data:
	354	for n in t.ng_data:
	355	outfd.write('%.4f,%.4f,%d,%d\n' %
	356	(invall_data[n].lon, invall_data[n].lat, t.ipt, n))
	357	outfd.close()
	358
	359	# write out quake probabilities
	360	try:
	361	outfd = open(QuakeProbFilename, "w")
	362	except IOError, msg:
	363	raise RuntimeError(1, "Error opening output file: %s" % msg)
	364
	365	outfd.write('Quake_ID,Ann_Prob,z_max(m),Mag\n')
	366	for t in tstar_data:
	367	outfd.write('%d,%.5G,%.5f,%.2f\n' % (t.ipt, t.zprob, t.zquake, t.mag))
	368	outfd.close()
	369
	370	################################################################################
	371	# This code does exactly what David Burbidge's 'get_multimux' program does.
	372	################################################################################
	373
	374	def get_multimux(quake_ID, multimux_dir, fault_event_filename):
	375	# get the fault name data
	376	filename = os.path.join(multimux_dir, FaultNameFilename)
	377	try:
	378	fd = open(filename, "r")
	379	fault_names = [ fn.strip() for fn in fd.readlines() ]
	380	fd.close()
	381	except IOError, msg:
	382	raise RuntimeError(1, "Error reading file: %s" % msg)
	383
	384	# open the output file
	385	try:
	386	outfd = open(fault_event_filename, "w")
	387	except IOError, msg:
	388	raise RuntimeError(1, "Error opening output file: %s" % msg)
	389
	390	# handle each fault
	391	nquake = 0
	392	for fn in fault_names:
	393	# create the filename for the multimux data file
	394	mmx_filename = 'i_multimux-%s' % fn
	395	mmx_filename = os.path.join(multimux_dir, mmx_filename)
	396
	397	# Read all data in file, checking as we go
	398	try:
	399	infd = open(mmx_filename, "r")
	400	except IOError, msg:
	401	raise RuntimeError(1, "Error opening file: %s" % msg)
	402
	403	# check fault name in file is as expected
	404	mux_faultname = infd.readline().strip()
	405	if mux_faultname != fn:
	406	raise RuntimeError(1, "Error reading file")
	407
	408	# read data
	409	while True:
	410	# get number of subfaults, EOF means finished
	411	try:
	412	nsubfault = infd.readline()
	413	except IOError:
	414	raise RuntimeError(1, "Error reading file")
	415
	416	if not nsubfault:
	417	break
	418	nsubfault = int(nsubfault)
	419
	420	nquake += 1
	421	if nquake == quake_ID:
	422	outfd.write(' %d\n' % nsubfault)
	423	for i in range(nsubfault):
	424	line = infd.readline()
	425	(subfaultname, slip, prob, mag, _) = \
	426	SpacesPattern.split(line, maxsplit=4)
	427	subfaultname = subfaultname.strip()
	428	slip = float(slip)
	429	outfd.write(" %s %g\n" % (subfaultname, slip))
	430	else:
	431	for i in range(nsubfault):
	432	try:
	433	infd.readline()
	434	except IOError:
	435	raise RuntimeError(1,
	436	"Something wrong at bottom of file %s" %
	437	mux_faultname)
	438
	439	infd.close()
	440	outfd.close()
	441
	442
	443	################################################################################
	444	# Code to do the GUI
	445	#
	446	# The Frame code is buried under the "if __name__ ..." test since we are going
	447	# to import this file into verify.py to test the list_quake() and get_multimux()
	448	# functions.
	449	################################################################################
	450
	451	if __name__ == '__main__':
	452
	453	class MyFrame(wx.Frame):
	454	def __init__(self, parent, id, title, pos=wx.DefaultPosition,
	455	size=wx.DefaultSize, style=wx.DEFAULT_FRAME_STYLE):
	456	# Make the frame
	457	wx.Frame.__init__(self, parent, id, title, pos=(50, 50),
	458	size=(FORM_WIDTH, FORM_HEIGHT),
	459	style=(wx.DEFAULT_FRAME_STYLE &
	460	~ (wx.RESIZE_BOX \| wx.MAXIMIZE_BOX \|
	461	wx.RESIZE_BORDER)))
	462
	463	p = self.panel = wx.Panel(self, -1)
	464
	465	if Imported_PyEmbeddedImage:
	466	tsunami = getIcon()
	467	icon = tsunami.GetIcon()
	468	self.SetIcon(icon)
	469
	470	self.Center()
	471	self.Show(True)
	472
	473	# place the 'output dir' textbox
	474	Y_OFFSET = START_YOFFSET
	475	wx.StaticBox(p, -1, 'output base directory', (3, 1),
	476	size=(BOX_WIDTH, OUTPUT_BOX_HEIGHT))
	477	self.txtOutputDir = wx.TextCtrl(p, -1, size=(BOX_WIDTH-10, 20),
	478	pos=(OUT_TEXT_XOFFSET,
	479	Y_OFFSET+OUT_TEXT_YOFFSET))
	480	self.txtOutputDir.SetToolTip(wx.ToolTip("Where your data files will be written"))
	481	self.txtOutputDir.Bind(wx.EVT_LEFT_DOWN, self.OnOutputDirSelect)
	482	Y_OFFSET += 50
	483
	484	# start laying out list_quake controls
	485	wx.StaticBox(p, -1, 'list_quakes', (3, Y_OFFSET),
	486	size=(BOX_WIDTH, BOX_HEIGHT))
	487	sampleList = [ x for (x, y) in Regions ]
	488	Y_OFFSET += GEN_YOFFSET
	489	wx.StaticText(p, -1, 'Region', pos=(GEN_LABELXOFFSET, Y_OFFSET),
	490	size=(CTL_WIDTH, CTL_HEIGHT), style=wx.ALIGN_RIGHT)
	491	self.cbRegion = wx.ComboBox(p, 500,
	492	pos=(GEN_TEXTXOFFSET, Y_OFFSET-COMBO_FUDGE),
	493	size=(100,20),
	494	choices=sampleList,
	495	style=wx.CB_DROPDOWN)
	496	self.cbRegion.SetEditable(False)
	497	Y_OFFSET += GEN_DELTAY
	498	wx.StaticText(p, -1, 'Hazard index',
	499	pos=(GEN_LABELXOFFSET, Y_OFFSET),
	500	size=(CTL_WIDTH, CTL_HEIGHT), style=wx.ALIGN_RIGHT)
	501	self.txtTGN = wx.TextCtrl(p, -1, size=(100, 20),
	502	pos=(GEN_TEXTXOFFSET, Y_OFFSET))
	503	Y_OFFSET += GEN_DELTAY
	504	wx.StaticText(p, -1, 'Minimum wave height',
	505	pos=(GEN_LABELXOFFSET, Y_OFFSET),
	506	size=(CTL_WIDTH, CTL_HEIGHT), style=wx.ALIGN_RIGHT)
	507	self.txtMinH = wx.TextCtrl(p, -1, size=(100, 20),
	508	pos=(GEN_TEXTXOFFSET, Y_OFFSET))
	509	Y_OFFSET += GEN_DELTAY
	510	wx.StaticText(p, -1, 'Maximum wave height',
	511	pos=(GEN_LABELXOFFSET, Y_OFFSET),
	512	size=(CTL_WIDTH, CTL_HEIGHT), style=wx.ALIGN_RIGHT)
	513	self.txtMaxH = wx.TextCtrl(p, -1, size=(100, 20),
	514	pos=(GEN_TEXTXOFFSET, Y_OFFSET))
	515	Y_OFFSET += GEN_DELTAY
	516	x = FORM_WIDTH/2 - BUTTON_WIDTH/2
	517	self.btnGenerate = wx.Button(p, label="List",
	518	pos=(x, Y_OFFSET),
	519	size=(100, BUTTON_HEIGHT))
	520	Y_OFFSET += GEN_DELTAY + BUTTON_HEIGHT/2
	521
	522	self.lstGenerate = wx.ListCtrl(p, -1, pos=(8, Y_OFFSET),
	523	size=(TEXTLIST_WIDTH,TEXTLIST_HEIGHT),
	524	style=wx.LC_REPORT\|wx.LC_SINGLE_SEL)
	525
	526	Y_OFFSET += TEXTLIST_HEIGHT + 20
	527	self.Bind(wx.EVT_BUTTON, self.GenerateClick, self.btnGenerate)
	528	self.lstGenerate.Bind(wx.EVT_LIST_ITEM_SELECTED,
	529	self.OnItemSelected, self.lstGenerate)
	530
	531	# start laying out get_multimux controls
	532	wx.StaticBox(p, -1, 'multimux_grid', (3, Y_OFFSET),
	533	size=(BOX_WIDTH, BOX2_HEIGHT))
	534	Y_OFFSET += 18
	535	wx.StaticText(p, -1, 'Quake ID', pos=(GEN_LABELXOFFSET, Y_OFFSET),
	536	size=(CTL_WIDTH, CTL_HEIGHT), style=wx.ALIGN_RIGHT)
	537	self.txtQuakeID = wx.TextCtrl(p, -1, size=(100, 20),
[7200]	538	pos=(GEN_TEXTXOFFSET, Y_OFFSET),
	539	style=wx.TE_READONLY)
[6251]	540	Y_OFFSET += GEN_DELTAY
	541	x = FORM_WIDTH/2 - BUTTON_WIDTH - DOUBLE_BUTTON_OFFSET
	542	self.btnMultimux = wx.Button(p, label="Multimux", pos=(x, Y_OFFSET),
	543	size=(100, BUTTON_HEIGHT))
	544	x = FORM_WIDTH/2 + DOUBLE_BUTTON_OFFSET
	545	self.btnGrid = wx.Button(p, label="Grid", pos=(x, Y_OFFSET),
	546	size=(100, BUTTON_HEIGHT))
	547	Y_OFFSET += GEN_DELTAY + BUTTON_HEIGHT/2
	548	self.lstMultimux = wx.ListCtrl(p, -1, pos=(8, Y_OFFSET),
	549	size=(TEXTLIST_WIDTH,TEXTLIST_HEIGHT),
	550	style=wx.LC_REPORT\|wx.LC_SINGLE_SEL)
	551	Y_OFFSET += TEXTLIST_HEIGHT + 20
	552	self.Bind(wx.EVT_BUTTON, self.MultimuxClick, self.btnMultimux)
	553	self.Bind(wx.EVT_BUTTON, self.GridClick, self.btnGrid)
	554
	555	self.Bind(wx.EVT_CLOSE, self.doStateSave)
	556	self.doStateRestore()
	557
	558	def OnOutputDirSelect(self, event):
	559	dlg = wx.DirDialog(self, "Choose an output directory:",
	560	style=wx.DD_DEFAULT_STYLE
	561	#\| wx.DD_DIR_MUST_EXIST
	562	#\| wx.DD_CHANGE_DIR
	563	)
	564
	565	# If the user selects OK, then we process the dialog's data.
	566	# This is done by getting the path data from the dialog - BEFORE
	567	# we destroy it.
	568	if dlg.ShowModal() == wx.ID_OK:
	569	self.txtOutputDir.Clear()
	570	self.txtOutputDir.WriteText(dlg.GetPath())
	571	self.cbRegion.SetFocus()
	572	# Only destroy a dialog after you're done with it.
	573	dlg.Destroy()
	574
	575
	576	def doStateSave(self, event):
	577	"""
	578	Save state here.
	579	"""
	580	self.cfg['OutputDirectory'] = self.txtOutputDir.GetValue()
	581	self.cfg['Region'] = self.cbRegion.GetValue()
	582	self.cfg['GaugeNumber'] = self.txtTGN.GetValue()
	583	self.cfg['MinimumHeight'] = self.txtMinH.GetValue()
	584	self.cfg['MaximumHeight'] = self.txtMaxH.GetValue()
	585	self.cfg.save()
	586
	587	event.Skip(True)
	588
	589	def doStateRestore(self):
	590	"""
	591	Restore state here - globals have been set and tested sane
	592	"""
	593	global Region, GaugeNumber, MinimumHeight, MaximumHeight
	594
	595	cfg = Config(ConfigFilename)
	596	output_dir = cfg['OutputDirectory']
	597	Region = cfg['Region']
	598	GaugeNumber = cfg['GaugeNumber']
	599	if GaugeNumber:
	600	GaugeNumber = int(GaugeNumber)
	601	MinimumHeight = cfg['MinimumHeight']
	602	if MinimumHeight:
	603	MinimumHeight = float(MinimumHeight)
	604	MaximumHeight = cfg['MaximumHeight']
	605	if MaximumHeight:
	606	MaximumHeight = float(MaximumHeight)
	607
	608	self.cfg = cfg
	609
	610	region_name = None
	611	for (name, dir) in Regions:
	612	if name == Region:
	613	region_name = name
	614	if not region_name:
	615	Region = ''
	616
	617	if output_dir:
	618	self.txtOutputDir.WriteText(output_dir)
	619	self.cbRegion.SetStringSelection(Region)
	620	if GaugeNumber:
	621	self.txtTGN.WriteText(str(GaugeNumber))
	622	if MinimumHeight:
	623	self.txtMinH.WriteText(str(MinimumHeight))
	624	if MaximumHeight:
	625	self.txtMaxH.WriteText(str(MaximumHeight))
	626
	627
	628	def OnItemSelected(self, event):
	629	item = self.lstGenerate.GetItem(event.m_itemIndex, 0)
	630	self.txtQuakeID.SetValue(item.GetText())
	631
	632	def error(self, msg):
	633	dlg = wx.MessageDialog(self, msg, 'Error',
	634	wx.OK \| wx.ICON_INFORMATION)
	635	dlg.ShowModal()
	636	dlg.Destroy()
	637
	638	######
	639	# Call the list_quakes function.
	640	######
	641	def GenerateClick(self, event):
	642	global ResultSubdir, Region, GaugeNumber, MinimumHeight, MaximumHeight
	643
	644	# check that everything has been entered correctly
	645	output_base_dir = self.txtOutputDir.GetValue()
	646	if len(output_base_dir) == 0:
	647	self.error('You must select an output base directory first')
	648	return
	649
	650	region_select = self.cbRegion.GetValue()
	651	if len(region_select) == 0:
	652	self.error('You must select a region first')
	653	return
	654
	655	try:
	656	gauge_number = int(self.txtTGN.GetValue())
	657	except:
	658	self.error("You must enter a Hazard Index Number")
	659	return
	660
	661	try:
	662	min_height = float(self.txtMinH .GetValue())
	663	except:
	664	self.error("You must enter a minimum wave height in metres (eg, 1.3)")
	665	return
	666
	667	try:
	668	max_height = float(self.txtMaxH.GetValue())
	669	except:
	670	self.error("You must enter a maximum wave height in metres (eg, 1.8)")
	671	return
	672	if min_height > max_height:
	673	self.error("Minimum wave height must be less than maximum height")
	674	return
	675
	676	# set directory names
	677	basename = None
	678	region_name = None
	679	for (name, dir) in Regions:
	680	if name == region_select:
	681	region_name = name
	682	basename = dir
	683	base_directory = os.path.join(BaseDirectory, basename)
	684	if not os.path.exists(base_directory):
	685	self.error("Sorry, region %s doesn't exist yet (%s)" %
	686	(region_name, base_directory))
	687	return
	688	tFilesDir = os.path.join(base_directory, 'Tfiles')
	689	multimuxDir = os.path.join(base_directory, 'multimux')
	690
	691	# create a sub-directory for the result files
	692	ResultSubdir = os.path.join(output_base_dir,
	693	'Results_%s_%d_%.2f_%.2f' %
	694	(region_select, gauge_number,
	695	min_height, max_height))
	696	try:
	697	shutil.rmtree(ResultSubdir)
	698	except:
	699	pass
	700	try:
	701	os.mkdir(ResultSubdir)
	702	except WindowsError, msg:
	703	self.error("Sorry, don't seem to be able to make "
	704	"the result directory '%s':\n%s" %
	705	(ResultSubdir, str(msg)))
	706	return
	707	except:
	708	self.error("Some sort of error making the result directory '%s'" %
	709	ResultSubdir)
	710	return
	711
	712	# set names of the output files (in dir '<wherever>/Results*')
	713	faultxy_filename = os.path.join(ResultSubdir, FaultXYFilename)
	714	quakeprob_filename = os.path.join(ResultSubdir, QuakeProbFilename)
	715
	716	# now check that gauge number is valid
	717	gaugeFile = os.path.join(tFilesDir, 'T-%05d' % gauge_number)
	718	if not os.path.exists(gaugeFile):
	719	self.error("T file for hazard %d doesn't exist" % gauge_number)
	720	return
	721
	722	# check we have an i_invall file
	723	invallFile = os.path.join(multimuxDir, 'i_invall')
	724	if not os.path.exists(invallFile):
	725	self.error("'i_invall' file doesn't exist: %s" % invallFile)
	726	return
	727
	728	# save region, hazard #, min and max heights to globals
	729	Region = region_select
	730	GaugeNumber = gauge_number
	731	MinimumHeight = min_height
	732	MaximumHeight = max_height
	733
	734	# clear the Quake ID textbox
	735	self.txtQuakeID.SetValue('')
	736
	737	# now process the data we have
	738	# clear output list
	739	self.lstGenerate.ClearAll()
	740	# delete output files
	741	try:
	742	os.unlink(FaultXYFilename)
	743	except:
	744	pass
	745	try:
	746	os.unlink(QuakeProbFilename)
	747	except:
	748	pass
	749
	750	####################################################################
	751	# reset button label, hourglass the cursor
	752	self.btnGenerate.SetLabel('Listing ...')
	753	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
	754
	755	wx.Yield() # allow GUI refresh
	756
	757	# generate output files
	758	try:
	759	list_quakes(gauge_number, min_height, max_height, invallFile,
	760	gaugeFile, faultxy_filename, quakeprob_filename)
	761	except RuntimeError, msg:
	762	self.error("Error in list_quakes module: %s" % msg)
	763	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	764	self.btnGenerate.SetLabel('List')
	765	return
	766	# put cursor back to normal, button label back to normal
	767	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	768	self.btnGenerate.SetLabel('List')
	769	####################################################################
	770
	771	# list result file, if any
	772	if not os.path.exists(quakeprob_filename):
	773	self.error("Error generating data: file '%s' not found" %
	774	quakeprob_filename)
	775	return
	776	in_fd = open(quakeprob_filename, 'r')
	777	lines = in_fd.readlines()
	778	in_fd.close()
	779	hdr = lines[0].split(',')
	780	for (i, h) in enumerate(hdr):
	781	self.lstGenerate.InsertColumn(i, h.strip())
	782	for (i, l) in enumerate(lines[1:]):
	783	l = l.split(',')
	784	index = self.lstGenerate.InsertStringItem(sys.maxint, l[0].strip())
	785	for (ii, ll) in enumerate(l[1:]):
	786	self.lstGenerate.SetStringItem(index, ii+1, ll.strip())
	787
	788	column_width = FORM_WIDTH / len(hdr) - 20
	789	for i in range(len(hdr)):
	790	self.lstGenerate.SetColumnWidth(i, column_width)
	791
	792	######
	793	# Call the get_multimux function.
	794	######
	795	def MultimuxClick(self, event):
[6362]	796	# check we have an output base directory
	797	output_base_dir = self.txtOutputDir.GetValue()
	798	if len(output_base_dir) == 0:
	799	self.error('You must select an output base directory first')
	800	return
	801
[6251]	802	# check we have a quake ID
	803	quake_ID = self.txtQuakeID.GetValue()
	804	if len(quake_ID) == 0:
[7214]	805	self.error("You must enter a quake ID by clicking in "
	806	"the 'list' result box")
[6251]	807	return
[7214]	808	quake_ID = int(quake_ID)
[6251]	809
	810	region_select = self.cbRegion.GetValue()
	811	if len(region_select) == 0:
	812	self.error('You must select a region first')
	813	return
	814
	815	region_dir_name = None
	816	for (disp, real) in Regions:
	817	if disp == region_select:
	818	region_dir_name = real
	819	break
	820	if region_dir_name is None:
	821	self.error('INTERNAL ERROR: region_dir_name is None?!')
	822	return
	823
	824	multimux_dir = os.path.join(BaseDirectory, region_dir_name, 'multimux')
	825
	826	####################################################################
	827	# clear list, set button label, hourglass the cursor
	828	self.lstMultimux.ClearAll()
	829	self.lstMultimux.InsertColumn(0, '')
	830	self.btnMultimux.SetLabel('Multimuxing ...')
	831	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
	832
[6364]	833	fault_event_dir = os.path.join(output_base_dir, str(quake_ID))
[6366]	834	if not os.path.exists(fault_event_dir):
	835	os.mkdir(fault_event_dir)
[6362]	836	fault_event_filename = os.path.join(fault_event_dir, 'event.list')
	837
[6251]	838	wx.Yield() # allow GUI refresh
	839
	840	try:
	841	get_multimux(quake_ID, multimux_dir, fault_event_filename)
	842	except RuntimeError, msg:
	843	self.error("Error in get_multimux module: %s" % msg)
	844	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	845	self.btnMultimux.SetLabel('Multimux')
	846	return
	847	# cursor and button label to normal
	848	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	849	self.btnMultimux.SetLabel('Multimux')
	850	####################################################################
	851
	852	# list resultant files, if any
	853	if not os.path.exists(fault_event_filename):
	854	self.error("Error generating data: file %s not found" %
	855	fault_event_filename)
	856	return
	857
	858	in_fd = open(fault_event_filename, 'r')
	859	lines = in_fd.readlines()
	860	in_fd.close()
	861	hdr = lines[0].strip()
	862	hdr = SpacesPattern.split(hdr)
	863	for (i, h) in enumerate(hdr):
	864	self.lstMultimux.InsertColumn(i, h)
	865	for (i, l) in enumerate(lines[1:]):
	866	l = l.strip()
	867	l = SpacesPattern.split(l)
	868	index = self.lstMultimux.InsertStringItem(sys.maxint, l[0])
	869	for (ii, ll) in enumerate(l[1:]):
	870	self.lstMultimux.SetStringItem(index, ii+1, ll)
	871	column_width = FORM_WIDTH / 2 - 30
	872	for i in range(2):
	873	self.lstMultimux.SetColumnWidth(i, column_width)
	874
	875	# generate the urs_filenames.py module from 'lines' data
	876	# get data from lines we read previously
	877	last_weight = None
	878	filenames = []
	879	for line in lines[1:]:
	880	(_, filename, weight) = SpacesPattern.split(line)
	881	weight = float(weight)
	882	if last_weight:
	883	if last_weight != weight:
	884	self.error("In file '%s', inconsistent weight, "
	885	"was %.4f, expected %.4f" % (weight, last_weight))
	886	return
	887	last_weight = weight
	888	filenames.append(filename)
[6362]	889
[6251]	890	######
	891	# Call the get_grid function.
	892	######
	893	def GridClick(self, event):
	894	# check we have a quake ID
	895	quake_ID = self.txtQuakeID.GetValue()
	896	if len(quake_ID) == 0:
[7214]	897	self.error("You must enter a quake ID by clicking in "
	898	"the 'list' result box")
[6251]	899	return
[7214]	900	quake_ID = int(quake_ID)
[6251]	901
	902	region_select = self.cbRegion.GetValue()
	903	if len(region_select) == 0:
	904	self.error('You must select a region first')
	905	return
	906
	907	region_dir_name = None
	908	for (disp, real) in Regions:
	909	if disp == region_select:
	910	region_dir_name = real
	911	break
	912	if region_dir_name is None:
	913	self.error('INTERNAL ERROR: region_dir_name is None?!')
	914	return
	915
	916	multimux_dir = os.path.join(BaseDirectory, region_dir_name, 'multimux')
	917	grid_dir = os.path.join(BaseDirectory, region_dir_name, 'static_2m')
[7214]	918
[7200]	919	# check that the sub-directory has been created
	920	if ResultSubdir == '' or not os.path.isdir(ResultSubdir):
	921	self.error("Sorry, you must 'list' earthquakes first.")
	922	return
[6251]	923
	924	####################################################################
	925	# clear list, set button label, hourglass the cursor
	926	self.lstMultimux.ClearAll()
	927	self.lstMultimux.InsertColumn(0, '')
	928	self.btnGrid.SetLabel('Gridding ...')
	929	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_WAIT))
	930
	931	fault_event_filename = ('%s/event_%6.6d.list' %
	932	(ResultSubdir, quake_ID))
	933	wx.Yield() # allow GUI refresh
	934
	935	try:
	936	get_multimux(quake_ID, multimux_dir, fault_event_filename)
	937	except RuntimeError, msg:
	938	self.error("Error in get_multimux module: %s" % msg)
	939	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	940	self.btnGrid.SetLabel('Grid')
	941	return
	942	# cursor and button label to normal
	943	self.panel.SetCursor(wx.StockCursor(wx.CURSOR_ARROW))
	944	self.btnGrid.SetLabel('Grid')
	945	####################################################################
	946
	947	# list resultant files, if any
	948	if not os.path.exists(fault_event_filename):
	949	self.error("Error generating data: file %s not found" %
	950	fault_event_filename)
	951	return
	952
	953	# read output data file
	954	in_fd = open(fault_event_filename, 'r')
	955	lines = in_fd.readlines()
	956	in_fd.close()
	957	# get header items, display them
	958	hdr = lines[0].strip()
	959	hdr = SpacesPattern.split(hdr)
	960	for (i, h) in enumerate(hdr):
	961	self.lstMultimux.InsertColumn(i, h)
	962	# display resultant data, collect filenames and weight_factor
	963	weight_factor = None
	964	wf_error = False # True if there was a weight_factor error
	965	filenames = []
	966	for (i, l) in enumerate(lines[1:]):
	967	l = l.strip()
	968	l = SpacesPattern.split(l)
	969	fname = l[0].replace('.grd-z-mux2', '.grd')
	970	filenames.append(fname)
	971	index = self.lstMultimux.InsertStringItem(sys.maxint, fname)
	972	for (ii, ll) in enumerate(l[1:]):
	973	self.lstMultimux.SetStringItem(index, ii+1, ll)
	974	if weight_factor:
	975	if weight_factor != float(l[1]):
	976	wf_error = True
	977	else:
	978	weight_factor = float(l[1])
	979	column_width = FORM_WIDTH / 2 - 30
	980	for i in range(2):
	981	self.lstMultimux.SetColumnWidth(i, column_width)
	982
	983	if wf_error:
	984	self.error('Weight factor is not the same for all files!')
	985	return
	986
	987	if not filenames:
	988	self.error('No grid files found!')
	989	return
	990
	991	# generate the bash script param_file.
	992	out_name = GridBashScriptName % quake_ID
	993	grid_filename = '%s/%s' % (ResultSubdir, out_name)
	994
	995	try:
	996	out_fd = open(grid_filename, "w")
	997	except IOError, msg:
	998	self.error("Error opening output file %s': %s" %
	999	(grid_filename, msg))
	1000	return
	1001
	1002	out_fd.write('GENERATED=%s\n' % time.asctime(time.localtime()))
	1003	out_fd.write('REGION=%s\n' % Region)
	1004	out_fd.write('HAZARD_INDEX=%d\n' % GaugeNumber)
	1005	out_fd.write('WAVE_HEIGHTS=[%.2f,%.2f]\n' % (MinimumHeight, MaximumHeight))
	1006	out_fd.write('QUAKE_ID=%d\n' % quake_ID)
	1007	out_fd.write('WEIGHT_FACTOR=%f\n' % weight_factor)
	1008	out_fd.write('DATA_PATH=%s\n' % grid_dir)
[7200]	1009	#for fn in filenames[1:]:
	1010	for fn in filenames:
[6251]	1011	owning_dir = filenames[0].split('-')[0]
	1012	out_fd.write('GRID=%s\n' % (os.path.join(owning_dir, fn)))
	1013	out_fd.write('FAULT=%s\n' % owning_dir)
	1014
	1015	app = wx.App()
	1016	frame = MyFrame(None, -1, '%s %s' % (APP_NAME, APP_VERSION),
	1017	size=(FORM_WIDTH, FORM_HEIGHT))
	1018	app.SetTopWindow(frame)
	1019	app.MainLoop()
[6330]	1020

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: branches/numpy_misc/tools/event_selection/EventSelection.py

Download in other formats: