Add init to AR3Ccf that extracts parameters from pickparam

This commit is contained in:
Darius Arnold 2019-04-09 18:19:26 +02:00
parent cd63896060
commit 281e92e2a9
2 changed files with 5 additions and 2 deletions

View File

@ -994,7 +994,7 @@ class AutopickStation(object):
if self.s_params.algoS == 'ARH': if self.s_params.algoS == 'ARH':
arhcf1 = ARHcf(h_copy, cuttimesh, self.s_params.tdet1h, self.s_params.tpred1h, self.pickparams) arhcf1 = ARHcf(h_copy, cuttimesh, self.s_params.tdet1h, self.s_params.tpred1h, self.pickparams)
elif self.s_params.algoS == 'AR3': elif self.s_params.algoS == 'AR3':
arhcf1 = AR3Ccf(h_copy, cuttimesh, self.s_params.tpred1h, self.s_params.Sarorder, self.s_params.tdet1h, self.p_params.addnoise) arhcf1 = AR3Ccf(h_copy, cuttimesh, self.s_params.tdet1h, self.s_params.tpred1h, self.pickparams)
return arhcf1 return arhcf1
def _calculate_aic_cf_s_pick(self, cuttimesh): def _calculate_aic_cf_s_pick(self, cuttimesh):
@ -1055,7 +1055,7 @@ class AutopickStation(object):
if self.s_params.algoS == 'ARH': if self.s_params.algoS == 'ARH':
arhcf2 = ARHcf(h_copy, cuttimesh2, self.s_params.tdet2h, self.s_params.tpred2h, self.pickparams) arhcf2 = ARHcf(h_copy, cuttimesh2, self.s_params.tdet2h, self.s_params.tpred2h, self.pickparams)
elif self.s_params.algoS == 'AR3': elif self.s_params.algoS == 'AR3':
arhcf2 = AR3Ccf(h_copy, cuttimesh2, self.s_params.tpred2h, self.s_params.Sarorder, self.s_params.tdet2h, self.p_params.addnoise) arhcf2 = AR3Ccf(h_copy, cuttimesh2, self.s_params.tdet2h, self.s_params.tpred2h, self.pickparams)
# save cf for later plotting # save cf for later plotting
self.arhcf2 = arhcf2 self.arhcf2 = arhcf2

View File

@ -593,6 +593,9 @@ class ARHcf(CharacteristicFunction):
class AR3Ccf(CharacteristicFunction): class AR3Ccf(CharacteristicFunction):
def __init__(self, data, cut, t1, t2, pickparams):
super(AR3Ccf, self).__init__(data, cut, t1=t1, t2=t2, order=pickparams["Sarorder"], fnoise=pickparams["addnoise"])
def calcCF(self, data): def calcCF(self, data):
""" """
Function to calculate a characteristic function using autoregressive modelling of the waveform of Function to calculate a characteristic function using autoregressive modelling of the waveform of