[new] added attributes, properties and special method __str__ to the Magnitude superclass -> improves significantly convenience of sub-class programming

This commit is contained in:
Sebastian Wehling-Benatelli 2016-09-26 15:55:56 +02:00
parent 9288a169a4
commit c52277e4a2

View File

@ -32,10 +32,26 @@ class Magnitude(object):
Base class object for Magnitude calculation within PyLoT. Base class object for Magnitude calculation within PyLoT.
""" """
def __init__(self, stream): def __init__(self, stream, event, verbosity=False):
self._plot_flag = False
self._verbosity = verbosity
self._event = event
self._stream = stream self._stream = stream
self._magnitudes = dict() self._magnitudes = dict()
def __str__(self):
print('number of stations used: {0}\n'.format(len(self.magnitudes.values())))
print('\tstation\tmagnitude')
for s, m in self.magnitudes.items(): print('\t{0}\t{1}'.format(s, m))
@property
def plot_flag(self):
return self._plot_flag
@plot_flag.setter
def plot_flag(self, value):
self._plot_flag = value
@property @property
def stream(self): def stream(self):
return self._stream return self._stream
@ -44,6 +60,14 @@ class Magnitude(object):
def stream(self, value): def stream(self, value):
self._stream = value self._stream = value
@property
def event(self):
return self._event
@property
def arrivals(self):
return self._event.origins[0].arrivals
@property @property
def magnitudes(self): def magnitudes(self):
return self._magnitudes return self._magnitudes
@ -233,8 +257,8 @@ class RichterMagnitude(Magnitude):
'sensitivity': 1 'sensitivity': 1
} }
def __init__(self, stream, t0, calc_win): def __init__(self, stream, event, t0, calc_win, verbosity=False):
super(RichterMagnitude, self).__init__(stream) super(RichterMagnitude, self).__init__(stream, event, verbosity)
self._t0 = t0 self._t0 = t0
self._calc_win = calc_win self._calc_win = calc_win