Changeset 3131 for trunk

Timestamp:

Aug 2, 2012, 1:58:53 PM (12 years ago)

Author:

ldelgass

Message:

First pass at atom label support in VTK molecules

Location:

trunk/packages/vizservers/vtkvis

Files:

• RpMolecule.cpp (modified) (12 diffs)
• RpMolecule.h (modified) (5 diffs)
• RpVtkGraphicsObject.h (modified) (4 diffs)
• RpVtkRendererGraphicsObjs.h (modified) (3 diffs)

trunk/packages/vizservers/vtkvis/RpMolecule.cpp

@@ -6 +6 @@
  */
 
+#include <cstdio>
 #include <cassert>
 
@@ -11 +12 @@
 #include <vtkPointData.h>
 #include <vtkFloatArray.h>
+#include <vtkStringArray.h>
 #include <vtkPolyData.h>
 #include <vtkPolyDataMapper.h>
@@ -18 +20 @@
 #include <vtkSphereSource.h>
 #include <vtkGlyph3D.h>
+#include <vtkPointSetToLabelHierarchy.h>
+#include <vtkLabelPlacementMapper.h>
+#include <vtkTextProperty.h>
 
 #include "RpMolecule.h"
@@ -29 +34 @@
     _radiusScale(0.3),
     _atomScaling(VAN_DER_WAALS_RADIUS),
-    _colorMap(NULL)
+    _colorMap(NULL),
+    _labelsOn(false)
 {
     _bondColor[0] = _bondColor[1] = _bondColor[2] = 1.0f;
@@ -75 +81 @@
             _bondProp->GetProperty()->LightingOff();
     }
+    if (_labelProp == NULL) {
+        _labelProp = vtkSmartPointer<vtkActor2D>::New();
+    }
     if (_prop == NULL) {
         _prop = vtkSmartPointer<vtkAssembly>::New();
@@ -99 +108 @@
         _bondMapper->SetResolveCoincidentTopologyToPolygonOffset();
         _bondMapper->ScalarVisibilityOn();
+    }
+    if (_labelMapper == NULL) {
+        _labelMapper = vtkSmartPointer<vtkLabelPlacementMapper>::New();
+        _labelMapper->SetShapeToRoundedRect();
+        _labelMapper->SetBackgroundColor(1.0, 1.0, 0.7);
+        _labelMapper->SetBackgroundOpacity(0.8);
+        _labelMapper->SetMargin(3);
     }
 
@@ -117 +133 @@
     initProp();
 
+    addLabelArray(ds);
     addRadiusArray(ds, _atomScaling, _radiusScale);
 
@@ -141 +158 @@
         }
         if (pd->GetNumberOfVerts() > 0) {
+            vtkSmartPointer<vtkPointSetToLabelHierarchy> hier = vtkSmartPointer<vtkPointSetToLabelHierarchy>::New();
+            hier->SetInput(pd);
+            hier->SetLabelArrayName("labels");
+            hier->GetTextProperty()->SetColor(0, 0, 0);
+            _labelMapper->SetInputConnection(hier->GetOutputPort());
+            _labelProp->SetMapper(_labelMapper);
+
             // Atoms
             vtkSmartPointer<vtkSphereSource> sphereSource = vtkSmartPointer<vtkSphereSource>::New();
@@ -171 +195 @@
     }
 
+    setAtomLabelVisibility(_labelsOn);
+
     _atomMapper->Update();
     _bondMapper->Update();
+    _labelMapper->Update();
 }
 
@@ -237 +264 @@
 
 /**
+ * \brief Turn on/off rendering of atom labels
+ */
+void Molecule::setAtomLabelVisibility(bool state)
+{
+    _labelsOn = state;
+    if (_labelProp != NULL) {
+        _labelProp->SetVisibility((state ? 1 : 0));
+    }
+}
+
+/**
  * \brief Turn on/off rendering of the atoms
  */
@@ -253 +291 @@
     if (_bondProp != NULL) {
         _bondProp->SetVisibility((state ? 1 : 0));
+    }
+}
+
+/**
+ * \brief Toggle visibility of the prop
+ */
+void Molecule::setVisibility(bool state)
+{
+    VtkGraphicsObject::setVisibility(state);
+    if (_labelProp != NULL) {
+        if (!state)
+            _labelProp->SetVisibility(0);
+        else
+            setAtomLabelVisibility(_labelsOn);
+    }
+}
+
+/**
+ * \brief Set opacity of molecule
+ */
+void Molecule::setOpacity(double opacity)
+{
+    VtkGraphicsObject::setOpacity(opacity);
+    if (_labelMapper != NULL) {
+        _labelMapper->SetBackgroundOpacity(opacity);
     }
 }
@@ -341 +404 @@
     if (_tuber != NULL)
         _tuber->SetRadius(scale);
+}
+
+void Molecule::addLabelArray(vtkDataSet *dataSet)
+{
+    vtkDataArray *elements = NULL;
+    if (dataSet->GetPointData() != NULL &&
+        dataSet->GetPointData()->GetScalars() != NULL &&
+        strcmp(dataSet->GetPointData()->GetScalars()->GetName(), "element") == 0) {
+        elements = dataSet->GetPointData()->GetScalars();
+    } else {
+        WARN("Can't label atoms without an element array");
+    }
+
+    vtkSmartPointer<vtkStringArray> labelArray = vtkSmartPointer<vtkStringArray>::New();
+    labelArray->SetName("labels");
+    vtkPolyData *pd = vtkPolyData::SafeDownCast(dataSet);
+    if (pd == NULL) {
+        ERROR("DataSet not a PolyData");
+        return;
+    }
+    for (int i = 0; i < pd->GetNumberOfVerts(); i++) {
+        char buf[32];
+        if (elements != NULL) {
+            int elt = (int)elements->GetComponent(i, 0);
+            sprintf(buf, "%s%d", g_elementNames[elt], i);
+        } else {
+            sprintf(buf, "%d", i);
+            labelArray->InsertNextValue(buf);
+        }
+        labelArray->InsertNextValue(buf);
+    }
+    dataSet->GetPointData()->AddArray(labelArray);
 }
 

trunk/packages/vizservers/vtkvis/RpMolecule.h

@@ -13 +13 @@
 #include <vtkPolyDataMapper.h>
 #include <vtkActor.h>
+#include <vtkActor2D.h>
 #include <vtkAssembly.h>
 #include <vtkTubeFilter.h>
 #include <vtkGlyph3D.h>
+#include <vtkLabelPlacementMapper.h>
 
 #include "ColorMap.h"
@@ -55 +57 @@
     }
 
+    virtual vtkProp *getOverlayProp()
+    {
+        return _labelProp;
+    }
+
     virtual void setClippingPlanes(vtkPlaneCollection *planes);
 
@@ -77 +84 @@
     void setBondRadiusScale(double scale);
 
+    virtual void setVisibility(bool state);
+
+    virtual void setOpacity(double opacity);
+
     void setAtomVisibility(bool state);
+
+    void setAtomLabelVisibility(bool state);
 
     void setBondVisibility(bool state);
@@ -91 +104 @@
     virtual void update();
 
+    static void addLabelArray(vtkDataSet *dataSet);
+
     static void addRadiusArray(vtkDataSet *dataSet, AtomScaling scaling, double scaleFactor);
 
@@ -97 +112 @@
     AtomScaling _atomScaling;
     ColorMap *_colorMap;
+    bool _labelsOn;
 
     vtkSmartPointer<vtkLookupTable> _lut;
     vtkSmartPointer<vtkActor> _atomProp;
     vtkSmartPointer<vtkActor> _bondProp;
+    vtkSmartPointer<vtkActor2D> _labelProp;
     vtkSmartPointer<vtkGlyph3D> _glypher;
     vtkSmartPointer<vtkTubeFilter> _tuber;
     vtkSmartPointer<vtkPolyDataMapper> _atomMapper;
     vtkSmartPointer<vtkPolyDataMapper> _bondMapper;
+    vtkSmartPointer<vtkLabelPlacementMapper> _labelMapper;
 };
 

trunk/packages/vizservers/vtkvis/RpVtkGraphicsObject.h

@@ -165 +165 @@
     }
 
+    virtual vtkProp *getOverlayProp()
+    {
+        return NULL;
+    }
+
     /**
      * \brief Set an additional transform on the prop
@@ -242 +247 @@
 
     /**
+     * \brief Set 2D aspect ratio scaling
+     *
+     * \param aspect 0=no scaling, otherwise aspect
+     * is horiz/vertical ratio
+     */
+    virtual void setAspect(double aspect)
+    {
+        ;
+    }
+
+    /**
      * \brief Set the prop scaling
      *
@@ -251 +267 @@
             getProp3D()->SetScale(scale);
         }
+    }
+
+    /**
+     * \brief Get the physical bounds of the prop
+     *
+     * If the prop is scaled, these bounds will be
+     * scaled as well
+     */
+    virtual double *getBounds()
+    {
+        if (getProp() != NULL)
+            return getProp()->GetBounds();
+        else
+            return NULL;
+    }
+
+    /**
+     * \brief Get the data bounds of the prop
+     *
+     * If the prop is scaled, these bounds will NOT
+     * be scaled, they will reflect the unscaled data
+     * bounds.
+     */
+    virtual double *getUnscaledBounds()
+    {
+        if (getProp3D() != NULL) {
+            double scale[3];
+            getProp3D()->GetScale(scale);
+            if (scale[0] == scale[1] && scale[1] == scale[2] &&
+                scale[0] == 1.0) {
+                return getBounds();
+            } else if (getBounds() != NULL) {
+                _unscaledBounds[0] = getBounds()[0] / scale[0];
+                _unscaledBounds[1] = getBounds()[1] / scale[0];
+                _unscaledBounds[2] = getBounds()[2] / scale[1];
+                _unscaledBounds[3] = getBounds()[3] / scale[1];
+                _unscaledBounds[4] = getBounds()[4] / scale[2];
+                _unscaledBounds[5] = getBounds()[5] / scale[2];
+                return _unscaledBounds;
+            }
+        }
+        return getBounds();
     }
 
@@ -743 +801 @@
     DataSet *_dataSet;
     double _dataRange[2];
+    double _unscaledBounds[6];
     double _opacity;
     float _color[3];

trunk/packages/vizservers/vtkvis/RpVtkRendererGraphicsObjs.h

@@ -60 +60 @@
         if (gobj->getProp())
             _renderer->RemoveViewProp(gobj->getProp());
+        if (gobj->getOverlayProp())
+            _renderer->RemoveViewProp(gobj->getOverlayProp());
         delete gobj;
 
@@ -100 +102 @@
         gobj->setDataSet(ds, this);
 
-        if (gobj->getProp() == NULL) {
+        if (gobj->getProp() == NULL &&
+            gobj->getOverlayProp() == NULL) {
             delete gobj;
             return false;
         } else {
-            _renderer->AddViewProp(gobj->getProp());
+            if (gobj->getProp())
+                _renderer->AddViewProp(gobj->getProp());
+            if (gobj->getOverlayProp())
+                _renderer->AddViewProp(gobj->getOverlayProp());
         }
 
@@ -233 +239 @@
     do {
         itr->second->setPosition(pos);
+    } while (doAll && ++itr != hashmap.end());
+
+    resetAxes();
+    _needsRedraw = true;
+}
+
+/**
+ * \brief Set the prop scaling by 2D aspect ratio
+ */
+template<class GraphicsObject>
+void Renderer::setGraphicsObjectAspect(const DataSetId& id, double aspect)
+{
+    std::tr1::unordered_map<DataSetId, GraphicsObject *>& hashmap = 
+        getGraphicsObjectHashmap<GraphicsObject>();
+    typename std::tr1::unordered_map<DataSetId, GraphicsObject *>::iterator itr;
+
+    bool doAll = false;
+
+    if (id.compare("all") == 0) {
+        itr = hashmap.begin();
+        doAll = true;
+    } else {
+        itr = hashmap.find(id);
+    }
+    if (itr == hashmap.end()) {
+        ERROR("%s not found: %s", typeid(GraphicsObject).name(), id.c_str());
+        return;
+    }
+
+    do {
+        itr->second->setAspect(aspect);
     } while (doAll && ++itr != hashmap.end());