From 614a1f9d5a9d87dce230fffe1f59f81c0fe6025a Mon Sep 17 00:00:00 2001
From: Charles Blake <cblake@csail.mit.edu>
Date: Mon, 16 Feb 2015 06:02:31 -0500
Subject: [PATCH] Add TableRef version of mgetOrPut.

---
 lib/pure/collections/tables.nim | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/pure/collections/tables.nim b/lib/pure/collections/tables.nim
index 96189baff..6719c79de 100644
--- a/lib/pure/collections/tables.nim
+++ b/lib/pure/collections/tables.nim
@@ -397,6 +397,11 @@ proc mget*[A, B](t: TableRef[A, B], key: A): var B =
   ## If `key` is not in `t`, the ``EInvalidKey`` exception is raised.
   t[].mget(key)
 
+proc mgetOrPut*[A, B](t: TableRef[A, B], key: A, value: B): var B =
+  ## retrieves value at ``t[key]`` or puts ``value`` if not present, either way
+  ## returning a value which can be modified.
+  t[].mgetOrPut(key, value)
+
 proc hasKey*[A, B](t: TableRef[A, B], key: A): bool =
   ## returns true iff `key` is in the table `t`.
   result = t[].hasKey(key)