feat(Controller): Allow controlling for output node

CooLearning · Apr 4, 2022 · 7d9db58 · 7d9db58
1 parent a41ee72
commit 7d9db58
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/app/devices/controller.device.ts b/src/app/devices/controller.device.ts
@@ -245,7 +245,15 @@ controllerDevice.attachButtonsToNeuron = function(): void {
  * @param {number} selectedNode - The selected node.
  */
 controllerDevice.attachControlsToNeuron = function(selectedNode: number): void {
-  const {neuron} = networkState.getNeuron(selectedNode);
+  let neuron;
+
+  if (networkState.isOutputNode(selectedNode)) {
+    neuron = networkState.getOutputNode();
+  }
+  else {
+    neuron = networkState.getNeuron(selectedNode).neuron;
+  }
+
   const links = neuron.inputLinks;
 
   // first draw

diff --git a/src/app/state/network.state.ts b/src/app/state/network.state.ts
@@ -191,7 +191,12 @@ networkState.getSelectedNeurons = function() {
     return;
   }
   else if (selectedNodes.length === 1) {
-    targets = [this.getNeuron(selectedNodes[0]).neuron];
+    if (this.isOutputNode(selectedNodes[0])) {
+      targets = [this.getOutputNode()];
+    }
+    else {
+      targets = [this.getNeuron(selectedNodes[0]).neuron];
+    }
   }
   else {
     targets = selectedNodes.map((i) => this.getNeuron(i).neuron);