input_file_generation.py

import os

import f90nml  # type: ignore


def generate_input_files(input_file_path, param_group, param_name, values, output_dir):
    nml = f90nml.read(input_file_path)

    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    for i, value in enumerate(values):

        nml[param_group][param_name] = value

        new_filename = f"{param_name}_{value}.in"
        new_file_path = os.path.join(output_dir, new_filename)

        nml.write(new_file_path, force=True)

    print(
        f"Generated {len(values)} input files with variations of {param_name} in '{output_dir}' directory."
    )


# input_file_path = "./parameter_scans/input_with_comments.in"
# input_file_path = "./parameter_scans/input_with_comments_aky_0.55.in"
# input_file_path = "./parameter_scans/input_with_comments_slowingdowndist.in"
input_file_path = "./parameter_scans/input_with_comments_slowingdowndist_tprim0.in"
input_file_path = "./parameter_scans/input_with_comments_slowingdowndist_fprim0.in"
# input_file_path = "./parameter_scans/input_with_comments_no_high_alphas.in"
# input_file_path = (
#     "./parameter_scans/input_with_comments_no_high_alphas_weird_density.in"
# )
param_group = "kt_grids_single_parameters"
param_name = "aky"
# param_group = "le_grids_knobs"
# param_name = "npassing"
values = [x / 1000 for x in range(10, 310, 10)]
# values = [0.05]
# output_dir = "./parameter_scans/beta_ky_scan_extras/"
# output_dir = "./parameter_scans/resolution_scans/npassing/"
# output_dir = "./parameter_scans/sdanalytic_scans/aky/"
# output_dir = "./parameter_scans/sdanalytic_tprim0_scans/aky/"
output_dir = "./parameter_scans/sdanalytic_fprim0_scans/aky/"
# output_dir = "./parameter_scans/no_high_alphas/aky/"
# output_dir = "./parameter_scans/maxwellian_scans/aky/"
# output_dir = "./parameter_scans/no_high_alphas_lower_ash_density/aky/"


generate_input_files(input_file_path, param_group, param_name, values, output_dir)