jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[read] more fixup for visualize_edges

Browse Source
This commit is contained in:
Jan Petykiewicz 2025-09-22 22:24:15 -07:00
parent f4818fd554
commit 32b6c207dc
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gridlock/read.py
View File

@ -147,6 +147,7 @@ class GridReadMixin(GridPosMixin):
cell_data: NDArray, cell_data: NDArray,
plane: PlaneProtocol | PlaneDict, plane: PlaneProtocol | PlaneDict,
which_shifts: int = 0, which_shifts: int = 0,
sample_period: int = 1,
finalize: bool = True, finalize: bool = True,
contour_args: dict[str, Any] | None = None, contour_args: dict[str, Any] | None = None,
ax: 'matplotlib.axes.Axes | None' = None, ax: 'matplotlib.axes.Axes | None' = None,
@ -207,7 +208,7 @@ class GridReadMixin(GridPosMixin):
fig, ax = pyplot.subplots() fig, ax = pyplot.subplots()
else: else:
fig = ax.figure fig = ax.figure
xc, yc = (self.shifted_exyz(which_shifts)[a] for a in surface) xc, yc = (self.shifted_xyz(which_shifts)[a] for a in surface)
xcmesh, ycmesh = numpy.meshgrid(xc, yc, indexing='ij') xcmesh, ycmesh = numpy.meshgrid(xc, yc, indexing='ij')
mappable = ax.contour(xcmesh, ycmesh, grid_slice, levels=levels, **contour_args) mappable = ax.contour(xcmesh, ycmesh, grid_slice, levels=levels, **contour_args)