coganlab / bids_coding Goto Github PK

def prep_tsv(file_path: PathLike, task: str, pmatchz: str, ieeg_config: dict,
             bids_dir: PathLike) -> (str, pd.DataFrame):
    df = None
    for name, var in ieeg_config["headerData"]["channels"].items():
        if name in file_path:
            df = mat2df(file_path, var)
            if "highpass_cutoff" in df.columns.to_list():
                df = df.rename(columns={"highpass_cutoff": "high_cutoff"})
            if "lowpass_cutoff" in df.columns.to_list():
                df = df.rename(columns={"lowpass_cutoff": "low_cutoff"})
    df["type"] = ieeg_config["type"] #uhhhh wtf is this???
    df["units"] = ieeg_config["headerData"]["units"]
    new_row = pd.DataFrame(
        {"name": ["Trigger"], "high_cutoff": [1], "low_cutoff": [1000],
         "type": ["TRIG"], "units": ["uV"]})
    df = pd.concat([df, new_row], ignore_index=True)
    df = pd.concat([df["name"], df["type"], df["units"], df["low_cutoff"],
                    df["high_cutoff"]], axis=1)
    filename = op.join(bids_dir, "sub-{}".format(pmatchz),
                       "sub-" + pmatchz + "_task-{}".format(
                           task) + "_channels.tsv")
    return filename, df