Apply suggestions

JuliaEarth · Oct 27, 2023 · 914a5aa · 914a5aa
1 parent 7e621da
commit 914a5aa
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 22 deletions.
diff --git a/src/operations.jl b/src/operations.jl
@@ -21,3 +21,25 @@ function thin end
 #-----------------
 
 include("thinning/random.jl")
+
+#---------------
+# PROCESS UNION
+#---------------
+
+"""
+    p₁ ∪ p₂
+
+Return the union of point processes `p₁` and `p₂`.
+"""
+Base.union(p₁::PointProcess, p₂::PointProcess) = UnionProcess(p₁, p₂)
+
+Base.union(p₁::PoissonProcess{<:Real}, p₂::PoissonProcess{<:Real}) = PoissonProcess(p₁.λ + p₂.λ)
+
+Base.union(p₁::PoissonProcess{<:Function}, p₂::PoissonProcess{<:Function}) = PoissonProcess(x -> p₁.λ(x) + p₂.λ(x))
+
+Base.union(p₁::PoissonProcess{<:Real}, p₂::PoissonProcess{<:Function}) = PoissonProcess(x -> p₁.λ + p₂.λ(x))
+
+Base.union(p₁::PoissonProcess{<:Function}, p₂::PoissonProcess{<:Real}) = PoissonProcess(x -> p₁.λ(x) + p₂.λ)
+
+Base.union(p₁::PoissonProcess{<:AbstractVector}, p₂::PoissonProcess{<:AbstractVector}) =
+  PoissonProcess(x -> p₁.λ + p₂.λ)
diff --git a/src/point/union.jl b/src/point/union.jl
@@ -19,25 +19,3 @@ function randsingle(rng::AbstractRNG, p::UnionProcess, g)
   pp₂ = rand(rng, p.p₂, g)
   PointSet([collect(pp₁); collect(pp₂)])
 end
-
-# ----------
-# OPERATION
-# ----------
-
-"""
-    p₁ ∪ p₂
-
-Return the union of point processes `p₁` and `p₂`.
-"""
-Base.union(p₁::PointProcess, p₂::PointProcess) = UnionProcess(p₁, p₂)
-
-Base.union(p₁::PoissonProcess{<:Real}, p₂::PoissonProcess{<:Real}) = PoissonProcess(p₁.λ + p₂.λ)
-
-Base.union(p₁::PoissonProcess{<:Function}, p₂::PoissonProcess{<:Function}) = PoissonProcess(x -> p₁.λ(x) + p₂.λ(x))
-
-Base.union(p₁::PoissonProcess{<:Real}, p₂::PoissonProcess{<:Function}) = PoissonProcess(x -> p₁.λ + p₂.λ(x))
-
-Base.union(p₁::PoissonProcess{<:Function}, p₂::PoissonProcess{<:Real}) = PoissonProcess(x -> p₁.λ(x) + p₂.λ)
-
-Base.union(p₁::PoissonProcess{<:AbstractVector}, p₂::PoissonProcess{<:AbstractVector}) =
-  PoissonProcess(x -> p₁.λ + p₂.λ)