subjectinfo.py

#
# This is kind of a mashup with 
# https://github.com/miykael/nipype-beginner-s-guide/blob/master/scripts/example_fMRI_1_first_level.py
#

def subjectinfo(subject_id):

    import pandas as pd
    from nipype.interfaces.base import Bunch
    
    path = '/data/sub-%s/func' % subject_id

    subject_info = []
    for run in run_list:
        trialfile = 'sub-%s_task-%s_run-%s_events.tsv' % (subject_id, task_list[0], run)
        trialinfo = pd.read_table(opj(path, trialfile))
        conditions = []
        onsets = []
        durations = []

        for trial_type in trialinfo.groupby('trial_type'):
            conditions.append(trial_type[0])
            onsets.append(list(round(trial_type[1].onset,2)))
            durations.append(round(trial_type[1].duration).tolist())

            
        subject_info.insert(int(run), 
                           Bunch(conditions=conditions,
                                onsets=onsets,
                                durations=durations,
                                amplitudes=None,
                                tmod=None,
                                pmod=None,
                                regressor_names=None,
                                regressors=None))
    
    return subject_info  # this output will later be returned to infosource


# Get Subject Info - get subject specific condition information
getsubjectinfo = Node(Function(input_names=['subject_id'],
                               output_names=['subject_info'],
                               function=subjectinfo),
                      name='getsubjectinfo')