Changeset 3641 for trunk

Timestamp:

Apr 28, 2013, 5:29:39 PM (11 years ago)

Author:

ldelgass

Message:

Add protocol to set field used for atom labels in vtkvis molecules. Also use
point data for atoms even if VTK file doesn't specify vertices.

Location:

trunk/packages/vizservers/vtkvis

Files:

• Molecule.cpp (modified) (7 diffs)
• Molecule.h (modified) (3 diffs)
• Renderer.h (modified) (1 diff)
• RendererCmd.cpp (modified) (2 diffs)
• RendererGraphicsObjs.cpp (modified) (1 diff)
• protocol.txt (modified) (1 diff)

trunk/packages/vizservers/vtkvis/Molecule.cpp

@@ -181 +181 @@
     vtkPolyData *pd = vtkPolyData::SafeDownCast(ds);
     if (pd) {
-        TRACE("Verts: %d Lines: %d Polys: %d Strips: %d",
-                  pd->GetNumberOfVerts(),
-                  pd->GetNumberOfLines(),
-                  pd->GetNumberOfPolys(),
-                  pd->GetNumberOfStrips());
+        TRACE("Points: %d Verts: %d Lines: %d Polys: %d Strips: %d",
+              pd->GetNumberOfPoints(),
+              pd->GetNumberOfVerts(),
+              pd->GetNumberOfLines(),
+              pd->GetNumberOfPolys(),
+              pd->GetNumberOfStrips());
         // DataSet is a vtkPolyData
         if (pd->GetNumberOfLines() > 0) {
@@ -226 +227 @@
             getAssembly()->AddPart(_bondProp);
         }
-        if (pd->GetNumberOfVerts() > 0) {
-            vtkSmartPointer<vtkPointSetToLabelHierarchy> hier = vtkSmartPointer<vtkPointSetToLabelHierarchy>::New();
+        if (pd->GetNumberOfPoints() > 0) {
+            if (_labelHierarchy == NULL) {
+                _labelHierarchy = vtkSmartPointer<vtkPointSetToLabelHierarchy>::New();
+            }
 #ifdef USE_VTK6
-            hier->SetInputData(pd);
+            _labelHierarchy->SetInputData(pd);
 #else
-            hier->SetInput(pd);
+            _labelHierarchy->SetInput(pd);
 #endif
-            hier->SetLabelArrayName("labels");
-            hier->GetTextProperty()->SetColor(0, 0, 0);
-            _labelMapper->SetInputConnection(hier->GetOutputPort());
+            _labelHierarchy->SetLabelArrayName("_atom_labels");
+            _labelHierarchy->GetTextProperty()->SetColor(0, 0, 0);
+            _labelMapper->SetInputConnection(_labelHierarchy->GetOutputPort());
             _labelProp->SetMapper(_labelMapper);
 
@@ -588 +591 @@
 }
 
+void Molecule::setAtomLabelField(const char *fieldName)
+{
+    if (_labelHierarchy != NULL) {
+        if (strcmp(fieldName, "default") == 0) {
+            _labelHierarchy->SetLabelArrayName("_atom_labels");
+        } else {
+            _labelHierarchy->SetLabelArrayName(fieldName);
+        }
+    }
+}
+
 /**
  * \brief Turn on/off rendering of atom labels
@@ -866 +880 @@
 
     vtkSmartPointer<vtkStringArray> labelArray = vtkSmartPointer<vtkStringArray>::New();
-    labelArray->SetName("labels");
+    labelArray->SetName("_atom_labels");
     vtkPolyData *pd = vtkPolyData::SafeDownCast(dataSet);
     if (pd == NULL) {
@@ -872 +886 @@
         return;
     }
-    for (int i = 0; i < pd->GetNumberOfVerts(); i++) {
+    for (int i = 0; i < pd->GetNumberOfPoints(); i++) {
         char buf[32];
         if (elements != NULL) {
@@ -915 +929 @@
     }
     vtkSmartPointer<vtkFloatArray> radii = vtkSmartPointer<vtkFloatArray>::New();
-    radii->SetName("radius");
+    radii->SetName("_radii");
     radii->SetNumberOfComponents(3);
     vtkPolyData *pd = vtkPolyData::SafeDownCast(dataSet);
@@ -922 +936 @@
         return;
     }
-    for (int i = 0; i < pd->GetNumberOfVerts(); i++) {
+    for (int i = 0; i < pd->GetNumberOfPoints(); i++) {
         float tuple[3];
         tuple[1] = tuple[2] = 0;

trunk/packages/vizservers/vtkvis/Molecule.h

@@ -19 +19 @@
 #include <vtkCylinderSource.h>
 #include <vtkLineSource.h>
+#include <vtkPointSetToLabelHierarchy.h>
 #include <vtkLabelPlacementMapper.h>
 
@@ -117 +118 @@
     void setAtomLabelVisibility(bool state);
 
+    void setAtomLabelField(const char *fieldName);
+
     void setBondVisibility(bool state);
 
@@ -160 +163 @@
     vtkSmartPointer<vtkGlyph3DMapper> _atomMapper;
     vtkSmartPointer<vtkGlyph3DMapper> _bondMapper;
+    vtkSmartPointer<vtkPointSetToLabelHierarchy> _labelHierarchy;
     vtkSmartPointer<vtkLabelPlacementMapper> _labelMapper;
 };

trunk/packages/vizservers/vtkvis/Renderer.h

@@ -703 +703 @@
     void setMoleculeAtomVisibility(const DataSetId& id, bool state);
 
+    void setMoleculeAtomLabelField(const DataSetId& id, const char *fieldName);
+
     void setMoleculeAtomLabelVisibility(const DataSetId& id, bool state);
 

trunk/packages/vizservers/vtkvis/RendererCmd.cpp

@@ -6598 +6598 @@
 
 static int
+MoleculeAtomLabelFieldOp(ClientData clientData, Tcl_Interp *interp, int objc, 
+                         Tcl_Obj *const *objv)
+{
+    const char *fieldName = Tcl_GetString(objv[2]);
+    if (objc == 4) {
+        const char *name = Tcl_GetString(objv[3]);
+        g_renderer->setMoleculeAtomLabelField(name, fieldName);
+    } else {
+        g_renderer->setMoleculeAtomLabelField("all", fieldName);
+    }
+    return TCL_OK;
+}
+
+
+static int
 MoleculeAtomLabelVisibilityOp(ClientData clientData, Tcl_Interp *interp, int objc, 
                               Tcl_Obj *const *objv)
@@ -7043 +7058 @@
     {"edges",        1, MoleculeEdgeVisibilityOp, 3, 4, "bool ?dataSetName?"},
     {"labels",       2, MoleculeAtomLabelVisibilityOp, 3, 4, "bool ?dataSetName?"},
+    {"lfield",       2, MoleculeAtomLabelFieldOp, 3, 4, "fieldName ?dataSetName?"},
     {"lighting",     3, MoleculeLightingOp, 3, 4, "bool ?dataSetName?"},
     {"linecolor",    5, MoleculeLineColorOp, 5, 6, "r g b ?dataSetName?"},

trunk/packages/vizservers/vtkvis/RendererGraphicsObjs.cpp

@@ -1953 +1953 @@
 }
 
+void Renderer::setMoleculeAtomLabelField(const DataSetId& id, const char *fieldName)
+{
+    MoleculeHashmap::iterator itr;
+
+    bool doAll = false;
+
+    if (id.compare("all") == 0) {
+        itr = _molecules.begin();
+        if (itr == _molecules.end())
+            return;
+        doAll = true;
+    } else {
+        itr = _molecules.find(id);
+    }
+    if (itr == _molecules.end()) {
+        ERROR("Molecule not found: %s", id.c_str());
+        return;
+    }
+
+    do {
+        itr->second->setAtomLabelField(fieldName);
+    } while (doAll && ++itr != _molecules.end());
+
+    _needsRedraw = true;
+}
+
 /**
  * \brief Turn on/off rendering of the Molecule atom labels for the given DataSet

trunk/packages/vizservers/vtkvis/protocol.txt

@@ -573 +573 @@
 molecule edges <bool> <?datasetName?>
 molecule labels <bool> <?datasetName?>
+         Toggle display of atom labels
+molecule lfield <fieldName> <?datasetName?>
+         Set the field used for atom labels
 molecule lighting <bool> <?datasetName?>
 molecule linecolor <r> <g> <b> <?datasetName?>