Add point count and chunk count to lazwriter (#135)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+### Added
+- **\[Python/C++\]** Added `PointCount` and `ChunkCount` to LazWriter
+- **\[Python/C++\]** Added `FilePath` property to LazWriter and FileWriter
+
 ## [2.5.0] - 2022-10-17
 
 ### Fixed

cpp/include/copc-lib/io/copc_writer.hpp

@@ -90,6 +90,8 @@ class FileWriter : public Writer, laz::BaseFileWriter
 
     void Close() override;
     ~FileWriter() { Close(); }
+
+    std::string FilePath() { return file_path_; }
 };
 
 } // namespace copc

cpp/include/copc-lib/io/laz_base_writer.hpp

@@ -63,6 +63,7 @@ class BaseFileWriter
 
   protected:
     std::fstream f_stream_;
+    std::string file_path_;
 };
 
 } // namespace copc::laz

cpp/include/copc-lib/io/laz_writer.hpp

@@ -33,6 +33,9 @@ class LazWriter : public BaseWriter
     {
         return std::dynamic_pointer_cast<las::LazConfigWriter>(config_);
     }
+
+    uint64_t PointCount() { return point_count_; }
+    uint64_t ChunkCount() { return chunks_.size(); }
 };
 
 class LazFileWriter : BaseFileWriter, public LazWriter
@@ -49,6 +52,8 @@ class LazFileWriter : BaseFileWriter, public LazWriter
         BaseFileWriter::Close();
     };
     ~LazFileWriter() { Close(); };
+
+    std::string FilePath() { return file_path_; }
 };
 
 } // namespace copc::laz

cpp/src/io/laz_base_writer.cpp

@@ -161,6 +161,7 @@ void BaseWriter::Close()
 
 BaseFileWriter::BaseFileWriter(const std::string &file_path)
 {
+    file_path_ = file_path;
     f_stream_.open(file_path.c_str(), std::ios::out | std::ios::binary);
     if (!f_stream_.good())
         throw std::runtime_error("FileWriterBase: Error while opening file path.");

python/bindings.cpp

@@ -419,6 +419,7 @@ PYBIND11_MODULE(_core, m)
             py::arg("scale") = py::none(), py::arg("offset") = py::none(), py::arg("wkt") = py::none(),
             py::arg("extra_bytes_vlr") = py::none(), py::arg("has_extended_stats") = py::none())
         .def_property_readonly("copc_config", &Writer::CopcConfig)
+        .def_property_readonly("path", &FileWriter::FilePath)
         .def("FindNode", &Writer::FindNode)
         .def("Close", &FileWriter::Close)
         .def("AddNode", py::overload_cast<const VoxelKey &, const las::Points &, const VoxelKey &>(&Writer::AddNode),
@@ -439,6 +440,9 @@ PYBIND11_MODULE(_core, m)
     py::class_<laz::LazFileWriter>(m, "LazWriter")
         .def(py::init<const std::string &, const las::LazConfigWriter &>(), py::arg("file_path"), py::arg("config"))
         .def_property_readonly("laz_config", &laz::LazWriter::LazConfig)
+        .def_property_readonly("point_count", &laz::LazWriter::PointCount)
+        .def_property_readonly("chunk_count", &laz::LazWriter::ChunkCount)
+        .def_property_readonly("path", &laz::LazFileWriter::FilePath)
         .def("Close", &laz::LazFileWriter::Close)
         .def("WritePoints", py::overload_cast<const las::Points &>(&laz::LazWriter::WritePoints), py::arg("points"))
         .def("WritePointsCompressed", &laz::LazWriter::WritePointsCompressed, py::arg("compressed_data"),

test/laz_writer_test.cpp

@@ -232,7 +232,11 @@ TEST_CASE("LAZ Write Points", "[LAZ Writer]")
     points.AddPoint(point);
 
     writer.WritePoints(points);
+    REQUIRE(writer.PointCount() == 2);
+    REQUIRE(writer.ChunkCount() == 1);
     writer.WritePoints(points);
+    REQUIRE(writer.PointCount() == 4);
+    REQUIRE(writer.ChunkCount() == 2);
     writer.Close();
 
     // Validate

test/laz_writer_test.py

@@ -213,7 +213,11 @@ def test_write():
     points.AddPoint(p)
 
     writer.WritePoints(points)
+    assert writer.point_count == 2
+    assert writer.chunk_count == 1
     writer.WritePoints(points)
+    assert writer.point_count == 4
+    assert writer.chunk_count == 2
     writer.Close()
 
     reader = copc.LazReader(file_path)