Skip to content

Use context manager in imread #3963

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

Merged
timothy-nunn merged 1 commit into from
Oct 30, 2025
Merged

Use context manager in imread #3963

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
53 changes: 22 additions & 31 deletions process/io/plot_proc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -396,9 +396,8 @@ def plot_main_power_flow(
# ===========================================

# Load the plasma image
plasma = mpimg.imread(
resources.path("process.io", "plasma.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "plasma.png") as img_path:
plasma = mpimg.imread(img_path.open("rb"))

# Display the plasma image over the figure, not the axes
new_ax = axis.inset_axes(
Expand All @@ -420,7 +419,8 @@ def plot_main_power_flow(
fontsize=11,
)
# Load the neutron image
neutron = mpimg.imread(resources.path("process.io", "neutron.png"))
with resources.path("process.io", "neutron.png") as img_path:
neutron = mpimg.imread(img_path.open("rb"))

new_ax = axis.inset_axes(
[0.2, 0.85, 0.03, 0.03], transform=axis.transAxes, zorder=10
Expand Down Expand Up @@ -566,9 +566,8 @@ def plot_main_power_flow(
)

# Load the alpha particle image
alpha = mpimg.imread(
resources.path("process.io", "alpha_particle.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "alpha_particle.png") as img_path:
alpha = mpimg.imread(img_path.open("rb"))

# Display the alpha particle image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -653,9 +652,8 @@ def plot_main_power_flow(
)

# Load the HCD injector image
hcd_injector_1 = hcd_injector_2 = mpimg.imread(
resources.path("process.io", "hcd_injector.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "hcd_injector.png") as img_path:
hcd_injector_1 = hcd_injector_2 = mpimg.imread(img_path.open("rb"))

# Display the injector image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -924,9 +922,8 @@ def plot_main_power_flow(
# ===========================================

# Load the turbine image
turbine = mpimg.imread(
resources.path("process.io", "turbine.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "turbine.png") as img_path:
turbine = mpimg.imread(img_path.open("rb"))

# Display the turbine image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -983,9 +980,8 @@ def plot_main_power_flow(
)

# Load the generator image
generator = mpimg.imread(
resources.path("process.io", "generator.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "generator.png") as img_path:
generator = mpimg.imread(img_path.open("rb"))

# Display the generator image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -1053,9 +1049,8 @@ def plot_main_power_flow(
)

# Load the pylon image
pylon = mpimg.imread(
resources.path("process.io", "pylon.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "pylon.png") as img_path:
pylon = mpimg.imread(img_path.open("rb"))

# Display the pylon image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -1263,9 +1258,8 @@ def plot_main_power_flow(
# ================================

# Load the first wall image
fw = mpimg.imread(
resources.path("process.io", "fw.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "fw.png") as img_path:
fw = mpimg.imread(img_path.open("rb"))

# Display the first wall image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -1522,9 +1516,8 @@ def plot_main_power_flow(
)

# Load the blanket image
blanket = mpimg.imread(
resources.path("process.io", "blanket_with_coolant.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "blanket_with_coolant.png") as img_path:
blanket = mpimg.imread(img_path.open("rb"))

# Display the blanket image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -1566,9 +1559,8 @@ def plot_main_power_flow(
)

# Load the vacuum vessel image
vv = mpimg.imread(
resources.path("process.io", "vv.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "vv.png") as img_path:
vv = mpimg.imread(img_path.open("rb"))

# Display the vacuum vessel image over the figure, not the axes
new_ax = axis.inset_axes(
Expand Down Expand Up @@ -1671,9 +1663,8 @@ def plot_main_power_flow(
)

# Load the divertor image
divertor = mpimg.imread(
resources.path("process.io", "divertor.png")
) # Use importlib.resources to locate the image
with resources.path("process.io", "divertor.png") as img_path:
divertor = mpimg.imread(img_path.open("rb"))

# Display the divertor image over the figure, not the axes
new_ax = axis.inset_axes([0.1, 0.4, 0.3, 0.25], transform=axis.transAxes, zorder=10)
Expand Down