codec_support/vq: do not move empty group centroid

[nihav.git] / nihav-codec-support / src / vq / generic_elbg.rs

diff --git a/nihav-codec-support/src/vq/generic_elbg.rs b/nihav-codec-support/src/vq/generic_elbg.rs
index b2c838dc24236cfc15b30e13e4888ffd5b535ef9..3c3e3c075f480043b3834e88e24ba6eb7bbc89fc 100644 (file)
--- a/nihav-codec-support/src/vq/generic_elbg.rs
+++ b/nihav-codec-support/src/vq/generic_elbg.rs
@@ -51,7 +51,9 @@ impl<T: VQElement, TS: VQElementSum<T>> Cluster<T, TS> {
         self.dist += u64::from(self.centroid.dist(entry.val)) * entry.count;
     }
     fn calc_centroid(&mut self) {
-        self.centroid = self.sum.get_centroid();
+        if self.count > 0 {
+            self.centroid = self.sum.get_centroid();
+        }
     }
     fn calc_dist(&mut self) {
         if self.count != 0 {