def precreation_hybridict_ordered(donor_dict_dis,acceptor_dict_dis,time_size_dict_dis):
''' Creates one dictionary hybdatadict out of three
The inputs are:
donordict = {'donor': 'n6mab031109', 'donorshanknum': 1, 'donorcluster': 54,
'donor_path':'/chandelierhome/skadir/hybrid_analysis/mariano/donors/',
'experiment_path': '/chandelierhome/skadir/hybrid_analysis/mariano/', 'donorcluid': 'MKKdistfloat'}
time_size_dict = {'amplitude_generating_function_args':[0.5, 1.5],'amplitude_generating_function':make_uniform_amplitudes,
'donorspike_timeseries_generating_function':create_time_series_constant,
'sampling_rate':20000, 'firing_rate':3, 'start_time':10,'end_time':None,
'donorspike_timeseries_arguments': 'arg'}
acceptor_dict = {'acceptor_path':'/chandelierhome/skadir/hybrid_analysis/mariano/acceptors/',
'acceptor': 'n6mab041109_60sec.dat','numchannels':32,
'output_path':'/chandelierhome/skadir/hybrid_analysis/mariano/',
}
'''
donor_dict = hash_utils.order_dictionary(donor_dict_dis)
acceptor_dict = hash_utils.order_dictionary(acceptor_dict_dis)
time_size_dict = hash_utils.order_dictionary(time_size_dict_dis)
hashDlist = hash_utils.get_product_hashlist([donor_dict,acceptor_dict,time_size_dict])
hashD = hash_utils.make_concatenated_filename(hashDlist)
hybdatadict_dis = merge_input_dicts(donor_dict,merge_input_dicts(acceptor_dict,time_size_dict))
hybdatadict_dis['hashD']= hashD
hybdatadict = hash_utils.order_dictionary(hybdatadict_dis)
return hybdatadict
def test_detection_algorithm(hybdatadict, SDparams,prb, detectioncrit):
'''
It will query whether the cached function:
hybridata_creation_lib.create_hybrid_kwdfile(hybdatadict),
has been called already with those arguments using `joblib_utils.is_cached`,
If it has, it calls it to obtain creation_groundtruth.
else if the hybrid dataset does not exist, it will raise an Error.
creation_groundtruth consists of the equivalent to the old:
GroundtruthResfile,GroundtruthClufile,... (i.e. the times and the cluster labels for the
added hybrid spikes.
detection criteria include:
allowed_discrepancy,CSthreshold
This function will call SpikeSimilarityMeasure(a,b)
and output the file: Hash(hybdatadict)_Hash(sdparams)_Hash(detectioncrit).kwik
It will return detcrit_groundtruth, the groundtruth relative to the criteria, detectioncrit.
This will be an ordered dictionary so that hashing can work!'''
if ju.is_cached(hcl.create_hybrid_kwdfile,hybdatadict):
print 'Yes, this hybrid dataset exists, I shall now check if you have run SD \n'
meanwaveform,kwdoutputname, creation_groundtruth, amplitude = hcl.create_hybrid_kwdfile(hybdatadict)
#Take the means of the binary donor masks of the donor cluster
binmeanmask = hcl.make_average_datamask_from_mean(hybdatadict, fmask= False)
argSD = [hybdatadict,SDparams,prb]
if ju.is_cached(rsd.run_spikedetekt,*argSD):
print 'Yes, SD has been run \n'
hash_hyb_SD = rsd.run_spikedetekt(hybdatadict,SDparams,prb)
else:
print 'You need to run Spikedetekt before attempting to analyse results '
DIRPATH = hybdatadict['output_path']
with Experiment(hash_hyb_SD, dir= DIRPATH, mode='a') as expt:
res_int= expt.channel_groups[0].spikes.time_samples
res_frac = expt.channel_groups[0].spikes.time_fractional
res_int_arr = res_int[:]
res_frac_arr = res_frac[:]
detected_times = res_int_arr + res_frac_arr
#Masks
fmask = expt.channel_groups[0].spikes.features_masks
#Spikes within time window
existencewin = np.zeros_like(creation_groundtruth)
#Mean binary mask for hybrid cluster
if 'manual_meanmask' in detectioncrit.keys():
binmeanmask = detectioncrit['manual_meanmask']
else:
binmeanmask = hcl.make_average_datamask_from_mean(hybdatadict, fmask= False)
indices_in_timewindow = hash_utils.order_dictionary({})
#indices_in_timewindow = {0 (this is the 1st hybrid spike): (array([0, 1, 3]),),
#1: (array([89, 90]),),
#2: (array([154, 156, 157]),),
#3: (array([191]),),
#4: (array([259, 260, 261]),),
num_spikes_in_timewindow = hash_utils.order_dictionary({})
CauchySchwarz = hash_utils.order_dictionary({})
detected = hash_utils.order_dictionary({})
NumHybSpikes = creation_groundtruth.shape[0]
trivialmainclustering = np.zeros_like(detected_times)
detectedgroundtruth = np.zeros_like(detected_times)
print detectedgroundtruth.shape
for k in np.arange(NumHybSpikes):
list_of_differences = np.zeros((detected_times.shape[0]))
list_of_differences[:]= detected_times[:] - creation_groundtruth[k]
indices_in_timewindow[k] = np.nonzero(np.absolute(list_of_differences)<=detectioncrit['allowed_discrepancy'])
num_spikes_in_timewindow[k] = indices_in_timewindow[k][0].shape[0]
for j in np.arange(num_spikes_in_timewindow[k]):
CauchySchwarz[k,j] = SpikeSimilarityMeasure(fmask[indices_in_timewindow[k][0][j],:,1],binmeanmask[0:3*hybdatadict['numchannels']])
if CauchySchwarz[k,j] > detectioncrit['CSthreshold']:
detected[k,j] = 1
else:
detected[k,j] = 0
detectedgroundtruth[indices_in_timewindow[k][0][j]]= detected[k,j]
#Store detectedgroundtruth in a clustering
detectionhashname = hash_utils.hash_dictionary_md5(detectioncrit)
#kwikfilename = DIRPATH + hash_hyb_SD
#+ '.kwik'
#add_clustering_kwik(kwikfilename, detectedgroundtruth, detectionhashname)
kwikfiles = open_files(hash_hyb_SD,dir=DIRPATH, mode='a')
add_clustering(kwikfiles,name = detectionhashname, spike_clusters=detectedgroundtruth,overwrite = True )
print 'Added a clustering called ', detectionhashname
add_clustering(kwikfiles,name = 'main', spike_clusters= trivialmainclustering, overwrite = True)
close_files(kwikfiles)
#clusters = '/channel_groups[0]/spikes/clusters'
#detectionhash = hash_dictionary_md5(detectioncrit)
#expt.createEArray(clusters, detectionhash, tb.UInt32Atom(), (0,),
# expectedrows=1000000)
#percentage_detected = float(sum(detected.values()))/NumHybSpikes
detcrit_groundtruth_pre ={'detection_hashname':
detectionhashname,'binmeanmask': binmeanmask,'indices_in_timewindow':indices_in_timewindow, 'numspikes_in_timeswindow': num_spikes_in_timewindow,
'Cauchy_Schwarz':CauchySchwarz,'detected': detected,'detected_groundtruth': detectedgroundtruth}
detcrit_groundtruth = hash_utils.order_dictionary(detcrit_groundtruth_pre)
return detcrit_groundtruth
