osm-io 0.3.0

Read and write OSM data
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use crate::osm::model::way::Way;
use crate::osm::pbf::string_table_builder::StringTableBuilder;
use crate::osmpbf;
use crate::osmpbf::PrimitiveGroup;

pub(crate) struct WaysGroupBuilder {
    ways: Option<Vec<osmpbf::Way>>,
    date_granularity: i32,
}

impl WaysGroupBuilder {
    pub(crate) fn new(date_granularity: i32, way: &Way, string_table_builder: &mut StringTableBuilder) -> WaysGroupBuilder {
        let mut ways = Some(Vec::<osmpbf::Way>::with_capacity(8000));
        ways.as_mut().unwrap().push(Self::convert(way, date_granularity, string_table_builder));

        WaysGroupBuilder {
            ways,
            date_granularity,
        }
    }

    pub(crate) fn add(&mut self, way: &Way, string_table_builder: &mut StringTableBuilder) {
        self.ways.as_mut().unwrap().push(Self::convert(way, self.date_granularity, string_table_builder));
    }

    #[allow(clippy::field_reassign_with_default)]
    fn convert(way: &Way, date_granularity: i32, string_table_builder: &mut StringTableBuilder) -> osmpbf::Way {
        let mut w = osmpbf::Way::default();
        w.id = way.id();
        let mut last_ref = 0;
        for r in way.refs() {
            w.refs.push(r - last_ref);
            last_ref = *r;
        }

        for tag in way.tags() {
            let key_index = string_table_builder.add(tag.k());
            let val_index = string_table_builder.add(tag.v());
            w.keys.push(key_index as u32);
            w.vals.push(val_index as u32)
        }
        w.info = Some(osmpbf::Info::default());
        w.info.as_mut().unwrap().visible = Some(way.visible());
        w.info.as_mut().unwrap().uid = Some(way.uid());
        w.info.as_mut().unwrap().changeset = Some(way.changeset());
        w.info.as_mut().unwrap().timestamp = Some(way.timestamp() / date_granularity as i64);
        w.info.as_mut().unwrap().version = Some(way.version());
        w.info.as_mut().unwrap().user_sid = Some(string_table_builder.add(way.user()) as u32);
        w
    }

    #[allow(clippy::field_reassign_with_default)]
    pub(crate) fn build(&mut self) -> PrimitiveGroup {
        let mut primitive_group = PrimitiveGroup::default();
        primitive_group.ways = self.ways.replace(Vec::<osmpbf::Way>::new()).unwrap();
        primitive_group
    }
}