From c69068faa67e6cf1f85bd7e97ca528e3a801b995 Mon Sep 17 00:00:00 2001
From: Luis Fernando Molina <Luis-Fernando-Molina@users.noreply.github.com>
Date: Thu, 6 Jun 2019 14:30:42 +0200
Subject: [PATCH] Update PartsFromNodes.js

---
 .../Workspace/FunctionLibraries/PartsFromNodes.js  | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js b/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
index 9550324e9..33e9ee2b4 100644
--- a/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
+++ b/StrategySpace/Workspace/FunctionLibraries/PartsFromNodes.js
@@ -251,6 +251,11 @@ function newPartsFromNodes () {
         },
         takePosition: {
           situations: []
+        },
+        positionSize: {
+          formula: {
+            code: DEFAULT_FORMULA_TEXT
+          }
         }
       },
       openStage: {
@@ -291,6 +296,8 @@ function newPartsFromNodes () {
     createPart('Take Profit', '', strategy.manageStage.takeProfit, strategy.manageStage, strategy.manageStage)
     createPart('Stop', 'Initial Stop', strategy.openStage.initialDefinition.stopLoss, strategy.openStage.initialDefinition, strategy.openStage.initialDefinition)
     createPart('Take Profit', 'Initial Take Profit', strategy.openStage.initialDefinition.takeProfit, strategy.openStage.initialDefinition, strategy.openStage.initialDefinition)
+    createPart('Position Size', '', strategy.triggerStage.positionSize, strategy.triggerStage, strategy.triggerStage)
+    createPart('Formula', '', strategy.triggerStage.positionSize.formula, strategy.triggerStage.positionSize, strategy.triggerStage.positionSize)
   }
 
   function addParameters (node) {
@@ -327,12 +334,19 @@ function newPartsFromNodes () {
         },
         takePosition: {
           situations: []
+        },
+        positionSize: {
+          formula: {
+            code: DEFAULT_FORMULA_TEXT
+          }
         }
       }
       createPart('Trigger Stage', '', node.triggerStage, node, node, 'Trigger Stage')
       createPart('Trigger On Event', '', node.triggerStage.triggerOn, node.triggerStage, node.triggerStage)
       createPart('Trigger Off Event', '', node.triggerStage.triggerOff, node.triggerStage, node.triggerStage)
       createPart('Take Position Event', '', node.triggerStage.takePosition, node.triggerStage, node.triggerStage)
+      createPart('Position Size', '', node.triggerStage.positionSize, node.triggerStage, node.triggerStage)
+      createPart('Formula', '', node.triggerStage.positionSize.formula, node.triggerStage.positionSize, node.triggerStage.positionSize)
     }
     if (node.openStage === undefined) {
       node.openStage = {