sd.py

import os
from time import localtime, strftime

from preproc.init_from_efit import CDBManager
from preproc.renate_od.interface import SetProfiles
from preproc.renate_od.utils import get_home_directory
from plotter.detector import detector
from plotter.detector_plane import detector_plane
from plotter.traj_plotter import traj_plotter


class SD:
    def __init__(self, shot_number, time, species, energy, detector_par,
                 runnumber=None, is_trajectory_detailed=False, title=None):
        self.shot_number = str(int(shot_number))
        self.time = str(int(time))
        self.species = species
        self.energy = str(int(energy))
        self.detector_par = detector_par
        self.runnumber = runnumber
        self.is_trajectory_detailed = is_trajectory_detailed
        self.title = title
        self.is_run_simulation = True
        self.set_runnumber()
        if self.is_run_simulation:
            self.parameter_file = os.path.join(get_home_directory(),
                                               "parameter_" + self.runnumber + ".sh")

            self.write_parameter_file()
            self.preproc()
            self.run_core()
            self.delete_parameter_file()
        self.run_plotters()

    def preproc(self):
        CDBManager(self.shot_number, self.time)
        SetProfiles(self.shot_number, self.time, self.species, self.energy)

    def run_core(self):
        os.chdir(get_home_directory())
        command = "./taiga_renate.exe -p=" + self.parameter_file + " -r=" + self.runnumber
        if self.is_trajectory_detailed:
            command += " --fulltrace"
        print(command)
        os.system(command)

    def run_plotters(self):
        os.chdir(get_home_directory())
        detector(self.shot_number, self.time, self.runnumber, self.title)
        detector_plane(self.shot_number, self.time, self.runnumber, self.detector_par, self.title)
        if self.is_trajectory_detailed:
            traj_plotter(self.shot_number, self.time, self.runnumber, self.detector_par, self.species, self.energy, self.title)

    def set_runnumber(self):
        if self.runnumber is None:
            self.runnumber = strftime("%Y%m%d%H%M%S", localtime())
        else:
            self.is_run_simulation = False

    def write_parameter_file(self):
        f = open(self.parameter_file, "w")
        f.write("##!/usr/bin/bash\n")
        f.write("shotnumber='" + self.shot_number + "'\n")
        f.write("time='" + self.time + "'\n")
        f.write("beammatter='" + self.species + "'\n")
        f.write("energy=" + self.energy + " #keV\n")
        f.write("toroidal_deflection=0 #degree; + 200 V deflection\n")
        f.write("diameter=5 #mm\n")
        f.write("particles=1000\n")
        f.write("detector='" + self.detector_par + "'\n")
        f.write("electric_field_module=0\n")
        f.write("detector_mask='final'\n")
        f.write("solver='rk'\n")
        f.write("secondary_ionisation=1")
        f.close()

    def delete_parameter_file(self):
        os.remove(self.parameter_file)


if __name__ == "__main__":
    a_shot_number = 17178
    a_time = 1097
    a_species = 'Li'
    an_energy = 70
    z_det = 22
    t_det = -3.0
    a_detector = f"0.6846,{z_det/100},{t_det/100},38,0"
    a_runnumber = None#'20240530015119'
    is_plot_trajectory = True
    a_title = f"reference magnetic field\n{a_species} beam, $E ={an_energy}$ keV\n$Z_D={z_det}$ cm, $T_D={t_det}$ cm"
    SD(a_shot_number, a_time, a_species, an_energy, a_detector, a_runnumber, is_trajectory_detailed=is_plot_trajectory, title=a_title)