making a start on the DomainSets.jl interface

Author: valentinsulzer

Project.toml

@@ -14,6 +14,7 @@ DiffRules = "b552c78f-8df3-52c6-915a-8e097449b14b"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
+DomainSets = "5b8099bc-c8ec-5219-889f-1d9e522a28bf"
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"
 LabelledArrays = "2ee39098-c373-598a-b85f-a56591580800"
 Latexify = "23fbe1c1-3f47-55db-b15f-69d7ec21a316"

src/domains.jl

@@ -1,11 +1,13 @@
-abstract type AbstractDomain{T,N} end
+using DomainSets
+
+abstract type AbstractDomain{T,N} <: Domain{T} end
 
 struct VarDomainPairing
   variables
-  domain::AbstractDomain
+  domain::Domain
 end
-Base.:∈(variable::ModelingToolkit.Num,domain::AbstractDomain) = VarDomainPairing(value(variable),domain)
-Base.:∈(variables::NTuple{N,ModelingToolkit.Num},domain::AbstractDomain) where N = VarDomainPairing(value.(variables),domain)
+Base.:∈(variable::ModelingToolkit.Num,domain::Domain) = VarDomainPairing(value(variable),domain)
+Base.:∈(variables::NTuple{N,ModelingToolkit.Num},domain::Domain) where N = VarDomainPairing(value.(variables),domain)
 
 ## Specific Domains
 

test/domains.jl

@@ -11,3 +11,10 @@ z ∈ (IntervalDomain(0.0,1.0) ⊗ IntervalDomain(0.0,1.0))
 (x,y) ∈ CircleDomain()
 @parameters r θ
 (r,θ) ∈ CircleDomain(true)
+
+(x,y) ∈ UnitDisk()
+(r,θ) ∈ UnitDisk()
+
+(x,y,z) ∈ UnitBall()
+@parameters ϕ
+(r,θ,ϕ) ∈ UnitBall()