Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add center slicing in plt_slices

Browse Source 
Also fix bug of hard coded `size`
This commit is contained in:
Yin Li 2020-08-11 13:05:38 -07:00
parent 9c8b331bf5
commit 61ca400942
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
map2map/data/figures.py
View File

@ -26,7 +26,6 @@ def plt_slices(*fields, size=64, title=None, cmap=None, norm=None):
nc = max(field.shape[0] for field in fields)
nf = len(fields)
nd = fields[0].ndim - 1
if title is not None:
assert len(title) == nf
@ -77,8 +76,19 @@ def plt_slices(*fields, size=64, title=None, cmap=None, norm=None):
norm_col = SymLogNorm(linthresh=linthresh, vmin=-vlim, vmax=vlim)
for c in range(field.shape[0]):
s = (c,) + (0,) * (nd - 2) + (slice(64),) * 2
axes[c, f].pcolormesh(field[s], cmap=cmap_col, norm=norm_col)
s0 = (c,) + tuple(d // 2 for d in field.shape[1:-2])
s1 = (
slice(
(field.shape[-2] - size) // 2,
(field.shape[-2] + size) // 2,
),
slice(
(field.shape[-1] - size) // 2,
(field.shape[-1] + size) // 2,
),
)
axes[c, f].pcolormesh(field[s0 + s1], cmap=cmap_col, norm=norm_col)
axes[c, f].set_aspect('equal')