GeoPackage出力ドライバーの実装

Cargo.toml

@@ -5,6 +5,7 @@ members = [
     "nusamai-geometry",
     "nusamai-gltf",
     "nusamai-geojson",
+    "nusamai-geopackage",
     "nusamai-plateau",
     "nusamai-mvt",
     "nusamai-projection",

nusamai-geopackage/Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "nusamai-geopackage"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+nusamai-geometry = { path = "../nusamai-geometry" }
+nusamai-plateau = { path = "../nusamai-plateau" }
+citygml = {path = "../nusamai-plateau/citygml" }
+
+[dev-dependencies]
+clap = { version = "4.4.11", features = ["derive"] }
+quick-xml = "0.31.0"

nusamai-geopackage/examples/gml2gpkg.rs

@@ -0,0 +1,84 @@
+use citygml::{CityGMLElement, CityGMLReader, Geometries, ParseError, SubTreeReader};
+use clap::Parser;
+use nusamai_plateau::models::CityObject;
+use std::io::BufRead;
+
+#[derive(Parser)]
+struct Args {
+    #[clap(required = true)]
+    filename: String,
+}
+
+struct TopLevelCityObject<'a> {
+    cityobj: citygml::object::FeatureOrData<'a>,
+    geometries: Geometries,
+}
+
+fn toplevel_dispatcher<R: BufRead>(
+    st: &mut SubTreeReader<R>,
+) -> Result<Vec<(CityObject, Geometries)>, ParseError> {
+    let mut items: Vec<(CityObject, Geometries)> = vec![];
+
+    match st.parse_children(|st| match st.current_path() {
+        b"core:cityObjectMember" => {
+            let mut cityobj: CityObject = Default::default();
+            cityobj.parse(st)?;
+            let geometries = st.collect_geometries();
+            items.push((cityobj, geometries));
+            Ok(())
+        }
+        b"gml:boundedBy" | b"app:appearanceMember" => {
+            st.skip_current_element()?;
+            Ok(())
+        }
+        other => Err(ParseError::SchemaViolation(format!(
+            "Unrecognized element {}",
+            String::from_utf8_lossy(other)
+        ))),
+    }) {
+        Ok(_) => Ok(items),
+        Err(e) => {
+            println!("Err: {:?}", e);
+            Err(e)
+        }
+    }
+}
+
+fn main() {
+    let args = Args::parse();
+
+    let reader = std::io::BufReader::new(std::fs::File::open(args.filename).unwrap());
+    let mut xml_reader = quick_xml::NsReader::from_reader(reader);
+
+    let items = match CityGMLReader::new().start_root(&mut xml_reader) {
+        Ok(mut st) => match toplevel_dispatcher(&mut st) {
+            Ok(items) => items,
+            Err(e) => panic!("Err: {:?}", e),
+        },
+        Err(e) => panic!("Err: {:?}", e),
+    };
+
+    let tlc_objs: Vec<_> = items
+        .iter()
+        .map(|(o, g)| {
+            let cityobj = match o.objectify().unwrap() {
+                citygml::object::ObjectValue::FeatureOrData(fod) => fod,
+                _ => panic!("Not a FeatureOrData"),
+            };
+
+            TopLevelCityObject {
+                cityobj,
+                geometries: g.clone(),
+            }
+        })
+        .collect();
+
+    for (i, o) in tlc_objs.iter().enumerate() {
+        println!();
+        println!("{}", i);
+        println!("{:?}", o.cityobj);
+        println!("{:?}", o.geometries);
+    }
+
+    // TODO: tlc_objs を 受け取り、 GeoPackage にして、書き出す
+}

nusamai-geopackage/src/lib.rs

@@ -0,0 +1,14 @@
+pub fn add(left: usize, right: usize) -> usize {
+    left + right
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn it_works() {
+        let result = add(2, 2);
+        assert_eq!(result, 4);
+    }
+}

nusamai-plateau/citygml/src/geometric.rs

@@ -34,7 +34,7 @@
 
 /// Geometries in a toplevel city object and its children.
 #[cfg_attr(feature = "serde", derive(serde::Deserialize, serde::Serialize))]
-#[derive(Debug, Default)]
+#[derive(Debug, Default, Clone)]
 pub struct Geometries {
     pub vertices: Vec<[f64; 3]>,
     pub multipolygon: MultiPolygon<'static, 1, u32>,