From 3647a74c0571103649a255cc17bd55d99662e0d6 Mon Sep 17 00:00:00 2001
From: Xiao Gui <xgui3783@gmail.com>
Date: Mon, 9 Mar 2020 12:05:13 +0100
Subject: [PATCH] chore: screenshot perf

---
 src/ui/nehubaContainer/nehubaContainer.component.ts | 9 ---------
 src/ui/takeScreenshot/takeScreenshot.component.ts   | 1 +
 2 files changed, 1 insertion(+), 9 deletions(-)

diff --git a/src/ui/nehubaContainer/nehubaContainer.component.ts b/src/ui/nehubaContainer/nehubaContainer.component.ts
index 2b194f084..5aad10033 100644
--- a/src/ui/nehubaContainer/nehubaContainer.component.ts
+++ b/src/ui/nehubaContainer/nehubaContainer.component.ts
@@ -776,15 +776,6 @@ export class NehubaContainer implements OnInit, OnChanges, OnDestroy {
         }
       }),
     )
-
-    // To get WebGL content when taking screenshot
-    HTMLCanvasElement.prototype.getContext = function(origFn) {
-      return function(type, attribs) {
-        attribs = attribs || {}
-        attribs.preserveDrawingBuffer = true
-        return origFn.call(this, type, attribs)
-      }
-    }(HTMLCanvasElement.prototype.getContext)
   }
 
   // datasetViewerRegistry : Set<string> = new Set()
diff --git a/src/ui/takeScreenshot/takeScreenshot.component.ts b/src/ui/takeScreenshot/takeScreenshot.component.ts
index ca8ce9972..b10f94a06 100644
--- a/src/ui/takeScreenshot/takeScreenshot.component.ts
+++ b/src/ui/takeScreenshot/takeScreenshot.component.ts
@@ -189,6 +189,7 @@ export class TakeScreenshotComponent implements OnInit, OnDestroy {
     }
 
     public mouseUp(_event: MouseEvent) {
+      (window as any).viewer.display.update()
       this.borderWidth = '0'
 
       this.isDragging = false
-- 
GitLab