Clojure interface to editable vectors - mutable,immutable,conj!,pop!,assoc!,get!,nth!

1 files changed, 45 insertions, 0 deletions

diff --git a/src/clj/clojure/core.clj b/src/clj/clojure/core.clj
index 8ed4aaf1..9e742659 100644
--- a/src/clj/clojure/core.clj
+++ b/src/clj/clojure/core.clj
@@ -4315,3 +4315,48 @@
   Delivers the supplied value to the promise, releasing any pending
   derefs. A subsequent call to deliver on a promise will throw an exception."  
   [promise val] (promise val))
+
+;;;;;;;;;;;;;;;;;;;;; editable collections ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
+(defn mutable 
+  "Returns a new, mutable version of the collection, in constant time."
+  [#^clojure.lang.IEditableCollection coll] 
+  (.mutable coll))
+
+(defn immutable 
+  "Returns a new, immutable version of the mutable collection, in constant time."
+  [#^clojure.lang.IMutableCollection coll] 
+  (.immutable coll))
+
+(defn conj!
+  "Adds x to the mutable collection, and return coll. The 'addition'
+  may happen at different 'places' depending on the concrete type."
+  [#^clojure.lang.IMutableCollection coll x]
+  (.conj coll x))
+
+(defn assoc!
+  "When applied to a mutable map, adds mapping of key(s) to
+  val(s). When applied to a mutable vector, sets the val at index.
+  Note - index must be <= (count vector). Returns coll."
+  ([#^clojure.lang.IMutableAssociative coll key val] (.assoc coll key val))
+  ([#^clojure.lang.IMutableAssociative coll key val & kvs]
+   (let [ret (.assoc coll key val)]
+     (if kvs
+       (recur ret (first kvs) (second kvs) (nnext kvs))
+       ret))))
+
+(defn pop!
+  "Removes the last item from a mutable vector. If
+  the collection is empty, throws an exception. Returns coll"
+  [#^clojure.lang.IMutableVector coll] 
+  (.pop coll))  
+
+(defn nth!
+  "Returns the value at the index. get! returns nil if index out of
+  bounds, nth! throws an exception"
+  [#^clojure.lang.IMutableVector coll index] 
+  (.nth coll index))  
+
+(defn get!
+  "Returns the value mapped to key, nil if key not present."
+  [#^clojure.lang.IMutableAssociative coll key]
+  (.valAt coll key))