Skip to content

More minor pep8 updates for AlterETM.py #115

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
espg wants to merge 10 commits into from
Closed

More minor pep8 updates for AlterETM.py #115

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
66607ee
PEP8 improvements
MarkusHackspacher Oct 11, 2024
07342fa
PEP8 improvements
MarkusHackspacher Oct 11, 2024
277575a
more pep8 updates
espg Oct 11, 2024
d3ecee5
rolling back for auto-merge
espg Oct 11, 2024
bd24cbb
one more roll back for auto-merge
espg Oct 11, 2024
908bf54
eq.date.year, eq.date.doy
MarkusHackspacher Oct 12, 2024
3ee130f
Merge branch 'patch-1' into update
espg Oct 14, 2024
561b7ab
Merge branch 'demiangomez:master' into update
espg Oct 14, 2024
7c27492
minor update
espg Oct 14, 2024
255fedf
easy flake8 fixes
espg Oct 14, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
158 changes: 102 additions & 56 deletions com/AlterETM.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
# app
from pgamit import dbConnection
from pgamit import Utils
from pgamit.Utils import required_length, process_date,station_list_help
from pgamit.Utils import required_length, process_date, station_list_help
from pgamit.pyBunch import Bunch
from pgamit import pyETM

Expand All @@ -27,46 +27,73 @@ def _split_lines(self, text, width):

def main():

parser = argparse.ArgumentParser(description='Program to alter the default ETM parameters for each station. '
'The command can be executed on several stations at the same time. '
'It is also possible to alter parameters for PPP and GAMIT '
'simultaneously.', formatter_class=SmartFormatter)
parser = argparse.ArgumentParser(description='Program to alter the default'
' ETM parameters for each '
'station. The command can be '
'executed on several stations'
' at the same time. It is '
'also possible to alter '
'parameters for PPP and GAMIT'
' simultaneously.',
formatter_class=SmartFormatter)

parser.add_argument('stnlist', type=str, nargs='+', metavar='all|net.stnm',
help=station_list_help())

parser.add_argument('-fun', '--function_type', nargs='+', metavar=('function', 'argument'), default=[],
help="R|Specifies the type of function to work with. Can be polynomial (p), jump (j), "
"periodic (q) or bulk earthquake jump removal (t). Each one accepts a list of arguments.\n"
"p {terms} where terms equals the number of polynomial terms in the ETM, i.e. "
"terms = 2 is constant velocity and terms = 3 is velocity + acceleration, etc.\n"
"j {action} {type} {date} {relax} where action can be + or -. A + indicates that a jump "
"should be added while a - means that an existing jump should be removed; "
"type = 0 is a mechanic jump and 1 is a geophysical jump; "
"date is the date of the event in all the accepted formats "
"(yyyy/mm/dd yyyy_doy gpswk-wkday fyear); and relax is a list of relaxation times for the "
"logarithmic decays (only used when type = 1, they are ignored when type = 0).\n"
"q {periods} where periods is a list expressed in days (1 yr = 365.25).\n"
"t {max_magnitude} {stack_name} removes any earthquake Mw <= max_magnitude from "
"the specified stations' trajectory models; if GAMIT solutions are invoked, provide the "
"stack_name to obtain the ETMs of the stations.\n"
"m {stack_name} [start_date] [end_date|days] removes mechanical jumps between given dates "
"from the specified stations' trajectory models; if no dates are provided, remove all "
"mechanical jumps. If only first date is provided, remove starting at that date until "
"today. Can also specify {start_date} {days} to add to {start_date}. "
"Provide the stack_name to obtain the ETMs of the stations. "
"If PPP solutions only (-soln ppp), stack_name is ignored. If both solutions are "
"indicated (or -soln is not specified) then stack_name must be provided.")

parser.add_argument('-soln', '--solution_type', nargs='+', choices=['ppp', 'gamit'],
default=['ppp', 'gamit'], action=required_length(1, 2),
help="Specifies the type of solution that this command will affect. If left empty, the ETMs "
"for both PPP and GAMIT will be affected. Otherwise, specify gamit to insert or "
"remove the function on GAMIT ETMs only or ppp to insert or remove the function on PPP "
parser.add_argument('-fun', '--function_type', nargs='+',
metavar=('function', 'argument'), default=[],
help="R|Specifies the type of function to work with. "
"Can be polynomial (p), jump (j), periodic (q) "
"or bulk earthquake jump removal (t). Each one "
"accepts a list of arguments.\n"
"p {terms} where terms equals the number of "
"polynomial terms in the ETM, i.e. terms = 2 "
"is constant velocity and terms = 3 is velocity "
"+ acceleration, etc.\n"
"j {action} {type} {date} {relax} where action "
"can be + or -. A + indicates that a jump "
"should be added while a - means that an "
"existing jump should be removed; type = 0 is a "
"mechanic jump and 1 is a geophysical jump; date "
"is the date of the event in all the accepted "
"formats (yyyy/mm/dd yyyy_doy gpswk-wkday fyear);"
" and relax is a list of relaxation times for the"
" logarithmic decays (only used when type = 1, "
"they are ignored when type = 0).\n"
"q {periods} where periods is a list expressed "
"in days (1 yr = 365.25).\n"
"t {max_magnitude} {stack_name} removes any "
"earthquake Mw <= max_magnitude from the "
"specified stations' trajectory models; if GAMIT "
"solutions are invoked, provide the stack_name "
"to obtain the ETMs of the stations.\n"
"m {stack_name} [start_date] [end_date|days] "
"removes mechanical jumps between given dates "
"from the specified stations' trajectory models; "
"if no dates are provided, remove all mechanical "
"jumps. If only first date is provided, remove "
"starting at that date until today. Can also "
"specify {start_date} {days} to add to "
"{start_date}. Provide the stack_name to obtain "
"the ETMs of the stations. If PPP solutions only "
"(-soln ppp), stack_name is ignored. If both "
"solutions are indicated (or -soln is not "
"specified) then stack_name must be provided.")

parser.add_argument('-soln', '--solution_type', nargs='+',
choices=['ppp', 'gamit'], default=['ppp', 'gamit'],
action=required_length(1, 2),
help="Specifies the type of solution that this "
"command will affect. If left empty, the ETMs "
"for both PPP and GAMIT will be affected. "
"Otherwise, specify gamit to insert or "
"remove the function on GAMIT ETMs only "
"or ppp to insert or remove the function on PPP "
"ETMs only.")

parser.add_argument('-print', '--print_params', action='store_true',
help="Print the parameters present in the database for the selected stations.")
help="Print the parameters present in the database " +
"for the selected stations.")

args = parser.parse_args()

Expand All @@ -84,11 +111,14 @@ def print_params(cnn, stnlist):

for station in stnlist:
params = cnn.query_float('SELECT * FROM etm_params WHERE "NetworkCode" = \'%s\' AND "StationCode" = \'%s\''
% (station['NetworkCode'], station['StationCode']), as_dict=True)
% (station['NetworkCode'],
station['StationCode']), as_dict=True)

for p in params:
print(' %s.%s %-5s %-5s %2i' % (station['NetworkCode'], station['StationCode'],
p['soln'], p['object'], p['terms']))
print(' %s.%s %-5s %-5s %2i' % (station['NetworkCode'],
station['StationCode'],
p['soln'], p['object'],
p['terms']))


def insert_modify_param(parser, cnn, stnlist, args):
Expand All @@ -98,13 +128,16 @@ def insert_modify_param(parser, cnn, stnlist, args):
parser.error('invalid number of arguments')

elif args.function_type[0] not in ('p', 'j', 'q', 't', 'm'):
parser.error('function type should be one of the following: polynomial (p), jump (j), periodic (q), '
'bulk geophysical jump removal (t), bulk mechanical jump removal (m).')
parser.error('function type should be one of the following:' +
'polynomial (p), jump (j), periodic (q), ' +
'bulk geophysical jump removal (t), ' +
'bulk mechanical jump removal (m).')

# create a bunch object to save all the params that will enter the database
tpar = Bunch()
tpar.NetworkCode = tpar.StationCode = tpar.soln = tpar.object = tpar.terms = tpar.frequencies = None
tpar.jump_type = tpar.relaxation = tpar.Year = tpar.DOY = tpar.action = None
tpar.NetworkCode = tpar.StationCode = tpar.soln = tpar.object = None
tpar.terms = tpar.frequencies = tpar.action = None
tpar.jump_type = tpar.relaxation = tpar.Year = tpar.DOY = None

ftype = args.function_type[0]
remove_eq = remove_mec = False
Expand All @@ -123,7 +156,8 @@ def insert_modify_param(parser, cnn, stnlist, args):
tpar.action = args.function_type[1]

if tpar.action not in ('+', '-'):
parser.error('action for function type jump (j) should be + or -')
parser.error('action for function type jump' +
'(j) should be + or -')

# jump type
tpar.jump_type = int(args.function_type[2])
Expand All @@ -147,11 +181,13 @@ def insert_modify_param(parser, cnn, stnlist, args):
if tpar.action == '-':
tpar.relaxation = None
elif tpar.action == '+':
parser.error('jump type == 1 but no relaxation parameter, please specify relaxation')
parser.error('jump type == 1 but no relaxation' +
'parameter, please specify relaxation')

elif ftype == 'q':
tpar.object = 'periodic'
tpar.frequencies = [float(1/float(p)) for p in args.function_type[1:]]
tpar.frequencies = [float(1/float(p)) for p
in args.function_type[1:]]

elif ftype == 't':
tpar.object = 'jump'
Expand All @@ -169,20 +205,24 @@ def insert_modify_param(parser, cnn, stnlist, args):
tpar.StationCode = station['StationCode']
tpar.soln = soln

station_soln = "%s.%s (%s)" % (station['NetworkCode'], station['StationCode'], soln)
station_soln = "%s.%s (%s)" % (station['NetworkCode'],
station['StationCode'], soln)

if remove_eq:
# load the ETM parameters for this station
print(' >> Obtaining ETM parameters for ' + station_soln)

if soln == 'ppp':
etm = pyETM.PPPETM(cnn, station['NetworkCode'], station['StationCode'])
etm = pyETM.PPPETM(cnn, station['NetworkCode'],
station['StationCode'])
else:
etm = pyETM.GamitETM(cnn, station['NetworkCode'], station['StationCode'],
etm = pyETM.GamitETM(cnn, station['NetworkCode'],
station['StationCode'],
stack_name=args.function_type[2])

for eq in [e for e in etm.Jumps.table
if e.p.jump_type in (pyETM.CO_SEISMIC_DECAY, pyETM.CO_SEISMIC_JUMP_DECAY,
if e.p.jump_type in (pyETM.CO_SEISMIC_DECAY,
pyETM.CO_SEISMIC_JUMP_DECAY,
pyETM.CO_SEISMIC_JUMP)]:
if eq.magnitude <= float(args.function_type[1]):
# this earthquake should be removed, fill in the data
Expand All @@ -197,15 +237,18 @@ def insert_modify_param(parser, cnn, stnlist, args):
print(' >> Obtaining ETM parameters for ' + station_soln)

if soln == 'ppp':
etm = pyETM.PPPETM(cnn, station['NetworkCode'], station['StationCode'])
etm = pyETM.PPPETM(cnn, station['NetworkCode'],
station['StationCode'])
else:
etm = pyETM.GamitETM(cnn, station['NetworkCode'], station['StationCode'],
etm = pyETM.GamitETM(cnn, station['NetworkCode'],
station['StationCode'],
stack_name=args.function_type[1])

for jump in [e for e in etm.Jumps.table
if e.p.jump_type in (pyETM.ANTENNA_CHANGE, pyETM.GENERIC_JUMP)]:
if e.p.jump_type in (pyETM.ANTENNA_CHANGE,
pyETM.GENERIC_JUMP)]:

# process the dates. If no date, returns from 1980 until now
# process the dates. If no date, returns from 1980 til now
sdate = process_date(args.function_type[2:])

if sdate[0] <= jump.date <= sdate[1]:
Expand All @@ -224,14 +267,17 @@ def apply_change(cnn, station, tpar, soln):
ppar = copy.deepcopy(dict(tpar))
ppar = {k: v
for k, v in ppar.items()
if v not in (None, []) and k not in ('action', 'relaxation', 'jump_type',
'terms', 'frequencies')}
if v not in (None, []) and k not in ('action', 'relaxation',
'jump_type', 'terms',
'frequencies')}

station_soln = "%s.%s (%s)" % (station['NetworkCode'], station['StationCode'], soln)
station_soln = "%s.%s (%s)" % (station['NetworkCode'],
station['StationCode'], soln)
# check if solution exists for this station
try:
epar = cnn.get('etm_params', ppar, list(ppar.keys()))
print(' >> Found a set of matching parameters for station ' + station_soln)
print(' >> Found a set of matching parameters for station '
+ station_soln)
print(' -- Deleting ' + station_soln)
cnn.delete('etm_params', **epar)

Expand Down