use crate::{
ast::{
join_fmt, CodeFormatter, GroupComments, GroupFn, Indentation, ParseScope, SimpleAttri,
},
ArcStr,
};
use core::{
cmp::Ordering,
fmt::{self, Write},
hash,
str::FromStr,
};
use itertools::Itertools;
use ordered_float::NotNan;
use std::collections::HashSet;
use strum_macros::{Display, EnumString};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[derive(Default, EnumString, Display)]
#[derive(serde::Serialize, serde::Deserialize)]
pub enum SdfEdgeType {
#[default]
#[strum(serialize = "noedge")]
Noedge,
#[strum(serialize = "start_edge")]
StartEdge,
#[strum(serialize = "end_edge")]
EndEdge,
#[strum(serialize = "both_edges")]
BothEdges,
}
#[derive(Debug, Clone, Default)]
#[mut_set::derive::item(
sort,
macro(derive(Debug, Clone, Default);)
)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct IdVector {
#[id]
pub id: usize,
pub vec: Vec<NotNan<f64>>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[derive(Display, EnumString)]
#[derive(serde::Serialize, serde::Deserialize)]
pub enum VariableType {
#[strum(serialize = "input_net_transition")]
InputNetTransition,
#[strum(serialize = "normalized_voltage")]
NormalizedVoltage,
#[strum(serialize = "total_output_net_capacitance")]
TotalOutputNetCapacitance,
#[strum(serialize = "related_out_total_output_net_capacitance")]
RelatedOutTotalOutputNetCapacitance,
#[strum(serialize = "constrained_pin_transition")]
ConstrainedPinTransition,
#[strum(serialize = "fanout_number")]
FanoutNumber,
#[strum(serialize = "fanout_pin_capacitance")]
FanoutPinCapacitance,
#[strum(serialize = "driver_slew")]
DriverSlew,
#[strum(serialize = "input_transition_time")]
InputTransitionTime,
}
impl SimpleAttri for VariableType {
#[inline]
fn nom_parse<'a>(
i: &'a str,
scope: &mut ParseScope,
) -> crate::ast::SimpleParseRes<'a, Self> {
crate::ast::nom_parse_from_str(i, scope)
}
}
#[derive(Debug, Default, Clone)]
#[derive(liberty_macros::Group)]
#[mut_set::derive::item(
sort,
macro(derive(Debug, Clone,Default);)
)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct Domain {
#[liberty(name)]
#[id(borrow = "&str")]
pub name: ArcStr,
#[liberty(comments)]
pub comments: GroupComments<Self>,
#[liberty(attributes)]
pub attributes: crate::ast::Attributes,
pub group_name: ArcStr,
pub calc_mode: Option<ArcStr>,
#[liberty(simple(type = Option))]
pub variable_1: Option<VariableType>,
pub variable_2: Option<VariableType>,
pub variable_3: Option<VariableType>,
pub index_1: Vec<f64>,
pub index_2: Vec<f64>,
pub index_3: Vec<f64>,
}
impl GroupFn for Domain {}
#[derive(Debug, Default, Clone, Eq, PartialEq)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct WordSet {
pub inner: HashSet<ArcStr>,
}
impl fmt::Display for WordSet {
#[expect(clippy::unwrap_in_result)]
#[expect(clippy::unwrap_used)]
#[inline]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self.inner.len() {
0 => Ok(()),
1 => write!(f, "{}", self.inner.iter().next().unwrap()),
_ => join_fmt(self.inner.iter().sorted(), f, |s, ff| ff.write_str(s.as_str()), " "),
}
}
}
impl Ord for WordSet {
#[inline]
fn cmp(&self, other: &Self) -> Ordering {
if self.inner.is_subset(&other.inner) {
Ordering::Less
} else if self.inner.is_superset(&other.inner) {
Ordering::Greater
} else {
Ordering::Equal
}
}
}
#[expect(clippy::non_canonical_partial_ord_impl)]
impl PartialOrd for WordSet {
#[inline]
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
if self.inner.len() == other.inner.len() {
self.inner.iter().sorted().partial_cmp(other.inner.iter().sorted())
} else if self.inner.is_subset(&other.inner) {
Some(Ordering::Less)
} else if self.inner.is_superset(&other.inner) {
Some(Ordering::Greater)
} else {
None
}
}
}
impl SimpleAttri for WordSet {
#[inline]
fn nom_parse<'a>(
i: &'a str,
scope: &mut ParseScope,
) -> crate::ast::SimpleParseRes<'a, Self> {
crate::ast::nom_parse_from_str(i, scope)
}
#[inline]
fn is_set(&self) -> bool {
!self.inner.is_empty()
}
#[inline]
fn fmt_self<T: Write, I: Indentation>(
&self,
f: &mut CodeFormatter<'_, T, I>,
) -> fmt::Result {
write!(f, "{self}")
}
}
impl hash::Hash for WordSet {
#[inline]
fn hash<H: hash::Hasher>(&self, state: &mut H) {
let mut sum = 0_u64;
for item in &self.inner {
let mut hasher = std::hash::DefaultHasher::new();
item.hash(&mut hasher);
sum = sum.wrapping_add(hash::Hasher::finish(&hasher));
}
state.write_u64(sum);
}
}
impl FromStr for WordSet {
type Err = fmt::Error;
#[inline]
fn from_str(s: &str) -> Result<Self, Self::Err> {
Ok(Self { inner: s.split(' ').map(ArcStr::from).collect() })
}
}
#[derive(Debug, Default, Clone)]
#[derive(liberty_macros::Group)]
#[mut_set::derive::item(
sort,
macro(derive(Debug, Clone,Default);)
)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct DummyGroup {
#[liberty(name)]
#[id(borrow = "Option<&str>", check_fn = "mut_set::borrow_option!")]
name: Option<ArcStr>,
#[liberty(comments)]
pub comments: GroupComments<Self>,
#[liberty(attributes)]
pub attributes: crate::ast::Attributes,
}
impl GroupFn for DummyGroup {}
#[derive(Debug, Default, Clone)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct Formula(pub ArcStr);
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[derive(serde::Serialize, serde::Deserialize)]
pub enum NameList {
Name(ArcStr),
List(WordSet),
}
impl Default for NameList {
#[inline]
fn default() -> Self {
Self::Name(ArcStr::new())
}
}