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)
