Merge pull request #687 from SciML/distributions

register a bunch of distributions

Author: ChrisRackauckas

Project.toml

@@ -10,6 +10,7 @@ DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"
 DiffEqJump = "c894b116-72e5-5b58-be3c-e6d8d4ac2b12"
 DiffRules = "b552c78f-8df3-52c6-915a-8e097449b14b"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
+Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"
 LabelledArrays = "2ee39098-c373-598a-b85f-a56591580800"
@@ -37,6 +38,7 @@ DataStructures = "0.17, 0.18"
 DiffEqBase = "6.48.1"
 DiffEqJump = "6.7.5"
 DiffRules = "0.1, 1.0"
+Distributions = "0.24"
 DocStringExtensions = "0.7, 0.8"
 IfElse = "0.1"
 LabelledArrays = "1.3"

src/ModelingToolkit.jl

@@ -16,6 +16,8 @@ using DocStringExtensions
 using Base: RefValue
 import IfElse
 
+import Distributions
+
 RuntimeGeneratedFunctions.init(@__MODULE__)
 
 using RecursiveArrayTools

src/extra_functions.jl

@@ -11,3 +11,15 @@ ModelingToolkit.derivative(::typeof(abs), args::NTuple{1,Any}, ::Val{1}) = IfEls
 ModelingToolkit.derivative(::typeof(IfElse.ifelse), args::NTuple{3,Any}, ::Val{1}) = 0
 ModelingToolkit.derivative(::typeof(IfElse.ifelse), args::NTuple{3,Any}, ::Val{2}) = IfElse.ifelse(args[1],1,0)
 ModelingToolkit.derivative(::typeof(IfElse.ifelse), args::NTuple{3,Any}, ::Val{3}) = IfElse.ifelse(args[1],0,1)
+
+ModelingToolkit.@register Base.rand(x)
+ModelingToolkit.@register Base.randn(x)
+
+ModelingToolkit.@register Distributions.pdf(dist,x)
+ModelingToolkit.@register Distributions.logpdf(dist,x)
+ModelingToolkit.@register Distributions.cdf(dist,x)
+ModelingToolkit.@register Distributions.logcdf(dist,x)
+ModelingToolkit.@register Distributions.quantile(dist,x)
+
+ModelingToolkit.@register Distributions.Uniform(mu,sigma)
+ModelingToolkit.@register Distributions.Normal(mu,sigma)