ecitygml-core 0.0.1-alpha.13

Core primitives and operations for processing CityGML data.
Documentation
use crate::model::common::LevelOfDetail;
use crate::model::core::{
    AbstractSpace, AsAbstractCityObject, AsAbstractFeature, AsAbstractSpace, AsAbstractSpaceMut,
    ImplicitGeometry,
};
use egml::model::geometry::Envelope;
use nalgebra::Isometry3;
use std::collections::HashMap;

#[derive(Debug, Clone, PartialEq)]
pub struct AbstractOccupiedSpace {
    pub(crate) abstract_space: AbstractSpace,
    pub lod1_implicit_representation: Option<ImplicitGeometry>,
    pub lod2_implicit_representation: Option<ImplicitGeometry>,
    pub lod3_implicit_representation: Option<ImplicitGeometry>,
}

impl AbstractOccupiedSpace {
    pub fn new(abstract_space: AbstractSpace) -> Self {
        Self {
            abstract_space,
            lod1_implicit_representation: None,
            lod2_implicit_representation: None,
            lod3_implicit_representation: None,
        }
    }
}

pub trait AsAbstractOccupiedSpace: AsAbstractSpace {
    fn abstract_occupied_space(&self) -> &AbstractOccupiedSpace;

    fn lod1_implicit_representation(&self) -> Option<&ImplicitGeometry> {
        self.abstract_occupied_space()
            .lod1_implicit_representation
            .as_ref()
    }

    fn lod2_implicit_representation(&self) -> Option<&ImplicitGeometry> {
        self.abstract_occupied_space()
            .lod2_implicit_representation
            .as_ref()
    }

    fn lod3_implicit_representation(&self) -> Option<&ImplicitGeometry> {
        self.abstract_occupied_space()
            .lod3_implicit_representation
            .as_ref()
    }

    fn implicit_representations_by_lod(&self) -> HashMap<LevelOfDetail, &ImplicitGeometry> {
        let mut map = HashMap::new();
        if let Some(x) = self.lod1_implicit_representation() {
            map.insert(LevelOfDetail::Zero, x);
        }
        if let Some(x) = self.lod2_implicit_representation() {
            map.insert(LevelOfDetail::One, x);
        }
        if let Some(x) = self.lod3_implicit_representation() {
            map.insert(LevelOfDetail::Two, x);
        }
        map
    }

    fn compute_envelope(&self) -> Option<Envelope> {
        let envelopes: Vec<Envelope> = vec![
            self.abstract_space().compute_envelope(),
            self.lod1_implicit_representation()
                .and_then(|x| x.compute_envelope()),
            self.lod2_implicit_representation()
                .and_then(|x| x.compute_envelope()),
            self.lod3_implicit_representation()
                .and_then(|x| x.compute_envelope()),
        ]
        .into_iter()
        .flatten()
        .collect();

        let refs: Vec<&Envelope> = envelopes.iter().collect();
        Envelope::from_envelopes(&refs)
    }
}

pub trait AsAbstractOccupiedSpaceMut: AsAbstractSpaceMut + AsAbstractOccupiedSpace {
    fn abstract_occupied_space_mut(&mut self) -> &mut AbstractOccupiedSpace;

    fn refresh_bounded_by(&mut self) {
        let envelope = AsAbstractOccupiedSpace::compute_envelope(self);
        self.set_bounded_by(envelope);
    }

    fn apply_transform(&mut self, m: &Isometry3<f64>) {
        AsAbstractSpaceMut::apply_transform(self, m);

        if let Some(g) = &mut self
            .abstract_occupied_space_mut()
            .lod1_implicit_representation
        {
            g.apply_transform(m);
        }
        if let Some(g) = &mut self
            .abstract_occupied_space_mut()
            .lod2_implicit_representation
        {
            g.apply_transform(m);
        }
        if let Some(g) = &mut self
            .abstract_occupied_space_mut()
            .lod3_implicit_representation
        {
            g.apply_transform(m);
        }
    }
}

impl AsAbstractOccupiedSpace for AbstractOccupiedSpace {
    fn abstract_occupied_space(&self) -> &AbstractOccupiedSpace {
        self
    }
}

impl AsAbstractOccupiedSpaceMut for AbstractOccupiedSpace {
    fn abstract_occupied_space_mut(&mut self) -> &mut AbstractOccupiedSpace {
        self
    }
}

#[macro_export]
macro_rules! impl_abstract_occupied_space_traits {
    ($type:ty) => {
        $crate::impl_abstract_space_traits!($type);

        impl $crate::model::core::AsAbstractSpace for $type {
            fn abstract_space(&self) -> &$crate::model::core::AbstractSpace {
                use $crate::model::core::AsAbstractOccupiedSpace;
                &self.abstract_occupied_space().abstract_space
            }
        }

        impl $crate::model::core::AsAbstractSpaceMut for $type {
            fn abstract_space_mut(&mut self) -> &mut $crate::model::core::AbstractSpace {
                use $crate::model::core::AsAbstractOccupiedSpaceMut;
                &mut self.abstract_occupied_space_mut().abstract_space
            }
        }
    };
}

impl_abstract_occupied_space_traits!(AbstractOccupiedSpace);