facof/parser.py at main · aridaibordon/facof · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import argparse
import os.path

from compute import compute_lev, compute_tr, compute_rr

PARSER_ARGS = [
    (
        ["-l", "-lev"],
        2,
        str,
        ("ELEM", "N_ELEC"),
        "compute atomic level structure",
        compute_lev,
    ),
    (
        ["-tr"],
        2,
        str,
        ("ELEM", "N_ELEC"),
        "compute atomic transition rates",
        compute_tr,
    ),
    (
        ["-rr"],
        2,
        str,
        ("ELEM", "N_ELEC"),
        "compute atomic RR and PI cross-section",
        compute_rr,
    ),
]


def get_facof_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        prog="FACOF v(1.0.0)",
        description=f"FAC on terminal framework (in {os.path.dirname(os.path.realpath(__file__))}).",
    )

    for args in PARSER_ARGS:
        flag, nargs, type, metavar, help, _ = args
        parser.add_argument(*flag, nargs=nargs, type=type, metavar=metavar, help=help)

    return parser