Changeset 5938 for geovis/trunk/Util.h

Timestamp:

Dec 3, 2015 12:30:50 PM (8 years ago)

Author:

ldelgass

Message:

Try finding FeatureSourceIndexNodes? in scene graph to lookup global ID for a
feature ID.

File:

• geovis/trunk/Util.h (modified) (1 diff)

geovis/trunk/Util.h

@@ -8 +8 @@
 #define GEOVIS_UTIL_H
 
+#include <cstring>
 #include <string>
 
+#include <osg/NodeVisitor>
+#include <osgEarthFeatures/FeatureIndex>
+#include <osgEarthFeatures/FeatureSourceIndexNode>
+
 namespace GeoVis {
+
+class FindFeatureSourceIndexNodeVisitor : public osg::NodeVisitor {
+public:
+    FindFeatureSourceIndexNodeVisitor() :
+        osg::NodeVisitor(TRAVERSE_ALL_CHILDREN),
+        source(NULL)
+    {
+    }
+
+    virtual ~FindFeatureSourceIndexNodeVisitor() {}
+
+    virtual void apply(osg::Node& node)
+    {
+        if (strcmp(node.className(), "FeatureSourceIndexNode") == 0) {
+            osgEarth::Features::FeatureSourceIndexNode *fsin = dynamic_cast<osgEarth::Features::FeatureSourceIndexNode *>(&node);
+            osgEarth::Features::FeatureSourceIndex *index = fsin->getIndex();
+            if (source == NULL || index->getFeatureSource() == source) {
+                nodes.push_back(fsin);
+            }
+        }
+        traverse(node);
+    }
+
+    osgEarth::Features::FeatureSource *source;
+    std::vector<osgEarth::Features::FeatureSourceIndexNode *> nodes;
+};
 
 extern std::string loadCSVLayer(const char *filename, const char *longitudeColName = "longitude", const char *latitudeColName = "latitude");