From be48d1df05e8bf8e0eada95d9a8dd806c4a95200 Mon Sep 17 00:00:00 2001
From: Stanislav Ponkrashov <s.ponkrashov@samsung.com>
Date: Wed, 8 Feb 2023 15:15:11 +0300
Subject: [PATCH] Enable VisualTensorView only for CircleGraph

---
 media/CircleGraph/index.html       | 2 ++
 media/CircleGraph/view-sidebar.js  | 2 +-
 src/CircleGraph/CircleGraphCtrl.ts | 8 ++++++++
 src/CircleGraph/CircleViewer.ts    | 1 +
 4 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/media/CircleGraph/index.html b/media/CircleGraph/index.html
index 9072ea67..146195b7 100644
--- a/media/CircleGraph/index.html
+++ b/media/CircleGraph/index.html
@@ -72,6 +72,8 @@
 <script type="text/javascript" nonce="%nonce%">
 // TODO find better way to set initial view mode
 var __viewMode = '%viewMode%';
+// TODO find better way to enable VisualTensorView
+var __enableVisualTensorView = '%enableVisualTensorView%';
 </script>
 <style>
 html { touch-action: none; overflow: hidden; width: 100%; height: 100%; -ms-text-size-adjust: 100%; -webkit-text-size-adjust: 100%; text-rendering: optimizeLegibility; -webkit-text-rendering: optimizeLegibility; -moz-text-rendering: optimizeLegibility; -ms-text-rendering: optimizeLegibility; -o-text-rendering: optimizeLegibility; -webkit-font-smoothing: antialiased; -moz-font-smoothing: antialiased; -ms-font-smoothing: antialiased; -o-font-smoothing: antialiased; }
diff --git a/media/CircleGraph/view-sidebar.js b/media/CircleGraph/view-sidebar.js
index 69ea8cb0..c7e5009f 100644
--- a/media/CircleGraph/view-sidebar.js
+++ b/media/CircleGraph/view-sidebar.js
@@ -1012,7 +1012,7 @@ sidebar.ArgumentView = class {
           const valueLine = this._host.document.createElement("div");
           try {
             const state = initializer.state;
-            if (!state) {
+            if (!state && __enableVisualTensorView === "true") {
               this._element.appendChild(new sidebar.VisualTensorView(this._host, JSON.parse(initializer.toString()))._element);
             }
             if (
diff --git a/src/CircleGraph/CircleGraphCtrl.ts b/src/CircleGraph/CircleGraphCtrl.ts
index c8152c05..5312e422 100644
--- a/src/CircleGraph/CircleGraphCtrl.ts
+++ b/src/CircleGraph/CircleGraphCtrl.ts
@@ -83,6 +83,7 @@ export class CircleGraphCtrl {
   protected _selectionNames: string[] | undefined;
   protected _state: CtrlStatus;
   protected _viewMode: string;
+  protected _enableVisualTensorView: boolean;
 
   private _ctrlDisposables: vscode.Disposable[] = [];
 
@@ -94,6 +95,7 @@ export class CircleGraphCtrl {
     this._eventHandler = undefined;
     this._state = CtrlStatus.init;
     this._viewMode = "viewer";
+    this._enableVisualTensorView = false;
   }
 
   public initGraphCtrl(
@@ -136,6 +138,10 @@ export class CircleGraphCtrl {
     this._modelToLoad = model;
   }
 
+  public enableVisualTensorView() {
+    this._enableVisualTensorView = true;
+  }
+
   /**
    * @brief set initial selection of nodes
    * @note  if called before loading, will be applied after load is finished
@@ -462,6 +468,8 @@ export class CircleGraphCtrl {
     html = this.updateUri(html, webview, "%view.js%", "view.js");
     // viewMode
     html = html.replace("%viewMode%", this._viewMode);
+    // VisualTensorView
+    html = html.replace("%enableVisualTensorView%", this._enableVisualTensorView ? "true" : "false");
 
     return html;
   }
diff --git a/src/CircleGraph/CircleViewer.ts b/src/CircleGraph/CircleViewer.ts
index 9d1c62a2..7dce2076 100644
--- a/src/CircleGraph/CircleViewer.ts
+++ b/src/CircleGraph/CircleViewer.ts
@@ -28,6 +28,7 @@ class CircleViewer extends CircleGraphCtrl {
   constructor(panel: vscode.WebviewPanel, extensionUri: vscode.Uri) {
     super(extensionUri, panel.webview);
     this._panel = panel;
+    this.enableVisualTensorView();
   }
 
   public owner(panel: vscode.WebviewPanel) {