def print_exe_time(start, end):
'''Function print_exe_time(start, end) prints the time between two datetime values
Parameters
----------
start : datetime
starting time
end : datetime
ending time
Return
------
nothing
'''
import iwfm as iwfm
diff = str(end - start).split(':')
secs = str(round(float(diff[2]), 1))
if int(diff[0]) > 0:
hours = iwfm.pad_front(str(int(diff[0])), 2, '0') + ':'
mins = iwfm.pad_front(str(int(diff[1])), 2, '0') + ':'
elif int(diff[1]) > 0:
hours = ''
mins = str(int(diff[1])) + ' min '
secs += ' sec'
else:
hours, mins = '', ''
secs += ' seconds'
print(f' Execution time: {hours}{mins}{secs}\n')
def dss_date(date):
''' dss_date() - Convert datetime object 'date' into a string in DSS
format MM/DD/YYYY_HH:MM with HH in 24-hour format,
midnight = 24:00
Parameters
----------
date : datetime
date as datetime object
Returns
-------
date : str
date as string in DSS format
'''
import iwfm as iwfm
mo = iwfm.pad_front(date.month, 2, '0') # expand to 2-digits
dy = iwfm.pad_front(date.day, 2, '0') # expand to 2-digits
hour = date.hour
if hour == 0: # if midnight then put in DSS value
hour = 24
hr = iwfm.pad_front(hour, 2, '0') # expand to 2-digits
mn = iwfm.pad_front(date.minute, 2, '0') # expand to 2-digits
return mo + '/' + dy + '/' + str(date.year) + '_' + hr + ':' + mn
def print_exe_time(start, end, verbose=1):
'''Function print_exe_time(start, end) returns the time between two datetime values
as a sting hh:mm:ss,
Parameters
----------
start : datetime
starting time
end : datetime
ending time
verbose : bool, default=False
True = print exe_time
Return
------
exe_time : str
execution time as a sting, format dependes on exe_time value
if exe_time >= 1 hour:
'hh:mm:ss'
elif exe_time >1 minute:
'mm min ss sec'
else:
'ss.s seconds'
'''
import iwfm as iwfm
diff = str(end - start).split(':')
secs = str(round(float(diff[2]), 1))
if int(diff[0]) > 0:
hours = iwfm.pad_front(str(int(diff[0])), 2, '0') + ':'
mins = iwfm.pad_front(str(int(diff[1])), 2, '0') + ':'
elif int(diff[1]) > 0:
hours = ''
mins = str(int(diff[1])) + ' min '
secs += ' sec'
else:
hours, mins = '', ''
secs += ' seconds'
exe_time = '' + hours + mins + secs
if verbose:
print(f' Execution time: {exe_time}\n')
return exe_time
def exe_time():
''' print_exe_time() - Tracks the time since the first call, and prints
the elapsed time on subsequent calls
Parameters
----------
nothing
Return
------
nothing
'''
import datetime
import iwfm as iwfm
if not hasattr(exe_time, 'start'):
exe_time.start = datetime.datetime.now()
return
else:
end = datetime.datetime.now()
diff = str(end - exe_time.start).split(':')
secs = str(round(float(diff[2]), 1))
if int(diff[0]) > 0:
hours = iwfm.pad_front(str(int(diff[0])), 2, '0') + ':'
mins = iwfm.pad_front(str(int(diff[1])), 2, '0') + ':'
elif int(diff[1]) > 0:
hours = ''
mins = str(int(diff[1])) + ' min '
secs += ' sec'
else:
hours, mins = '', ''
secs += ' seconds'
print(f' Elapsed time: {hours}{mins}{secs}\n')
return
def gw_plot_obs_draw(well_name,
date,
meas,
no_hyds,
gwhyd_obs,
gwhyd_name,
well_info,
start_date,
title_words,
yaxis_width=-1):
''' gw_plot_obs_draw() - Create a PDF file with a graph of the simulated data
vs time for all hydrographs as lines, with observed values vs time as
dots, saved as the_well_name.pdf
Parameters
----------
well_name : str
well name, often state well number
date : list
list of dates (paired with meas)
meas : list
list of observed values (paired with date)
no_hyds : int
number of simulation time series to be graphed
gwhyd_obs : list
simulated IWFM groundwater hydrographs
[0]==dates, [1 to no_hyds]==datasets
gwhyd_name : list
hydrograph names from PEST observations file
well_info : list
well data from Groundwater.dat file
start_date : str
first date in simulation hydrograph files
title_words : str
plot title words
yaxis_width : int, default=-1
minimum y-axis width, -1 for automatic
Return
------
nothing
'''
import datetime
import matplotlib
import iwfm as iwfm
# Force matplotlib to not use any Xwindows backend.
matplotlib.use('TkAgg') # Set to TkAgg ...
matplotlib.use('Agg') # ... then reset to Agg
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
from matplotlib.backends.backend_pdf import PdfPages
line_colors = [
'b-', 'y-', 'r-', 'g-', 'c-', 'm-', 'k-', 'b--', 'y--', 'r--', 'g--',
'c--', 'm--', 'k--', 'b:', 'y:', 'r:', 'g:', 'c:', 'm:', 'k:'
]
# 'r-' = red line, 'bo' = blue dots, 'r--' = red dashes,
# 'r:' = red dotted line, 'bs' = blue squares, 'g^' = green triangles, etc
col = well_info[0]
dates = []
for i in range(0, len(gwhyd_obs)):
dates.append(datetime.datetime.strptime(gwhyd_obs[i][0], '%m/%d/%Y'))
ymin, ymax, sim_heads, sim_dates = 1e6, -1e6, [], []
for j in range(0, no_hyds):
sim_heads.append([float(x) for x in gwhyd_obs[j][1:]])
ymin = min(ymin, min(min(sim_heads)), min(meas))
ymax = max(ymax, max(max(sim_heads)), max(meas))
meas_dates = []
for i in range(0, len(date)):
meas_dates.append(datetime.datetime.strptime(date[i], '%m/%d/%Y'))
years = mdates.YearLocator()
months = mdates.MonthLocator()
yearsFmt = mdates.DateFormatter('%Y')
# plot simulated vs sim_dates as line, and meas vs specific dates as points, on one plot
with PdfPages(well_name + '_' + iwfm.pad_front(col, 4, '0') +
'.pdf') as pdf:
fig = plt.figure(figsize=(10, 7.5))
ax = plt.subplot(111)
ax.xaxis_date()
plt.grid(linestyle='dashed')
ax.yaxis.grid(True)
ax.xaxis.grid(True)
ax.xaxis.set_minor_locator(years)
plt.xlabel('Date')
plt.ylabel('Head (ft msl)')
plt.title(title_words + ': ' + well_name.upper() + ' Layer ' +
str(well_info[3]))
plt.plot(meas_dates, meas, 'bo', label='Observed')
# if minimum y axis width was set by user, check and set if necessary
if yaxis_width > 0:
if ymax > ymin:
if ymax - ymin < yaxis_width: # set minimum and maximum values
center = (ymax - ymin) / 2 + ymin
plt.ylim(center - yaxis_width / 2,
center + yaxis_width / 2)
for j in range(0, no_hyds):
plt.plot(sim_dates[j],
sim_heads[j],
line_colors[j],
label=gwhyd_name[j])
leg = ax.legend(frameon=1, facecolor='white')
pdf.savefig()
plt.close()
return