Update mappingx

Author: FredericWantiez

src/tapedfunction.jl

@@ -373,9 +373,16 @@ function copy_bindings(old::Dict{Symbol, Any})
     return newb
 end
 
-function Base.copy(tf::TapedFunction)
+function Base.copy(tf::TapedFunction, args...)
     # create a new uninitialized TapedFunction
     new_tf = TapedFunction(tf)
     new_tf.bindings = copy_bindings(tf.bindings)
+    if !isempty(args)
+        length(args) > tf.arity && throw("Number of updates for `args` is greater than generated function arity $(tf.arity)")
+        for (i, arg) in enumerate(args)
+            slot = Symbol("_", i + 1)
+            haskey(tf.bindings, slot) && _update_var!(tf, slot, arg)
+       end
+    end
     return new_tf
 end

src/tapedtask.jl

@@ -159,9 +159,9 @@ Base.IteratorEltype(::Type{<:TapedTask}) = Base.EltypeUnknown()
 
 # copy the task
 
-function Base.copy(t::TapedTask)
-    tf = copy(t.tf)
-    new_t = TapedTask(tf)
+function Base.copy(t::TapedTask, args...)
+    tf = copy(t.tf, args...)
+    new_t = TapedTask(tf, args...)
     storage = t.task.storage::IdDict{Any,Any}
     new_t.task.storage = copy(storage)
     new_t.task.storage[:tapedtask] = new_t