implement QuickInfo support (#109)

Author: hobu

pdal/PyPipeline.cpp

@@ -34,6 +34,7 @@
 
 #include "PyArray.hpp"
 #include "PyPipeline.hpp"
+#include <pdal/util/Utils.hpp>
 
 #ifndef _WIN32
 #include <dlfcn.h>
@@ -132,6 +133,74 @@ std::string PipelineExecutor::getSchema() const
 }
 
 
+MetadataNode computePreview(Stage* stage)
+{
+    if (!stage)
+        throw pdal_error("no valid stage in QuickInfo");
+
+    QuickInfo qi = stage->preview();
+    if (!qi.valid())
+        throw pdal_error("No summary data available for stage '" + stage->getName()+"'" );
+
+    std::stringstream strm;
+    MetadataNode summary(stage->getName());
+    summary.add("num_points", qi.m_pointCount);
+    if (qi.m_srs.valid())
+    {
+        MetadataNode srs = qi.m_srs.toMetadata();
+        summary.add(srs);
+    }
+    if (qi.m_bounds.valid())
+    {
+        MetadataNode bounds = Utils::toMetadata(qi.m_bounds);
+        summary.add(bounds.clone("bounds"));
+    }
+
+    std::string dims;
+    auto di = qi.m_dimNames.begin();
+    while (di != qi.m_dimNames.end())
+    {
+        dims += *di;
+        ++di;
+        if (di != qi.m_dimNames.end())
+           dims += ", ";
+    }
+    if (dims.size())
+        summary.add("dimensions", dims);
+    pdal::Utils::toJSON(summary, strm);
+    return summary;
+
+}
+
+
+std::string PipelineExecutor::getQuickInfo() const
+{
+
+    Stage* stage(nullptr);
+    std::vector<Stage *> stages = m_manager.stages();
+    std::vector<Stage *> previewStages;
+
+    for (auto const& s: stages)
+    {
+        auto n = s->getName();
+        auto v = pdal::Utils::split2(n,'.');
+        if (v.size() > 0)
+            if (pdal::Utils::iequals(v[0], "readers"))
+                previewStages.push_back(s);
+    }
+
+    MetadataNode summary;
+    for (auto const& stage: previewStages)
+    {
+        MetadataNode n = computePreview(stage);
+        summary.add(n);
+    }
+
+    std::stringstream strm;
+    pdal::Utils::toJSON(summary, strm);
+    return strm.str();
+}
+
 void PipelineExecutor::addArrayReaders(std::vector<std::shared_ptr<Array>> arrays)
 {
     // Make the symbols in pdal_base global so that they're accessible

pdal/PyPipeline.hpp

@@ -59,6 +59,7 @@ class PDAL_DLL PipelineExecutor {
     const PointViewSet& views() const;
     std::string getPipeline() const;
     std::string getMetadata() const;
+    std::string getQuickInfo() const;
     std::string getSchema() const;
     std::string getLog() const { return m_logStream.str(); }
 

pdal/libpdalpython.cpp

@@ -95,6 +95,8 @@ namespace pdal {
 
         std::string getPipeline() { return getExecutor()->getPipeline(); }
 
+        std::string getQuickInfo() { return getExecutor()->getQuickInfo(); }
+
         std::string getMetadata() { return getExecutor()->getMetadata(); }
 
         py::object getSchema() {
@@ -161,6 +163,7 @@ namespace pdal {
         .def_property_readonly("log", &Pipeline::getLog)
         .def_property_readonly("schema", &Pipeline::getSchema)
         .def_property_readonly("pipeline", &Pipeline::getPipeline)
+        .def_property_readonly("quickinfo", &Pipeline::getQuickInfo)
         .def_property_readonly("metadata", &Pipeline::getMetadata)
         .def_property_readonly("arrays", &Pipeline::getArrays)
         .def_property_readonly("meshes", &Pipeline::getMeshes)

test/test_pipeline.py

@@ -362,6 +362,12 @@ def test_only_readers(self):
         assert count2 == 2 * count1
         np.testing.assert_array_equal(np.concatenate([array1, array1]), array2)
 
+    def test_quickinfo(self):
+        r = pdal.Reader("test/data/autzen-utm.las")
+        p = r.pipeline()
+        info = json.loads(p.quickinfo)
+        assert 'readers.las' in info.keys()
+        assert info['readers.las']['num_points'] == 1065
 
 class TestArrayLoad:
     def test_merged_arrays(self):