def permutation_hark_tf(tf_filename): config = read_source.read_hark_tf_source(tf_filename) # main tf_dict = {} tf_pair = [] with zipfile.ZipFile(tf_filename, "r") as zf: for el in config["positions"]: index = el[0] theta = math.atan2(el[2], el[1]) # -pi ~ pi tf_pair.append((theta, index)) sorted_pairs = sorted(tf_pair) for index, pair in enumerate(sorted_pairs): tf_dict[index] = pair[1] return tf_dict
def read_hark_tf(tf_filename): config=read_source.read_hark_tf_source(tf_filename) # main tf_dict={} with zipfile.ZipFile(tf_filename, 'r') as zf: for el in config["positions"]: index=el[0] path=TF_FILE%index #print path fp=zf.open(path) np_mat,info=parse_mat(fp) #print info #print np_mat.shape tf_dict[index]={"info":info,"mat":np_mat,"position":el[1:4]} config["tf"]=tf_dict #print "# config:",config return config