Source code for prsctrl.scripts.offset_measurement

from ..data import PrsData
import numpy as np
import matplotlib.pyplot as plt
import os

import logging
log = logging.getLogger(__name__)



from ..data.plot import plot_offset_data, plot_offset_transients
from ..data.offset import merge_offset_transients, get_offset_data




def plot_offset_in_place(data_or_dir, title="Offset during measurement"):
    if isinstance(data_or_dir, PrsData):
        data: PrsData = data_or_dir
        dirpath: str = data.dirpath
    else:
        data: PrsData = PrsData(data_or_dir)
        dirpath: str = data_or_dir
    # data.remove_calculated_values()
    plot_base_path = os.path.join(dirpath, os.path.basename(dirpath))
    # fig, ax = plot_offset_transients(data, what=["lock-in-R_raw"], convert_idx2time=True, subplots_kw=dict(figsize=(5.2, 4.8)))
    # if title:
    #     fig.suptitle(title)
    # fig.tight_layout()
    # fig.savefig(plot_base_path + "-offset.pdf")

    fig, ax = plot_offset_data(data, what=["lock-in-R", "lock-in-theta", "lock-in-aux"], relative_time=True, with_errors=True, subplots_kw=dict(figsize=(5.2, 8)))
    if title:
        fig.suptitle(title)
    fig.tight_layout()
    fig.savefig(plot_base_path + "-offset-all.pdf")




#
# OFFSET TRANSIENTS
# For measurements where only offsets were recorded over long times
#




def plot_offset_transients_in_place(datadir, title=None):
    data = PrsData(datadir)
    plot_base_path = os.path.join(datadir, os.path.basename(datadir))
    fig, ax = plot_offset_transients(data, what=["lock-in-R_raw"], convert_idx2time=True, subplots_kw=dict(figsize=(5.2, 4.8)))
    if title:
        fig.suptitle(title)
    fig.tight_layout()
    fig.savefig(plot_base_path + ".pdf")

    fig, ax = plot_offset_transients(data, what=["lock-in-R_raw", "lock-in-theta_raw", "lock-in-aux_raw"], convert_idx2time=False, subplots_kw=dict(figsize=(5.2, 8)))
    if title:
        fig.suptitle(title)
    fig.tight_layout()
    fig.savefig(plot_base_path + "-all.pdf")