Compress compiled block using { unsued: "keep_assign" }

lib/cwise-transform.js

@@ -11,7 +11,7 @@ var OPTIONAL_FIELDS = [ "pre", "post", "printCode", "funcName", "blockSize" ]
 
 function processFunc(func) {
   var codeStr = "var X=" + func
-  var minified = uglify.minify(codeStr, {fromString: true}).code
+  var minified = uglify.minify(codeStr, {fromString: true, compress: { unused: "keep_assign" }}).code
   var code = minified.substr(6, minified.length-7)
   return parse(code)
 }
@@ -44,4 +44,4 @@ function cwiseTransform(file, opts) {
     }
   })
   return sm
-}
+}

test/browserify.js

@@ -5,7 +5,7 @@ var vm = require("vm")
 var path = require("path")
 var tape = require("tape")
 
-var cases = [ "unary", "binary", "offset" ]
+var cases = [ "unary", "binary", "offset", "fill" ]
 
 bundleCasesFrom(0)
 

test/fill.js

@@ -0,0 +1,42 @@
+var cwise = require("cwise")
+var ndarray = require("ndarray")
+
+if(typeof test === "undefined") {
+  test = require("tape")
+}
+
+test("fill", function(t) {
+
+  var fill = cwise({
+    args: ["index", "array", "scalar"],
+    body: function(idx, out, f) {
+      out = f.apply(undefined, idx)
+    }
+  })
+
+  var xlen = 10
+  var ylen = 5
+  var array = ndarray(new Float32Array(xlen * ylen), [xlen, ylen])
+
+  fill(array, function(row, col) {
+    return 0
+  })
+
+  for(var i = 0; i < xlen; i++) {
+    for(var j = 0; j < ylen; j++) {
+      t.equals(array.get(i, j), 0, 'fill ('+ i + ',' + j + ')')
+    }
+  }
+
+  fill(array, function(row, col) {
+    return 10 * (row + col)
+  })
+
+  for(var i = 0; i < xlen; i++) {
+    for(var j = 0; j < ylen; j++) {
+      t.equals(array.get(i, j), 10 * (i + j), 'fill ('+ i + ',' + j + ')')
+    }
+  }
+
+  t.end()
+})