Changeset 2542 for branches/blt4/packages/vizservers/vtkvis/RpStreamlines.cpp

Timestamp:

Sep 21, 2011, 2:27:44 PM (13 years ago)

Author:

gah

Message:

update from trunk

File:

• branches/blt4/packages/vizservers/vtkvis/RpStreamlines.cpp (modified) (16 diffs)

branches/blt4/packages/vizservers/vtkvis/RpStreamlines.cpp

@@ -36 +36 @@
     _lineType(LINES),
     _colorMode(COLOR_BY_VECTOR_MAGNITUDE),
-    _seedVisible(true)
+    _colorMap(NULL),
+    _seedVisible(true),
+    _dataScale(1)
 {
     _faceCulling = true;
@@ -51 +53 @@
 Streamlines::~Streamlines()
 {
+}
+
+void Streamlines::setDataSet(DataSet *dataSet,
+                             bool useCumulative,
+                             double scalarRange[2],
+                             double vectorMagnitudeRange[2],
+                             double vectorComponentRange[3][2])
+{
+    if (_dataSet != dataSet) {
+        _dataSet = dataSet;
+
+        if (useCumulative) {
+            _dataRange[0] = scalarRange[0];
+            _dataRange[1] = scalarRange[1];
+            _vectorMagnitudeRange[0] = vectorMagnitudeRange[0];
+            _vectorMagnitudeRange[1] = vectorMagnitudeRange[1];
+            for (int i = 0; i < 3; i++) {
+                _vectorComponentRange[i][0] = vectorComponentRange[i][0];
+                _vectorComponentRange[i][1] = vectorComponentRange[i][1];
+            }
+        } else {
+            _dataSet->getScalarRange(_dataRange);
+            _dataSet->getVectorRange(_vectorMagnitudeRange);
+            for (int i = 0; i < 3; i++) {
+                _dataSet->getVectorRange(_vectorComponentRange[i], i);
+            }
+        }
+
+        update();
+    }
 }
 
@@ -96 +128 @@
         _seedActor->GetProperty()->SetRepresentationToWireframe();
         _seedActor->GetProperty()->LightingOff();
+        setSeedVisibility(_seedVisible);
     }
     if (_prop == NULL) {
@@ -201 +234 @@
 
     vtkDataSet *ds = _dataSet->getVtkDataSet();
-    double dataRange[2];
-    _dataSet->getVectorMagnitudeRange(dataRange);
+
     double bounds[6];
     _dataSet->getBounds(bounds);
@@ -215 +247 @@
         maxBound = bounds[5] - bounds[4];
     }
+
+    double cellSizeRange[2];
+    double avgSize;
+    _dataSet->getCellSizeRange(cellSizeRange, &avgSize);
+    _dataScale = avgSize / 8.;
 
     vtkSmartPointer<vtkCellDataToPointData> cellToPtData;
@@ -294 +331 @@
     _seedActor->SetMapper(_seedMapper);
 
-    _lut = vtkSmartPointer<vtkLookupTable>::New();
-    _lut->SetRange(dataRange);
-    _lut->SetVectorModeToMagnitude();
-
-    _pdMapper->SetScalarModeToUsePointFieldData();
-    if (ds->GetPointData() != NULL &&
-        ds->GetPointData()->GetVectors() != NULL) {
-        TRACE("Vector name: '%s'", ds->GetPointData()->GetVectors()->GetName());
-        _pdMapper->SelectColorArray(ds->GetPointData()->GetVectors()->GetName());
-    }
-    _pdMapper->SetColorModeToMapScalars();
-    _pdMapper->UseLookupTableScalarRangeOn();
-    _pdMapper->SetLookupTable(_lut);
+    if (_lut == NULL) {
+        setColorMap(ColorMap::getDefault());
+    }
+
+    setColorMode(_colorMode);
 
     _linesActor->SetMapper(_pdMapper);
@@ -724 +753 @@
             numSides = 3;
         tubeFilter->SetNumberOfSides(numSides);
-        tubeFilter->SetRadius(radius);
+        tubeFilter->SetRadius(_dataScale * radius);
         _pdMapper->SetInputConnection(_lineFilter->GetOutputPort());
         if (_faceCulling && _opacity == 1.0)
@@ -749 +778 @@
         }
         vtkRibbonFilter *ribbonFilter = vtkRibbonFilter::SafeDownCast(_lineFilter);
-        ribbonFilter->SetWidth(width);
+        ribbonFilter->SetWidth(_dataScale * width);
         ribbonFilter->SetAngle(angle);
         ribbonFilter->UseDefaultNormalOn();
@@ -759 +788 @@
 }
 
+void Streamlines::updateRanges(bool useCumulative,
+                               double scalarRange[2],
+                               double vectorMagnitudeRange[2],
+                               double vectorComponentRange[3][2])
+{
+    if (useCumulative) {
+        _dataRange[0] = scalarRange[0];
+        _dataRange[1] = scalarRange[1];
+        _vectorMagnitudeRange[0] = vectorMagnitudeRange[0];
+        _vectorMagnitudeRange[1] = vectorMagnitudeRange[1];
+        for (int i = 0; i < 3; i++) {
+            _vectorComponentRange[i][0] = vectorComponentRange[i][0];
+            _vectorComponentRange[i][1] = vectorComponentRange[i][1];
+        }
+    } else {
+        _dataSet->getScalarRange(_dataRange);
+        _dataSet->getVectorRange(_vectorMagnitudeRange);
+        for (int i = 0; i < 3; i++) {
+            _dataSet->getVectorRange(_vectorComponentRange[i], i);
+        }
+    }
+
+    // Need to update color map ranges and/or active vector field
+    setColorMode(_colorMode);
+}
+
 void Streamlines::setColorMode(ColorMode mode)
 {
@@ -772 +827 @@
         _pdMapper->SetScalarModeToDefault();
         if (_lut != NULL) {
-            double dataRange[2];
-            _dataSet->getDataRange(dataRange);
-            _lut->SetRange(dataRange);
+            _lut->SetRange(_dataRange);
         }
     }
@@ -786 +839 @@
         }
         if (_lut != NULL) {
-            double dataRange[2];
-            _dataSet->getVectorMagnitudeRange(dataRange);
-            TRACE("vmag range: %g %g", dataRange[0], dataRange[1]);
-            _lut->SetRange(dataRange);
+            _lut->SetRange(_vectorMagnitudeRange);
             _lut->SetVectorModeToMagnitude();
         }
@@ -802 +852 @@
         }
         if (_lut != NULL) {
-            double dataRange[2];
-            _dataSet->getVectorComponentRange(dataRange, 0);
-            _lut->SetRange(dataRange);
+            _lut->SetRange(_vectorComponentRange[0]);
             _lut->SetVectorModeToComponent();
             _lut->SetVectorComponent(0);
@@ -817 +865 @@
         }
         if (_lut != NULL) {
-            double dataRange[2];
-            _dataSet->getVectorComponentRange(dataRange, 1);
-            _lut->SetRange(dataRange);
+            _lut->SetRange(_vectorComponentRange[1]);
             _lut->SetVectorModeToComponent();
             _lut->SetVectorComponent(1);
@@ -832 +878 @@
         }
         if (_lut != NULL) {
-            double dataRange[2];
-            _dataSet->getVectorComponentRange(dataRange, 2);
-            TRACE("vz range: %g %g", dataRange[0], dataRange[1]);
-            _lut->SetRange(dataRange);
+            _lut->SetRange(_vectorComponentRange[2]);
             _lut->SetVectorModeToComponent();
             _lut->SetVectorComponent(2);
@@ -848 +891 @@
 
 /**
- * \brief Get the VTK colormap lookup table in use
- */
-vtkLookupTable *Streamlines::getLookupTable()
-{ 
-    return _lut;
+ * \brief Called when the color map has been edited
+ */
+void Streamlines::updateColorMap()
+{
+    setColorMap(_colorMap);
 }
 
@@ -858 +901 @@
  * \brief Associate a colormap lookup table with the DataSet
  */
-void Streamlines::setLookupTable(vtkLookupTable *lut)
-{
-    if (lut == NULL) {
+void Streamlines::setColorMap(ColorMap *cmap)
+{
+    if (cmap == NULL)
+        return;
+
+    _colorMap = cmap;
+ 
+    if (_lut == NULL) {
         _lut = vtkSmartPointer<vtkLookupTable>::New();
-    } else {
-        _lut = lut;
-    }
+        if (_pdMapper != NULL) {
+            _pdMapper->UseLookupTableScalarRangeOn();
+            _pdMapper->SetLookupTable(_lut);
+        }
+    }
+
+    _lut->DeepCopy(cmap->getLookupTable());
 
     switch (_colorMode) {
-    case COLOR_BY_VECTOR_MAGNITUDE: {
-        double dataRange[2];
-        _dataSet->getVectorMagnitudeRange(dataRange);
+    case COLOR_CONSTANT:
+    case COLOR_BY_SCALAR:
+        _lut->SetRange(_dataRange);
+        break;
+    case COLOR_BY_VECTOR_MAGNITUDE:
         _lut->SetVectorModeToMagnitude();
-        _lut->SetRange(dataRange);
-    }
-        break;
-    case COLOR_BY_VECTOR_X: {
-        double dataRange[2];
-        _dataSet->getVectorComponentRange(dataRange, 0);
+        _lut->SetRange(_vectorMagnitudeRange);
+        break;
+    case COLOR_BY_VECTOR_X:
         _lut->SetVectorModeToComponent();
         _lut->SetVectorComponent(0);
-        _lut->SetRange(dataRange);
-    }
-        break;
-    case COLOR_BY_VECTOR_Y: {
-        double dataRange[2];
-        _dataSet->getVectorComponentRange(dataRange, 1);
+        _lut->SetRange(_vectorComponentRange[0]);
+        break;
+    case COLOR_BY_VECTOR_Y:
         _lut->SetVectorModeToComponent();
         _lut->SetVectorComponent(1);
-        _lut->SetRange(dataRange);
-    }
-        break;
-    case COLOR_BY_VECTOR_Z: {
-        double dataRange[2];
-        _dataSet->getVectorComponentRange(dataRange, 2);
+        _lut->SetRange(_vectorComponentRange[1]);
+        break;
+    case COLOR_BY_VECTOR_Z:
         _lut->SetVectorModeToComponent();
         _lut->SetVectorComponent(2);
-        _lut->SetRange(dataRange);
-    }
+        _lut->SetRange(_vectorComponentRange[2]);
         break;
     default:
          break;
-    }
-
-    if (_pdMapper != NULL) {
-        _pdMapper->SetLookupTable(_lut);
     }
 }