mirror of
https://github.com/Findus23/halo_comparison.git
synced 2024-09-19 16:03:50 +02:00
quick rename
This commit is contained in:
parent
7e1bb2fd6b
commit
2a70aa9abe
1 changed files with 5 additions and 5 deletions
10
halo_plot.py
10
halo_plot.py
|
@ -36,19 +36,19 @@ def in_area(coords: Coords, xobj, yobj, zobj, factor=1.3) -> bool:
|
|||
|
||||
def main():
|
||||
offset = 2
|
||||
columns = [128, 256, 512]
|
||||
resolutions = [128, 256, 512]
|
||||
initial_halo_id = int(argv[1])
|
||||
if has_1024_simulations:
|
||||
columns.append(1024)
|
||||
resolutions.append(1024)
|
||||
fig: Figure = plt.figure(figsize=figsize_from_page_fraction(columns=2, height_to_width=1))
|
||||
axes: List[List[Axes]] = fig.subplots(len(waveforms), len(columns), sharex="row", sharey="row")
|
||||
axes: List[List[Axes]] = fig.subplots(len(waveforms), len(resolutions), sharex="row", sharey="row")
|
||||
with h5py.File(vis_datafile) as vis_out:
|
||||
halo_group = vis_out[str(initial_halo_id)]
|
||||
|
||||
vmin, vmax = halo_group["vmin_vmax"]
|
||||
print(vmin, vmax)
|
||||
for i, waveform in enumerate(waveforms):
|
||||
for j, resolution in enumerate(columns):
|
||||
for j, resolution in enumerate(resolutions):
|
||||
dir = base_dir / f"{waveform}_{resolution}_100"
|
||||
halos = read_velo_halos(dir)
|
||||
ax = axes[i][j]
|
||||
|
@ -102,7 +102,7 @@ def main():
|
|||
# break
|
||||
# break
|
||||
rowcolumn_labels(axes, waveforms, isrow=True)
|
||||
rowcolumn_labels(axes, columns, isrow=False)
|
||||
rowcolumn_labels(axes, resolutions, isrow=False)
|
||||
|
||||
fig.tight_layout()
|
||||
fig.subplots_adjust(right=0.825)
|
||||
|
|
Loading…
Reference in a new issue