use bacnet_encoding::constructed::{decode_shed_level, encode_shed_level};
use bacnet_types::constructed::BACnetShedLevel;
use bacnet_types::enums::{EventState, ObjectType, PropertyIdentifier, Reliability};
use bacnet_types::error::Error;
use bacnet_types::primitives::{Date, ObjectIdentifier, PropertyValue, StatusFlags, Time};
use bytes::BytesMut;
use std::borrow::Cow;
use crate::common::{self, read_common_properties};
use crate::traits::BACnetObject;
mod metadata;
#[cfg(test)]
mod tests;
pub struct LoadControlObject {
oid: ObjectIdentifier,
name: String,
description: String,
present_value: u32,
requested_shed_level: BACnetShedLevel,
expected_shed_level: BACnetShedLevel,
actual_shed_level: BACnetShedLevel,
shed_duration: u64,
start_time: (Date, Time),
status_flags: StatusFlags,
event_state: EventState,
reliability: Reliability,
}
impl LoadControlObject {
pub fn new(instance: u32, name: impl Into<String>) -> Result<Self, Error> {
let oid = ObjectIdentifier::new(ObjectType::LOAD_CONTROL, instance)?;
Ok(Self {
oid,
name: name.into(),
description: String::new(),
present_value: 0,
requested_shed_level: BACnetShedLevel::Level(0),
expected_shed_level: BACnetShedLevel::Level(0),
actual_shed_level: BACnetShedLevel::Level(0),
shed_duration: 0,
start_time: (
Date {
year: 0xFF,
month: 0xFF,
day: 0xFF,
day_of_week: 0xFF,
},
Time {
hour: 0xFF,
minute: 0xFF,
second: 0xFF,
hundredths: 0xFF,
},
),
status_flags: StatusFlags::empty(),
event_state: EventState::NORMAL,
reliability: Reliability::NO_FAULT_DETECTED,
})
}
pub fn set_requested_shed_level(&mut self, level: BACnetShedLevel) -> Result<(), Error> {
match level {
BACnetShedLevel::Percent(percent) if percent > 100 => {
return Err(common::value_out_of_range_error());
}
BACnetShedLevel::Amount(amount) => {
common::reject_non_finite(amount)?;
if amount < 0.0 {
return Err(common::value_out_of_range_error());
}
}
_ => {}
}
let default = choice_default(&level);
self.expected_shed_level = default.clone();
self.actual_shed_level = default;
self.requested_shed_level = level;
Ok(())
}
pub fn set_actual_shed_level(&mut self, level: BACnetShedLevel) -> Result<(), Error> {
if std::mem::discriminant(&level) != std::mem::discriminant(&self.requested_shed_level) {
return Err(common::value_out_of_range_error());
}
if let BACnetShedLevel::Amount(amount) = level {
common::reject_non_finite(amount)?;
}
self.actual_shed_level = level;
Ok(())
}
}
fn choice_default(level: &BACnetShedLevel) -> BACnetShedLevel {
match level {
BACnetShedLevel::Percent(_) => BACnetShedLevel::Percent(100),
BACnetShedLevel::Level(_) => BACnetShedLevel::Level(0),
BACnetShedLevel::Amount(_) => BACnetShedLevel::Amount(0.0),
}
}
fn shed_level_value(level: &BACnetShedLevel) -> PropertyValue {
let mut buf = BytesMut::new();
encode_shed_level(&mut buf, level);
PropertyValue::ApplicationData(buf.to_vec())
}
fn decode_shed_level_write(value: PropertyValue) -> Result<BACnetShedLevel, Error> {
let PropertyValue::ApplicationData(bytes) = value else {
return Err(common::invalid_data_type_error());
};
match decode_shed_level(&bytes, 0) {
Ok((level, end)) if end == bytes.len() => Ok(level),
_ => Err(common::invalid_data_type_error()),
}
}
impl BACnetObject for LoadControlObject {
fn object_identifier(&self) -> ObjectIdentifier {
self.oid
}
fn object_name(&self) -> &str {
&self.name
}
fn read_property(
&self,
property: PropertyIdentifier,
array_index: Option<u32>,
) -> Result<PropertyValue, Error> {
if let Some(result) =
read_common_properties!(self, property, array_index, no_out_of_service)
{
return result;
}
match property {
p if p == PropertyIdentifier::OBJECT_TYPE => {
Ok(PropertyValue::Enumerated(ObjectType::LOAD_CONTROL.to_raw()))
}
p if p == PropertyIdentifier::PRESENT_VALUE => {
Ok(PropertyValue::Enumerated(self.present_value))
}
p if p == PropertyIdentifier::REQUESTED_SHED_LEVEL => {
Ok(shed_level_value(&self.requested_shed_level))
}
p if p == PropertyIdentifier::EXPECTED_SHED_LEVEL => {
Ok(shed_level_value(&self.expected_shed_level))
}
p if p == PropertyIdentifier::ACTUAL_SHED_LEVEL => {
Ok(shed_level_value(&self.actual_shed_level))
}
p if p == PropertyIdentifier::SHED_DURATION => {
Ok(PropertyValue::Unsigned(self.shed_duration))
}
p if p == PropertyIdentifier::START_TIME => Ok(PropertyValue::List(vec![
PropertyValue::Date(self.start_time.0),
PropertyValue::Time(self.start_time.1),
])),
p if p == PropertyIdentifier::EVENT_STATE => {
Ok(PropertyValue::Enumerated(self.event_state.to_raw()))
}
_ => Err(common::unknown_property_error()),
}
}
fn write_property(
&mut self,
property: PropertyIdentifier,
_array_index: Option<u32>,
value: PropertyValue,
_priority: Option<u8>,
) -> Result<(), Error> {
if let Some(result) = common::write_description(&mut self.description, property, &value) {
return result;
}
match property {
p if p == PropertyIdentifier::SHED_DURATION => {
if let PropertyValue::Unsigned(v) = value {
self.shed_duration = v;
Ok(())
} else {
Err(common::invalid_data_type_error())
}
}
p if p == PropertyIdentifier::REQUESTED_SHED_LEVEL => {
self.set_requested_shed_level(decode_shed_level_write(value)?)
}
_ => Err(crate::common::unhandled_write_error(
self.property_metadata().as_ref(),
property,
_array_index,
)),
}
}
fn property_metadata(&self) -> Cow<'_, [crate::property_metadata::PropertyMetadata]> {
metadata::for_object(self)
}
fn property_list(&self) -> Cow<'static, [PropertyIdentifier]> {
crate::property_metadata::property_list_from_metadata(self.property_metadata().as_ref())
}
fn supports_cov(&self) -> bool {
true
}
}