Changeset 4073 for trunk/packages/vizservers

Timestamp:

Dec 7, 2013 12:34:21 AM (10 years ago)

Author:

ldelgass

Message:

Add software rendering of legends, enabled by default. Should work with vtk
6.0 release and up (i.e. work around the bug with the scalar bar actor rendering
with newer VTK).

Location:

trunk/packages/vizservers/vtkvis

Files:

• ColorMap.cpp (modified) (1 diff)
• ColorMap.h (modified) (2 diffs)
• Makefile.in (modified) (2 diffs)
• Renderer.cpp (modified) (4 diffs)

trunk/packages/vizservers/vtkvis/ColorMap.cpp

@@ -488 +488 @@
     return _elementDefault;
 }
+
+/**
+ * \brief Render (using CPU) color map to an image
+ */
+void ColorMap::renderColorMap(ColorMap *map, int width, int height,
+                              vtkUnsignedCharArray *imgData,
+                              bool opaque, float bgColor[3],
+                              bool bgr, int bytesPerPixel)
+{
+    int size = bytesPerPixel * width * height;
+    if (imgData->GetMaxId() + 1 != size) {
+        imgData->SetNumberOfComponents(bytesPerPixel);
+        imgData->SetNumberOfValues(size);
+    }
+    unsigned char *dst = imgData->GetPointer(0);
+    vtkLookupTable *table = map->getLookupTable();
+    if (height > width) {
+        for (int i = 0; i < height; i++) {
+            double x = (double)i/(height-1);
+            double rgb[3];
+            table->GetColor(x, rgb);
+            unsigned char color[3];
+            if (opaque) {
+                color[0] = (unsigned char)(255. * (bgr ? rgb[2] : rgb[0]));
+                color[1] = (unsigned char)(255. * rgb[1]);
+                color[2] = (unsigned char)(255. * (bgr ? rgb[0] : rgb[2]));
+            } else {
+                double opacity = table->GetOpacity(x);
+                color[0] = (unsigned char)(255. * (bgColor[0] * (1.0 - opacity) + (bgr ? rgb[2] : rgb[0]) * opacity));
+                color[1] = (unsigned char)(255. * (bgColor[1] * (1.0 - opacity) + rgb[1] * opacity));
+                color[2] = (unsigned char)(255. * (bgColor[2] * (1.0 - opacity) + (bgr ? rgb[0] : rgb[2]) * opacity));
+            }
+            for (int j = 0; j < width; j++) {
+                memcpy(dst, color, 3);
+                dst += 3;
+                if (bytesPerPixel == 4) {
+                    *(dst++) = (unsigned char)255.;
+                }
+            }
+        }
+    } else {
+        for (int i = 0; i < height; i++) {
+            for (int j = 0; j < width; j++) {
+                double x = (double)j/(width-1);
+                double rgb[3];
+                table->GetColor(x, rgb);
+                unsigned char color[3];
+                if (opaque) {
+                    color[0] = (unsigned char)(255. * (bgr ? rgb[2] : rgb[0]));
+                    color[1] = (unsigned char)(255. * rgb[1]);
+                    color[2] = (unsigned char)(255. * (bgr ? rgb[0] : rgb[2]));
+                } else {
+                    double opacity = table->GetOpacity(x);
+                    color[0] = (unsigned char)(255. * (bgColor[0] * (1.0 - opacity) + (bgr ? rgb[2] : rgb[0]) * opacity));
+                    color[1] = (unsigned char)(255. * (bgColor[1] * (1.0 - opacity) + rgb[1] * opacity));
+                    color[2] = (unsigned char)(255. * (bgColor[2] * (1.0 - opacity) + (bgr ? rgb[0] : rgb[2]) * opacity));
+                }
+                memcpy(dst, color, 3);
+                dst += 3;
+                if (bytesPerPixel == 4) {
+                    *(dst++) = (unsigned char)255.;
+                }
+            }
+        }
+    }
+}

trunk/packages/vizservers/vtkvis/ColorMap.h

@@ -17 +17 @@
 #include <vtkPiecewiseFunction.h>
 #include <vtkLookupTable.h>
+#include <vtkUnsignedCharArray.h>
 
 namespace VtkVis {
@@ -115 +116 @@
     static ColorMap *getElementDefault();
 
+    static void renderColorMap(ColorMap *map, int width, int height,
+                               vtkUnsignedCharArray *imgData,
+                               bool opaque, float bgColor[3],
+                               bool bgr = false,
+                               int bytesPerPixel = 3);
+
 private:
     static ColorMap *_default;

trunk/packages/vizservers/vtkvis/Makefile.in

@@ -6 +6 @@
 USE_OFFSCREEN_RENDERING = #yes
 USE_THREADS             = yes
+USE_CPU_LEGEND_RENDER   = yes
 NEW_SCALAR_BAR          = #yes
 
@@ -125 +126 @@
 ifdef USE_THREADS
 DEFINES         += -DUSE_THREADS
+endif
+ifdef USE_CPU_LEGEND_RENDER
+DEFINES         += -DLEGEND_SOFTWARE_RENDER
 endif
 ifdef NEW_SCALAR_BAR

trunk/packages/vizservers/vtkvis/Renderer.cpp

@@ -1919 +1919 @@
     if (colorMap == NULL)
         return false;
-
+#ifdef LEGEND_SOFTWARE_RENDER
+    ColorMap::renderColorMap(colorMap, width, height, imgData, opaque, _bgColor,
+#ifdef RENDER_TARGA
+                             true, TARGA_BYTES_PER_PIXEL
+#else
+                             false
+#endif
+                             );
+#else
     if (_legendRenderWindow == NULL) {
         _legendRenderWindow = vtkSmartPointer<vtkRenderWindow>::New();
@@ -2212 +2220 @@
                                       imgData);
 #endif
+#endif
     TRACE("Leave");
     return true;
@@ -2225 +2234 @@
     if (colorMap == NULL)
         return false;
-
+#ifdef LEGEND_SOFTWARE_RENDER
+    ColorMap::renderColorMap(colorMap, width, height, imgData, opaque, _bgColor,
+#ifdef RENDER_TARGA
+                             true, TARGA_BYTES_PER_PIXEL
+#else
+                             false
+#endif
+                             );
+#else
     if (_legendRenderWindow == NULL) {
         _legendRenderWindow = vtkSmartPointer<vtkRenderWindow>::New();
@@ -2354 +2371 @@
                                       !_legendRenderWindow->GetDoubleBuffer(),
                                       imgData);
+#endif
 #endif
     TRACE("Leave");