Updating plotting app.
This commit is contained in:
parent
52b55705b8
commit
b074f0e695
Notes:
subgit
2018-03-07 17:59:10 +01:00
r824 www/trunk
1
.gitignore
vendored
1
.gitignore
vendored
@ -1,3 +1,4 @@
|
||||
wsgi/.idea
|
||||
www/dlsv
|
||||
www/event.xml
|
||||
www/events.xml
|
||||
|
||||
147
wsgi/plotFDSN.py
147
wsgi/plotFDSN.py
@ -2,7 +2,7 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Get waveform data from FDSN web service and create a fancy plot
|
||||
This programm runs as a script or as a WSGI application.
|
||||
This programme runs as a script or as a WSGI application.
|
||||
|
||||
Subversion information:
|
||||
$Id$
|
||||
@ -41,6 +41,7 @@ def utc2local(utctime, timezone='Europe/Berlin'):
|
||||
utctime = utc.localize(utctime.datetime)
|
||||
return local.normalize(utctime.astimezone(local))
|
||||
|
||||
|
||||
def fancy_plot(st, wsgi=False, img_format='png', color=True):
|
||||
"""
|
||||
creating fancy plot from ObsPy stream st
|
||||
@ -131,6 +132,61 @@ def fancy_plot(st, wsgi=False, img_format='png', color=True):
|
||||
plt.show()
|
||||
|
||||
|
||||
def trace_dayplot(st, deltat = None,
|
||||
ftype='bandpass', fmin=1.0, fmax=7.0,
|
||||
col=('b', 'r', 'g'), interval=20, outpattern='',
|
||||
wsgi=False):
|
||||
"""
|
||||
|
||||
:type st: object
|
||||
:type ftype: str
|
||||
:type fmin: float
|
||||
:type fmax: float
|
||||
:type col: tuple
|
||||
:type interval: float
|
||||
:type outpattern: str
|
||||
:type wsgi: bool
|
||||
"""
|
||||
|
||||
if wsgi:
|
||||
try:
|
||||
import cStringIO as StringIO
|
||||
except ImportError:
|
||||
import StringIO as StringIO
|
||||
|
||||
# filter
|
||||
if (ftype == 'bandpass') or (ftype == 'bandstop'):
|
||||
st.filter(ftype, freqmin=fmin, freqmax=fmax)
|
||||
elif (ftype == 'lowpass') or (ftype == 'highpass'):
|
||||
st.filter(ftype, freq=fmin)
|
||||
st.merge()
|
||||
|
||||
stime = st[0].stats.starttime
|
||||
if deltat:
|
||||
etime = stime + deltat
|
||||
else:
|
||||
etime = st[0].stats.endtime
|
||||
|
||||
# plot
|
||||
for i in range(0, len(st)):
|
||||
if wsgi:
|
||||
buf = StringIO.StringIO()
|
||||
st[i].plot(outfile=buf, format=outpattern, type='dayplot', color=col, interval=interval,
|
||||
starttime=stime, endtime=etime)
|
||||
return buf
|
||||
else:
|
||||
if outpattern != '':
|
||||
imagefile = outpattern.format(st[i].stats.network,
|
||||
st[i].stats.station,
|
||||
st[i].stats.channel,
|
||||
st[i].stats.starttime.year,
|
||||
st[i].stats.starttime.julday)
|
||||
st[i].plot(type='dayplot', color=col, interval=interval,
|
||||
outfile=imagefile)
|
||||
else:
|
||||
st[i].plot(type='dayplot', color=col, interval=interval)
|
||||
|
||||
|
||||
def get_fdsn(bulk_request, output='VEL', base_url='https://ariadne.geophysik.ruhr-uni-bochum.de'):
|
||||
"""
|
||||
Fetches waveform data from FDSN web service and returns
|
||||
@ -172,18 +228,18 @@ def get_fdsn(bulk_request, output='VEL', base_url='https://ariadne.geophysik.ruh
|
||||
return None
|
||||
|
||||
|
||||
def main(backend=None, cla=None, wsgi=False):
|
||||
def main(backend=None, args=None, wsgi=False):
|
||||
"""
|
||||
Main function to create a waveform plot.
|
||||
This functions calls get_fdsn and fancy_plot to create the figure.
|
||||
If wsgi == True it returns an ioString object containing the figure.
|
||||
|
||||
:type backend: str
|
||||
:type cla: dict
|
||||
:type args: dict
|
||||
:type wsgi: bool
|
||||
:rtype : object
|
||||
:param backend:
|
||||
:param cla:
|
||||
:param args:
|
||||
:param wsgi:
|
||||
:return: ioString object with created image
|
||||
"""
|
||||
@ -199,32 +255,48 @@ def main(backend=None, cla=None, wsgi=False):
|
||||
mpl.use(backend)
|
||||
from obspy import UTCDateTime
|
||||
|
||||
if cla:
|
||||
deltat = cla['length']
|
||||
if cla['stime']:
|
||||
otime = UTCDateTime(cla['stime'])
|
||||
if args:
|
||||
deltat = args['length']
|
||||
if args['stime']:
|
||||
otime = UTCDateTime(args['stime'])
|
||||
else:
|
||||
otime = UTCDateTime() - 3600 - deltat
|
||||
network = cla['station'].split('.')[0]
|
||||
station = cla['station'].split('.')[1]
|
||||
network = args['station'].split('.')[0]
|
||||
station = args['station'].split('.')[1]
|
||||
if args['type'] == 'dayplot':
|
||||
channel = 'BHZ'
|
||||
else:
|
||||
if args['length'] < 3600:
|
||||
channel = 'HH?'
|
||||
else:
|
||||
channel = 'BH?'
|
||||
else:
|
||||
otime = UTCDateTime('2014-11-15T11:35:25Z')
|
||||
deltat = 30
|
||||
network = 'GR'
|
||||
station = 'BUG'
|
||||
channel = 'HH?'
|
||||
|
||||
st = get_fdsn([(network, station, '*', 'HH?', otime, otime + deltat)], base_url=cla['server'])
|
||||
st = get_fdsn([(network, station, '*', channel, otime, otime + deltat)], base_url=args['server'])
|
||||
if st is not None:
|
||||
stime = max(st[0].stats.starttime, otime)
|
||||
etime = min(st[0].stats.endtime, otime + deltat)
|
||||
st.trim(starttime=stime, endtime=etime)
|
||||
|
||||
if wsgi:
|
||||
# noinspection PyUnresolvedReferences
|
||||
return fancy_plot(st, wsgi=wsgi, img_format=cla['format'], color=cla['color'])
|
||||
if args['type'] == 'dayplot':
|
||||
with warnings.catch_warnings():
|
||||
warnings.simplefilter("ignore")
|
||||
return trace_dayplot(st, outpattern=args['format'], wsgi=wsgi, deltat=deltat)
|
||||
else:
|
||||
# noinspection PyUnresolvedReferences
|
||||
fancy_plot(st, color=cla['color'])
|
||||
return fancy_plot(st, wsgi=wsgi, img_format=args['format'], color=args['color'])
|
||||
else:
|
||||
if args['type'] == 'dayplot':
|
||||
with warnings.catch_warnings():
|
||||
warnings.simplefilter("ignore")
|
||||
trace_dayplot(st, wsgi=wsgi)
|
||||
else:
|
||||
fancy_plot(st, color=args['color'])
|
||||
elif not wsgi:
|
||||
warnings.warn('No data available!')
|
||||
|
||||
@ -253,34 +325,39 @@ def application(environ, start_response):
|
||||
|
||||
# fake command line arguments
|
||||
# setting needed default values
|
||||
cla = {'server': 'http://localhost/'}
|
||||
wsgi_args = {'server': 'http://localhost/'}
|
||||
|
||||
# fill cla with environment variables
|
||||
# fill wsgi_args with environment variables
|
||||
form = FieldStorage(fp=environ['wsgi.input'], environ=environ)
|
||||
cla['station'] = form.getfirst('station', 'GR.BUG').upper()
|
||||
cla['stime'] = form.getfirst('start_time', [])
|
||||
cla['length'] = form.getfirst('length', '30')
|
||||
cla['length'] = abs(int(cla['length']))
|
||||
cla['format'] = form.getfirst('format', 'png').lower()
|
||||
if cla['format'] != 'png' and cla['format'] != 'svg':
|
||||
cla['format'] = 'png'
|
||||
cla['color'] = form.getfirst('color', 'TRUE').upper()
|
||||
if cla['color'] == 'FALSE':
|
||||
cla['color'] = False
|
||||
wsgi_args['station'] = form.getfirst('station', 'GR.BUG').upper()
|
||||
wsgi_args['stime'] = form.getfirst('start_time', [])
|
||||
wsgi_args['length'] = form.getfirst('length', '30')
|
||||
wsgi_args['length'] = abs(int(wsgi_args['length']))
|
||||
wsgi_args['format'] = form.getfirst('format', 'png').lower()
|
||||
if wsgi_args['format'] != 'png' and wsgi_args['format'] != 'svg':
|
||||
wsgi_args['format'] = 'png'
|
||||
wsgi_args['color'] = form.getfirst('color', 'TRUE').upper()
|
||||
if wsgi_args['color'] == 'FALSE':
|
||||
wsgi_args['color'] = False
|
||||
else:
|
||||
cla['color'] = True
|
||||
wsgi_args['color'] = True
|
||||
wsgi_args['type'] = form.getfirst('type', '').lower()
|
||||
if (wsgi_args['type'] != 'dayplot' and wsgi_args['type'] != 'normal') or wsgi_args['type'] == '':
|
||||
if wsgi_args['length'] < 43200:
|
||||
wsgi_args['type'] = 'normal'
|
||||
else:
|
||||
wsgi_args['type'] = 'dayplot'
|
||||
|
||||
# process the request
|
||||
buf = main(cla=cla, backend='Agg', wsgi=True)
|
||||
buf = main(args=wsgi_args, backend='Agg', wsgi=True)
|
||||
if buf is not None:
|
||||
data = buf.getvalue()
|
||||
# noinspection PyUnresolvedReferences
|
||||
buf.close()
|
||||
data_length = len(data)
|
||||
|
||||
if cla['format'] == u'svg':
|
||||
cla['format'] = u'svg+xml'
|
||||
start_response('200 OK', [('Content-Type', 'image/%s' % cla['format']), ('Content-Length', '%d' % data_length)])
|
||||
if wsgi_args['format'] == 'svg':
|
||||
wsgi_args['format'] = 'svg+xml'
|
||||
start_response('200 OK', [('Content-Type', 'image/%s' % wsgi_args['format']), ('Content-Length', '%d' % data_length)])
|
||||
return [data]
|
||||
else:
|
||||
start_response('400 Bad Request', [])
|
||||
@ -301,6 +378,8 @@ if __name__ == "__main__":
|
||||
parser.add_argument(u'-u', u'--url', action='store', dest='server',
|
||||
default=u'https://ariadne.geophysik.ruhr-uni-bochum.de',
|
||||
help=u'Base URL of the FDSN web service (https://ariadne.geophysik.ruhr-uni-bochum.de).')
|
||||
parser.add_argument(u'-t', u'--type', action='store', dest='type', metavar='TYPE',
|
||||
help=u'Type of plot: normal or dayplot.')
|
||||
parser.add_argument(u'-f', u'--file', action='store', dest='filename', metavar='FILENAME',
|
||||
help=u'Save plot to file FILENAME')
|
||||
parser.add_argument(u'-c', u'--color', action='store_true', help=u'Create color plot.')
|
||||
@ -317,6 +396,6 @@ if __name__ == "__main__":
|
||||
cla = vars(parser.parse_args())
|
||||
|
||||
if os.getenv('DISPLAY'):
|
||||
main(cla=cla)
|
||||
main(args=cla)
|
||||
else:
|
||||
main('Agg', cla)
|
||||
@ -1,13 +1,13 @@
|
||||
#! /usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Produce a dayplot
|
||||
Produce a dayplot from seismogram recordings
|
||||
|
||||
Subversion information:
|
||||
$Id$
|
||||
"""
|
||||
"""
|
||||
Copyright 2012 Seismological Observatory, Ruhr-University Bochum
|
||||
|
||||
license: gpl3
|
||||
Copyright 2012-2015 Seismological Observatory, Ruhr-University Bochum
|
||||
http://www.gmg.ruhr-uni-bochum.de/geophysik/seisobs
|
||||
Contributors:
|
||||
Martina Rische <martina.rische@rub.de>
|
||||
@ -28,28 +28,30 @@ You should have received a copy of the GNU General Public License along
|
||||
with this program. If not, see http://www.gnu.org/licenses/.
|
||||
"""
|
||||
|
||||
def traceDayplot(datafile, ftype='bandpass', fmin=1.0, fmax=7.0,
|
||||
|
||||
def trace_dayplot(st, ftype='bandpass', fmin=1.0, fmax=7.0,
|
||||
col=('b', 'r', 'g'), interval=20.0, outpattern=''):
|
||||
"""
|
||||
|
||||
'''
|
||||
'''
|
||||
|
||||
|
||||
from obspy.core import read
|
||||
|
||||
st = read(datafile)
|
||||
:type st: object
|
||||
:type ftype: str
|
||||
:type fmin: float
|
||||
:type fmax: float
|
||||
:type col: tuple
|
||||
:type interval: float
|
||||
:type outpattern: str
|
||||
"""
|
||||
|
||||
# filter
|
||||
if (ftype == 'bandpass') or (ftype == 'bandstop'):
|
||||
st.filter(ftype, freqmin=fmin, freqmax=fmax)
|
||||
elif (ftype == 'lowpass') or (ftype == 'highpass'):
|
||||
st.filter(ftype, freq=fmin)
|
||||
|
||||
st.merge()
|
||||
|
||||
# plot
|
||||
for i in range(0, len(st)):
|
||||
if (outpattern != ''):
|
||||
if outpattern != '':
|
||||
imagefile = outpattern.format(st[i].stats.network,
|
||||
st[i].stats.station,
|
||||
st[i].stats.channel,
|
||||
@ -66,6 +68,8 @@ if __name__ == "__main__":
|
||||
|
||||
# parse arguments
|
||||
import argparse
|
||||
from obspy.core import read
|
||||
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Produce filtered 24h-plot (dayplot).',
|
||||
epilog='$Rev: 403 $ ($Date: 2012-04-13 12:16:22 +0200 (Fri, 13 Apr 2012) $, $Author: kasper $)')
|
||||
@ -96,7 +100,7 @@ if __name__ == "__main__":
|
||||
help="Output filename pattern for the plot. (default: unset, show plot on screen only), \
|
||||
Use {0} to substitute network code, \
|
||||
{1} to substitute station code, \
|
||||
{2} to subsitute station channel, \
|
||||
{2} to substitute station channel, \
|
||||
{3} to substitute year, \
|
||||
{4} to substitute doy number \
|
||||
.format (supported formats: emf, eps, pdf, png, ps, raw, rgba, svg, svgz)")
|
||||
@ -107,8 +111,7 @@ if __name__ == "__main__":
|
||||
if cla.fmax < 0:
|
||||
cla.fmax = -cla.fmax
|
||||
|
||||
# call traceDayplot(...) for each file
|
||||
# call trace_dayplot(...) for each file
|
||||
for datafile in cla.file:
|
||||
traceDayplot(datafile, cla.ftype, cla.fmin, cla.fmax, cla.color,
|
||||
trace_dayplot(read(datafile), cla.ftype, cla.fmin, cla.fmax, cla.color,
|
||||
cla.interval, cla.outpattern)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user