From 0bb1c71b877e44513e79f4c0c3c87924e1ed011e Mon Sep 17 00:00:00 2001
From: open-trade <info@opentradesolutions.com>
Date: Thu, 19 Nov 2020 00:53:10 +0800
Subject: [PATCH] wait for image

---
 flutter_hbb/lib/model.dart | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/flutter_hbb/lib/model.dart b/flutter_hbb/lib/model.dart
index 8a4132e00..8ab833b78 100644
--- a/flutter_hbb/lib/model.dart
+++ b/flutter_hbb/lib/model.dart
@@ -26,6 +26,7 @@ class FfiModel with ChangeNotifier {
   PeerInfo _pi = PeerInfo();
   Display _display = Display();
   bool _decoding = false;
+  bool _waitForImage = false;
 
   FfiModel() {
     init();
@@ -61,6 +62,10 @@ class FfiModel with ChangeNotifier {
     if (!_decoding) {
       var rgba = FFI.getRgba();
       if (rgba != null) {
+        if (_waitForImage) {
+          _waitForImage = false;
+          dismissLoading();
+        }
         _decoding = true;
         ui.decodeImageFromPixels(
             rgba, _display.width, _display.height, ui.PixelFormat.bgra8888,
@@ -120,6 +125,10 @@ class FfiModel with ChangeNotifier {
       _display = _pi.displays[_pi.currentDisplay];
       FFI.cursorModel.updateDisplayOrigin(_display.x, _display.y);
     }
+    if (displays.length > 1) {
+      showLoading('Waiting for image...');
+      _waitForImage = true;
+    }
   }
 }