__author__ = "Joshua Zosky"

"""
    Copyright 2015 Joshua Zosky
    joshua.e.zosky@gmail.com

    This file is part of "RetroTS".
    "RetroTS" is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
    "RetroTS" is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
    You should have received a copy of the GNU General Public License
    along with "RetroTS".  If not, see <http://www.gnu.org/licenses/>.
"""

from matplotlib.pyplot import (
    figure,
    plot,
    subplot,
    text,
    xlabel,
    title,
    legend,
    grid,
    show,
)
from numpy import real, spacing, pi
from .zscale import z_scale


def show_rvt_peak(r, fg):
    """

    :param r:
    :param fg:
    :return:
    """
    rvt_peak_plot = figure(fg)
    rvt_peak_plot.clf()
    # set(fg, 'KeyPressFcn', @afni_fig_interface)  # Appears to be unnecessary to retain from MATLAB code
    subplot(211)
    plot(r["t"], real(r["x"]), "g")

    if r["rvt"].any():
        subplot(211)
        plot(
            r["t_mid_prd"], z_scale(r["rvt"], min(r["p_trace"]), max(r["p_trace"])), "k"
        )
    plot(r["tp_trace"], r["p_trace"], "ro", r["tp_trace"], r["p_trace"], "r")
    plot(r["tn_trace"], r["n_trace"], "bo", r["tn_trace"], r["n_trace"], "b")
    plot(r["t_mid_prd"], r["p_trace_mid_prd"], "kx")
    for i in range(len(r["prd"])):
        text(r["t_mid_prd"][i], r["p_trace_mid_prd"][i], "%.2f" % r["prd"][i])
    if r["tR"]:
        if r["p_trace_r"].any():
            plot(r["tR"], r["p_trace_r"], "m")
            plot(r["tR"], r["n_trace_r"], "y")
        if r["rvtrs"].any():
            plot(
                r["tR"], z_scale(r["rvtrs"], min(r["p_trace"]), max(r["p_trace"])), "k."
            )
    xlabel("time (sec)")
    title(r["v_name"])  # , 'Interpreter', 'None')
    subplot(413)
    vn = real(r["x"]) / (abs(r["x"]) + spacing(1))
    plot(r["t"], vn, "g")
    plot(r["t"], r["phase"] / 2 / pi, "m")
    if "phase_r" in r:
        plot(r["tR"], r["phase_r"] / 2 / pi, "m-.")
    xlabel("time (sec)")
    title("Scaled by magnitude of analytical signal")  # , 'Interpreter', 'None')
    legend(["Scaled signal", "phase"])
    subplot(414)
    plot(
        r["time_series_time"], r["phase_slice"][:, 0], "ro"
    )  # This one isn't any different than the next plot line, check out why values are funky
    plot(r["time_series_time"], r["phase_slice"][:, 1], "bo")
    plot(r["time_series_time"], r["phase_slice"][:, 1], "b-")
    plot(r["t"], r["phase"], "k")
    grid("on")
    xlabel("time (sec)")
    title("Phase sampled at slice acquisition time")
    legend(["slice 0", "slice 1", "slice 1", "original phase"])
    # plotsign2(fg)  # Another AFNI function, not crucial at the moment to making one figure
    show()