Override StreamableExecutor::getSchema

Author: gsakkis

pdal/StreamableExecutor.cpp

@@ -229,5 +229,16 @@ void StreamableExecutor::done()
     m_executed = true;
 }
 
+std::string StreamableExecutor::getSchema() const
+{
+    if (!m_executed)
+        throw pdal_error("Pipeline has not been executed!");
+
+    std::stringstream strm;
+    MetadataNode root = m_table.layout()->toMetadata().clone("schema");
+    pdal::Utils::toJSON(root, strm);
+    return strm.str();
+}
+
 } // namespace python
 } // namespace pdal

pdal/StreamableExecutor.hpp

@@ -88,6 +88,7 @@ class StreamableExecutor : public PipelineExecutor
     StreamableExecutor(std::string const& json, point_count_t chunkSize, int prefetch);
     ~StreamableExecutor();
 
+    std::string getSchema() const;
     PyArrayObject* executeNext();
     void stop();
 

pdal/libpdalpython.pyx

@@ -248,6 +248,10 @@ cdef class PipelineIterator(PipelineResultsMixin):
     def __dealloc__(self):
         self._executor.get().stop()
 
+    @property
+    def schema(self):
+        return json.loads(self._executor.get().getSchema())
+
     def __iter__(self):
         return self
 

test/test_pipeline.py

@@ -445,7 +445,6 @@ def test_metadata(self):
 
         assert r.metadata == it.metadata
 
-    @pytest.mark.xfail
     def test_schema(self):
         """Fetching a schema works"""
         r = get_pipeline("range.json")