def generate_input(self):
self.objs = []
for x in range(1, self.tests + 1):
log("Generating inputs %d/%d" % (x, self.tests), end='\r')
obj = {}
if self.density is not None:
d = self.density
obj["values"] = []
obj["data"] = []
for r in self.nodes_plus_links:
"""
n = floor(
(-2 + d + sqrt(4 - 4 * d + d * d + 8 * d * r)) / (2 * d)
)
m = r - n
"""
n = (-2 + d + d * sqrt(
(d * d + 8 * d * r - 4 * d + 4) / (d * d))) / (2 * d)
n = floor(n)
m = r - n
obj["values"].append(r)
obj["data"].append(gen.get_input(n, m))
elif type(self.links) == int:
obj["values"] = self.nodes
obj['data'] = [
gen.get_input(n, self.links) for n in self.nodes
]
else:
obj["values"] = self.links
obj['data'] = [
gen.get_input(self.nodes, l) for l in self.links
]
self.objs.append(obj)
log()
self.save()
def main():
args = init()
if args.job == 'gen':
gen_exp()
elif args.job == 'recg':
output = subprocess.check_output('')
elif args.job == 'splearn':
Is = get_input(args.dat)
T, A, E, Ts, D, converge, t, in_iter, out_iter = robust_space_alignment(
Is)
def plot_cmp(fptr, idir, odir, nfiles):
odatdir = odir.replace("plots", "output")
print odatdir
fip = gen.get_input(odatdir + "/input")
topo_name = gen.get_param(fip, "topo_name")
init_wt = gen.get_param(fip, "Initial Wt setting")
algo = gen.get_param(fip, "link weight update")
tm_name = gen.get_param(fip, "tm_name")
fo = open("tmp", "a")
if algo.lower() == 'keymetric_ecmp':
fo.write(" km %s\n" % (tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'inv_cap':
fo.write(" inv_cap %s\n" % (tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'base_rand':
fo.write(" thorup %s\n" % (tm_name))
fo.close()
def plot_cmp(fptr,idir,odir,nfiles):
odatdir = odir.replace("plots","output")
print odatdir
fip = gen.get_input(odatdir+"/input")
topo_name = gen.get_param(fip,"topo_name")
init_wt = gen.get_param(fip,"Initial Wt setting")
algo = gen.get_param(fip,"link weight update")
tm_name = gen.get_param(fip,"tm_name")
fo = open("tmp","a")
if algo.lower() == 'keymetric_ecmp':
fo.write(" km %s\n"%(tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'inv_cap':
fo.write(" inv_cap %s\n"%(tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'base_rand':
fo.write(" thorup %s\n"%(tm_name))
fo.close()
init_wt = gen.get_param(fip, "Initial Wt setting")
algo = gen.get_param(fip, "link weight update")
tm_name = gen.get_param(fip, "tm_name")
fo = open("tmp", "a")
if algo.lower() == 'keymetric_ecmp':
fo.write(" km %s\n" % (tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'inv_cap':
fo.write(" inv_cap %s\n" % (tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'base_rand':
fo.write(" thorup %s\n" % (tm_name))
fo.close()
if __name__ == '__main__':
start = int(sys.argv[1])
end = int(sys.argv[2])
#fb = "data/output/exp5/pste/rocketfuel_abilene/run"
#fb = "data/output/conext/mate/pste/mate/run"
fb = "data/output/exp/pste/test_flow/run"
for i in range(start, end + 1):
file = fb + str(i) + "/input"
fptr = gen.get_input(file)
fo = open("tmp", "a")
fo.write("%s " % (i))
fo.close()
plot(fptr, i)
if algo == 'inv_cap':
fptr = gen.modify_param(fptr, 'initial wt setting', 'inv_cap')
fptr = gen.modify_param(fptr, 'link weight update', 'static')
fptr = gen.modify_param(fptr, 'numslices', 1)
out = gen.get_param(fptr, "outdir")
fptr = gen.modify_param(fptr, 'outdir', out + algo + "/")
return fptr
def change_topo(fptr, tid, fail=''):
fptr = gen.modify_param(fptr, 'ip_file', 'tm' + fail + str(tid) + '_ips')
fptr = gen.modify_param(fptr, 'topo_file',
'tm' + fail + str(tid) + '_topo')
fptr = gen.modify_param(fptr, 'tm_name', 'tm' + str(tid))
return fptr
if __name__ == '__main__':
fptr = gen.get_input()
#fptr = gen.modify_param(fptr,'duplex','no')
#for algo in ['pste']:#,'inv_cap']:
for algo in ['pste']: # ,'inv_cap']:
fptr = algo_setting(fptr, algo)
for tid in range(0, 1):
print 'Topo ID %s\n' % (tid)
for i in range(0, 1):
initialize(fptr)
#plot.plot(fptr)
fip = gen.get_input(odatdir+"/input")
topo_name = gen.get_param(fip,"topo_name")
init_wt = gen.get_param(fip,"Initial Wt setting")
algo = gen.get_param(fip,"link weight update")
tm_name = gen.get_param(fip,"tm_name")
fo = open("tmp","a")
if algo.lower() == 'keymetric_ecmp':
fo.write(" km %s\n"%(tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'inv_cap':
fo.write(" inv_cap %s\n"%(tm_name))
if algo.lower() == 'static' and init_wt.lower() == 'base_rand':
fo.write(" thorup %s\n"%(tm_name))
fo.close()
if __name__ == '__main__':
start = int(sys.argv[1])
end = int(sys.argv[2])
#fb = "data/output/exp5/pste/rocketfuel_abilene/run"
fb = "data/output/conext/mate/pste/mate/run"
for i in range(start,end+1):
file = fb + str(i) + "/input"
fptr = gen.get_input(file)
fo = open("tmp","a")
fo.write("%s "%(i))
fo.close()
plot(fptr,i)