good

cubao · Nov 12, 2023 · 5df343c · 5df343c
1 parent d68e73d
commit 5df343c
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 2 deletions.
diff --git a/headers b/headers
diff --git a/src/geobuf/geobuf_plus.hpp b/src/geobuf/geobuf_plus.hpp
@@ -3,12 +3,31 @@
 #include "planet.hpp"
 #include <spdlog/spdlog.h>
 
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <mio/mio.hpp>
+
 namespace cubao
 {
 
 struct GeobufPlus
 {
     GeobufPlus() = default;
+    mio::shared_ummap_source mmap;
+
+    bool mmap_init(const std::string &path) {
+        mmap = std::make_shared<mio::ummap_source>(path);
+        int cursor = 10;
+        if (std::string((const char *)mmap.data(), cursor) != "GeobufPlus") {
+            spdlog::error("invalid geobuf plus file, wrong magic");
+            return false;
+        }
+        auto xx =  mmap[cursor];
+        int num_features = *(const int*)(mmap.data() + cursor);
+        spdlog::info("#features: {}", num_features);
+        return true;
+    }
 
     void init(const std::string &header_bytes) {}
     mapbox::geojson::feature_collection decode(const std::string &bytes) const

diff --git a/src/main.cpp b/src/main.cpp
@@ -280,6 +280,7 @@ PYBIND11_MODULE(_pybind11_geobuf, m)
     using GeobufPlus = cubao::GeobufPlus;
     py::class_<GeobufPlus>(m, "GeobufPlus", py::module_local()) //
         .def(py::init<>())
+        .def("mmap_init", &GeobufPlus::mmap_init, "path"_a)
         .def("init", &GeobufPlus::init, "header_bytes"_a)
         //
         .def_static(

diff --git a/tests/test_geobuf.py b/tests/test_geobuf.py
@@ -1893,7 +1893,10 @@ def test_query():
     from pybind11_geobuf import GeobufPlus
 
     ipath = f"{__pwd}/../data/suzhoubeizhan.json"
-    print(GeobufPlus.encode(ipath, f"{__pwd}/../build/export.pbf"))
+    opath = f"{__pwd}/../build/export.pbf"
+    # print(GeobufPlus.encode(ipath, opath))
+    gbp = GeobufPlus()
+    print(gbp.mmap_init(opath))
 
     np.set_printoptions(suppress=True)
     pwd = os.path.abspath(os.path.dirname(__file__))