def run(self): subjpath = os.path.join(self.pipeline.mount_point,'data/eeg',self.pipeline.subject) bipolar_pairs = get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False) monopolar_channels = get_single_elecs_from_bps(bipolar_pairs) print len(monopolar_channels), 'single electrodes', len(bipolar_pairs), 'bipolar pairs' self.pass_object('bipolar_pairs', bipolar_pairs) self.pass_object('monopolar_channels', monopolar_channels) for i,bp in enumerate(bipolar_pairs): bipolar_pairs.tagName[i] = bp.tagName.upper() loc_tag = dict(zip(bipolar_pairs.tagName, bipolar_pairs.locTag)) try: stim_bipolar_pairs = get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False, bipolfileend='_talLocs_database_stimOnly.mat') for i,bp in enumerate(stim_bipolar_pairs): stim_bipolar_pairs.tagName[i] = bp.tagName.upper() loc_tag.update(dict(zip(stim_bipolar_pairs.tagName, stim_bipolar_pairs.locTag))) except: pass self.pass_object('loc_tag', loc_tag) joblib.dump(loc_tag, self.get_path_to_resource_in_workspace(self.pipeline.subject+'-loc_tag.pkl'))
def get_bps(self, events): #dataroot = get_dataroot(events) #subjpath = os.path.dirname(os.path.dirname(dataroot)) subjpath = os.path.join(self.pipeline.mount_point, 'data/eeg', self.pipeline.subject) return get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False)
def get_bps(self, events): subjpath = os.path.join(self.pipeline.mount_point,'data/eeg',self.pipeline.subject) return get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False)
def get_bps(events): dataroot = get_dataroot(events) subjpath = os.path.dirname(os.path.dirname(dataroot)) return get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False)
def get_bps(self, events): #dataroot = get_dataroot(events) #subjpath = os.path.dirname(os.path.dirname(dataroot)) subjpath = os.path.join(self.pipeline.mount_point,'data/eeg',self.pipeline.subject) return get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False)
def get_bp_tal_data(sub, rhino_prefix='/'): from get_bipolar_subj_elecs import get_bipolar_subj_elecs subjpath = rhino_prefix+'/data/eeg/'+sub return(get_bipolar_subj_elecs(subjpath))
if num_mp_procs > 0: po = mp.Pool(num_mp_procs) res = [] for subj in np.unique(ev['subject']): for sess in np.unique(ev[ev['subject']==subj]['session']): #sessions: indx = ((ev['subject']==subj) & (ev['session']==sess)) dataroot = np.unique([es.dataroot for es in ev[indx]['esrc']]) if len(dataroot)!=1: raise ValueError('Invalid number of dataroots: '+str(dataroot)) else: dataroot = dataroot[0] subjpath = os.path.dirname(os.path.dirname(dataroot)) talinfo = get_bipolar_subj_elecs( subjpath,leadsonly=True,exclude_bad_leads=False) if num_mp_procs > 0: res.append(po.apply_async( proc_sess, [subj, sess, talinfo, pow_params[exp]['output_path']])) else: proc_sess(subj, sess, talinfo, pow_params[exp]['output_path']) if num_mp_procs > 0: po.close() # track results for i,r in enumerate(res): sys.stdout.write('%d '%i) sys.stdout.flush() r.get()
def get_bps(self, events): subjpath = os.path.join(self.pipeline.mount_point, 'data/eeg', self.pipeline.subject) return get_bipolar_subj_elecs(subjpath, leadsonly=True, exclude_bad_leads=False)