Add implicitly mapped measures and kernels

.github/workflows/Breakage.yml

@@ -18,14 +18,14 @@ jobs:
         pkgversion: [latest, stable]
 
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
 
       # Install Julia
-      - uses: julia-actions/setup-julia@v1
+      - uses: julia-actions/setup-julia@v2
         with:
-          version: 1
+          version: '1'
           arch: x64
-      - uses: actions/cache@v1
+      - uses: actions/cache@v3
         env:
           cache-name: cache-artifacts
         with:
@@ -79,7 +79,7 @@ jobs:
               end;
             end'
 
-      - uses: actions/upload-artifact@v2
+      - uses: actions/upload-artifact@v3
         with:
           name: pr
           path: pr/
@@ -88,9 +88,9 @@ jobs:
     needs: break
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v4
 
-      - uses: actions/download-artifact@v2
+      - uses: actions/download-artifact@v3
         with:
           name: pr
           path: pr/
@@ -121,7 +121,7 @@ jobs:
             fi
           done >> MSG
 
-      - uses: actions/upload-artifact@v2
+      - uses: actions/upload-artifact@v3
         with:
           name: pr
           path: pr/

.github/workflows/CI.yml

@@ -25,7 +25,7 @@ jobs:
       fail-fast: false
       matrix:
         version:
-          - '1.6'
+          - '1.10'
           - '1'
           - 'pre'
         os:

Project.toml

@@ -4,7 +4,6 @@ authors = ["Chad Scherrer <[email protected]>", "Oliver Schulz <oschulz@mp
 version = "0.14.12"
 
 [deps]
-ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 ChangesOfVariables = "9e997f8a-9a97-42d5-a9f1-ce6bfc15e2c0"
 Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
 ConstantRNGs = "aa9b60e7-6b1c-4c29-a6e5-e43521412437"
@@ -22,14 +21,22 @@ LogarithmicNumbers = "aa2f6b4e-9042-5d33-9679-40d3a6b85899"
 MappedArrays = "dbb5928d-eab1-5f90-85c2-b9b0edb7c900"
 NaNMath = "77ba4419-2d1f-58cd-9bb1-8ffee604a2e3"
 PrettyPrinting = "54e16d92-306c-5ea0-a30b-337be88ac337"
+PropertyFunctions = "09e99361-2bb8-48a2-a80f-de58f0739eb4"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 Static = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"
+StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Tricks = "410a4b4d-49e4-4fbc-ab6d-cb71b17b3775"
 
+[weakdeps]
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+
+[extensions]
+MeasureBaseChainRulesCoreExt = "ChainRulesCore"
+
 [compat]
 ChainRulesCore = "1"
 ChangesOfVariables = "0.1.3"
@@ -49,11 +56,13 @@ LogarithmicNumbers = "1"
 MappedArrays = "0.4"
 NaNMath = "0.3, 1"
 PrettyPrinting = "0.3, 0.4"
+PropertyFunctions = "0.2.2"
 Random = "1"
 Reexport = "1"
 SpecialFunctions = "2"
 Static = "0.8, 1"
+StaticArrays = "1.5"
 Statistics = "1"
 Test = "1"
 Tricks = "0.1"
-julia = "1.6"
+julia = "1.10"

docs/make.jl

@@ -8,25 +8,16 @@ using Documenter
 using MeasureBase
 
 # Doctest setup
-DocMeta.setdocmeta!(
-    MeasureBase,
-    :DocTestSetup,
-    :(using MeasureBase);
-    recursive=true,
-)
+DocMeta.setdocmeta!(MeasureBase, :DocTestSetup, :(using MeasureBase); recursive = true)
 
 makedocs(
     sitename = "MeasureBase",
     modules = [MeasureBase],
     format = Documenter.HTML(
         prettyurls = !("local" in ARGS),
-        canonical = "https://juliamath.github.io/MeasureBase.jl/stable/"
+        canonical = "https://juliamath.github.io/MeasureBase.jl/stable/",
     ),
-    pages = [
-        "Home" => "index.md",
-        "API" => "api.md",
-        "LICENSE" => "LICENSE.md",
-    ],
+    pages = ["Home" => "index.md", "API" => "api.md", "LICENSE" => "LICENSE.md"],
     doctest = ("fixdoctests" in ARGS) ? :fix : true,
     linkcheck = !("nonstrict" in ARGS),
     warnonly = ("nonstrict" in ARGS),

ext/MeasureBaseChainRulesCoreExt.jl

@@ -0,0 +1,54 @@
+# This file is a part of MeasureBase.jl, licensed under the MIT License (MIT).
+
+module MeasureBaseChainRulesCoreExt
+
+using MeasureBase
+using ChainRulesCore: NoTangent, ZeroTangent
+import ChainRulesCore
+
+# = utils ====================================================================
+
+using MeasureBase: isneginf, isposinf
+
+_isneginf_pullback(::Any) = (NoTangent(), ZeroTangent())
+ChainRulesCore.rrule(::typeof(isneginf), x) = isneginf(x), _logdensityof_rt_pullback
+
+_isposinf_pullback(::Any) = (NoTangent(), ZeroTangent())
+ChainRulesCore.rrule(::typeof(isposinf), x) = isposinf(x), _isposinf_pullback
+
+# = insupport & friends ======================================================
+
+using MeasureBase: check_dof, require_insupport, checked_arg, _checksupport, _origin_depth
+
+@inline function ChainRulesCore.rrule(::typeof(_checksupport), cond, result)
+    y = _checksupport(cond, result)
+    function _checksupport_pullback(ȳ)
+        return NoTangent(), ZeroTangent(), one(ȳ)
+    end
+    y, _checksupport_pullback
+end
+
+_require_insupport_pullback(ΔΩ) = NoTangent(), ZeroTangent()
+function ChainRulesCore.rrule(::typeof(require_insupport), μ, x)
+    return require_insupport(μ, x), _require_insupport_pullback
+end
+
+_origin_depth_pullback(ΔΩ) = NoTangent(), NoTangent()
+ChainRulesCore.rrule(::typeof(_origin_depth), ν) = _origin_depth(ν), _origin_depth_pullback
+
+_check_dof_pullback(ΔΩ) = NoTangent(), NoTangent(), NoTangent()
+ChainRulesCore.rrule(::typeof(check_dof), ν, μ) = check_dof(ν, μ), _check_dof_pullback
+
+_checked_arg_pullback(ΔΩ) = NoTangent(), NoTangent(), ΔΩ
+ChainRulesCore.rrule(::typeof(checked_arg), ν, x) = checked_arg(ν, x), _checked_arg_pullback
+
+# = return type inference ====================================================
+
+using MeasureBase: logdensityof_rt
+
+_logdensityof_rt_pullback(::Any) = (NoTangent(), NoTangent(), ZeroTangent())
+function ChainRulesCore.rrule(::typeof(logdensityof_rt), target, v)
+    logdensityof_rt(target, v), _logdensityof_rt_pullback
+end
+
+end # module MeasureBaseChainRulesCoreExt

src/MeasureBase.jl

@@ -1,6 +1,7 @@
 module MeasureBase
 
 using Base: @propagate_inbounds
+using Base: OneTo
 
 using Random
 import Random: rand!
@@ -21,6 +22,7 @@ using DensityInterface: FuncDensity, LogFuncDensity
 using DensityInterface
 
 using InverseFunctions
+using InverseFunctions: FunctionWithInverse
 using ChangesOfVariables
 using ConstantRNGs
 
@@ -29,14 +31,17 @@ import ConstructionBase
 using ConstructionBase: constructorof
 using IntervalSets
 
+using StaticArrays:
+    StaticArray, StaticVector, StaticMatrix, SArray, SVector, SMatrix, SOneTo
+
 using PrettyPrinting
 const Pretty = PrettyPrinting
 
-using ChainRulesCore
 import FillArrays
 using Static
 using Static: StaticInteger
 using FunctionChains
+using PropertyFunctions: PropSelFunction
 
 export gentype
 export rebase
@@ -106,6 +111,7 @@ function logdensity_def end
 using Compat
 
 using IrrationalConstants
+using IrrationalConstants: loghalf
 
 include("static.jl")
 include("smf.jl")
@@ -139,6 +145,7 @@ include("combinators/restricted.jl")
 include("combinators/smart-constructors.jl")
 include("combinators/powerweighted.jl")
 include("combinators/conditional.jl")
+include("combinators/implicitlymapped.jl")
 
 include("standard/stdmeasure.jl")
 include("standard/stduniform.jl")
@@ -147,6 +154,8 @@ include("standard/stdlogistic.jl")
 include("standard/stdnormal.jl")
 include("combinators/half.jl")
 
+#include("implicitmaps.jl")
+
 include("rand.jl")
 
 include("density.jl")

src/combinators/half.jl

@@ -19,6 +19,11 @@ function Base.rand(rng::AbstractRNG, ::Type{T}, μ::Half) where {T}
     return abs(rand(rng, T, unhalf(μ)))
 end
 
+function logdensityof(μ::Half, x)
+    ld = logdensityof(unhalf(μ), x) - loghalf
+    return x ≥ 0 ? ld : oftype(ld, -Inf)
+end
+
 logdensity_def(μ::Half, x) = logdensity_def(unhalf(μ), x)
 
 @inline function insupport(d::Half, x)