[Merged by Bors] - Add @addlogprob! macro

src/DynamicPPL.jl

@@ -93,7 +93,9 @@ export  AbstractVarInfo,
         @logprob_str,
 # Convenience functions
         logprior,
-        logjoint
+        logjoint,
+# Convenience macros
+        @addlogprob!
 
 # Reexport
 using Distributions: loglikelihood

src/utils.jl

@@ -1,3 +1,14 @@
+"""
+    @addlogprob!(ex)
+
+Add the result of the evaluation of `ex` to the joint log probability.
+"""
+macro addlogprob!(ex)
+    return quote
+        acclogp!($(esc(:(_varinfo))), $(esc(ex)))
+    end
+end
+
 """
     getargs_dottilde(x)
 

test/utils.jl

@@ -3,6 +3,20 @@ using DynamicPPL: getargs_dottilde, getargs_tilde
 
 using Test
 
+@testset "addlogprob!" begin
+    @model function testmodel()
+        global lp_before = getlogp(_varinfo)
+        @addlogprob!(42)
+        global lp_after = getlogp(_varinfo)
+    end
+
+    model = testmodel()
+    varinfo = DynamicPPL.VarInfo(model)
+    model(varinfo)
+    @test iszero(lp_before)
+    @test getlogp(varinfo) == lp_after == 42
+end
+
 @testset "getargs_dottilde" begin
     # Some things that are not expressions.
     @test getargs_dottilde(:x) === nothing