Added debugging view. Packet loss suspected.
4 from operator import attrgetter
8 def help(socket, client, worker):
9 socket.sendto("\n".join([key + ": " + value[1]
10 for key, value in commands.iteritems()]) + "\n", client)
12 def createModel(socket, client, worker, (name,)):
13 model.createModel(name)
14 socket.sendto("model %s created\n" % name, client)
16 def deleteModel(socket, client, worker, (name,)):
17 if model.deleteModel(name):
18 socket.sendto("model %s deleted\n" % name, client)
20 def saveModel(socket, client, worker, (name,)):
21 model = lookupModel(socket, client, name)
23 pickle.dump(model, open(name + ".model", "w"))
24 socket.sendto("model %s saved\n" % name, client)
26 def loadModel(socket, client, worker, (name,)):
27 model.addModel(pickle.load(open(name + ".model", "r")))
28 socket.sendto("model %s loaded\n" % name, client)
31 if name in model.models:
32 return model.models[name]
34 def lookupModel(socket, client, name):
35 if name not in model.models:
36 socket.sendto("model %s not found\n" % name, client)
39 return model.models[name]
41 def addObserver(socket, client, worker, (modelName, eventType)):
42 model = lookupModel(socket, client, modelName)
44 model.register(socket, client, eventType)
45 socket.sendto("observer for event type %s added: " % eventType
46 + str(client) + "\n", client)
48 def setResolution(socket, client, worker, (modelName, x, y)):
49 model = lookupModel(socket, client, modelName)
51 model.setResolution(int(x), int(y))
52 socket.sendto("model %s resolution set to %s, %s\n" %
53 (modelName, x, y), client)
55 def setRange(socket, client, worker, (modelName, begin, end)):
56 model = lookupModel(socket, client, modelName)
58 model.setRange(complex(begin), complex(end))
59 socket.sendto("model %s range set to %s, %s\n" %
60 (modelName, begin, end), client)
62 def update(socket, client, worker, (modelName, iterations)):
63 model = lookupModel(socket, client, modelName)
65 old_res = model.resolution
66 begin_y = model.range[0].imag
67 end_y = model.range[1].imag
68 begin_x = model.range[0].real
69 end_x = model.range[1].real
70 step_x = (end_x - begin_x) / len(worker)
71 model.resolution = (old_res[0] / len(worker), old_res[1])
73 for i, w in enumerate(worker):
74 new_begin_x = begin_x + step_x * i
75 new_end_x = begin_x + step_x * (i + 1)
76 model.range = (complex(new_begin_x, begin_y),
77 complex(new_end_x, end_y))
79 print model.resolution
81 dump = pickle.dumps(model)
82 socket.sendto("%010i%s%010i" % (len(dump), dump,
84 socket.sendto("model %s updated with max %s iterations\n"
85 % (modelName, iterations), client)
86 model.range = (complex(begin_x, begin_y), complex(end_x, end_y))
87 model.resolution = old_res
89 def view(socket, client, worker, (name,)):
90 model = lookupModel(socket, client, name)
92 keys = model.cells.keys()
93 keys.sort(key=attrgetter("real", "imag"))
96 cells[k] = model.cells[k]
97 print "size: %i hash: %s" % (len(cells),
98 hashlib.sha256(str(cells)).hexdigest())
100 commands = {"help": (help, "help"),
101 "create model": (createModel, "creates a new model"),
102 "delete model": (deleteModel, "deletes a model"),
103 "save model": (saveModel, "makes a persistent copy of the model"),
104 "load model": (loadModel, "loads a persistent model"),
105 "set resolution": (setResolution, "sets the model resolution"),
106 "set range": (setRange, "sets the model range"),
107 "update": (update, "updates the model data"),
108 "view": (view, "shows the model data"),
109 "add observer": (addObserver, "adds an observer to a model's events"),}