gfor.js invented

Author: unbornchikken

lib/es6/complex.js

@@ -31,7 +31,7 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 "use strict";
 
-let lodash = require("lodash");
+let _ = require("lodash");
 let assert = require("better-assert");
 
 function Complex(real, imag) {

lib/es6/dim4.js

@@ -31,7 +31,7 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 "use strict";
 
-let lodash = require("lodash");
+let _ = require("lodash");
 let assert = require("better-assert");
 
 function Dim4(dim0, dim1, dim2, dim3) {

lib/es6/ext.js

@@ -38,13 +38,13 @@ function synchronify(fire, f) {
     return function () {
         var err;
         var res;
+        var done = false;
         let cb = function (e, r) {
             err = e;
             res = r;
             done = true;
         };
 
-        let done = false;
         let args = _.toArray(arguments).concat(cb);
 
         f.apply(this, args);
@@ -54,7 +54,7 @@ function synchronify(fire, f) {
         if (err) throw err;
 
         return res;
-    }
+    };
 }
 
 function installAsyncAndSync(fire, obj, name) {
@@ -121,7 +121,8 @@ function Ext(self) {
             finally {
                 this.setDevice(current);
             }
-        }
+        },
+        gfor: require("./makeGfor")(self)
     });
 }
 

lib/es6/makeGfor.js

@@ -0,0 +1,78 @@
+/*
+ Copyright (c) 2014-2015, ArrayFire
+ Copyright (c) 2015 Gábor Mező aka unbornchikken ([email protected])
+ All rights reserved.
+
+ Redistribution and use in source and binary forms, with or without modification,
+ are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright notice, this
+ list of conditions and the following disclaimer.
+
+ * Redistributions in binary form must reproduce the above copyright notice, this
+ list of conditions and the following disclaimer in the documentation and/or
+ other materials provided with the distribution.
+
+ * Neither the name of the ArrayFire nor the names of its
+ contributors may be used to endorse or promote products derived from
+ this software without specific prior written permission.
+
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
+ ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+"use strict";
+
+let _ = require("lodash");
+let assert = require("better-assert");
+let Seq = require("./seq");
+
+function makeGfor(af) {
+    return function gfor() {
+        let fPos = -1;
+        for(let i = 0; i < arguments.length; i++) {
+            if (_.isFunction(arguments[i])) {
+                fPos = i;
+                break;
+            }
+        }
+        if (fPos === -1) {
+            throw new Error("Body function argument expected.");
+        }
+        if (fPos === 0) {
+            throw new Error("Seq arguments expected.");
+        }
+        let thisArg = arguments[fPos];
+        if (arguments.length > fPos + 1 && _.isObject(fPos + 1)) {
+            thisArg = arguments[fPos + 1];
+        }
+        let seq;
+        if (fPos === 1) {
+            seq = new Seq(arguments[0]);
+        }
+        else if (fPos === 2) {
+            seq = new Seq(arguments[0], arguments[1]);
+        }
+        else {
+            throw new Error("Invalid number of Seq arguments.");
+        }
+        seq.isGFor = true;
+        af._gforToggle();
+        try {
+            arguments[fPos].call(thisArg, seq);
+        }
+        finally {
+            af._gforToggle();
+        }
+    }
+}
+
+module.exports = makeGfor;

lib/es6/seq.js

@@ -31,20 +31,26 @@ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 "use strict";
 
-let lodash = require("lodash");
+let _ = require("lodash");
 let assert = require("better-assert");
 
-function Seq(begin, end, step, isGFor) {
+function Seq(begin, end, step) {
     assert(_.isNumber(begin));
-    assert(_.isNumber(end));
+    if (_.isUndefined(end)) {
+        begin = 0;
+        end = begin;
+    }
+    else {
+        assert(_.isNumber(end));
+    }
     step = step || 1;
     assert(_.isNumber(step));
-    isGFor = !!isGFor;
 
     this.begin = begin;
     this.end = end;
     this.step = step;
-    this.isGFor = isGFor;
+
+    this.isGFor = false;
 }
 
 module.exports = Seq;

src/fire.cpp

@@ -55,6 +55,13 @@ NAN_METHOD(_DoEvents)
     NanReturnUndefined();
 }
 
+NAN_METHOD(_GforToggle)
+{
+    NanScope();
+    af::gforToggle();
+    NanReturnUndefined();
+}
+
 void Init(v8::Handle<v8::Object> exports)
 {
     Symbols::Init();
@@ -73,4 +80,5 @@ void Init(v8::Handle<v8::Object> exports)
 
     // Helpers:
     exports->Set(NanNew("_doEvents"), NanNew<FunctionTemplate>(_DoEvents)->GetFunction());
+    exports->Set(NanNew("_gforToggle"), NanNew<FunctionTemplate>(_GforToggle)->GetFunction());
 }