Additional functions for STArray

See ethul/purescript-angular#5

Author: ethul

README.md

@@ -116,21 +116,31 @@
 
 ### Types
 
+    type Assoc a = { index :: Number, value :: a }
+
     data STArray :: * -> * -> *
 
 
 ### Values
 
     emptySTArray :: forall a h r. Eff (st :: ST h | r) (STArray h a)
 
+    getAssocs :: forall a h r. STArray h a -> Eff (st :: ST h | r) [Assoc a]
+
+    getElems :: forall a h r. STArray h a -> Eff (st :: ST h | r) [a]
+
     peekSTArray :: forall a h r. STArray h a -> Number -> Eff (st :: ST h | r) (Maybe a)
 
     pokeSTArray :: forall a h r. STArray h a -> Number -> a -> Eff (st :: ST h | r) Boolean
 
-    pushSTArray :: forall a h r. STArray h a -> a -> Eff (st :: ST h | r) Unit
+    pushAllSTArray :: forall a h r. STArray h a -> [a] -> Eff (st :: ST h | r) Number
+
+    pushSTArray :: forall a h r. STArray h a -> a -> Eff (st :: ST h | r) Number
 
     runSTArray :: forall a r. (forall h. Eff (st :: ST h | r) (STArray h a)) -> Eff r [a]
 
+    spliceSTArray :: forall a h r. STArray h a -> Number -> Number -> [a] -> Eff (st :: ST h | r) [a]
+
 
 ## Module Data.Array.Unsafe
 

src/Data/Array/ST.purs

@@ -1,10 +1,15 @@
 module Data.Array.ST
   ( STArray(..)
+  , Assoc()
   , runSTArray
   , emptySTArray
   , peekSTArray
   , pokeSTArray
   , pushSTArray
+  , pushAllSTArray
+  , spliceSTArray
+  , getElems
+  , getAssocs
   ) where
 
 import Data.Maybe
@@ -15,6 +20,8 @@ import Control.Monad.ST (ST())
 
 foreign import data STArray :: * -> * -> *
 
+type Assoc a = { value :: a, index :: Number }
+
 foreign import runSTArray """
   function runSTArray(f) {
     return f;
@@ -62,15 +69,57 @@ foreign import pokeSTArrayImpl """
 pokeSTArray :: forall a h r. STArray h a -> Number -> a -> Eff (st :: ST h | r) Boolean
 pokeSTArray arr i a = runFn3 pokeSTArrayImpl arr i a
 
-foreign import pushSTArrayImpl """
-  function pushSTArrayImpl(arr, a) {
-    return function() {
-      arr.push(a);
-      return {};
+foreign import pushAllSTArrayImpl """
+  function pushAllSTArrayImpl(arr, as) {
+    return function(){
+      return arr.push.apply(arr, as);
     };
-  }""" :: forall a h e. Fn2 (STArray h a)
-                            a
-                            (Eff (st :: ST h | e) Unit)
+  }""" :: forall a h r. Fn2 (STArray h a)
+                            [a]
+                            (Eff (st :: ST h | r) Number)
+
+pushAllSTArray :: forall a h r. STArray h a -> [a] -> Eff (st :: ST h | r) Number
+pushAllSTArray = runFn2 pushAllSTArrayImpl
 
-pushSTArray :: forall a h r. STArray h a -> a -> Eff (st :: ST h | r) Unit
-pushSTArray arr a = runFn2 pushSTArrayImpl arr a
+pushSTArray :: forall a h r. STArray h a -> a -> Eff (st :: ST h | r) Number
+pushSTArray arr a = pushAllSTArray arr [a]
+
+foreign import spliceSTArrayImpl """
+  function spliceSTArrayImpl(arr, index, howMany, bs) {
+    return function(){
+      return arr.splice.apply(arr, [index, howMany].concat(bs));
+    };
+  }""" :: forall a h r. Fn4 (STArray h a)
+                            Number
+                            Number
+                            [a]
+                            (Eff (st :: ST h | r) [a])
+
+spliceSTArray :: forall a h r. STArray h a -> Number -> Number -> [a] -> Eff (st :: ST h | r) [a]
+spliceSTArray = runFn4 spliceSTArrayImpl
+
+foreign import getElems """
+  function getElems(arr) {
+    return function(){
+      var as = [];
+      var i = -1;
+      var n = arr.length;
+      while(++i < n) {
+        as[i] = arr[i];
+      }
+      return as;
+    };
+  }""" :: forall a h r. STArray h a -> Eff (st :: ST h | r) [a]
+
+foreign import getAssocs """
+  function getAssocs(arr) {
+    return function(){
+      var as = [];
+      var i = -1;
+      var n = arr.length;
+      while(++i < n) {
+        as[i] = {value: arr[i], index: i};
+      }
+      return as;
+    };
+  }""" :: forall a h r. STArray h a -> Eff (st :: ST h | r) [Assoc a]