From 44dfc0f57e9dd9f4ff53f954a0e186248a71e026 Mon Sep 17 00:00:00 2001
From: Luis Fernando Molina <Luis-Fernando-Molina@users.noreply.github.com>
Date: Mon, 27 May 2019 15:51:47 +0200
Subject: [PATCH] addMissingEvents

---
 .../FunctionLibraries/PartsFromNodes.js         | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js b/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
index c72444f80..bcbd862d1 100644
--- a/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
+++ b/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
@@ -261,16 +261,25 @@ function newPartsFromNodes () {
     }
   }
 
-  function addMissingEvents () {
+  function addMissingEvents (node) {
     if (node.entryPoint === undefined) {
       node.entryPoint = {
         situations: []
       }
       createPart('Trigger On Event', '', node.entryPoint, node, node)
     }
-
-    createPart('Trigger Off Event', '', node.triggerStage.exitPoint, node.triggerStage, node.triggerStage)
-    createPart('Take Position Event', '', node.triggerStage.sellPoint, node.triggerStage, node.triggerStage)
+    if (node.exitPoint === undefined) {
+      node.exitPoint = {
+        situations: []
+      }
+      createPart('Trigger Off Event', '', node.exitPoint, node, node)
+    }
+    if (node.sellPoint === undefined) {
+      node.sellPoint = {
+        situations: []
+      }
+      createPart('Take Position Event', '', node.sellPoint, node, node)
+    }
   }
 
   function addPhase (parentNode) {