From a587dc4e07504691349d3b7f3faac48275da4741 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dennis=20Gl=C3=A4ser?= <dennis.glaeser@iws.uni-stuttgart.de>
Date: Sat, 21 Oct 2023 12:19:41 +0200
Subject: [PATCH] [vtk-view] add color bar

---
 .../src/components/vtk-plots/VtkComponent.vue | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/viplab-standalone-frontend-vue/src/components/vtk-plots/VtkComponent.vue b/viplab-standalone-frontend-vue/src/components/vtk-plots/VtkComponent.vue
index 4484c0f..b566522 100644
--- a/viplab-standalone-frontend-vue/src/components/vtk-plots/VtkComponent.vue
+++ b/viplab-standalone-frontend-vue/src/components/vtk-plots/VtkComponent.vue
@@ -91,6 +91,7 @@ import 'vtk.js/Sources/Rendering/Profiles/Geometry';
 
 import vtkFullScreenRenderWindow from 'vtk.js/Sources/Rendering/Misc/FullScreenRenderWindow';
 import vtkActor from 'vtk.js/Sources/Rendering/Core/Actor';
+import vtkScalarBarActor from 'vtk.js/Sources/Rendering/Core/ScalarBarActor';
 import vtkMapper from 'vtk.js/Sources/Rendering/Core/Mapper';
 import vtkColorTransferFunction from 'vtk.js/Sources/Rendering/Core/ColorTransferFunction';
 
@@ -125,6 +126,7 @@ export default {
       renderer: Object,
       renderWindow: Object,
       actor: Object,
+      barActor: null,
       representation: '1:2:0',
       representationOptions: [
         {
@@ -207,6 +209,12 @@ export default {
       this.mapper.setLookupTable(this.lookupTable);
       this.applyPreset();
 
+      const [location, name] = color.split(':');
+      if (location == "SolidColor")
+          this.removeBarActor();
+      else
+          this.makeBarActor(name);
+
       this.renderWindow.render();
     },
     getMapperConfig(value) {
@@ -244,6 +252,29 @@ export default {
       this.lookupTable.updateRange();
       // console.log(this.lookupTable);
     },
+    removeBarActor() {
+      if (this.barActor != null) {
+        this.renderer.removeActor(this.barActor);
+        this.barActor.delete();
+      }
+    },
+    makeBarActor(fieldName) {
+      this.removeBarActor();
+      this.barActor = vtkScalarBarActor.newInstance();
+      this.barActor.setMapper(this.mapper);
+      this.barActor.setAxisLabel(fieldName);
+      this.barActor.setScalarsToColors(this.lookupTable);
+      this.barActor.setVisibility(true);
+      this.barActor.setAutomated(false);
+      this.barActor.setBoxPosition([0.66, -0.5]);
+      let boxSize = this.barActor.getBoxSize();
+      let tickStyle = this.barActor.getAxisTextStyle();
+      boxSize[0] *= 3.0;
+      tickStyle.fontSize = 35;
+      this.barActor.setBoxSize(boxSize);
+      this.barActor.setTickTextStyle(tickStyle);
+      this.renderer.addActor(this.barActor);
+    },
     decreaseFileIndex() {
       this.fileIndex = Math.max(this.fileIndex - 1, 0);
       return this.fileIndex;