Add ApplyToObjectField to Object.m

ianhinder · Sep 13, 2013 · 57d2efa · 57d2efa
1 parent 745868e
commit 57d2efa
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/Tests/Object.mt b/Tests/Object.mt
@@ -69,3 +69,11 @@ Test[
   ,
   TestID->"JoinObjectField-Exists"
 ]
+
+Test[
+  ApplyToObjectField[NewObject[Obj, {"Key1" -> 1, "Key2" -> 2}], "Key1", g]
+  ,
+  Obj["Key1" -> g[1], "Key2" -> 2]
+  ,
+  TestID->"ApplyToObjectField"
+]
diff --git a/Tools/CodeGen/Object.m b/Tools/CodeGen/Object.m
@@ -25,6 +25,7 @@
 GetObjectField;
 AppendObjectField;
 JoinObjectField;
+ApplyToObjectField;
 
 Begin["`Private`"];
 
@@ -62,6 +63,11 @@
   JoinObjectField[obj_, field_String, values_List] :=
   SetObjectField[obj, field, Join[GetObjectField[obj, field], values]]];
 
+DefFn[
+  ApplyToObjectField[obj_, field_String, f_] :=
+  SetObjectField[obj, field, f[GetObjectField[obj, field]]]];
+
+
 End[];
 
 EndPackage[];