use crate::core::additional_data::AdditionalData;
use crate::lane::road_mark::weight::Weight;
use crate::object::corner_reference::CornerReference;
use crate::object::road_mark_color::RoadMarkColor;
use crate::object::side_type::SideType;
use std::borrow::Cow;
use uom::si::f64::Length;
use uom::si::length::meter;
#[derive(Debug, Clone, PartialEq)]
pub struct Marking {
pub corner_reference: Vec<CornerReference>,
pub color: RoadMarkColor,
pub line_length: Length,
pub side: Option<SideType>,
pub space_length: Length,
pub start_offset: Length,
pub stop_offset: Length,
pub weight: Option<Weight>,
pub width: Option<Length>,
pub z_offset: Option<Length>,
pub additional_data: AdditionalData,
}
impl Marking {
pub fn visit_attributes(
&self,
visitor: impl for<'b> FnOnce(
Cow<'b, [xml::attribute::Attribute<'b>]>,
) -> xml::writer::Result<()>,
) -> xml::writer::Result<()> {
visit_attributes_flatten!(
visitor,
"color" => Some(self.color.as_str()),
"lineLength" => Some(self.line_length.value.to_scientific_string()).as_deref(),
"side" => self.side.as_ref().map(SideType::as_str),
"spaceLength" => Some(self.space_length.value.to_scientific_string()).as_deref(),
"startOffset" => Some(self.start_offset.value.to_scientific_string()).as_deref(),
"stopOffset" => Some(self.stop_offset.value.to_scientific_string()).as_deref(),
"weight" => self.weight.as_ref().map(Weight::as_str),
"width" => self.width.map(|v| v.value.to_scientific_string()).as_deref(),
"zOffset" => self.z_offset.map(|v| v.value.to_scientific_string()).as_deref(),
)
}
pub fn visit_children(
&self,
mut visitor: impl FnMut(xml::writer::XmlEvent) -> xml::writer::Result<()>,
) -> xml::writer::Result<()> {
for corner_reference in &self.corner_reference {
visit_children!(visitor, "cornerReference" => corner_reference);
}
self.additional_data.append_children(visitor)
}
}
impl<'a, I> TryFrom<crate::parser::ReadContext<'a, I>> for Marking
where
I: Iterator<Item = xml::reader::Result<xml::reader::XmlEvent>>,
{
type Error = Box<crate::parser::Error>;
fn try_from(mut read: crate::parser::ReadContext<'a, I>) -> Result<Self, Self::Error> {
let mut corner_reference = Vec::new();
let mut additional_data = AdditionalData::default();
match_child_eq_ignore_ascii_case!(
read,
"cornerReference" => CornerReference => |v| corner_reference.push(v),
_ => |_name, context| additional_data.fill(context),
);
Ok(Self {
corner_reference,
color: read.attribute("color")?,
line_length: read.attribute("lineLength").map(Length::new::<meter>)?,
side: read.attribute_opt("side")?,
space_length: read.attribute("spaceLength").map(Length::new::<meter>)?,
start_offset: read.attribute("startOffset").map(Length::new::<meter>)?,
stop_offset: read.attribute("stopOffset").map(Length::new::<meter>)?,
weight: read.attribute_opt("weight")?,
width: read.attribute_opt("width")?.map(Length::new::<meter>),
z_offset: read.attribute_opt("zOffset")?.map(Length::new::<meter>),
additional_data,
})
}
}
#[cfg(feature = "fuzzing")]
impl arbitrary::Arbitrary<'_> for Marking {
fn arbitrary(u: &mut arbitrary::Unstructured) -> arbitrary::Result<Self> {
use crate::fuzzing::NotNan;
Ok(Self {
corner_reference: u.arbitrary()?,
color: u.arbitrary()?,
line_length: Length::new::<meter>(u.not_nan_f64()?),
side: u.arbitrary()?,
space_length: Length::new::<meter>(u.not_nan_f64()?),
start_offset: Length::new::<meter>(u.not_nan_f64()?),
stop_offset: Length::new::<meter>(u.not_nan_f64()?),
weight: u.arbitrary()?,
width: u
.arbitrary::<Option<()>>()?
.map(|_| u.not_nan_f64())
.transpose()?
.map(Length::new::<meter>),
z_offset: u
.arbitrary::<Option<()>>()?
.map(|_| u.not_nan_f64())
.transpose()?
.map(Length::new::<meter>),
additional_data: u.arbitrary()?,
})
}
}