From c4d625da095456e081d788b5ebda15c2b10496b8 Mon Sep 17 00:00:00 2001
From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Wed, 24 Mar 2021 20:30:29 +0000
Subject: [PATCH] deck: Do update count

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---
 src/deck.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/deck.rs b/src/deck.rs
index eddaaa62..0a57b7ab 100644
--- 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))
+      }
+    ))
+  }
 }
-- 
2.30.2