23 lines
548 B
Python
23 lines
548 B
Python
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() |