def outputOBSPY(hp, event=None, only_fm_picks=False):
"""
Make an Event which includes the current focal mechanism information from HASH
Use the 'only_fm_picks' flag to only include the picks HASH used for the FocalMechanism.
This flag will replace the 'picks' and 'arrivals' lists of existing events with new ones.
Inputs
-------
hp : hashpy.HashPype instance
event : obspy.core.event.Event
only_fm_picks : bool of whether to overwrite the picks/arrivals lists
Returns
-------
obspy.core.event.Event
Event will be new if no event was input, FocalMech added to existing event
"""
# Returns new (or updates existing) Event with HASH solution
n = hp.npol
if event is None:
event = Event(focal_mechanisms=[], picks=[], origins=[])
origin = Origin(arrivals=[])
origin.time = UTCDateTime(hp.tstamp)
origin.latitude = hp.qlat
origin.longitude = hp.qlon
origin.depth = hp.qdep
origin.creation_info = CreationInfo(version=hp.icusp)
origin.resource_id = ResourceIdentifier('smi:hash/Origin/{0}'.format(
hp.icusp))
for _i in range(n):
p = Pick()
p.creation_info = CreationInfo(version=hp.arid[_i])
p.resource_id = ResourceIdentifier('smi:nsl/Pick/{0}'.format(
p.creation_info.version))
p.waveform_id = WaveformStreamID(network_code=hp.snet[_i],
station_code=hp.sname[_i],
channel_code=hp.scomp[_i])
if hp.p_pol[_i] > 0:
p.polarity = 'positive'
else:
p.polarity = 'negative'
a = Arrival()
a.creation_info = CreationInfo(version=hp.arid[_i])
a.resource_id = ResourceIdentifier('smi:nsl/Arrival/{0}'.format(
p.creation_info.version))
a.azimuth = hp.p_azi_mc[_i, 0]
a.takeoff_angle = 180. - hp.p_the_mc[_i, 0]
a.pick_id = p.resource_id
origin.arrivals.append(a)
event.picks.append(p)
event.origins.append(origin)
event.preferred_origin_id = origin.resource_id.resource_id
else: # just update the changes
origin = event.preferred_origin()
picks = []
arrivals = []
for _i in range(n):
ind = hp.p_index[_i]
a = origin.arrivals[ind]
p = a.pick_id.getReferredObject()
a.takeoff_angle = hp.p_the_mc[_i, 0]
picks.append(p)
arrivals.append(a)
if only_fm_picks:
origin.arrivals = arrivals
event.picks = picks
# Use me double couple calculator and populate planes/axes etc
x = hp._best_quality_index
# Put all the mechanisms into the 'focal_mechanisms' list, mark "best" as preferred
for s in range(hp.nmult):
dc = DoubleCouple([hp.str_avg[s], hp.dip_avg[s], hp.rak_avg[s]])
ax = dc.axis
focal_mech = FocalMechanism()
focal_mech.creation_info = CreationInfo(creation_time=UTCDateTime(),
author=hp.author)
focal_mech.triggering_origin_id = origin.resource_id
focal_mech.resource_id = ResourceIdentifier(
'smi:hash/FocalMechanism/{0}/{1}'.format(hp.icusp, s + 1))
focal_mech.method_id = ResourceIdentifier('HASH')
focal_mech.nodal_planes = NodalPlanes()
focal_mech.nodal_planes.nodal_plane_1 = NodalPlane(*dc.plane1)
focal_mech.nodal_planes.nodal_plane_2 = NodalPlane(*dc.plane2)
focal_mech.principal_axes = PrincipalAxes()
focal_mech.principal_axes.t_axis = Axis(azimuth=ax['T']['azimuth'],
plunge=ax['T']['dip'])
focal_mech.principal_axes.p_axis = Axis(azimuth=ax['P']['azimuth'],
plunge=ax['P']['dip'])
focal_mech.station_polarity_count = n
focal_mech.azimuthal_gap = hp.magap
focal_mech.misfit = hp.mfrac[s]
focal_mech.station_distribution_ratio = hp.stdr[s]
focal_mech.comments.append(
Comment(
hp.qual[s],
resource_id=ResourceIdentifier(
focal_mech.resource_id.resource_id + '/comment/quality')))
#----------------------------------------
event.focal_mechanisms.append(focal_mech)
if s == x:
event.preferred_focal_mechanism_id = focal_mech.resource_id.resource_id
return event
def output(hp, dbout=None, solution=0, schema="css3.0"):
'''Write the preferred HASH solution to Datascope database.
This writes the strike, dip, rakes to 'fplane', arids used for a
given mech in 'predmech' and the takeoffs in 'predarr'.
Input
-----
dbout : str or antelope.datascope.Dbptr to database
solution : <STUB> int of desired solution.
'''
from hashpy.doublecouple import DoubleCouple
x = solution
dc = DoubleCouple([hp.str_avg[x], hp.dip_avg[x], hp.rak_avg[x]])
str1, dip1, rak1 = dc.plane1
str2, dip2, rak2 = dc.plane2
axes = dc.axis
if dbout is not None:
db = Dbptr(dbout, perm='r+')
else:
db = dbtmp(schema)
mechid = db.nextid('mechid')
dbfpln = dblookup(db,table='fplane')
dbfpln.record = dbfpln.addnull()
dbfpln.putv(
'orid' , hp.icusp,
'str1' , round(str1,1) ,
'dip1' , round(dip1,1) ,
'rake1', round(rak1,1),
'algorithm', "HASH",
'mechid', mechid,
'auth', 'hashpy:'+ hp.author,
'str2' , round(str2,1) ,
'dip2' , round(dip2,1) ,
'rake2', round(rak2,1),
'taxazm', round(axes['T']['azimuth'],1),
'taxplg', round(axes['T']['dip'],1),
'paxazm', round(axes['P']['azimuth'],1),
'paxplg', round(axes['P']['dip'],1),
)
dbpmec = dblookup(db,table='predmech')
dbparr = dblookup(db,table='predarr')
for k in range(hp.npol):
if hp.p_pol[k] > 0:
fm = 'U'
else:
fm = 'D'
dbpmec.record = dbpmec.addnull()
dbpmec.putv('arid', int(hp.arid[k]) ,
'orid', hp.icusp,
'mechid', mechid,
'fm', fm,
)
dbparr.record = dbparr.addnull()
dbparr.putv('arid', int(hp.arid[k]),
'orid', hp.icusp,
'esaz', hp.qazi[k],
'dip' , hp.p_the_mc[k,0],
)
return db