def test_max_value(self):
parser = FileParser('./test-input.txt')
max_register_value = parser.get_max_register_value()
self.assertEqual(0, max_register_value)
parser.run()
max_register_value = parser.get_max_register_value()
self.assertEqual(10, max_register_value)
def main(args):
if "filename" in args:
parser = FileParser(args.filename)
else:
parser = NetParser(args.host, args.port)
renderer = Renderer(args.pixels, args.mass, parser)
window = Window(args.fps, renderer)
window.mainloop()
def _read_file(self, file, data_size):
"""[summary]
Arguments:
file {[type]}
"""
parser = FileParser(self.seq_len, file)
d_size = len(parser.tours)
if d_size < data_size:
self.data_size = d_size
points_list = np.array(parser.coords[: self.data_size])
solutions = np.array(parser.tours[: self.data_size])
return {"Points_List": points_list, "Solutions": solutions}
def get_parser(self, url: str):
# first try supported hosts
host = urlparse(url).netloc
if host == Anime1Parser.host:
return Anime1Parser()
# finally try file parser by parsing file extention
file_ext_match = re.match(r'.*\.(\w+$)', url)
if file_ext_match and file_ext_match.group(
1) in FileParser.support_extensions:
return FileParser()
# nothing we can do...
raise Exception("url not support: %s" % url)
def new_file(self):
self.master.filename = filedialog.askopenfilename(
initialdir="./json",
title="Select file",
filetypes=(("json files", "*.json"), ("all files", "*.*")))
self.parser = FileParser(self.master.filename)
self.keylist = self.parser.findKeys()
self.key_search['values'] = self.keylist
self.key_extract['values'] = self.keylist
self.key_search.current(0)
self.key_extract.current(0)
file = os.path.basename(self.master.filename)
self.name.config(text="Current File: " + file)
def serve(sock, filename):
try:
parser = FileParser(filename)
sock.send("{}\n".format(parser.n))
sock.send("{}\n".format(parser.L))
sock.send("{}\n".format(parser.N))
sock.send("{}\n".format(parser.N_p))
for m in parser.masses:
sock.send("{}\n".format(m))
for _ in range(parser.N_p):
msg = sock.recv(3)
if msg != "g\r\n":
print "msg = '{}'".format(msg)
return
for (x, y) in parser.next():
sock.send("{} {} 0.0 0.0\n".format(x, y))
except Exception:
traceback.print_exc()
def main():
ip_file = None
myargs = getopts(argv)
if "-f" in myargs:
ip_file = myargs["-f"]
else:
print "Usage: python mission_assign.py -f <input file>"
exit()
op_file = ip_file.split(".")[0] + "_output.txt"
# create parser object
parser = FileParser(filepath=ip_file)
# flatten the uas indices into instances, taking into account uas count per uas
parser.create_uas_index_to_instance_map()
# setup mission map where each entry is a mission id - compatible uas list map.
missions = parser.get_domain_for_missions()
# setup pilot map where each entry is a pilot name - compatible uas list map.
pilots, pilots_pref = parser.get_domain_for_pilots()
uas_max = parser.get_uas_instance_count()
uas_names = parser.get_uas_names()
# create the solver object
solver = Solver(pilot_map=pilots,
mission_map=missions,
uas_max=uas_max,
pilot_prefs={})
# build the domain variables
solver.build_variables()
# build the constraints
solver.build_constraints()
# start the timer for 90 seconds
t = threading.Timer(90.0, timeout)
t.start()
# solve the 'problem'
solution = solver.get_solution()
# solution was found, timer can be stopped
t.cancel()
if solution:
# pretty print logic follows
print "Solution found! Writing to file..." + op_file
pretty_map = {}
for key, value in solution.iteritems():
if type(key) == int:
if value in pretty_map:
pretty_map[value]["mission"].append(key)
else:
pretty_map[value] = {
"mission": [key],
"pilot": None,
"uas":
uas_names[parser.get_uas_index_from_instance(value)],
"fav": None
}
sorted_pretty_list = [None] * len(missions)
for key, value in solution.iteritems():
if type(key) != int:
pretty_map[value]["pilot"] = key
pretty_map[value][
"fav"] = "Yes" if value in pilots_pref[key] else "No"
for uas, value in pretty_map.iteritems():
missions = value["mission"]
for mission in missions:
sorted_pretty_list[mission] = str("M" + str(mission + 1) +
" " + value["pilot"] + " " +
value["uas"] + " " +
value["fav"])
with open(op_file, 'w') as f:
for assignment in sorted_pretty_list:
f.write(assignment + "\n")
print assignment
else:
print "No solution found!"
with open(op_file, 'w') as f:
f.write("No solution found!" + "\n")
