Skip to content
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

Autoplot mscolab for sideview #2573

Merged
merged 3 commits into from
Nov 13, 2024
Merged

Autoplot mscolab for sideview #2573

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7fe8a97
sideview can use mscolab
ReimarBauer Nov 5, 2024
7977280
added mscolab for sideview
ReimarBauer Nov 8, 2024
b8cf060
slugify added
ReimarBauer Nov 8, 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
124 changes: 84 additions & 40 deletions mslib/utils/mssautoplot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@
import defusedxml.ElementTree as etree
import PIL.Image
import matplotlib
from slugify import slugify
from fs import open_fs

import mslib
Expand Down Expand Up @@ -238,6 +239,36 @@ def __init__(self, cpath, msc_url=None, msc_auth_password=None, username=None, p
raise SystemExit("Filename {} doesn't exist".format(filename))
self.read_ftml(filename)

def setup(self):
pass

def update_path(self, filename=None):
"""
Update the path by reading the FTML data from the given filename
and redrawing the path based on the updated waypoints model data.

Parameters:
:filename: The name of the file to read FTML data from.

Returns:
None
"""
# plot path and label
if filename != "":
self.read_ftml(filename)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
self.plotter.redraw_path(waypoints_model_data=self.wp_model_data)

def update_path_ops(self, filename=None):
self.setup()
# plot path and label
if filename != "":
self.read_operation(filename, self.url, self.msc_auth, self.username, self.password)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
self.plotter.redraw_path(waypoints_model_data=self.wp_model_data)

def read_ftml(self, filename):
self.wps, self.wp_model_data = load_from_ftml(filename)
self.wp_lats, self.wp_lons, self.wp_locs = [[x[i] for x in self.wps] for i in [0, 1, 3]]
Expand Down Expand Up @@ -278,37 +309,18 @@ def __init__(self, cpath, msc_url, msc_auth_password, msc_username, msc_password
self.msc_auth = msc_auth_password
self.url = msc_url

def update_path(self, filename=None):
"""
Update the path by reading the FTML data from the given filename
and redrawing the path based on the updated waypoints model data.

Parameters:
:filename: The name of the file to read FTML data from.

Returns:
None
"""
# plot path and label
if filename != "":
self.read_ftml(filename)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
self.plotter.redraw_path(waypoints_model_data=self.wp_model_data)

def update_path_ops(self, filename=None):
# plot path and label
if filename != "":
self.read_operation(filename, self.url, self.msc_auth, self.username, self.password)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
self.plotter.redraw_path(waypoints_model_data=self.wp_model_data)
def setup(self):
pass

def draw(self, flight, section, vertical, filename, init_time, time, url, layer, style, elevation, no_of_plots):
if filename != "" and filename == flight:
self.update_path_ops(filename)
elif filename != "":
self.update_path(filename)
try:
self.update_path(filename)
except AttributeError as e:
logging.debug(e)
raise SystemExit("No FLIGHT Selected")

width, height = self.myfig.get_plot_size_in_px()
self.bbox = self.params['basemap']
Expand All @@ -330,24 +342,26 @@ def draw(self, flight, section, vertical, filename, init_time, time, url, layer,
}

auth_username, auth_password = get_auth_from_url_and_name(url, self.config["MSS_auth"])
# bbox for 1.3.0 needs a fix, swapped order
wms = MSUIWebMapService(url,
username=auth_username,
password=auth_password,
version='1.3.0')
version='1.1.1')

img = wms.getmap(**kwargs)
image_io = io.BytesIO(img.read())
img = PIL.Image.open(image_io)
self.myfig.draw_image(img)
t = str(time)
date_time = re.sub(r'\W+', '', t)
self.myfig.fig.savefig(f"{flight}_{layer}_{section}_{date_time}_{no_of_plots}_{elevation}.png")
plot_filename = slugify(f"{flight}_{layer}_{section}_{date_time}_{no_of_plots}_{elevation}") + ".png"
self.myfig.fig.savefig(plot_filename)
print(f"The image is saved at: {os.getcwd()}/{plot_filename}")


class SideViewPlotting(Plotting):
def __init__(self, cpath, msc_url, msc_auth_password, msc_username, msc_password, pdlg):
# ToDo Implement access to MSColab
super(SideViewPlotting, self).__init__(cpath, msc_url, msc_auth_password, msc_username, msc_password)
super(SideViewPlotting, self).__init__(cpath, msc_url, msc_auth_password, msc_username, msc_password, pdlg)
self.pdlg = pdlg
self.myfig = qt.SideViewPlotter()
self.ax = self.myfig.ax
Expand Down Expand Up @@ -376,8 +390,18 @@ def setup(self):
self.myfig.redraw_xaxis(self.lats, self.lons, times, times_visible)

def update_path(self, filename=None):
self.setup()
if filename is not None:
"""
Update the path by reading the FTML data from the given filename
and redrawing the path based on the updated waypoints model data.

Parameters:
:filename: The name of the file to read FTML data from.

Returns:
None
"""
# plot path and label
if filename != "":
self.read_ftml(filename)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
Expand All @@ -388,19 +412,27 @@ def update_path(self, filename=None):
self.myfig.draw_vertical_lines(highlight, self.lats, self.lons)

def update_path_ops(self, filename=None):
self.setup()
# plot path and label
if filename != "":
self.read_operation(filename, self.url, self.msc_auth, self.username, self.password)
self.fig.canvas.draw()
self.plotter.update_from_waypoints(self.wp_model_data)
self.plotter.redraw_path(waypoints_model_data=self.wp_model_data)
indices = list(zip(self.intermediate_indexes, self.wp_press))
self.plotter.redraw_path(vertices=indices,
waypoints_model_data=self.wp_model_data)
highlight = [[wp[0], wp[1]] for wp in self.wps]
self.myfig.draw_vertical_lines(highlight, self.lats, self.lons)

def draw(self, flight, section, vertical, filename, init_time, time, url, layer, style, elevation, no_of_plots):
try:
self.update_path(filename)
except AttributeError as e:
logging.debug(e)
raise SystemExit("No FLIGHT Selected")
if filename != "" and filename == flight:
self.update_path_ops(filename)
elif filename != "":
try:
self.update_path(filename)
except AttributeError as e:
logging.debug(e)
raise SystemExit("No FLIGHT Selected")
width, height = self.myfig.get_plot_size_in_px()
p_bot, p_top = [float(x) * 100 for x in vertical.split(",")]
self.bbox = tuple([x for x in (self.num_interpolation_points,
Expand All @@ -421,18 +453,25 @@ def draw(self, flight, section, vertical, filename, init_time, time, url, layer,
"format": "image/png",
"size": (width, height)
}

auth_username, auth_password = get_auth_from_url_and_name(url, self.config["MSS_auth"])
# bbox for sideview is correct
wms = MSUIWebMapService(url,
username=auth_username,
password=auth_password,
version='1.3.0')

img = wms.getmap(**kwargs)

image_io = io.BytesIO(img.read())
img = PIL.Image.open(image_io)
plot_filename = slugify(f"{flight}_{layer}_{time}_{no_of_plots}") + ".png"
self.myfig.setup_side_view()
self.myfig.draw_image(img)
self.ax.set_title(f"{flight}: {layer} \n{time} {no_of_plots}", horizontalalignment="left", x=0)
self.myfig.redraw_xaxis(self.lats, self.lons, None, False)
self.myfig.draw_image(img)
self.myfig.fig.savefig(f"{flight}_{layer}_{no_of_plots}.png", bbox_inches='tight')
self.myfig.fig.savefig(plot_filename, bbox_inches='tight')
print(f"The image is saved at: {os.getcwd()}/{plot_filename}")


class LinearViewPlotting(Plotting):
Expand Down Expand Up @@ -524,6 +563,8 @@ def draw(self):
@click.option('--etime', default="", help='Ending time for downloading multiple plots with a fixed interval.')
@click.pass_context
def main(ctx, cpath, view, ftrack, itime, vtime, intv, stime, etime):
pdlg = None

def close_process_dialog(pdlg):
pdlg.close()

Expand Down Expand Up @@ -575,6 +616,9 @@ def draw(no_of_plots):
elif view == "side":
side_view.draw(flight, section, vertical, filename, init_time, time,
url, layer, style, elevation, no_of_plots=no_of_plots)
else:
print("View is not available, Plot not created!")
return False
except Exception as e:
if "times" in str(e):
print("Invalid times and/or levels requested")
Expand Down
Loading