Source code for biomechzoo.utils.get_split_events

from biomechzoo.utils.findfield import findfield




def get_split_events(data, first_event_name):
    """ splits lengthy trials containing n cycles into n trials based on side"""

    # find all events, events should follow style name1, name2, etc..
    split_events = []
    _, channel_name = findfield(data, first_event_name)
    if channel_name is None:
        return None

    event_name_root = first_event_name[0:-1]
    first_event_number = int(first_event_name[-1])
    i = 1
    if first_event_number > 1:
        i = first_event_number

    while True:
        key = f"{event_name_root}{i}"
        if key in data[channel_name]['event']:
            split_events.append(key)
            i += 1
        else:
            break

    n_segments = len(split_events) - 1
    if n_segments < 1:
        print("Not enough {} events to split.".format(event_name_root))
        return

    return split_events