figures_fcst.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
"""
Created on Tue May  4 11:07:21 2021
 
@author: cd
"""
 
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import warnings
warnings.filterwarnings('error', category=RuntimeWarning)
 
def plot_fcst_hydrograph(*, ax, tFcst, fcstResults, parameter, units,
              fontSz=9, timeInterval=24, yMin=0, yMax=None, lineWidth=0.75,
              showXTicks=True, showTitle=True, colorOvrd=None, lineStyle='-'):
    if colorOvrd:
        ax.plot(tFcst, fcstResults, linewidth=lineWidth, linestyle=lineStyle, color=colorOvrd)
    else:
        ax.plot(tFcst, fcstResults, linewidth=lineWidth, linestyle=lineStyle)
    ax.relim()
    ax.autoscale()
    ax.set_xlim(tFcst[0], tFcst[-1])
    ax.set_ylim(bottom=yMin)
    if yMax is not None:
        ax.set_ylim(top=yMax)
    ax.get_yaxis().get_major_formatter().set_scientific(False)
    # Set the date tick interval
    # ax1.xaxis.set_minor_locator(mdates.HourLocator(interval=1))
    ax.xaxis.set_major_locator(mdates.HourLocator(interval=timeInterval))
    if showXTicks:
        ax.xaxis.set_major_formatter(mdates.DateFormatter('%m-%d-%Hz'))
        ax.xaxis.set_tick_params(rotation = 45)
        for label in ax.xaxis.get_majorticklabels():
            label.set_horizontalalignment('right')
    else:
        ax.set_xticklabels([])
    ax.xaxis.grid(True, which='major', linestyle=':')
    ax.yaxis.grid(True, which='major', linestyle=':')
    plt.setp(ax.get_yticklabels(), fontsize=fontSz)
    ax.set_ylabel(parameter+' ('+units+')', fontsize=fontSz)
    if showTitle: ax.set_title(parameter, fontsize=fontSz)
 
    
def plot_fcst_risk(*, ax, tFcst, riskTol, riskFcst,
                   nameThreshold='Storage Threshold', fontSz=9, timeInterval=24, yMin=0, yMax=None,
                   showXTicks=True, showTitle=True):
    ax.plot(tFcst, riskTol*100, 'b--', linewidth=1, label='Risk Tolerance')
    ax.plot(tFcst, riskFcst*100, 'r-', linewidth=1.5, label='Forecasted Risk')
    ax.set_xlim(tFcst[0], tFcst[-1])
    ax.set_ylim(bottom=0, top=yMax)
    # Set the date tick interval
    # ax2.xaxis.set_minor_locator(mdates.HourLocator(interval=1))
    ax.xaxis.set_major_locator(mdates.HourLocator(interval=timeInterval))
    if showXTicks:
        ax.xaxis.set_major_formatter(mdates.DateFormatter('%m-%d-%Hz'))
        ax.xaxis.set_tick_params(rotation = 45)
        for label in ax.xaxis.get_majorticklabels():
            label.set_horizontalalignment('right')
    else:
        ax.set_xticklabels([])
    ax.xaxis.grid(True, which='major', linestyle=':')
    ax.yaxis.grid(True, which='major', linestyle=':')
    plt.setp(ax.get_xticklabels(), fontsize=fontSz)
    plt.setp(ax.get_yticklabels(), fontsize=fontSz)
    ax.set_ylabel('Risk (%)', fontsize=fontSz)
    ax.set_xlabel('Date', fontsize=fontSz)
    ax.legend(prop=dict(size=fontSz), loc='upper left')
    if showTitle: ax.set_title('% of Ensemble Members > '+nameThreshold, fontsize=fontSz)