SpectrumBeam.py

#!/usr/bin/env python
"""
Executes Transcar with user-defined input particle flux spectrum

python SpectrumBeam.py flux.csv /tmp/spectrumout
"""

import logging
from pathlib import Path
import numpy as np
from pandas import read_csv
from argparse import ArgumentParser
import signal

#
import transcar


def main():
    signal.signal(signal.SIGINT, signal.SIG_DFL)
    p = ArgumentParser(description="parallel instance transcar runner")
    p.add_argument("fluxfn", help="particle flux csv filename")
    p.add_argument("outdir", help="simulation output directory")
    p.add_argument("-Q0", help="total particle flux", type=float, default=70114000000.0)
    p.add_argument("-infn", help="energy bin CSV file", default="BT_E1E2prev.csv")
    p.add_argument(
        "-datcar", help="DATCAR input file to copy", default="DATCAR_spectrum.asc"
    )
    p.add_argument(
        "-msgfn", help="file to write transcar messages to", default="transcar.log"
    )
    p.add_argument(
        "-errfn", help="file to write transcar Errors to", default="transcarError.log"
    )
    p = p.parse_args()

    rodir = Path(p.outdir).expanduser().resolve()
    infn = Path(p.infn).expanduser().resolve()
    fluxfn = Path(p.fluxfn).expanduser().resolve()

    params = {
        "rodir": rodir,
        "Q0": p.Q0,
        "msgfn": p.msgfn,
        "errfn": p.errfn,
        "datcar": p.datcar,
    }

    # logfn = rodir / "Beams.log"

    rodir.mkdir(parents=True, exist_ok=True)
    logging.basicConfig(
        format="%(asctime)s %(levelname)s %(message)s",
        datefmt="%H:%M:%S",
        level=logging.DEBUG,
    )

    beams = read_csv(infn, header=None, names=["E1", "E2", "pr1", "pr2"])
    beams["flux"] = np.loadtxt(fluxfn, delimiter=",")

    transcar.beam_spectrum_arbiter(beams, params)


if __name__ == "__main__":
    main()