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
