def main(sta1, sta2, filterid, components, mov_stack=1, ampli=5, seismic=False,
show=False, outfile=None, envelope=False, refilter=None):
db = connect()
maxlag = float(get_config(db, 'maxlag'))
samples = get_maxlag_samples(db)
cc_sampling_rate = float(get_config(db, 'cc_sampling_rate'))
start, end, datelist = build_movstack_datelist(db)
base = mdates.date2num(start)
plt.figure(figsize=(12, 9))
sta1 = sta1.replace('.', '_')
sta2 = sta2.replace('.', '_')
t = np.arange(samples)/cc_sampling_rate - maxlag
if refilter:
freqmin, freqmax = refilter.split(':')
freqmin = float(freqmin)
freqmax = float(freqmax)
if sta2 >= sta1:
pair = "%s:%s" % (sta1, sta2)
print("New Data for %s-%s-%i-%i" % (pair, components, filterid,
mov_stack))
nstack, stack_total = get_results(db, sta1, sta2, filterid, components,
datelist, mov_stack, format="matrix")
ax = plt.subplot(111)
for i, line in enumerate(stack_total):
if np.all(np.isnan(line)):
continue
if refilter:
line = bandpass(line, freqmin, freqmax, cc_sampling_rate,
zerophase=True)
if envelope:
line = obspy_envelope(line)
line /= line.max()
plt.plot(t, line * ampli + i + base, c='k')
if seismic:
y1 = np.ones(len(line)) * i
y2 = line*ampli + i + base
plt.fill_between(t, y1, y2, where=y2 >= y1, facecolor='k',
interpolate=True)
for filterdb in get_filters(db, all=True):
if filterid == filterdb.ref:
low = float(filterdb.low)
high = float(filterdb.high)
break
plt.xlabel("Lag Time (s)")
plt.axhline(0, lw=0.5, c='k')
plt.grid()
title = '%s : %s, %s, Filter %d (%.2f - %.2f Hz), Stack %d' %\
(sta1.replace('_', '.'), sta2.replace('_', '.'), components,
filterid, low, high, mov_stack)
if refilter:
title += ", Re-filtered (%.2f - %.2f Hz)" % (freqmin, freqmax)
plt.title(title)
plt.scatter(0, [start, ], alpha=0)
plt.ylim(start-datetime.timedelta(days=ampli),
end+datetime.timedelta(days=ampli))
plt.xlim(-maxlag, maxlag)
ax.fmt_ydata = mdates.DateFormatter('%Y-%m-%d')
cursor = Cursor(ax, useblit=True, color='red', linewidth=1.2)
if outfile:
if outfile.startswith("?"):
pair = pair.replace(':', '-')
outfile = outfile.replace('?', '%s-%s-f%i-m%i' % (pair,
components,
filterid,
mov_stack))
outfile = "ccftime " + outfile
print("output to:", outfile)
plt.savefig(outfile)
if show:
plt.show()
def main(sta1,
sta2,
filterid,
components,
mov_stack=1,
ampli=5,
seismic=False,
show=False,
outfile=None,
envelope=False,
refilter=None):
db = connect()
maxlag = float(get_config(db, 'maxlag'))
samples = get_maxlag_samples(db)
cc_sampling_rate = float(get_config(db, 'cc_sampling_rate'))
start, end, datelist = build_movstack_datelist(db)
base = mdates.date2num(start)
plt.figure(figsize=(12, 9))
sta1 = sta1.replace('.', '_')
sta2 = sta2.replace('.', '_')
t = np.arange(samples) / cc_sampling_rate - maxlag
if refilter:
freqmin, freqmax = refilter.split(':')
freqmin = float(freqmin)
freqmax = float(freqmax)
if sta2 >= sta1:
pair = "%s:%s" % (sta1, sta2)
print("New Data for %s-%s-%i-%i" %
(pair, components, filterid, mov_stack))
nstack, stack_total = get_results(db,
sta1,
sta2,
filterid,
components,
datelist,
mov_stack,
format="matrix")
ax = plt.subplot(111)
for i, line in enumerate(stack_total):
if np.all(np.isnan(line)):
continue
if refilter:
line = bandpass(line,
freqmin,
freqmax,
cc_sampling_rate,
zerophase=True)
if envelope:
line = obspy_envelope(line)
line /= line.max()
plt.plot(t, line * ampli + i + base, c='k')
if seismic:
y1 = np.ones(len(line)) * i
y2 = line * ampli + i + base
plt.fill_between(t,
y1,
y2,
where=y2 >= y1,
facecolor='k',
interpolate=True)
for filterdb in get_filters(db, all=True):
if filterid == filterdb.ref:
low = float(filterdb.low)
high = float(filterdb.high)
break
plt.xlabel("Lag Time (s)")
plt.axhline(0, lw=0.5, c='k')
plt.grid()
title = '%s : %s, %s, Filter %d (%.2f - %.2f Hz), Stack %d' %\
(sta1.replace('_', '.'), sta2.replace('_', '.'), components,
filterid, low, high, mov_stack)
if refilter:
title += ", Re-filtered (%.2f - %.2f Hz)" % (freqmin, freqmax)
plt.title(title)
plt.scatter(0, [
start,
], alpha=0)
plt.ylim(start - datetime.timedelta(days=ampli),
end + datetime.timedelta(days=ampli))
plt.xlim(-maxlag, maxlag)
ax.fmt_ydata = mdates.DateFormatter('%Y-%m-%d')
cursor = Cursor(ax, useblit=True, color='red', linewidth=1.2)
if outfile:
if outfile.startswith("?"):
pair = pair.replace(':', '-')
outfile = outfile.replace(
'?',
'%s-%s-f%i-m%i' % (pair, components, filterid, mov_stack))
outfile = "ccftime " + outfile
print("output to:", outfile)
plt.savefig(outfile)
if show:
plt.show()
def main(sta1,
sta2,
filterid,
components,
mov_stack=1,
ampli=5,
seismic=False,
show=False,
outfile=None,
envelope=False,
refilter=None,
normalize=None,
**kwargs):
db = connect()
maxlag = float(get_config(db, 'maxlag'))
samples = get_maxlag_samples(db)
cc_sampling_rate = float(get_config(db, 'cc_sampling_rate'))
start, end, datelist = build_movstack_datelist(db)
base = mdates.date2num(start)
plt.figure(figsize=(12, 9))
sta1 = sta1 #.replace('.', '_')
sta2 = sta2 #.replace('.', '_')
t = np.arange(samples) / cc_sampling_rate - maxlag
if refilter:
freqmin, freqmax = refilter.split(':')
freqmin = float(freqmin)
freqmax = float(freqmax)
if sta2 < sta1:
print("Stations STA1 STA2 should be sorted alphabetically")
return
sta1 = check_stations_uniqueness(db, sta1)
sta2 = check_stations_uniqueness(db, sta2)
pair = "%s:%s" % (sta1, sta2)
print("Fetching CCF data for %s-%s-%i-%i" %
(pair, components, filterid, mov_stack))
nstack, stack_total = get_results(db,
sta1,
sta2,
filterid,
components,
datelist,
mov_stack,
format="matrix")
if nstack == 0:
print("No CCF found for this request")
return
if normalize == "common":
stack_total /= np.nanmax(stack_total)
ax = plt.subplot(111)
for i, line in enumerate(stack_total):
if np.all(np.isnan(line)):
continue
if refilter:
line = bandpass(line,
freqmin,
freqmax,
cc_sampling_rate,
zerophase=True)
if envelope:
line = obspy_envelope(line)
if normalize == "individual":
line /= line.max()
plt.plot(t, line * ampli + i + base, c='k', lw=0.5)
if seismic:
y1 = np.ones(len(line)) * i + base
y2 = line * ampli + i + base
plt.fill_between(t,
y1,
y2,
where=y2 >= y1,
facecolor='k',
interpolate=True)
low = high = 0.0
for filterdb in get_filters(db, all=True):
if filterid == filterdb.ref:
low = float(filterdb.low)
high = float(filterdb.high)
break
plt.xlabel("Lag Time (s)")
plt.axhline(0, lw=0.5, c='k')
plt.grid()
title = '%s : %s, %s, Filter %d (%.2f - %.2f Hz), Stack %d' %\
(sta1, sta2, components,
filterid, low, high, mov_stack)
if refilter:
title += ", Re-filtered (%.2f - %.2f Hz)" % (freqmin, freqmax)
plt.title(title)
plt.scatter(0, [
start,
], alpha=0)
plt.xlabel("Time Lag (s)")
plt.ylim(start - datetime.timedelta(days=10),
end + datetime.timedelta(days=10))
if "xlim" in kwargs:
plt.xlim(kwargs["xlim"][0], kwargs["xlim"][1])
else:
plt.xlim(-maxlag, maxlag)
ax.fmt_ydata = mdates.DateFormatter('%Y-%m-%d')
cursor = Cursor(ax, useblit=True, color='red', linewidth=1.2)
plt.tight_layout()
if outfile:
if outfile.startswith("?"):
pair = pair.replace(':', '-')
outfile = outfile.replace(
'?', '%s-%s-f%i-m%i' % (pair, components, filterid, mov_stack))
outfile = "ccftime " + outfile
print("output to:", outfile)
plt.savefig(outfile)
if show:
plt.show()