def main(_):
action_reward_dict = {}
policy_sess = tf.Session()
#K.set_session(policy_sess)
manager = NetworkManager(FLAGS, clip_rewards=CLIP_REWARDS)
size = [len(LAYER_SIZES)] * NUM_LAYERS
reward_space = np.zeros((size))
#print(reward_space.shape)
for i in range(NUM_ENUM):
for idx, val in np.ndenumerate(reward_space):
action = [LAYER_SIZES[i] for i in idx]
#print(action)
with policy_sess.as_default():
_, acc = manager.get_rewards(model_fn, action)
print(action, acc)
acc = round(acc * JSON_SCALE, 2)
action = str(tuple(action))
if action not in action_reward_dict:
action_reward_dict[action] = [acc]
else:
action_reward_dict[action].append(acc)
action_average_reward_dict = {}
for k in action_reward_dict.keys():
action_average_reward_dict[k] = round(np.mean(action_reward_dict[k]),
2)
with open('action_reward_dict.json', 'w') as f:
json.dump(action_reward_dict, f)
f.close()
with open('action_average_reward_dict.json', 'w') as f:
json.dump(action_average_reward_dict, f)
f.close()
def main(_):
CLIP_REWARDS = False
value = [30,60,100,144]
state_space = [v for v in itertools.product(value, repeat=3)]
data = defaultdict(list)
for itr in xrange(500):
for state in state_space:
states = list(state)
manager = NetworkManager(FLAGS, clip_rewards=CLIP_REWARDS)
reward, previous_acc = manager.get_rewards(model_fn, states)
previous_acc = round(previous_acc*100000,2)
data[str(state)].append(previous_acc)
with open('data.json', 'w') as outfile:
json.dump(data, outfile)
def main(_):
CLIP_REWARDS = False
#filter_space = [v for v in itertools.product(filter_val, repeat=3)]
#stride_space = [v for v in itertools.product(stride_val, repeat=3)]
#kernal_space = [v for v in itertools.product(kernel_val, repeat=3)]
data = defaultdict(list)
with open('result.txt', 'w') as out:
for state in states:
print(state)
manager = NetworkManager(FLAGS, clip_rewards=CLIP_REWARDS)
reward, previous_acc = manager.get_rewards(model_fn_cnn, state)
previous_acc = round(previous_acc * 100000, 2)
print(previous_acc)
data[str(state)].append(previous_acc)
out.write("{} {}\n".format(state, previous_acc))
with open('data.json', 'w') as outfile:
json.dump(data, outfile)
def main(_):
CLIP_REWARDS = False
data = defaultdict(list)
with open('main_result.txt', 'w') as out:
for ite in xrange(3):
ite += 1
print('outter iteration:', ite)
iteration = 0
for state in states:
iteration += 1
print(iteration, state)
manager = NetworkManager(FLAGS, clip_rewards=CLIP_REWARDS)
reward, previous_acc = manager.get_rewards(model_fn_cnn, state)
previous_acc = round(previous_acc * 100000, 2)
print(previous_acc)
data[str(state)].append(previous_acc)
out.write("{} {}\n".format(state, previous_acc))
with open('main_sample.json', 'w') as outfile:
json.dump(data, outfile)
def main(_):
# create a shared session between Keras and Tensorflow
policy_sess = tf.Session()
K.set_session(policy_sess)
NUM_LAYERS = 3 # number of layers of the state space
MAX_TRIALS = 250 # maximum number of models generated
MAX_EPOCHS = 60 # maximum number of epochs to train
BATCHSIZE = 100 # batchsize
EXPLORATION = 0.5 # high exploration for the first 1000 steps
REGULARIZATION = 1e-3 # regularization strength
CONTROLLER_CELLS = 32 # number of cells in RNN controller
CLIP_REWARDS = False # clip rewards in the [-0.05, 0.05] range
RESTORE_CONTROLLER = True # restore controller to continue training
# construct a state space
state_space = StateSpace()
# add states
#state_space.add_state(name='kernel', values=[3])
state_space.add_state(name='filters', values=[30, 60, 100, 144])
#state_space.add_state(name='stride', values=[1])
# print the state space being searched
state_space.print_state_space()
previous_acc = 0.0
total_reward = 0.0
with policy_sess.as_default():
# create the Controller and build the internal policy network
controller = Controller(policy_sess,
NUM_LAYERS,
state_space,
reg_param=REGULARIZATION,
exploration=EXPLORATION,
controller_cells=CONTROLLER_CELLS,
restore_controller=RESTORE_CONTROLLER)
print('done')
# create the Network Manager
manager = NetworkManager(FLAGS, clip_rewards=CLIP_REWARDS)
# get an initial random state space if controller needs to predict an
# action from the initial state
state = state_space.get_random_state_space(NUM_LAYERS)
print("Initial Random State : ", state_space.parse_state_space_list(state))
#print()
# train for number of trails
for trial in range(MAX_TRIALS):
with policy_sess.as_default():
actions = controller.get_action(
state) # get an action for the previous state
# print the action probabilities
state_space.print_actions(actions)
print("Predicted actions : ",
state_space.parse_state_space_list(actions))
# build a model, train and get reward and accuracy from the network manager
reward, previous_acc = manager.get_rewards(
model_fn_cnn, state_space.parse_state_space_list(actions))
print("Rewards : ", reward, "Accuracy : ", previous_acc)
with policy_sess.as_default():
total_reward += reward
print("Total reward : ", total_reward)
# actions and states are equivalent, save the state and reward
state = actions
controller.store_rollout(state, reward)
# train the controller on the saved state and the discounted rewards
loss = controller.train_step()
print("Trial %d: Controller loss : %0.6f" % (trial + 1, loss))
# write the results of this trial into a file
with open('train_history.csv', mode='a+') as f:
data = [previous_acc, reward]
data.extend(state_space.parse_state_space_list(state))
writer = csv.writer(f)
writer.writerow(data)
print()
print("Total Reward : ", total_reward)
with policy_sess.as_default():
# create the Controller and build the internal policy network
controller = Controller(policy_sess,
NUM_LAYERS,
state_space,
reg_param=REGULARIZATION,
exploration=EXPLORATION,
controller_cells=CONTROLLER_CELLS,
embedding_dim=EMBEDDING_DIM,
restore_controller=RESTORE_CONTROLLER)
# create the Network Manager
manager = NetworkManager(dataset,
epochs=MAX_EPOCHS,
child_batchsize=CHILD_BATCHSIZE,
clip_rewards=CLIP_REWARDS,
acc_beta=ACCURACY_BETA)
# get an initial random state space if controller needs to predict an
# action from the initial state
state = state_space.get_random_state_space(NUM_LAYERS)
print("Initial Random State : ", state_space.parse_state_space_list(state))
print()
# clear the previous files
controller.remove_files()
# train for number of trails
for trial in range(MAX_TRIALS):
with policy_sess.as_default():
dataset = [x_train, y_train, x_test,
y_test] # pack the dataset for the NetworkManager
with policy_sess.as_default():
# create the Encoder and build the internal policy network
controller = Encoder(policy_sess,
state_space,
B=B,
K=K_,
train_iterations=RNN_TRAINING_EPOCHS,
reg_param=REGULARIZATION,
controller_cells=CONTROLLER_CELLS,
restore_controller=RESTORE_CONTROLLER)
# create the Network Manager
manager = NetworkManager(dataset, epochs=MAX_EPOCHS, batchsize=BATCHSIZE)
print()
# train for number of trails
for trial in range(B):
with policy_sess.as_default():
K.set_session(policy_sess)
if trial == 0:
k = None
else:
k = K_
actions = controller.get_actions(
top_k=k) # get all actions for the previous state
dataset = [x_train, y_train, x_test, y_test] # pack the dataset for the NetworkManager
previous_acc = 0.0
total_reward = 0.0
with policy_sess.as_default():
# create the Controller and build the internal policy network
controller = Controller(policy_sess, NUM_LAYERS, state_space,
reg_param=REGULARIZATION,
exploration=EXPLORATION,
controller_cells=CONTROLLER_CELLS,
restore_controller=RESTORE_CONTROLLER)
# create the Network Manager
manager = NetworkManager(dataset, epochs=MAX_EPOCHS, batchsize=BATCHSIZE, clip_rewards=CLIP_REWARDS)
# get an initial random state space if controller needs to predict an
# action from the initial state
state = state_space.get_random_state_space(NUM_LAYERS)
print("Initial Random State : ", state_space.parse_state_space_list(state))
print()
# train for number of trails
for trial in range(MAX_TRIALS):
with policy_sess.as_default():
K.set_session(policy_sess)
actions = controller.get_action(state) # get an action for the previous state
# print the action probabilities
state_space.print_actions(actions)
with policy_sess.as_default():
# create the Controller and build the internal policy network
controller = Controller(policy_sess,
NUM_LAYERS,
state_space,
reg_param=REGULARIZATION,
exploration=EXPLORATION,
controller_cells=CONTROLLER_CELLS,
embedding_dim=EMBEDDING_DIM,
restore_controller=RESTORE_CONTROLLER)
# create the Network Manager
manager = NetworkManager(dataset,
epochs=MAX_EPOCHS,
child_batchsize=CHILD_BATCHSIZE,
clip_rewards=CLIP_REWARDS,
acc_beta=ACCURACY_BETA)
# get an initial random state space if controller needs to predict an
# action from the initial state
state = state_space.get_random_state_space(NUM_LAYERS)
print("Initial Random State : ", state_space.parse_state_space_list(state))
print()
# clear the previous files
controller.remove_files()
old_acc = 0.7
ite_count = 0
acc_list = []
sip.setapi(name, API_VERSION)
from PyQt4 import QtCore
from info import ServerInfo
from manager import NetworkManager
from transfer_thread import TransferThread
global ess
ess = ServerInfo()
ess.fromAddr('misura.expertsystemsolutions.it:80')
ess.name = 'Simulation Server'
ess.host = 'misura.expertsystemsolutions.it'
ess.serial = 'ESimulServ'
manager = NetworkManager()
def simpleConnection(addr, user='', password='', mac='', save=True):
addr = str(addr)
logging.debug('simpleConnection', addr, user, password, mac, save)
if '@' in addr:
usrpass, addr = addr.replace('https://', '').split('@')
addr = addr.lower().replace('/rpc', '')
if ':' in usrpass:
user, password = usrpass.split(':')
else:
user = usrpass
if not addr.startswith('https'):
addr = 'https://' + addr
if not addr.endswith('/RPC'):
def main_training():
# Number of GPUs available. Use 0 for CPU mode.
ngpu = torch.cuda.device_count()
# Decide which device we want to run on
device = torch.device("cuda:0" if (
torch.cuda.is_available() and ngpu > 0) else "cpu")
train_loader, val_loader, test_loader = get_dataset(
CHILD_BATCHSIZE, '/tmp/datasets/cifar-10')
dataloaders = [train_loader, val_loader, test_loader]
# construct a state space
state_space = StateSpace()
# add states
state_space.add_state(name='kernel', values=[1, 3, 5, 7])
state_space.add_state(name='filters', values=[24, 36, 48, 64])
previous_acc = 0.0
total_reward = 0.0
controller = Controller(NUM_LAYERS, state_space)
manager = NetworkManager(dataloaders, device, epochs=2)
# get an initial random state space if controller needs to predict an action from the initial state
state = state_space.get_random_state_space(NUM_LAYERS)
print("Initial Random State : ", state_space.parse_state_space_list(state))
for trial in range(MAX_TRIALS):
actions, prob_actions = controller.get_action(
state) # get an action for the previous state
# print the action probabilities
state_space.print_actions(actions)
print("Predicted actions : ",
state_space.parse_state_space_list(actions))
# build a model, train and get reward and accuracy from the network manager
model = ChildNetwork(state_space.parse_state_space_list(actions))
reward, previous_acc = manager.get_rewards(model)
print("Rewards : ", reward, "Accuracy : ", previous_acc)
total_reward += reward
print("Total reward : ", total_reward)
# actions and states are equivalent, save the state and reward
state = actions
prob_state = prob_actions
controller.store_rollout(state, reward, prob_state)
# train the controller on the saved state and the discounted rewards
loss = controller.update_policy()
print("Trial %d: Controller loss : %0.6f" % (trial + 1, loss))
# write the results of this trial into a file
with open('train_history.csv', mode='a+') as f:
data = [previous_acc, reward]
data.extend(state_space.parse_state_space_list(state))
writer = csv.writer(f)
writer.writerow(data)
print()
print("Total Reward : ", total_reward)
with policy_sess.as_default():
# create the Encoder and build the internal policy network
controller = Encoder(policy_sess,
state_space,
B=B,
K=K_,
train_iterations=RNN_TRAINING_EPOCHS,
reg_param=REGULARIZATION,
controller_cells=CONTROLLER_CELLS,
restore_controller=RESTORE_CONTROLLER)
# create the Network Manager
manager = NetworkManager(dataset,
epochs=MAX_EPOCHS,
batchsize=BATCHSIZE,
cell_number=NORMAL_CELL_NUMBER,
filters=FIRST_LAYER_FILTERS)
print()
#reset all shared weights
if os.path.isdir('shared_weights') and CLEAR_SHARE_WEIGHTS:
shutil.rmtree('shared_weights')
# train for number of trails
for trial in range(B):
with policy_sess.as_default():
K.set_session(policy_sess)
if trial == 0:
k = None
def __init__(self, *args, **kwargs):
self.state = ToolState.EMPTY
# dimensions of the area where nodes can be placed
self.workAreaXDim = 720
self.workAreaYDim = 520
self.workAreaXPos = 40
self.workAreaYPos = 40
# multiple choice bubble menu for node addition
self.defaultBubble = Bubble(size_hint=(None, None), size=(250, 40))
self.newComputerButton = Button(text="Nowy komputer")
self.newRouterButton = Button(text="Nowy router")
self.defaultBubble.add_widget(self.newComputerButton)
self.defaultBubble.add_widget(self.newRouterButton)
# coordinates of node-button to be placed
self.pendingNodePosX = 0
self.pendingNodePosY = 0
# multiple choice bubble menu for node editing/removal
self.nodeBubble = Bubble(size_hint=(None, None), size=(250, 40))
self.newConnButton = Button(text="Dodaj połączenie")
self.deleteNodeButton = Button(text="Usuń węzeł")
self.nodeBubble.add_widget(self.newConnButton)
self.nodeBubble.add_widget(self.deleteNodeButton)
# reference to button being edited/removed
self.pendingNodeRef = Button()
# global state - adding connection or not
self.isInConnectionMode = False
self.netManager = NetworkManager(self.__deleteLine, self.appendLog,
self.animatePacket)
# define widgets of side panel and add them to window
self.sidePanelTabbedPanel = TabbedPanel(do_default_tab=False,
size=(200, 600),
pos=(800, 0),
background_color=(0, 0, 0, 0),
tab_width=100)
# define widgets of log display and add them to window
self.sidePanelLogTab = TabbedPanelItem(text="Symulacja")
self.sidePanelLogLayout = GridLayout(cols=1,
spacing=10,
size_hint=(None, None),
size=(200, 550))
self.logField = TextInput(padding=10,
readonly=True,
size_hint_max_y=None,
size_hint=(None, None),
size=(190, 550))
self.logField.bind(minimum_height=self.logField.setter('height'))
self.sidePanelLogScrollView = ScrollView(
size_hint=(None, None),
size=(200, 450),
scroll_type=['bars'],
bar_width=10,
bar_inactive_color=[1, 1, 1, .8],
bar_color=[0.3, 1, 1, 1])
self.sidePanelLogScrollView.add_widget(self.logField)
self.sidePanelLogLayout.add_widget(self.sidePanelLogScrollView)
self.sidePanelLogTab.add_widget(self.sidePanelLogLayout)
self.sidePanelTabbedPanel.add_widget(self.sidePanelLogTab)
# A welcome log
self.appendLog('A new workspace has been created.')
# define widgets of node edit panel to be added to window later
# when a node is selected, add sidePanelNodeTab with the proper tab name to sidePanelTabbedPanel
# and fill sidePanelNodeTab with content - interface configuration form
self.sidePanelNodeTab = TabbedPanelItem()
self.sidePanelNodeLayout = GridLayout(cols=1,
spacing=5,
size_hint=(None, None),
size=(200, 550))
self.sidePanelNodeLayout.bind(
minimum_height=self.sidePanelNodeLayout.setter('height'))
self.sidePanelNodeScrollView = ScrollView(
size_hint=(None, None),
size=(200, 550),
pos=(800, 0),
scroll_type=['bars'],
bar_width=10,
bar_inactive_color=[1, 1, 1, .8],
bar_color=[1, 1, 1, 1])
self.sidePanelNodeTab.add_widget(self.sidePanelNodeScrollView)
self.sidePanelNodeScrollView.add_widget(self.sidePanelNodeLayout)
# define simulation-related buttons and add them to the window
self.toggleSimButton = Button(text="Symuluj")
self.addPacketButton = Button(text="Nowy pakiet",
on_press=self.on_new_packet)
self.stepSimButton = Button(text=">|", on_press=self.on_step)
self.playSimButton = Button(text=">")
self.simButtonLayout = BoxLayout()
self.sidePanelLogLayout.add_widget(self.addPacketButton)
self.sidePanelLogLayout.add_widget(self.toggleSimButton)
self.sidePanelLogLayout.add_widget(self.simButtonLayout)
self.simButtonLayout.add_widget(self.playSimButton)
self.simButtonLayout.add_widget(self.stepSimButton)
self.packetNodes = []
# set window color
Window.clearcolor = (0.7, 0.7, 0.7, 1)
super(Demoer, self).__init__(*args, **kwargs)
self.add_widget(self.sidePanelTabbedPanel)
# create delimiter line around the work area
self.canvas.before.add(
Line(rectangle=(self.workAreaXPos, self.workAreaYPos,
self.workAreaXDim, self.workAreaYDim),
width=0.5))
# create background square behind side panel
with self.canvas.before:
Color(0.3, 0.3, 0.3, 1)
Rectangle(pos=(2 * self.workAreaXPos + self.workAreaXDim, 0),
size=(200, 600))
Color(1, 1, 1, 1)
class Demoer(FloatLayout):
def __init__(self, *args, **kwargs):
self.state = ToolState.EMPTY
# dimensions of the area where nodes can be placed
self.workAreaXDim = 720
self.workAreaYDim = 520
self.workAreaXPos = 40
self.workAreaYPos = 40
# multiple choice bubble menu for node addition
self.defaultBubble = Bubble(size_hint=(None, None), size=(250, 40))
self.newComputerButton = Button(text="Nowy komputer")
self.newRouterButton = Button(text="Nowy router")
self.defaultBubble.add_widget(self.newComputerButton)
self.defaultBubble.add_widget(self.newRouterButton)
# coordinates of node-button to be placed
self.pendingNodePosX = 0
self.pendingNodePosY = 0
# multiple choice bubble menu for node editing/removal
self.nodeBubble = Bubble(size_hint=(None, None), size=(250, 40))
self.newConnButton = Button(text="Dodaj połączenie")
self.deleteNodeButton = Button(text="Usuń węzeł")
self.nodeBubble.add_widget(self.newConnButton)
self.nodeBubble.add_widget(self.deleteNodeButton)
# reference to button being edited/removed
self.pendingNodeRef = Button()
# global state - adding connection or not
self.isInConnectionMode = False
self.netManager = NetworkManager(self.__deleteLine, self.appendLog,
self.animatePacket)
# define widgets of side panel and add them to window
self.sidePanelTabbedPanel = TabbedPanel(do_default_tab=False,
size=(200, 600),
pos=(800, 0),
background_color=(0, 0, 0, 0),
tab_width=100)
# define widgets of log display and add them to window
self.sidePanelLogTab = TabbedPanelItem(text="Symulacja")
self.sidePanelLogLayout = GridLayout(cols=1,
spacing=10,
size_hint=(None, None),
size=(200, 550))
self.logField = TextInput(padding=10,
readonly=True,
size_hint_max_y=None,
size_hint=(None, None),
size=(190, 550))
self.logField.bind(minimum_height=self.logField.setter('height'))
self.sidePanelLogScrollView = ScrollView(
size_hint=(None, None),
size=(200, 450),
scroll_type=['bars'],
bar_width=10,
bar_inactive_color=[1, 1, 1, .8],
bar_color=[0.3, 1, 1, 1])
self.sidePanelLogScrollView.add_widget(self.logField)
self.sidePanelLogLayout.add_widget(self.sidePanelLogScrollView)
self.sidePanelLogTab.add_widget(self.sidePanelLogLayout)
self.sidePanelTabbedPanel.add_widget(self.sidePanelLogTab)
# A welcome log
self.appendLog('A new workspace has been created.')
# define widgets of node edit panel to be added to window later
# when a node is selected, add sidePanelNodeTab with the proper tab name to sidePanelTabbedPanel
# and fill sidePanelNodeTab with content - interface configuration form
self.sidePanelNodeTab = TabbedPanelItem()
self.sidePanelNodeLayout = GridLayout(cols=1,
spacing=5,
size_hint=(None, None),
size=(200, 550))
self.sidePanelNodeLayout.bind(
minimum_height=self.sidePanelNodeLayout.setter('height'))
self.sidePanelNodeScrollView = ScrollView(
size_hint=(None, None),
size=(200, 550),
pos=(800, 0),
scroll_type=['bars'],
bar_width=10,
bar_inactive_color=[1, 1, 1, .8],
bar_color=[1, 1, 1, 1])
self.sidePanelNodeTab.add_widget(self.sidePanelNodeScrollView)
self.sidePanelNodeScrollView.add_widget(self.sidePanelNodeLayout)
# define simulation-related buttons and add them to the window
self.toggleSimButton = Button(text="Symuluj")
self.addPacketButton = Button(text="Nowy pakiet",
on_press=self.on_new_packet)
self.stepSimButton = Button(text=">|", on_press=self.on_step)
self.playSimButton = Button(text=">")
self.simButtonLayout = BoxLayout()
self.sidePanelLogLayout.add_widget(self.addPacketButton)
self.sidePanelLogLayout.add_widget(self.toggleSimButton)
self.sidePanelLogLayout.add_widget(self.simButtonLayout)
self.simButtonLayout.add_widget(self.playSimButton)
self.simButtonLayout.add_widget(self.stepSimButton)
self.packetNodes = []
# set window color
Window.clearcolor = (0.7, 0.7, 0.7, 1)
super(Demoer, self).__init__(*args, **kwargs)
self.add_widget(self.sidePanelTabbedPanel)
# create delimiter line around the work area
self.canvas.before.add(
Line(rectangle=(self.workAreaXPos, self.workAreaYPos,
self.workAreaXDim, self.workAreaYDim),
width=0.5))
# create background square behind side panel
with self.canvas.before:
Color(0.3, 0.3, 0.3, 1)
Rectangle(pos=(2 * self.workAreaXPos + self.workAreaXDim, 0),
size=(200, 600))
Color(1, 1, 1, 1)
# switch between placing connection or editing nodes
def toggleConnectionMode(self, instance):
self.clearBubbles()
if not self.isInConnectionMode:
self.pendingNodeRef.background_color = (1, 0.5, 0.5, 1)
self.isInConnectionMode = True
else:
self.pendingNodeRef.background_color = (1, 1, 1, 1)
self.isInConnectionMode = False
# display node property edit tab
def showNodeEditPanel(self, instance):
node_name = self.netManager.getNodeName(instance)
self.sidePanelNodeTab.text = node_name
if len(self.sidePanelTabbedPanel.children) == 2:
self.sidePanelTabbedPanel.add_widget(self.sidePanelNodeTab)
self.sidePanelTabbedPanel.switch_to(self.sidePanelNodeTab)
self.sidePanelNodeLayout.clear_widgets()
self.sidePanelNodeLayout.add_widget(
Label(text="Interfejsy węzła " + node_name + ":",
size_hint=(1, None)))
interfaces = self.netManager.getNodeInterfaces(instance)
for label, (ip, conn) in interfaces.items():
self.sidePanelNodeLayout.add_widget(
Label(text=label, size_hint=(1, None), height=30))
input_field = TextInput(text=ip,
multiline=False,
size_hint=(1, None),
height=30)
cb = functools.partial(self.on_node_edit, conn, instance)
input_field.bind(on_text_validate=cb)
self.sidePanelNodeLayout.add_widget(input_field)
# remove active bubble menus
def clearBubbles(self):
self.remove_widget(self.nodeBubble)
self.remove_widget(self.defaultBubble)
def __deleteLine(self, line):
self.canvas.before.remove(line)
# remove active node after clicking in bubble menu
def deleteNode(self, instance):
self.clearBubbles()
self.remove_widget(self.pendingNodeRef)
self.netManager.deleteNode(self.pendingNodeRef)
if self.sidePanelNodeTab.text == self.pendingNodeRef.children[0].text:
self.sidePanelTabbedPanel.remove_widget(self.sidePanelNodeTab)
self.sidePanelTabbedPanel.switch_to(self.sidePanelLogTab)
# add pending node as computer after clicking in bubble menu
def addComputer(self, instance):
nodeButton = Button(pos=(self.pendingNodePosY, self.pendingNodePosX),
size_hint=(None, None),
size=(40, 40))
nodeImg = Image(source="Images/computer.png")
nodeLabel = Label()
nodeButton.add_widget(nodeImg)
nodeButton.add_widget(nodeLabel)
nodeImg.center_x = nodeImg.parent.center_x
nodeImg.center_y = nodeImg.parent.center_y + 10
nodeLabel.center_x = nodeLabel.parent.center_x
nodeLabel.center_y = nodeLabel.parent.center_y - 10
nodeButton.bind(on_press=self.showNodeBubble)
self.add_widget(nodeButton)
self.netManager.addComputer(nodeButton)
nodeLabel.text = self.netManager.getNodeName(nodeButton)
# add pending node as router after clicking in bubble menu
def addRouter(self, instance):
nodeButton = Button(pos=(self.pendingNodePosY, self.pendingNodePosX),
size_hint=(None, None),
size=(40, 40))
nodeImg = Image(source="Images/router.png")
nodeLabel = Label()
nodeButton.add_widget(nodeImg)
nodeButton.add_widget(nodeLabel)
nodeImg.center_x = nodeImg.parent.center_x
nodeImg.center_y = nodeImg.parent.center_y + 10
nodeLabel.center_x = nodeLabel.parent.center_x
nodeLabel.center_y = nodeLabel.parent.center_y - 10
nodeButton.bind(on_press=self.showNodeBubble)
self.add_widget(nodeButton)
self.netManager.addRouter(nodeButton)
nodeLabel.text = self.netManager.getNodeName(nodeButton)
# show bubble menu on click on node
# OR create connection between active node and clicked node when in connection mode
# OR select nodes for packet transmission
def showNodeBubble(self, instance):
if self.state == ToolState.SELECTING_PACKET_NODE1:
self.packetNodes.append(instance)
instance.background_color = (1, 0.5, 0.5, 1)
self.state = ToolState.SELECTING_PACKET_NODE2
elif self.state == ToolState.SELECTING_PACKET_NODE2:
packetNodes = self.packetNodes.copy()
packetNodes.append(instance)
try:
self.netManager.preparePacket(tuple(packetNodes))
except DemoerException as e:
self.showPopup('Error', e.message)
else:
instance.background_color = (1, 0.5, 0.5, 1)
self.state = ToolState.EMPTY
self.packetNodes.clear()
else:
if not self.isInConnectionMode:
self.showNodeEditPanel(instance)
self.clearBubbles()
self.pendingNodeRef = instance
self.nodeBubble.pos = (instance.pos[0] - 105,
instance.pos[1] + 40)
self.deleteNodeButton.bind(on_press=self.deleteNode)
self.newConnButton.bind(on_press=self.toggleConnectionMode)
self.add_widget(self.nodeBubble)
else:
try:
connection = self.netManager.addConnection(
instance, self.pendingNodeRef)
except DemoerException as e:
self.showPopup('Error', e.message)
else:
line = Line(points=[
self.pendingNodeRef.pos[0] + 20,
self.pendingNodeRef.pos[1] + 20, instance.pos[0] + 20,
instance.pos[1] + 20
],
width=2)
self.canvas.before.add(line)
connection.setArg(line)
self.toggleConnectionMode(Button())
# show bubble on right-clicking canvas
def showDefaultBubble(self, posx, posy):
self.clearBubbles()
self.pendingNodePosY = posx - 20
self.pendingNodePosX = posy - 20
self.defaultBubble.pos = (posx - 125, posy)
self.newComputerButton.bind(on_press=self.addComputer)
self.newRouterButton.bind(on_press=self.addRouter)
self.add_widget(self.defaultBubble)
def animatePacket(self, sourceWidget, targetWidget):
print('Animating')
img = Image(source='Images/packet.png', pos=sourceWidget.pos)
anim = Animation(x=targetWidget.pos[0], y=targetWidget.pos[1])
self.add_widget(img)
anim.start(img)
def showPopup(self, title, content):
popup = Popup(title=title,
content=Label(text=content),
size_hint=(None, None),
size=(500, 200))
popup.open()
def appendLog(self, content):
self.logField.text = self.logField.text + content + '\n\n'
def on_step(self, instance):
try:
self.netManager.stepSimulation()
except DemoerException as e:
self.showPopup('Error', e.message)
def on_new_packet(self, instance):
if not self.state == ToolState.EMPTY:
return
self.state = ToolState.SELECTING_PACKET_NODE1
def on_node_edit(self, connection, node_button, text_input):
try:
self.netManager.setAddress(node_button, connection,
text_input.text)
except DemoerException as e:
self.showPopup('Error', e.message)
def on_touch_down(self, touch, after=False):
if after:
self.remove_widget(self.defaultBubble)
if self.workAreaXPos + self.workAreaXDim > touch.pos[0] > self.workAreaXPos \
and self.workAreaYPos + self.workAreaYDim > touch.pos[1] > self.workAreaYPos:
if touch.button == "right":
if self.state != ToolState.EMPTY:
self.state = ToolState.EMPTY
self.packetNodes.clear()
else:
if not self.isInConnectionMode:
self.showDefaultBubble(touch.x, touch.y)
else:
self.toggleConnectionMode(Button())
else:
Clock.schedule_once(lambda dt: self.on_touch_down(touch, True),
0.01)
return super(Demoer, self).on_touch_down(touch)
