def clean_bundles(bundles, length_thr):
new_bundles = []
for bundle in bundles:
lengths = length_distribution(bundle)
new_bundle = [s for (i, s) in enumerate(
bundle) if lengths[i] > length_thr]
new_bundles.append(new_bundle)
return new_bundles
def all_descriptors(bundles):
descr = {}
descr['lengths'] = []
descr['avg_streamline'] = []
descr['winding_angle'] = []
descr['dragons_hits'] = []
for bundle in bundles:
bundle = flip_to_source(bundle, bundle[0][0])
descr['lengths'].append(length_distribution(bundle))
avg = avg_streamline(bundle)
descr['avg_streamline'].append(avg)
descr['winding_angle'].append(winding_angles(bundle))
descr['dragons_hits'].append(dragons_hits(bundle, avg))
return descr
