Source code for idstools.view.distribution_sources

import itertools
import logging

import matplotlib.pyplot as plt

from idstools.compute.distribution_sources import DistributionSourcesCompute
from idstools.view.common import BasePlot

logger = logging.getLogger(f"module.{__name__}")


[docs]class DistributionSourcesView(BasePlot):
    def __init__(self, ids):
        self.distribution_sources_compute = DistributionSourcesCompute(ids)
        self.ids = ids

[docs]    def view_neutrons(self, ax: plt.axes, time_slice, source_index=0, **kwargs):
        rho_tor_norm = self.distribution_sources_compute.get_rho_tor_norm(time_slice, source_index)
        nrho = len(rho_tor_norm)
        if rho_tor_norm is not None and nrho == 0:
            logger.critical(
                f"distribution_sources.source[{source_index}].profiles_1d[{time_slice}].grid.rho_tor_norm) is empty"
            )
            return

        sources = self.distribution_sources_compute.get_source_info(time_slice)
        if len(sources) > 32:
            sources = dict(itertools.islice(sources.items(), 32))
        for key, source in sources.items():
            ax.plot(rho_tor_norm, source["particles"], label=source["label"])
            logger.info(
                f' {source["label"]}; P = ' + "%.2f" % (source["powerInKW"]) + " kW",
            )

        ax.set_xlim(rho_tor_norm[0], rho_tor_norm[nrho - 1])
        ax.set_xlabel(r"$\rho/\rho_0$", labelpad=1)
        ax.set_ylabel(r"Neutron rate ($s^{-1}.m{^{-3}}$)", labelpad=0)

        # set legend
        # legx_pos = 1.35
        # legy_pos = 1.05
        ax.legend()

[docs]    def view_time(self, ax: plt.axes, time: float):
        ax_time = ax.twiny()
        ymin, ymax = ax.get_ylim()
        ax_time.plot(
            [time, time],
            [ymin, ymax],
            color="gray",
            linestyle="--",
            linewidth=1,
            label=r"$t_{slice}$",
        )
        ax_time.set_ylim(ymin, ymax)

[docs]    def view_pulse_info(self, ax: plt.axes, title: str, hostdir: str, shot: int, run: int, t: float):
        self.database_info(ax, title, hostdir, shot, run, t)