use crate::core::additional_data::AdditionalData;
use crate::signal::control::Control;
use std::borrow::Cow;
use vec1::Vec1;
#[derive(Debug, Clone, PartialEq)]
pub struct Controller {
pub control: Vec1<Control>,
pub id: String,
pub name: Option<String>,
pub sequence: Option<u64>,
pub additional_data: AdditionalData,
}
impl Controller {
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,
"id" => Some(self.id.as_str()),
"name" => self.name.as_deref(),
"sequence" => self.sequence.map(|s| s.to_string()).as_deref(),
)
}
pub fn visit_children(
&self,
mut visitor: impl FnMut(xml::writer::XmlEvent) -> xml::writer::Result<()>,
) -> xml::writer::Result<()> {
for control in &self.control {
visit_children!(visitor, "control" => control);
}
self.additional_data.append_children(visitor)
}
}
impl<'a, I> TryFrom<crate::parser::ReadContext<'a, I>> for Controller
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 control = Vec::new();
let mut additional_data = AdditionalData::default();
match_child_eq_ignore_ascii_case!(
read,
"control" true => Control => |v| control.push(v),
_ => |_name, context| additional_data.fill(context),
);
Ok(Self {
control: Vec1::try_from_vec(control).unwrap(),
id: read.attribute("id")?,
name: read.attribute_opt("name")?,
sequence: read.attribute_opt("sequence")?,
additional_data,
})
}
}
#[cfg(feature = "fuzzing")]
impl arbitrary::Arbitrary<'_> for Controller {
fn arbitrary(u: &mut arbitrary::Unstructured) -> arbitrary::Result<Self> {
Ok(Self {
control: {
let mut vec1 = Vec1::new(u.arbitrary()?);
vec1.extend(u.arbitrary::<Vec<_>>()?);
vec1
},
id: u.arbitrary()?,
name: u.arbitrary()?,
sequence: u.arbitrary()?,
additional_data: u.arbitrary()?,
})
}
}