Source code for prsctrl.data.offset

from .prsdata import *
import numpy as np

#
# OFFSET
#


def get_offset_data(data: PrsData, what=["lock-in-R", "slock-in-R"]) -> tuple[list[str], np.ndarray]:
    """
    :return: np.ndarray of [index, [timestamp, *what]]
    """
    offset_keys = data._get_time_sorted_offset_keys()
    odata = np.empty((len(offset_keys), len(what)+1))
    for i, k in enumerate(offset_keys):
        odata[i,0] = data.data[k]["timestamp_start"]
        for j, col in enumerate(what):
            val = data.get_for_wl(k, col)
            odata[i,j+1] = val
    return offset_keys, odata



# OFFSET TRANSIENTS


def merge_offset_transients(data: PrsData, what=["lock-in-R_raw"]):
    """
    Merge the raw data of all offset measurements
    """
    offset_keys = data._get_time_sorted_offset_keys()
    odata = [np.empty(0) for _ in what]
    for k in offset_keys:
        for i, c in enumerate(what):
            print(odata[i].shape, data.data[k][c].shape)
            odata[i] = np.append(odata[i], data.data[k][c], axis=0)
    return offset_keys, odata