Add new function mapReplaceAdd

This is used in McLachlan to manipulate calculations
ianhinder · Sep 29, 2011 · bee7aeb · bee7aeb
1 parent 89a1637
commit bee7aeb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Tools/MathematicaMisc/MapLookup.m b/Tools/MathematicaMisc/MapLookup.m
@@ -4,6 +4,7 @@
 lookup::usage = "";
 mapContains::usage = "";
 mapReplace::usage = "";
+mapReplaceAdd::usage = "";
 mapValueMap::usage = "";
 lookupDefault::usage = "";
 mapValueMapMultiple::usage = "";
@@ -59,6 +60,13 @@
     VerifyMap[map];
     Map[If[First[#] === key, key -> value, #] &, map]];
 
+mapReplaceAdd[map_, key_, value_] :=
+  Module[{},
+    VerifyMap[map];
+  If[mapContains[map, key],
+     mapReplace[map, key, value],
+     mapAdd[map, key, value]]];
+
 mapAdd[map_, key_, value_] :=
   Module[{},
     VerifyMap[map];