def run(self, arg):
pygame.display.flip()
self.update()
self.render()
timerevent = eventqueue.event(.01667, self.run, (1) )
self.eventqueue.add(timerevent)
pygame.display.flip()
def startPropagating(popped_event, event_queue, links):
#update the flow rate when there is propagation in the flow's initial link
if popped_event.link == popped_event.flow.findFirstLink(links):
#if this is the first packet propagating, set the 'last propagation' time
if popped_event.flow.last_propagation < 0:
popped_event.flow.last_propagation = popped_event.time
#else calculate the flow's rate and update the 'last propagation' time
else:
popped_event.flow.flow_rate = 8 * popped_event.packet.size / \
((popped_event.time - popped_event.flow.last_propagation) * 1000000.)
popped_event.flow.last_propagation = popped_event.time
#add the time point to the flow's rate history
if popped_event.event_type != 'FAST' and popped_event.packet.type != 'message':
popped_event.flow.flow_rate_history.append((popped_event.time,
popped_event.flow.flow_rate))
#add delay to relevant packets in the same link buffer when
#the packet is transitioning from buffering to propagating
popped_event.link.updateBufferPackets(popped_event.packet)
#if this is the first packet propagating, set the link's 'last propagation'
if popped_event.link.last_propagation < 0:
popped_event.link.last_propagation = popped_event.time
#else calculate the flow's rate and update the last propagation time
else:
popped_event.link.current_rate = 8 * popped_event.packet.size / \
((popped_event.time - popped_event.link.last_propagation) * 1000000.)
popped_event.link.last_propagation = popped_event.time
#add the time point to the link's rate history
if popped_event.event_type != 'FAST' and popped_event.packet.type != 'message':
popped_event.link.link_rate_history.append((popped_event.time,
popped_event.link.current_rate))
#update the current time of the packet
popped_event.packet.current_time += popped_event.link.delay / 1000.
#insert the event of the packet propagating, adding the link delay time
event_queue.insert_event(event('Propagating',popped_event.packet.current_time,
popped_event.packet, popped_event.link,
popped_event.flow))
#if the window size is not filled in the flow, add a packet to the first link
if popped_event.flow.occupancy < int(floor(popped_event.flow.window)):
first_link = popped_event.flow.findFirstLink(links)
popped_event.flow.addPacket(event_queue, first_link, popped_event.time)
def render(self, images):
for image in self.images:
image = image.split(",")
facing = 0
if image[3] == "-1":
facing = 1
self.surface.blit(self.get_image(image[0])[facing],(int(image[1]),int(image[2]) ) ) #sorta ugly
img = self.font.render("Kills:" + str(self.kills) ,1, (255,255,255,255))
self.surface.blit(img,(0,600-60))
img = self.font.render("Deaths:" +str(self.deaths) ,1, (255,255,255,255))
self.surface.blit(img,(0,600-30))
self.surface.blit(self.hpbar,(14*3,52*3))
for x in range(1,self.hp*2+1,2):
self.surface.blit(self.hpinc,(4*3+14*3,52*3+36*3-x*3))
try:
img = self.font.render(self.texts[0] ,1, (255,255,255,255))
self.surface.blit(img,(200,600-130))
img = self.font.render(self.texts[1] ,1, (255,255,255,255))
self.surface.blit(img,(200,600-105))
img = self.font.render(self.texts[2] ,1, (255,255,255,255))
self.surface.blit(img,(200,600-80))
img = self.font.render(self.texts[3] ,1, (255,255,255,255))
self.surface.blit(img,(200,600-55))
except:
pass
if self.text != '':
self.surface.blit(self.chatbar,(200,600-30))
img = self.font.render("Say: " + self.text,1,(255,255,255,255))
self.surface.blit(img,(200,600-30))
if self.escapemenu:
self.renderer.add_widget(self.cancelbutton,self.quitbutton)
self.renderer.distribute_events(*self.event)
self.renderer.update()
self.renderer.draw(self.surface)
if self.tabbed:
self.tabrenderer.add_widget(self.name_label,self.k_label,self.d_label)
self.tabrenderer.add_widget(self.kill_label,self.death_label)
self.tabrenderer.update()
self.tabrenderer.draw(self.surface)
timerevent = eventqueue.event(0.01667, self.render, () )
self.eventqueue.add(timerevent)
pygame.display.update()
def addToBuffer(self, event_queue, time, packet, flow):
#check if the next packet will not overfill the link buffer
if self.buffer_occupancy + packet.size \
< self.buffer_capacity * 1000:
#set the number of packet drops to 0
self.packet_drops = 0
#calculate the transmission time for the packet on this new link
transmission = self.getTransmission(packet)
#if there is delay from other packets in the buffer, delay this packet
if len(self.buffer_elements) > 0:
#get the last packet on the first link's buffer
last = self.buffer_elements[len(self.buffer_elements) - 1]
#check whether there is half-duplex congestion
switch = last.current_router != packet.current_router and \
(packet.type == 'packet' and last.type == 'packet')
#if there is no half-duplex congestion, add transmission accordingly
if switch == 0:
if time < last.current_time:
#set the current time of the new packet with link congestion
packet.current_time = last.current_time + transmission
else:
#set the current time of the new packet without link congestion
packet.current_time = time + transmission
#if there is half-duplex congestion, also add propagation accordingly
else:
if time < last.current_time:
#set the current time of the new packet with link congestion
packet.current_time = last.current_time + transmission + \
self.rate / 1000.
else:
#set the current time of the new packet without link congestion
packet.current_time = time + transmission
else:
packet.current_time = time + transmission
#insert the buffering event for the new packet on the first link
event_queue.insert_event(
event('Buffering', packet.current_time, packet, self, flow))
#add the packet to the buffer and update its occupancy and elements
self.buffer_occupancy += packet.size
self.buffer_elements.append(packet)
#if the buffer is overfilled, then the packet is dropped
else:
self.packet_drops += 1
if flow.con_ctrl == 1:
flow.ss_threshold = flow.window / 2
flow.window /= 2.
if flow.window < 1.:
flow.window = 1.
flow.occupancy -= 1
if packet.type != 'message':
flow.window_history.append((time, flow.window))
def run(self):
timerevent = eventqueue.event(0.1667, self.render, () )
self.eventqueue.add(timerevent)
def run_simulation(event_queue, flows, links, routers, con_ctrl):
#add all of the packets for each flow, with the flow source put into route
for flow in flows:
flow.initializePackets(data_packet_size)
if con_ctrl == 2:
event_queue.insert_event(event('FAST', flow.start + fast_interval, -1, -1, flow))
# Initialize a message for each flow in the network in order to start
# dynamic routing
# for flow in flows:
# flow.initializeMessage(event_queue, links, dyn_rout_interval)
#initialize first link's buffer for each flow from host based on window size
for i in range(len(flows)):
#get the first link connected to the flow's source
first_link = flows[i].findFirstLink(links)
#fit however many packets into the first link and add buffering events
for j in range(int(floor(flows[i].window))):
flows[i].addPacket(event_queue, first_link, flows[i].start)
lastTime = dyn_rout_interval
#iterate through all packets, using the same sequential events in the queue
while not event_queue.is_empty():
#get event after popping from queue
popped_event = event_queue.pop_event()
# Add more messages for dynamic routing every time the dynamic routing
# interval has passed.
if(popped_event.time >= lastTime + dyn_rout_interval):
newTime = popped_event.time
for flow in flows:
flow.initializeMessage(event_queue, links, newTime)
lastTime = newTime
#perform the transition from buffering to propagating
if popped_event.event_type == 'Buffering':
if popped_event.packet.type != 'message':
popped_event.flow.time_packet_last_seen = popped_event.time
startPropagating(popped_event, event_queue, links)
#perform the transition from propagating to buffering
elif popped_event.event_type == 'FAST':
popped_event.flow.window = min(2 * popped_event.flow.window, \
(1 - gamma) * popped_event.flow.window + \
gamma * (popped_event.flow.window * \
popped_event.flow.base_rtt/popped_event.flow.last_rtt + alpha))
if popped_event.flow.time_packet_last_seen + fast_limit >= popped_event.time:
next_fast = event('FAST', popped_event.time + fast_interval, -1, -1, popped_event.flow)
popped_event.flow.packet_seen = 0
event_queue.insert_event(next_fast)
else:
#check whether the acknowledgment has returned to the source for a packet
if popped_event.event_type == 'Propagating':
if popped_event.event_type != 'FAST' and popped_event.packet.type != 'message':
popped_event.flow.time_packet_last_seen = popped_event.time
if popped_event.finishTrip() != 0:
if popped_event.flow.con_ctrl == 1:
if popped_event.flow.ss_threshold < popped_event.flow.window:
popped_event.flow.window += 1./popped_event.flow.window
else:
popped_event.flow.window += 1.
#update the packet delay for the flow when the RTT is finished
popped_event.flow.packet_delay = (popped_event.time \
- popped_event.packet.start_time) * 1000
#add the time point to the flow's packet delay history
if popped_event.event_type != 'FAST' and popped_event.packet.type != 'message':
popped_event.flow.packet_delay_history.append((popped_event.time,
popped_event.flow.packet_delay))
#decrement the flow occupancy after packet finishes
popped_event.flow.occupancy -= 1
#if the last packet finishes, skip to the next event on queue
if (popped_event.flow.current_packet
== len(popped_event.flow.packets) + 1):
continue
#get the first link of the flow to potentially add packets to
first_link = popped_event.flow.findFirstLink(links)
first_link_filled = 1
#while the occupancy has not reached the window, add packets to
#the first link if the link capacity is not met
while popped_event.flow.occupancy < \
int(floor(popped_event.flow.window)) and first_link_filled != 0:
first_link_filled = popped_event.flow.addPacket(event_queue,
first_link, popped_event.time)
#check whether the packet has reached its destination
elif popped_event.reachedDestination() != 0:
#create an acknowledgement packet based on the popped packet's info
ack = packet(popped_event.packet.id, popped_event.flow.src,
popped_event.flow.dest, 'ack', data_ack_size,
popped_event.flow.dest)
#update the route and current router of this new acknowledgment packet
ack.route_index = popped_event.packet.route_index
ack.route = popped_event.packet.route
ack.route.append(popped_event.flow.dest.id)
ack.current_router = popped_event.flow.dest
#make sure the start time of this ack is the corresponding packet's
ack.start_time = popped_event.packet.start_time
#add the buffering event of the new acknowledgement to the queue
popped_event.link.addToBuffer(event_queue, popped_event.time, ack,
popped_event.flow)
# check if the message has gotten to the router it was trying to reach
elif popped_event.messageReceived(popped_event.flow, links) != 0:
# For every router, call the updateTable function with the
# packets dictionary of paths taken
for router in routers:
router.updateTable(popped_event.packet.source, popped_event.packet.destDict)
#check if the packet is a packet routing in the middle of the flow
elif popped_event.packet.type == 'packet':
#route the packet and update the event queue
popped_event.routePacket(links, event_queue)
#check if the packet is an acknowledgement and is not at the flow source
elif popped_event.packet.type == 'ack' and \
popped_event.packet.route_index > 0:
#route the acknowledgement and update the event queue
popped_event.routeAcknowledgement(links, event_queue)
# Check if the packet is a message routing in the middle of the flow
elif popped_event.packet.type == 'message':
# Route the message and update the event queue
popped_event.routeMessage(links, event_queue)
#get more time points for each link's occupancy and packet drops
for i in range(len(links)):
if popped_event.event_type != 'FAST' and popped_event.packet.type != 'message':
links[i].buffer_occupancy_history.append((popped_event.time,
links[i].buffer_occupancy))
links[i].packet_drops_history.append((popped_event.time,
links[i].packet_drops))
legend = []
for i in range(len(flows)):
#graph each flow's rate over time
x,y = zip(*flows[i].flow_rate_history)
x_ = []
y_ = []
for e in range(0,len(x) - weight, weight):
x_.append(sum(x[e:e+ weight]) / weight)
y_.append(sum(y[e:e+ weight]) / weight)
plt.plot(x_,y_)
legend.append("Flow " + str(i + 1))
plt.legend(legend, loc='best')
plt.title('Flow Rate over Time')
plt.ylabel('Flow Rate')
plt.xlabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
for i in range(len(flows)):
#graph each flow's window size over time
x,y = zip(*flows[i].window_history)
plt.plot(x,y)
plt.legend(legend, loc='best')
plt.title('Flow Window Size over Time')
plt.ylabel('Window Size')
plt.ylabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
for i in range(len(flows)):
#graph each flow's packet delay over time
x,y = zip(*flows[i].packet_delay_history)
x_ = []
y_ = []
for e in range(0,len(x) - weight, weight):
x_.append(sum(x[e:e+ weight]) / weight)
y_.append(sum(y[e:e+ weight]) / weight)
plt.plot(x_,y_)
plt.legend(legend, loc='best')
plt.title('Flow Packet Delay over Time')
plt.ylabel('Packet Delay')
plt.ylabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
legend = []
for i in range(len(links)):
#graph each link's rate over time
if len(links[i].link_rate_history) > 0:
x,y = zip(*links[i].link_rate_history)
x_ = []
y_ = []
for e in range(0,len(x) - weight, weight):
x_.append(sum(x[e:e+ weight]) / weight)
y_.append(sum(y[e:e+ weight]) / weight)
plt.plot(x_,y_)
legend.append("Link " + str(i))
plt.legend(legend, loc='best')
plt.title('Link Rate over Time')
plt.ylabel('Link Rate')
plt.xlabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
for i in range(len(links)):
#graph each link's buffer occupancy over time
if len(links[i].buffer_occupancy_history) > 0:
x, y = zip(*links[i].buffer_occupancy_history)
x_ = []
y_ = []
for e in range(0,len(x) - 2 * weight, 2 * weight):
x_.append(sum(x[e:e+ 2 * weight]) / (2 * weight))
y_.append(sum(y[e:e+ 2 * weight]) / (2 * weight))
plt.plot(x_,y_)
plt.legend(legend, loc='best')
plt.title('Link Buffer Occupancy over Time')
plt.ylabel('Buffer Occupancy')
plt.xlabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
for i in range(len(links)):
#graph each link's packet drop count over time
if len(links[i].packet_drops_history) > 0:
x, y = zip(*links[i].packet_drops_history)
plt.plot(x,y)
plt.legend(legend, loc='best')
plt.title('Link Packet Drop Count')
plt.ylabel('Packet Drop Count')
plt.ylabel('Time')
plt.gca().set_ylim(bottom=0)
plt.show()
def timer(self):
if self.nodelay:
self.executebuffer(self.buffer)
else:
timerevent = eventqueue.event(.01667, self.postbuffer, self.buffer)
self.eventqueue.add(timerevent)
def addPacket(self, event_queue, first_link, time):
#check that there are still packets in the flow that need to be sent
if self.current_packet <= len(self.packets):
#access the next packet in the flow
new_packet = self.packets[self.current_packet - 1]
#check whether the buffer capacity allows for the packet to be added
if first_link.buffer_occupancy + new_packet.size \
< first_link.buffer_capacity * 1000:
#set this next packet's start time
new_packet.start_time = time
#calculate the transmission time for the first link of the new packet
transmission = first_link.getTransmission(new_packet)
#if there is delay from other packets in the buffer, delay this packet
if len(first_link.buffer_elements) > 0:
#get the last packet on the first link's buffer
last = first_link.buffer_elements[
len(first_link.buffer_elements) - 1]
#check whether there is half-duplex congestion
switch = last.current_router != new_packet.current_router \
and last.type == 'packet'
#if there is no half-duplex congestion, add transmission accordingly
if switch == 0:
if time < last.current_time:
#set the current time of the new packet with link congestion
new_packet.current_time = last.current_time + transmission
else:
#set the current time of the new packet without link congestion
new_packet.current_time = time + transmission
#if there is half-duplex congestion, also add propagation accordingly
else:
if time < last.current_time:
#set the current time of the new packet with link congestion
new_packet.current_time = last.current_time + transmission + \
first_link.rate / 1000.
else:
#set the current time of the new packet without link congestion
new_packet.current_time = time + transmission
else:
new_packet.current_time = time + transmission
#insert the buffering event for the new packet on the first link
event_queue.insert_event(
event('Buffering', new_packet.current_time, new_packet,
first_link, self))
#append the packet to the link buffer and increase buffer occupancy
first_link.buffer_occupancy += new_packet.size
first_link.buffer_elements.append(new_packet)
#update the current packet (not in the flow) and increment occupancy
self.current_packet += 1
self.occupancy += 1
#return 1 to indicate that a packet was added to the buffer
return 1
#return 0 to indicate that the buffer was too full to add a packet
else:
return 0
#return 0 to indicate that there are no more packets for the flow
else:
return 0