From dbb31462addae59796fe68d8a68dff1bfab10d37 Mon Sep 17 00:00:00 2001
From: Tim Mickel <tim.mickel@gmail.com>
Date: Mon, 6 Jun 2016 15:31:14 -0400
Subject: [PATCH] Extend getSubstack to support multiple substacks
---
src/engine/blocks.js | 13 ++++++++++---
1 file changed, 10 insertions(+), 3 deletions(-)
diff --git a/src/engine/blocks.js b/src/engine/blocks.js
index 26351c1b1..f7f6ddf9c 100644
--- a/src/engine/blocks.js
+++ b/src/engine/blocks.js
@@ -50,14 +50,21 @@
/**
* Get the substack for a particular C-shaped block
* @param {?string} id ID for block to get the substack for
+ * @param {?number} substackNum Which substack to select (e.g. for if-else)
* @return {?string} ID of block in the substack
*/
- Blocks.prototype.getSubstack = function (id) {
+ Blocks.prototype.getSubstack = function (id, substackNum) {
var block = this._blocks[id];
if (typeof block === 'undefined') return null;
+
+ var inputName = 'SUBSTACK';
+ if (substackNum > 0) {
+ inputName += substackNum;
+ }
+
// Empty C-block?
- if (!('SUBSTACK' in block.inputs)) return null;
- return block.inputs['SUBSTACK'].block;
+ if (!(inputName in block.inputs)) return null;
+ return block.inputs[inputName].block;
};
/**