220 lines
8.5 KiB
Python
220 lines
8.5 KiB
Python
## Plotting Mike21 SW results for SouthShore
|
|
# Author: AJMR
|
|
# December 22, 2021
|
|
|
|
# %% Setup Project
|
|
from mikeio import Dfsu, Mesh, Dfs2, Dfs0
|
|
from mikeio.eum import ItemInfo, EUMUnit, EUMType
|
|
|
|
import pandas as pd
|
|
import pathlib as pl
|
|
import numpy as np
|
|
import geopandas as gp
|
|
import math
|
|
import matplotlib.pyplot as plt
|
|
import contextily as ctx
|
|
from mpl_toolkits.axes_grid1.anchored_artists import AnchoredSizeBar, AnchoredDirectionArrows
|
|
import matplotlib.font_manager as fm
|
|
|
|
# %% Read Model Log
|
|
|
|
pth = pl.Path("//srv-mad3.baird.com/", "Projects", "13632.101 South Shore Breakwater", "06_Models",
|
|
"13632.101.W.AJMR.Rev0_ModelLog.xlsx")
|
|
runLog = pd.read_excel(pth.as_posix(), "StormRuns")
|
|
|
|
# %% Read Model Results
|
|
modelPlot = range(0, 4)
|
|
hmo_list = [None] * (max(modelPlot) + 1)
|
|
|
|
for m in modelPlot:
|
|
dfsIN = Dfsu(pl.Path(runLog['Run Location'][m], str(runLog['Number'][m]) + '_' +
|
|
runLog['Run Name'][m] + '.sw - Result Files', 'ResultsMap.dfsu').as_posix())
|
|
|
|
# Read Mesh for first model
|
|
mikeMeshIN = dfsIN.to_shapely()
|
|
|
|
# Convert to list of polygons
|
|
mikeMeshIN_list = [p for p in mikeMeshIN]
|
|
|
|
ds = dfsIN.read()
|
|
hm0 = ds["Sign. Wave Height"]
|
|
|
|
# Add to list
|
|
hmo_list[m] = gp.GeoDataFrame(hm0, geometry=mikeMeshIN_list)
|
|
|
|
print(m)
|
|
|
|
# Cleanup unnecessary variables
|
|
del mikeMeshIN
|
|
del mikeMeshIN_list
|
|
del hm0
|
|
|
|
# %% Plotting
|
|
# Shaded Water
|
|
mapbox = 'https://api.mapbox.com/styles/v1/alexander0042/ckemxgtk51fgp19nybfmdcb1e/tiles/256/{z}/{x}/{y}@2x?access_token=pk.eyJ1IjoiYWxleGFuZGVyMDA0MiIsImEiOiJjazVmdG4zbncwMHY4M2VrcThwZGUzZDFhIn0.w6oDHoo1eCeRlSBpwzwVtw'
|
|
x, y, arrow_length = 0.93, 0.95, 0.12
|
|
fontprops = fm.FontProperties(size=12)
|
|
|
|
for m in modelPlot:
|
|
fig, axes = plt.subplots(figsize=(6, 6))
|
|
|
|
plt.xlim([427800, 431000])
|
|
plt.ylim([4758000, 4762000])
|
|
|
|
hmo_list[m].plot(ax=axes, column=0, antialiased=False, scheme='equal_interval', k=7, cmap='turbo',
|
|
legend=True, legend_kwds={'fmt': "{:.2f}"}, vmax=6, vmin=0)
|
|
|
|
ctx.add_basemap(axes, source=mapbox, crs='EPSG:32616')
|
|
plt.suptitle(runLog['Short Description'][m])
|
|
axes.title.set_text('Significant Wave Height (m)')
|
|
axes.titlesize = 'small'
|
|
axes.set_xlabel('Easting [m]')
|
|
axes.set_ylabel('Northing [m]')
|
|
|
|
# # Add in scale bar and North Arrow
|
|
# scalebar = AnchoredSizeBar(axes.transData,
|
|
# 500, '500 m', 'lower right', pad=0.5, size_vertical=10, fontproperties=fontprops)
|
|
# axes.add_artist(scalebar)
|
|
|
|
axes.annotate('N', xy=(x, y), xytext=(x, y - arrow_length),
|
|
arrowprops=dict(facecolor='black', width=6, headwidth=20),
|
|
ha='center', va='center', fontsize=15,
|
|
xycoords=axes.transAxes)
|
|
|
|
plt.show()
|
|
|
|
fig.savefig(
|
|
'//srv-mad3.baird.com/Projects/13632.101 South Shore Breakwater/06_Models/02_Mike21SW/Images/Production/' +
|
|
runLog['Short Description'][m] + '_HM0.png', bbox_inches='tight')
|
|
|
|
# %% Read time series
|
|
MIKEds_list = [None] * (max(modelPlot) + 1)
|
|
MIKEdsT_list = [None] * (max(modelPlot) + 1)
|
|
|
|
for m in modelPlot:
|
|
dfsIN = Dfs0(pl.Path(runLog['Run Location'][m],
|
|
runLog['Run Name'][m] + '.sw - Result Files', 'BreakPts.dfs0').as_posix())
|
|
dfsIN_read = dfsIN.read()
|
|
MIKEds_list[m] = dfsIN_read.to_dataframe()
|
|
|
|
dfsTIN = Dfs0(pl.Path(runLog['Run Location'][m],
|
|
runLog['Run Name'][m] + '.sw - Result Files', 'TransectPTS.dfs0').as_posix())
|
|
dfsTIN_read = dfsTIN.read()
|
|
MIKEdsT_list[m] = dfsTIN_read.to_dataframe()
|
|
|
|
print(m)
|
|
|
|
# Cleanup unnecessary variables
|
|
del dfsIN_read
|
|
del dfsIN
|
|
del dfsTIN_read
|
|
del dfsTIN
|
|
|
|
# %% Read in Toe and Crest Shapefiles
|
|
breakwaterPTS = gp.read_file("//srv-mad3.baird.com/Projects/"
|
|
"13632.101 South Shore Breakwater/08_CADD/Outgoing/"
|
|
"20211211_Toe Extents (to Alexander)/"
|
|
"20211211_Toe_Extents_NAD83_WISStatePlaneSZn_USFt_Lines_OffshoreClipSimple_10m_vertexUTM.shp")
|
|
breakCrest = pd.read_csv(
|
|
'//srv-mad3.baird.com/Projects/13632.101 South Shore Breakwater/08_CADD/Outgoing/20211214_Crest Points (to Alexander)/20211214_Crest_Points_m.csv',
|
|
header=None, names=['x', 'y', 'z'])
|
|
breakCrest_gdf = gp.GeoDataFrame(breakCrest, crs='EPSG:32154',
|
|
geometry=gp.points_from_xy(breakCrest.x, breakCrest.y))
|
|
breakCrest_gdf.to_crs('EPSG:32616', inplace=True)
|
|
|
|
breakwaterPTS_Crest = breakwaterPTS.sjoin_nearest(breakCrest_gdf)
|
|
breakwaterPTS_Crest.rename(columns={'x': 'breakCrest_x', 'y': 'breakCrest_y', 'z': 'Crest'}, inplace=True)
|
|
|
|
# %% Merge with data
|
|
breakPointsOut = [None] * (max(modelPlot) + 1)
|
|
breakTimesOut = [None] * (max(modelPlot) + 1)
|
|
|
|
for m in modelPlot:
|
|
breakwaterPTS_times = None
|
|
|
|
for t in range(0, MIKEds_list[m].shape[0]):
|
|
breakwaterPTS_merge = None
|
|
breakwaterPTS_merge = breakwaterPTS_Crest
|
|
|
|
for i in range(0, MIKEds_list[m].shape[1]):
|
|
paramName = MIKEds_list[m].columns.values[i][MIKEds_list[m].columns.values[i].find('"', 5, -1) + 3:]
|
|
if paramName not in breakwaterPTS_merge.columns:
|
|
breakwaterPTS_merge[paramName] = np.full([breakwaterPTS_merge.shape[0], 1], np.nan)
|
|
tmpFID = int(MIKEds_list[m].columns.values[i][1:MIKEds_list[m].columns.values[i].find('"', 1)])
|
|
tmpIND = int(MIKEds_list[m].columns.values[i][5:MIKEds_list[m].columns.values[i].find('"', 5)])
|
|
|
|
breakwaterPTS_merge.loc[((breakwaterPTS_merge.FID == tmpFID) &
|
|
(breakwaterPTS_merge.vertex_ind == tmpIND)),
|
|
paramName] = MIKEds_list[m].iloc[t, i]
|
|
|
|
breakwaterPTS_merge['Time'] = MIKEds_list[m].index[t]
|
|
|
|
breakwaterPTS_times = pd.concat([breakwaterPTS_times, breakwaterPTS_merge], ignore_index=True)
|
|
|
|
print(i)
|
|
|
|
print(m)
|
|
|
|
breakPointsOut[m] = breakwaterPTS_merge
|
|
breakTimesOut[m] = breakwaterPTS_times
|
|
|
|
# %% Read in Breakwater Transect Points Sh
|
|
breakwaterT = pd.read_csv("C:/Users/arey/files/Projects/SouthShore/Bathy/BreakTransectPTS_Names.csv", sep='\t')
|
|
|
|
breakwaterT_PTS = gp.GeoDataFrame(breakwaterT, geometry=gp.points_from_xy(breakwaterT.X, breakwaterT.Y),
|
|
crs='EPSG:32616')
|
|
|
|
# %% Merge with data
|
|
breakPoints_TOut = [None] * (max(modelPlot) + 1)
|
|
breakTimes_TOut = [None] * (max(modelPlot) + 1)
|
|
|
|
for m in modelPlot:
|
|
breakwaterPTS_times = None
|
|
|
|
for t in range(0, MIKEdsT_list[m].shape[0]):
|
|
breakwaterPTS_merge = None
|
|
breakwaterPTS_merge = breakwaterT_PTS
|
|
|
|
for i in range(0, MIKEdsT_list[m].shape[1]):
|
|
paramName = MIKEdsT_list[m].columns.values[i][MIKEdsT_list[m].columns.values[i].find(':', 5, -1) + 2:]
|
|
if paramName not in breakwaterPTS_merge.columns:
|
|
breakwaterPTS_merge[paramName] = np.full([breakwaterPTS_merge.shape[0], 1], np.nan)
|
|
tmpName = MIKEdsT_list[m].columns.values[i][0:MIKEdsT_list[m].columns.values[i].find(':', 1)]
|
|
|
|
breakwaterPTS_merge.loc[(breakwaterPTS_merge.Name == tmpName),
|
|
paramName] = MIKEdsT_list[m].iloc[t, i]
|
|
|
|
breakwaterPTS_merge['Time'] = MIKEdsT_list[m].index[t]
|
|
|
|
breakwaterPTS_times = pd.concat([breakwaterPTS_times, breakwaterPTS_merge], ignore_index=True)
|
|
|
|
breakPoints_TOut[m] = breakwaterPTS_merge
|
|
breakTimes_TOut[m] = breakwaterPTS_times
|
|
|
|
|
|
#%% Format and save
|
|
|
|
for m in modelPlot:
|
|
saveTmp = breakPointsOut[m].copy()
|
|
saveTmp['X'] = saveTmp.geometry.x
|
|
saveTmp['Y'] = saveTmp.geometry.y
|
|
saveTmp.drop(['vertex_par', 'vertex_p_1', 'angle', 'geometry', 'index_right',
|
|
'breakCrest_x', 'breakCrest_y', 'Length'], axis=1, inplace=True)
|
|
saveTmp.sort_values(by=['FID', 'vertex_ind'], inplace=True, ignore_index=True)
|
|
|
|
# Reorder columns
|
|
colNames = saveTmp.columns.values
|
|
saveTmp = saveTmp[['X', 'Y', *colNames[0:-2]]]
|
|
saveTmpT = saveTmp.transpose(copy=True)
|
|
|
|
# Transect points
|
|
saveTmp2 = breakPoints_TOut[m].copy()
|
|
saveTmp2.drop(['geometry'], axis=1, inplace=True)
|
|
saveTmp2T = saveTmp2.transpose(copy=True)
|
|
|
|
|
|
saveTmpT.to_csv('//srv-mad3.baird.com/Projects/13632.101 South Shore Breakwater/06_Models/02_Mike21SW/Results/' +
|
|
'Toe' + runLog['Short Description'][m] + '.csv')
|
|
|
|
saveTmp2T.to_csv('//srv-mad3.baird.com/Projects/13632.101 South Shore Breakwater/06_Models/02_Mike21SW/Results/' +
|
|
'Transect' + runLog['Short Description'][m] + '.csv') |