Actually define promote_symtype

Author: YingboMa

src/equations.jl

@@ -14,6 +14,7 @@ struct Equation
 end
 Base.:(==)(a::Equation, b::Equation) = all(isequal.((a.lhs, a.rhs), (b.lhs, b.rhs)))
 Base.hash(a::Equation, salt::UInt) = hash(a.lhs, hash(a.rhs, salt))
+
 Base.show(io::IO, eq::Equation) = print(io, eq.lhs, " ~ ", eq.rhs)
 
 SymbolicUtils.simplify(x::Equation; kw...) = simplify(x.lhs; kw...) ~ simplify(x.rhs; kw...)

src/extra_functions.jl

@@ -1,5 +1,5 @@
-@register Base.getindex(x,i::Integer)
-@register Base.getindex(x,i)
+@register Base.getindex(x,i::Integer) false
+@register Base.getindex(x,i) false
 @register Base.binomial(n,k)
 
 @register Base.signbit(x)
@@ -36,8 +36,8 @@ 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)
+ModelingToolkit.@register Distributions.Uniform(mu,sigma) false
+ModelingToolkit.@register Distributions.Normal(mu,sigma) false
 
 @register ∈(x::Num, y::AbstractArray)
 @register ∪(x, y)

src/register_function.jl

@@ -1,5 +1,5 @@
 """
-    @register(expr, Ts = [Num, Symbolic, Real])
+    @register(expr, define_promotion, Ts = [Num, Symbolic, Real])
 
 Overload approperate methods such that ModelingToolkit can stop tracing into the
 registered function.
@@ -11,7 +11,7 @@ registered function.
 @register hoo(x, y)::Int # `hoo` returns `Int`
 ```
 """
-macro register(expr, Ts = [Num, Symbolic, Real])
+macro register(expr, define_promotion = true, Ts = [Num, Symbolic, Real])
     if expr.head === :(::)
         ret_type = expr.args[2]
         expr = expr.args[1]
@@ -48,7 +48,7 @@ macro register(expr, Ts = [Num, Symbolic, Real])
     push!(
           ex.args,
           quote
-              if $!($hasmethod($promote_symtype, $Tuple{$typeof($f), $Vararg}))
+              if $define_promotion
                   (::$typeof($promote_symtype))(::$typeof($f), args...) = $ret_type
               end
           end