Merge pull request #168 from YichengDWu/UnionDomain

Union domain

Author: YichengDWu

src/compact/NeuralPDE/pinnsampler.jl

@@ -109,6 +109,13 @@ function sample(d::DomainSets.GenericSphere{StaticArraysCore.SVector{2, T}, T},
     return data
 end
 
+function sample(d::DomainSets.UnitCircle, points::Int, alg::QuasiMonteCarlo.SamplingAlgorithm)
+    θ = sample(points, [0.0], [2π], alg)
+    data = [cos.(θ); sin.(θ)]
+    return data
+end
+
+
 function sample(d::DomainSets.GenericSphere{StaticArraysCore.SVector{3, T}, T}, points::Int,  alg::QuasiMonteCarlo.SamplingAlgorithm) where {T}
     (; center, radius) = d
     r = sample(points, [-1,-1], [1,1], alg)
@@ -130,6 +137,11 @@ function sample(d::SetdiffDomain{S, <:Tuple{<:Rectangle, F}}, points::Int, alg::
     return data[:, idx]
 end
 
+function sample(d::UnionDomain, points::Int, alg::QuasiMonteCarlo.SamplingAlgorithm) where {S, F}
+    data = mapreduce(x -> sample(x, points, alg), hcat, d.domains)
+    return data
+end
+
 function sample(d::Rectangle, points::Int, sampling_alg::QuasiMonteCarlo.SamplingAlgorithm)
     bounds = get_bounds(d)
     return sample(points, bounds[1], bounds[2], sampling_alg)
@@ -148,6 +160,7 @@ function sample(points::Int, lb::AbstractVector, ub::AbstractVector, ::SobolSamp
     return sobolsample(points, lb, ub)
 end
 
+
 function sobolsample(n::Int, lb, ub)
     s = cached_sobolseq(n, lb, ub)
     return reduce(hcat, [Sobol.next!(s) for i in 1:n])