mirror of
https://github.com/glatterf42/spectra_python_files.git
synced 2024-09-18 14:43:45 +02:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Created on Tue Mar 15 12:00:49 2022
|
|
|
|
@author: ben
|
|
"""
|
|
|
|
from sys import argv
|
|
from pathlib import Path
|
|
from write_spectra_jobs import write_spectra_jobs
|
|
from directories import monofonic_tests_basedir
|
|
|
|
def main(scale_factor: int, Nres1: int, Nres2: int, Lbox: float, waveforms: list):
|
|
a = [0.166666, 0.333333, 0.5, 0.666666, 1.0]
|
|
print(f"Chose scale factor a = {a[scale_factor]} (index {scale_factor} / {len(a) - 1})")
|
|
|
|
for wave in waveforms:
|
|
write_spectra_jobs(scale_factor, Nres1, Nres2, Lbox, wave,
|
|
output_basedir=monofonic_tests_basedir / "spectra/")
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if argv[5] == "all":
|
|
waveforms = ["DB2", "DB4", "DB6", "DB8", "DB10", "shannon"]
|
|
else:
|
|
waveforms = argv[5:len(argv)]
|
|
|
|
assert len(waveforms) <= 6
|
|
|
|
scale_factor = int(argv[1])
|
|
Nres1 = int(argv[2])
|
|
Nres2 = int(argv[3])
|
|
Lbox = float(argv[4])
|
|
|
|
main(
|
|
scale_factor=scale_factor,
|
|
Nres1=Nres1,
|
|
Nres2=Nres2,
|
|
Lbox=Lbox,
|
|
waveforms=waveforms
|
|
)
|
|
|