deck: Do update count

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>

diff --git a/src/deck.rs b/src/deck.rs
index eddaaa62808b24a827994170b08584112bb85a59..0a57b7ab108d43089fb490ebee2c6288a37b7b79 100644 (file)
--- a/src/deck.rs
+++ b/src/deck.rs
@@ -231,4 +231,13 @@ impl PieceTrait for Deck {
       ops: PUOs::Simple(PUO::Modify(())),
     }, xupdates, None)
   }
+
+  fn occultation_notify_hook(&self, piece: PieceId) -> UnpreparedUpdates {
+    Some(Box::new(
+      move |updates: &mut PrepareUpdatesBuffer| {
+        updates.piece_update_image(piece)
+          .unwrap_or_else(|e| error!("unable to send update! {:?}", e))
+      }
+    ))
+  }
 }