forcePlot.py

# REDPy - Repeating Earthquake Detector in Python
# Copyright (C) 2016-2020  Alicia Hotovec-Ellis (ahotovec-ellis@usgs.gov)
# Licensed under GNU GPLv3 (see LICENSE.txt)

import redpy.config
import redpy.table
import redpy.plotting
import argparse
import numpy as np
import os

"""
Run this script to force plotting. Can be used after killing mid-run or updating settings.
Note that the -s and -e settings follow Python convention for arrays, i.e., start at 0
and do not include the ending number. For example, -s 1 -e 5 would replot 1, 2, 3, and 4.

usage: forcePlot.py [-h] [-v] [-a] [-c CONFIGFILE]

optional arguments:
  -h, --help            show this help message and exit
  -v, --verbose         increase written print statements
  -a, --all             replot everything, not just updated families
  -r, --resetlp         reset the 'last print' column (use for 'missing file' errors)
  -s STARTFAM, --startfam STARTFAM
                        manual starting family to replot (assumes ENDFAM is -1 if not set)
  -e ENDFAM, --endfam ENDFAM
                        manual (noninclusive) ending family to replot (assumes STARTFAM is
                        0 if not set)
  -f, --famplot         only replot the family plots, not html files
  -l, --html            only render the html, not any images
  -c CONFIGFILE, --configfile CONFIGFILE
                        use configuration file named CONFIGFILE instead of
                        default settings.cfg
"""

parser = argparse.ArgumentParser(description=
    "Run this script to force plotting. Can be used after killing mid-run or updating settings.")
parser.add_argument("-v", "--verbose", action="count", default=0,
    help="increase written print statements")
parser.add_argument("-a", "--all", action="count", default=0,
    help="replot everything, not just updated families")
parser.add_argument("-r", "--resetlp", action="count", default=0,
    help="reset the 'last print' column (use for 'missing file' errors)")
parser.add_argument("-s", "--startfam", type=int, default=0,
    help="manual starting family to replot (assumes ENDFAM is -1 if not set)")
parser.add_argument("-e", "--endfam", type=int, default=0,
    help="manual (noninclusive) ending family to replot (assumes STARTFAM is 0 if not set)")
parser.add_argument("-f", "--famplot", action="count", default=0,
    help="only replot the family plots, not html files")
parser.add_argument("-l", "--html", action="count", default=0,
    help="only render the html, not any images")
parser.add_argument("-c", "--configfile",
    help="use configuration file named CONFIGFILE instead of default settings.cfg")
args = parser.parse_args()

if args.configfile:
    opt = redpy.config.Options(args.configfile)
    if args.verbose: print("Using config file: {0}".format(args.configfile))
else:
    opt = redpy.config.Options("settings.cfg")
    if args.verbose: print("Using config file: settings.cfg")

if args.verbose: print("Opening hdf5 table: {0}".format(opt.filename))
h5file, rtable, otable, ttable, ctable, jtable, dtable, ftable = redpy.table.openTable(opt)

if args.all:
    if args.verbose: print("Resetting plotting column...")
    ftable.cols.printme[0:ftable.attrs.nClust] = np.ones((ftable.attrs.nClust,))

if args.resetlp:
    if args.verbose: print("Resetting last print column...")
    ftable.cols.lastprint[:] = np.arange(len(ftable))
    
if args.startfam or args.endfam:
    if args.startfam and not args.endfam:
        ftable.cols.printme[args.startfam:ftable.attrs.nClust] = np.ones(
                                                     (ftable.attrs.nClust-args.startfam,))
    elif args.endfam and not args.startfam:
        ftable.cols.printme[0:args.endfam] = np.ones((args.endfam,))
    else:
        ftable.cols.printme[args.startfam:args.endfam] = np.ones(
                                                             (args.endfam-args.startfam,))

if args.verbose: print("Creating requested plots...")

if args.famplot:
    redpy.plotting.plotFamilies(rtable, ftable, ctable, opt)

if args.html:
    redpy.plotting.plotFamilyHTML(rtable, ftable, opt)

if args.html or args.famplot:
    ftable.cols.printme[:] = np.zeros((len(ftable),))
    ftable.cols.lastprint[:] = np.arange(len(ftable))
else:
    redpy.plotting.createPlots(rtable, ftable, ttable, ctable, otable, opt)


if args.verbose: print("Closing table...")
h5file.close()
if args.verbose: print("Done")