Support address splicing into the current namespace via the here constant.

Author: ztangent

Project.toml

@@ -1,7 +1,7 @@
 name = "TracedRandom"
 uuid = "03576162-b2ff-4022-a931-e67b9900fd45"
 authors = ["Xuan <[email protected]>"]
-version = "0.1.0"
+version = "0.1.1"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

src/TracedRandom.jl

@@ -2,12 +2,27 @@ module TracedRandom
 
 using Random
 
-export rand!, randn!, randexp, randexp!,
+export here, rand!, randn!, randexp, randexp!,
        bitrand, randstring, randsubseq, randsubseq!,
        shuffle, shuffle!, randperm, randperm!, randcycle, randcycle!
 
 const Address = Union{Symbol,Pair{Symbol}}
 
+struct Here end
+
+"""
+    here
+
+Special address that refers to the address namespace of the calling context.
+Supported when [`rand`](@ref) is called on a non-primitive stochastic function,
+causing all named random variables sampled by that function to be spliced into
+the address namespace of the calling context. 
+"""
+const here = Here()
+
+Base.rand(::Here, fn::Function, args...) = fn(args...)
+Base.rand(::AbstractRNG, ::Here, fn::Function, args...) = fn(args...)
+
 Base.rand(::Address, fn::Function, args...) = fn(args...)
 Base.rand(::AbstractRNG, ::Address, fn::Function, args...) = fn(args...)
 

test/runtests.jl

@@ -9,6 +9,12 @@ Random.seed!(0)
 untraced = gaussian_mixture([1, 10, 100])
 @test traced == untraced
 
+Random.seed!(0)
+traced = rand(here, gaussian_mixture, [1, 10, 100])
+Random.seed!(0)
+untraced = gaussian_mixture([1, 10, 100])
+@test traced == untraced
+
 end
 
 @testset "Addressed calls with global RNG" begin