Beispiel #1
0
def raw_add_annotations(raw: mne.io.RawArray, marker_streams=None) -> mne.io.RawArray:
    t_reference = raw._t_original
    if marker_streams:
        for stream in marker_streams:
            annotations = marker_stream2annotations(stream, t_reference=t_reference)
            raw.set_annotations(raw.annotations + annotations)

    return raw
Beispiel #2
0
def add_events_df_to_mne_raw(mne_raw: mne.io.RawArray, events_df, orig_time):
    if mne_raw.annotations:
        mne_raw.annotations.append(onset=events_df['onset'] + orig_time,
                                   duration=events_df['duration'],
                                   description=events_df['description'])
        return mne_raw
    return mne_raw.set_annotations(
        mne_annotations_from_dataframe(events_df, orig_time))