Module rpps.viz.plots.freq

Frequency domain visualizations

Functions

def mag(ax, samps)

Expand source code

def mag(ax, samps):
    """Plot FFT Magnitude"""
    if ax is None:
        fig, ax = plt.subplots()
    x = bins(len(samps))
    y = np.abs(fft(samps))
    if isinstance(ax, mpl.lines.Line2D):
        ax.set_data(x, y)
        line = ax
    else:
        line, = ax.plot(x, y, label="Magnitude")
        ax.title("Magnitude")
        plt.title("Freq Domain", loc="left")
        plt.title(f"{len(samps)} samples", loc="right")
        plt.xlabel("Index")
        plt.ylabel("Magnitude")
    return line

Plot FFT Magnitude

def mag_phase(ax_mag, ax_pha, samps)

Expand source code

def mag_phase(ax_mag, ax_pha, samps):
    fig, ax = plt.subplots(1,2)
    line_mag = mag(ax_mag, samps)
    line_pha = pha(ax_pga, samps)
    return line_mag, line_pha

def pha(ax, samps)

Expand source code

def pha(ax, samps):
    """Plot FFT phase"""
    if ax is None:
        fig, ax = plt.subplots()
    x = bins(len(samps))
    y = np.angle(fft(samps))

    if isinstance(ax, mpl.lines.Line2D):
        ax.set_data(x, y)
        line = ax
    else:
        line, = ax.plot(x, y, label="Phase")

        ax.title("Phase")
        ax.title("Freq Domain", loc="left")
        ax.title(f"{len(samps)} samples", loc="right")
        ax.xlabel("Bin")
        ax.ylabel("Phase")
    return line

Plot FFT phase

def psd(ax, samps, Fs=1, cf=0, vbw_hz=None)

Expand source code

def psd(ax, samps, Fs=1, cf=0, vbw_hz=None):
    if ax is None:
        fig, ax = plt.subplots()
    x = freq_axis(len(samps), Fs, cf)
    y = utils.psd(samps, Fs, vbw_hz)

    if isinstance(ax, mpl.lines.Line2D):
        ax.set_data(x, y)
        line = ax
    else:
        line, = ax.plot(x, y, c="y")
        ax.grid(True)
        ax.set_title("PSD")
        ax.set_title("Freq Domain", loc="left")
        ax.set_title(f"{len(samps)} samples", loc="right")
        ax.set_xlabel("Frequency [Hz]")
        ax.set_ylabel("Magnitude [dB]")
        ax.set_xlim(x[0], x[-1])
    return line

def psd_minmax(ax, samps, sr=1, vbw_hz=None, cache={})

Expand source code

def psd_minmax(ax, samps, sr=1, vbw_hz=None, cache={}):
    if isinstance(ax, tuple):
        line, line_min, line_max = ax
        line = psd(line, samps, sr, vbw_hz)
    else:
        line = psd(ax, samps, sr, vbw_hz)
    x = line.get_xdata()
    y = line.get_ydata()
    if cache.get("min", None) is None:
        cache["min"] = np.copy(y)
    if cache.get("max", None) is None:
        cache["max"] = np.copy(y)
    mask = cache["min"]>y
    cache["min"][mask] = y[mask]
    mask = cache["max"]<y
    cache["max"][mask] = y[mask]

    if isinstance(ax, tuple):
        line_min.set_data(x, cache["min"])
        line_max.set_data(x, cache["max"])
    else:
        line_min, = ax.plot(x, cache["min"], c="b")
        line_max, = ax.plot(x, cache["max"], c="r")
    return line, line_min, line_max

def psdreal(ax, samps, Fs=1, cf=0, vbw_hz=None)

Expand source code

def psdreal(ax, samps, Fs=1, cf=0, vbw_hz=None):
    if ax is None:
        fig, ax = plt.subplots()
    y = utils.psd(samps.real, Fs, vbw_hz)
    y = y[len(y)//2:]
    x = freq_axis(len(y), Fs, cf)
    # x = np.arange(0, Fs, Fs/len(y)) + cf

    if isinstance(ax, mpl.lines.Line2D):
        ax.set_data(x, y)
        line = ax
    else:
        line, = ax.plot(x, y, c="y")
        ax.grid(True)
        ax.set_title("PSD")
        ax.set_title(f"Freq Domain [{Fs} Hz]", loc="left")
        ax.set_title(f"{len(samps)} samples", loc="right")
        ax.set_xlabel("Frequency [Hz]")
        ax.set_ylabel("Magnitude [dB]")
        ax.set_xlim(x[0], x[-1])
    return line