if len(sys.argv) == 1:
raise IOError("Needs one argument, the template to check")
templatename = str(sys.argv[1])
from par import match_filter_par as defaults
from par import template_gen_par as tempdef
from par import bright_lights_par as brightdef
from obspy import read
from core.bright_lights import _read_tt, _resample_grid
template = read(templatename)
stations_in = []
for tr in template:
stations_in += [tr.stats.station]
stations_in = list(set(stations_in))
stations, nodes, lags = _read_tt(
brightdef.nllpath, stations_in, brightdef.phase, phaseout="S", ps_ratio=brightdef.ps_ratio
)
stations, nodes, lags = _resample_grid(stations, nodes, lags, brightdef.volume, brightdef.resolution)
# print np.shape(lags)
for station in stations:
if not "template_dummy" in locals():
template_dummy = template.select(station=station)
else:
template_dummy += template.select(station=station)
template = template_dummy
for tr in template:
for i in xrange(len(stations)):
if tr.stats.station == stations[i]:
if not "alllags" in locals():
alllags = [lags[i]]
else:
stations = []
print 'Template generation parameters are:'
print ' sfilebase: ' + templatedef.sfilebase
print ' samp_rate: ' + str(templatedef.samp_rate) + ' Hz'
print ' lowcut: ' + str(templatedef.lowcut) + ' Hz'
print ' highcut: ' + str(templatedef.highcut) + ' Hz'
print ' length: ' + str(templatedef.length) + ' s'
print ' swin: ' + templatedef.swin + '\n'
if not oldnodes:
# Use the brightness function to search for possible templates
# First read in the travel times
print 'Reading in the original grids'
stations, allnodes, alllags = \
bright_lights._read_tt(brightdef.nllpath,brightdef.stations,\
brightdef.phase, phaseout='S', \
ps_ratio=brightdef.ps_ratio)
print 'I have read in ' + str(len(allnodes)) + ' nodes'
# Resample the grid to allow us to run it quickly!
print 'Cutting the grid'
stations, nodes, lags = bright_lights._resample_grid(
stations, allnodes, alllags, brightdef.mindepth, brightdef.maxdepth,
brightdef.corners, brightdef.resolution)
del allnodes, alllags
# Check that we still have a grid!
if len(nodes) == 0:
raise IOError("You have no nodes left")
# Remove lags that have a similar network moveout, e.g. the sum of the
# differences in moveouts is small.
print "Removing simlar lags"
stations, nodes, lags = bright_lights._rm_similarlags(
from obspy import read
stream = read('test_data/*'+date.split('/')[0]+'-'+\
date.split('/')[1]+'-'+date.split('/')[2]+'-processed.ms')
if not len(stream) == 0:
import os
path = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0,path[0:len(path)-5])
from par import template_gen_par as templatedef
from par import bright_lights_par as brightdef
from core import bright_lights
from utils import EQcorrscan_plotting as plotting
# Use the brightness function to search for possible templates
# First read in the travel times
print 'Reading in the original grids'
stations, allnodes, alllags = \
bright_lights._read_tt(brightdef.nllpath,brightdef.stations,\
brightdef.phase)
# Resample the grid to allow us to run it quickly!
print 'Cutting the grid'
stations, nodes, lags = bright_lights._resample_grid(stations, allnodes,
alllags,
brightdef.volume,
brightdef.resolution)
# Remove lags that have a similar network moveout, e.g. the sum of the
# differences in moveouts is small.
print "Removing simlar lags"
stations, nodes, lags = bright_lights._rm_similarlags(stations, nodes, lags,
brightdef.nodesimthresh)
print "Plotting new grid"
plotting.threeD_gridplot(nodes)
dailycoherence = coherence_test(stream, stations, nodes, lags, \
templatedef.length)
stations=[]
print 'Template generation parameters are:'
print ' sfilebase: '+templatedef.sfilebase
print ' samp_rate: '+str(templatedef.samp_rate)+' Hz'
print ' lowcut: '+str(templatedef.lowcut)+' Hz'
print ' highcut: '+str(templatedef.highcut)+' Hz'
print ' length: '+str(templatedef.length)+' s'
print ' swin: '+templatedef.swin+'\n'
if not oldnodes:
# Use the brightness function to search for possible templates
# First read in the travel times
print 'Reading in the original grids'
stations, allnodes, alllags = \
bright_lights._read_tt(brightdef.nllpath,brightdef.stations,\
brightdef.phase, phaseout='S', \
ps_ratio=brightdef.ps_ratio)
print 'I have read in '+str(len(allnodes))+' nodes'
# Resample the grid to allow us to run it quickly!
print 'Cutting the grid'
stations, nodes, lags = bright_lights._resample_grid(stations, allnodes,
alllags,
brightdef.mindepth,
brightdef.maxdepth,
brightdef.corners,
brightdef.resolution)
del allnodes, alllags
# Check that we still have a grid!
if len(nodes) == 0:
raise IOError("You have no nodes left")
# Remove lags that have a similar network moveout, e.g. the sum of the
import sys
if len(sys.argv) == 1:
raise IOError("Needs one argument, the template to check")
templatename = str(sys.argv[1])
from par import match_filter_par as defaults
from par import template_gen_par as tempdef
from par import bright_lights_par as brightdef
from obspy import read
from core.bright_lights import _read_tt, _resample_grid
template = read(templatename)
stations_in = []
for tr in template:
stations_in += [tr.stats.station]
stations_in = list(set(stations_in))
stations, nodes, lags = _read_tt(brightdef.nllpath,stations_in,\
brightdef.phase, phaseout='S', \
ps_ratio=brightdef.ps_ratio)
stations, nodes, lags = _resample_grid(stations, nodes, lags, brightdef.volume,\
brightdef.resolution)
# print np.shape(lags)
for station in stations:
if not 'template_dummy' in locals():
template_dummy = template.select(station=station)
else:
template_dummy += template.select(station=station)
template = template_dummy
for tr in template:
for i in xrange(len(stations)):
if tr.stats.station == stations[i]:
if not 'alllags' in locals():
alllags = [lags[i]]
