squish/scripts/check_width_exists.py

from pathlib import Path
import numpy as np, pickle, sys

from squish import Simulation

def main():
	n = int(sys.argv[1])
	all_widths = set(np.round(np.arange(3, 10.05, 0.05), 2))
	for file in Path(f"squish_output/Radial[T]Search - N{n} - 500").iterdir():
		sim, frames = Simulation.load(file / 'data.squish')

		if sim.domain.n == n:
			try:
				all_widths.remove(next(frames)["domain"][1])
			except StopIteration:
				pass

	remain_widths = sorted(list(all_widths))[::-1]
	print("Remaining:", remain_widths)


if __name__ == "__main__":
	main()