better merge for Centroid with vec

joshday · Mar 17, 2020 · 2d655f7 · 2d655f7
1 parent 9f9f0c7
commit 2d655f7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/part.jl b/src/part.jl
@@ -43,11 +43,16 @@ Base.in(x, c::Centroid) = false
 Base.isless(a::Centroid, b::Centroid) = isless(a.center, b.center)
 Base.show(io::IO, c::Centroid) = print(io, "Centroid: $(c.center)")
 
-function Base.merge!(a::Centroid, b::Centroid, astat, bstat)
+function Base.merge!(a::Centroid{<:Number}, b::Centroid{<:Number}, astat, bstat)
     w = nobs(bstat) / nobs(astat)
     a.center = smooth(a.center, b.center, w)
     a
 end
+function Base.merge!(a::Centroid, b::Centroid, astat, bstat)
+    w = nobs(bstat) / nobs(astat)
+    smooth!(a.center, b.center, w)
+    a
+end
 
 #-----------------------------------------------------------------------------# ClosedInterval
 mutable struct ClosedInterval{T}