diff --git a/indica/plotters/plot_time_evolution.py b/indica/plotters/plot_time_evolution.py index 2c41cb2c..7bf950fa 100644 --- a/indica/plotters/plot_time_evolution.py +++ b/indica/plotters/plot_time_evolution.py @@ -30,8 +30,8 @@ # "cxff_pi:vtor", # "cxff_tws_c:ti", # "cxff_tws_c:vtor", - # "ts:te", - # "ts:ne", + "ts:te", + "ts:ne", # "lines", ] @@ -69,7 +69,10 @@ def plot_st40_data( set_plot_rcparams(fig_style) xr.set_options(keep_attrs=True) - colors = CMAP(np.linspace(0.75, 0.1, len(pulses), dtype=float)) + if len(pulses) > 1: + colors = CMAP(np.linspace(0.75, 0.1, len(pulses), dtype=float)) + else: + colors = ["blue"] raw: dict = {quant: {} for quant in quantities} binned = deepcopy(raw) @@ -104,6 +107,8 @@ def plot_st40_data( plt.autoscale() save_figure(fig_path, f"{quantity}", save_fig=save_fig) + return data, st40 + def plot_data(data, quantity: str, pulse: int, tplot: float, key="raw", color=None): str_to_add = "" diff --git a/indica/workflows/load_modelling_plasma.py b/indica/workflows/load_modelling_plasma.py index d3167b0b..2b530a42 100644 --- a/indica/workflows/load_modelling_plasma.py +++ b/indica/workflows/load_modelling_plasma.py @@ -617,7 +617,7 @@ def plot_data_bckc_comparison( (_bckc).plot( label=label, color=COLORS["bckc"], - linewidth=rcParams["lines.linewidth"] * 1.5, + linewidth=rcParams["lines.linewidth"], alpha=alpha, ) del label @@ -699,6 +699,18 @@ def example_params(example: str, all_runs: bool = False): tend=0.11, tplot=0.1, ), + "michail_10014": dict( + comment="predictive ASTRA", + pulse_code=36010014, + pulse=10014, + equil="efit", + equil_run=0, + code="astra", + runs=["RUN24"], + tstart=0.03, + tend=0.1, + tplot=0.07, + ), "aleksei_11228": dict( comment="ASTRA using TS and invented Ti shapes", pulse=11228, @@ -803,5 +815,5 @@ def example_params(example: str, all_runs: bool = False): if __name__ == "__main__": plt.ioff() - example_run() + example_run(example="alsu_11314") plt.show()