def pct_updig(checked, kbouts=2, maxsep=50): ad = submit_summarize_runs.analysis_dir_from_donebase( checked.rsplit('.', 1)[0]) #config = eval(open(checked.split('l1800')[0]+'config.dict').read()) newact_tarf = glob(ad + '/newact_ols_postsummary-*.tar')[0] newact_tarh = tarfile.open(newact_tarf) newacts = Util.tar2obj_all(newact_tarh) SHAPE = eval(open(ad + '/SHAPE').read()) cm = vidtools.calc_coordMat(SHAPE) centroids = [ vidtools.centroid(reduce(lambda x, y: x + y, l), SHAPE, cm) for l in newacts if len(l) > 0 ] #pdy = config['predug'][1][1] #return len(filter(lambda x:x<pdy, \ # [y for x,y in centroids]))/float(len(centroids)) k = trial_from_filename(glob(ad + '/*hwc4*hmcdNone*dig_areas.list')[0]) y_offsets = [ p1[1] - p2[1] for p1, p2 in zip(centroids[:-kbouts], centroids[kbouts:]) if vidtools.hypotenuse(p1, p2) < maxsep ] return k, len(filter(lambda x: x > 0, y_offsets)) / float(len(y_offsets))
def load_all(ROOT='/n/hoekstrafs2/burrowing/antfarms/data'): areas = {} checked = glob(os.path.join(ROOT,'*/*/*.checked')) for chk in checked: ad = submit_summarize_runs.analysis_dir_from_donebase(chk.rsplit('.',1)[0]) digfs = glob(ad+'/*hwc4*hmcdNone*dig_areas.list') areas[digfs[0]] = eval(open(digfs[0]).read()) digs = {} print >> sys.stderr, 'updigs' digs['udigs'] = load_udigs(checked) print >> sys.stderr, 'pct' digs['pct'] = load_digs_pct(areas) print >> sys.stderr, 'peak_hour' digs['peak_hour'] = load_digs_peak_hour(areas) print >> sys.stderr, 'area_mean' digs['area_mean'] = load_digs_area_mean(areas) return digs
def load_all(ROOT='/n/hoekstrafs2/burrowing/antfarms/data'): areas = {} checked = glob(os.path.join(ROOT, '*/*/*.checked')) for chk in checked: ad = submit_summarize_runs.analysis_dir_from_donebase( chk.rsplit('.', 1)[0]) digfs = glob(ad + '/*hwc4*hmcdNone*dig_areas.list') areas[digfs[0]] = eval(open(digfs[0]).read()) digs = {} print >> sys.stderr, 'updigs' digs['udigs'] = load_udigs(checked) print >> sys.stderr, 'pct' digs['pct'] = load_digs_pct(areas) print >> sys.stderr, 'peak_hour' digs['peak_hour'] = load_digs_peak_hour(areas) print >> sys.stderr, 'area_mean' digs['area_mean'] = load_digs_area_mean(areas) return digs
def pct_updig(checked,kbouts=2,maxsep=50): ad = submit_summarize_runs.analysis_dir_from_donebase(checked.rsplit('.',1)[0]) #config = eval(open(checked.split('l1800')[0]+'config.dict').read()) newact_tarf = glob(ad+'/newact_ols_postsummary-*.tar')[0] newact_tarh = tarfile.open(newact_tarf) newacts = Util.tar2obj_all(newact_tarh) SHAPE = eval(open(ad+'/SHAPE').read()) cm = vidtools.calc_coordMat(SHAPE) centroids = [vidtools.centroid(reduce(lambda x,y:x+y, l),SHAPE,cm) for l in newacts if len(l)>0] #pdy = config['predug'][1][1] #return len(filter(lambda x:x<pdy, \ # [y for x,y in centroids]))/float(len(centroids)) k = trial_from_filename(glob(ad+'/*hwc4*hmcdNone*dig_areas.list')[0]) y_offsets = [p1[1]-p2[1] for p1,p2 in zip(centroids[:-kbouts],centroids[kbouts:]) if vidtools.hypotenuse(p1,p2)<maxsep] return k,len(filter(lambda x:x>0,y_offsets))/float(len(y_offsets))