123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337 |
- use std::{
- fmt::{Debug, Display},
- ops::{Add, AddAssign, Div, Mul, Sub, SubAssign},
- };
- use binrw::{binrw, BinRead, BinWrite};
- use diff::{Diff, VecDiff};
- use num_enum::{IntoPrimitive, TryFromPrimitive};
- use rand::{thread_rng, Rng};
- use serde::{Deserialize, Serialize};
- use serde_repr::{Deserialize_repr, Serialize_repr};
- use strum::EnumIter;
- use crate::{array_of::ArrayOfPrimitive, field_of::FieldOf};
- /// Generic field with structure of length + data
- pub type Field = ArrayOfPrimitive<u8>;
- pub type U16 = FieldOf<u16>;
- pub type I16 = FieldOf<i16>;
- pub type U32 = FieldOf<u32>;
- pub type I32 = FieldOf<i32>;
- pub type F64 = FieldOf<f64>;
- pub type Bool = FieldOf<u32>;
- impl From<FieldOf<u32>> for bool {
- fn from(value: FieldOf<u32>) -> Self {
- value.value != 0
- }
- }
- impl From<bool> for FieldOf<u32> {
- fn from(value: bool) -> Self {
- Self {
- value: match value {
- true => 1,
- false => 0,
- },
- }
- }
- }
- #[binrw]
- #[derive(Clone, Default, PartialEq)]
- pub struct WString {
- // Temporary variable that holds number of elements
- #[br(temp)]
- #[bw(try_calc(u32::try_from(value.len() * core::mem::size_of::<u16>())))]
- size: u32,
- #[br(count = size / core::mem::size_of::<u16>() as u32)]
- pub value: Vec<u16>,
- }
- impl From<&WString> for String {
- fn from(value: &WString) -> Self {
- String::from_utf16_lossy(&value.value)
- }
- }
- impl From<String> for WString {
- fn from(value: String) -> Self {
- let value: Vec<u16> = value.encode_utf16().collect();
- Self { value }
- }
- }
- // Custom Diff implementation to only diff internal value
- impl Diff for WString {
- type Repr = VecDiff<u16>;
- fn diff(&self, other: &Self) -> Self::Repr {
- self.value.diff(&other.value)
- }
- fn apply(&mut self, diff: &Self::Repr) {
- self.value.apply(diff)
- }
- fn identity() -> Self {
- Self { value: vec![] }
- }
- }
- impl Debug for WString {
- fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
- write!(f, "\"{}\"", String::from(self))
- }
- }
- impl Display for WString {
- fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
- write!(f, "{}", String::from(self))
- }
- }
- #[derive(Clone, Debug, Default, Diff, PartialEq, BinRead, BinWrite)]
- #[diff(attr(
- #[derive(Debug, PartialEq)]
- ))]
- pub struct Rgba {
- pub red: u8,
- pub green: u8,
- pub blue: u8,
- pub alpha: u8,
- }
- impl From<(u8, u8, u8)> for Rgba {
- fn from(value: (u8, u8, u8)) -> Self {
- Self {
- red: value.0,
- green: value.1,
- blue: value.2,
- alpha: 0,
- }
- }
- }
- impl Rgba {
- pub fn random() -> Self {
- Self {
- red: thread_rng().gen(),
- green: thread_rng().gen(),
- blue: thread_rng().gen(),
- alpha: 0,
- }
- }
- }
- #[derive(
- Copy, Clone, Debug, Default, Diff, PartialEq, BinRead, BinWrite, Serialize, Deserialize,
- )]
- #[diff(attr(
- #[derive(Debug, PartialEq)]
- ))]
- #[serde(rename_all = "PascalCase")]
- pub struct Coordinate {
- pub x: f64,
- pub y: f64,
- }
- impl Display for Coordinate {
- fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
- write!(f, "({:.2}, {:.2})", self.x, self.y)
- }
- }
- impl Sub for Coordinate {
- type Output = Coordinate;
- fn sub(self, rhs: Self) -> Self::Output {
- Coordinate {
- x: self.x - rhs.x,
- y: self.y - rhs.y,
- }
- }
- }
- impl SubAssign for Coordinate {
- fn sub_assign(&mut self, rhs: Self) {
- self.x -= rhs.x;
- self.y -= rhs.y;
- }
- }
- impl Add for Coordinate {
- type Output = Coordinate;
- fn add(self, rhs: Self) -> Self::Output {
- Coordinate {
- x: self.x + rhs.x,
- y: self.y + rhs.y,
- }
- }
- }
- impl AddAssign for Coordinate {
- fn add_assign(&mut self, rhs: Self) {
- self.x += rhs.x;
- self.y += rhs.y;
- }
- }
- impl Mul for Coordinate {
- type Output = Coordinate;
- fn mul(self, rhs: Self) -> Self::Output {
- Coordinate {
- x: self.x * rhs.x,
- y: self.y * rhs.y,
- }
- }
- }
- impl Div for Coordinate {
- type Output = Coordinate;
- fn div(self, rhs: Self) -> Self::Output {
- Coordinate {
- x: self.x / rhs.x,
- y: self.y / rhs.y,
- }
- }
- }
- impl From<(f64, f64)> for Coordinate {
- fn from(value: (f64, f64)) -> Self {
- Self {
- x: value.0,
- y: value.1,
- }
- }
- }
- #[derive(Clone, Debug, Diff, PartialEq, BinRead, BinWrite, strum::Display)]
- #[diff(attr(
- #[derive(Debug, PartialEq)]
- ))]
- #[brw(repr(u32))]
- #[repr(u32)]
- pub enum WobbleType {
- Spiral = 0,
- Sinusoidal = 1,
- Ellipse = 2,
- Vert8 = 3,
- Hori8 = 4,
- Unknown = 5,
- }
- impl Default for WobbleType {
- fn default() -> Self {
- WobbleType::Unknown
- }
- }
- #[derive(Clone, Debug, Diff, PartialEq, BinRead, BinWrite, strum::Display)]
- #[diff(attr(
- #[derive(Debug, PartialEq)]
- ))]
- #[brw(repr(u16))]
- #[repr(u16)]
- pub enum ObjectType {
- Unknown = 0,
- Curve = 1,
- Point = 2,
- Rectangle = 3,
- Circle = 4,
- Ellipse = 5,
- Polygon = 6,
- HatchLine = 16,
- Hatch = 32,
- }
- impl Default for ObjectType {
- fn default() -> Self {
- ObjectType::Unknown
- }
- }
- #[derive(
- Copy,
- Clone,
- Debug,
- Diff,
- PartialEq,
- strum::Display,
- Serialize_repr,
- Deserialize_repr,
- EnumIter,
- IntoPrimitive,
- TryFromPrimitive,
- )]
- #[diff(attr(
- #[derive(Debug, PartialEq)]
- ))]
- #[repr(u32)]
- pub enum PulseWidth {
- Ns2 = 2,
- Ns4 = 4,
- Ns6 = 6,
- Ns8 = 8,
- Ns12 = 12,
- Ns20 = 20,
- Ns30 = 30,
- Ns45 = 45,
- Ns60 = 60,
- Ns80 = 80,
- Ns100 = 100,
- Ns150 = 150,
- Ns200 = 200,
- Ns250 = 250,
- Ns350 = 350,
- Ns500 = 500,
- }
- impl PulseWidth {
- pub fn min_freq(&self) -> u32 {
- match self {
- PulseWidth::Ns2 => 850_000,
- PulseWidth::Ns4 => 500_000,
- PulseWidth::Ns6 => 320_000,
- PulseWidth::Ns8 => 250_000,
- PulseWidth::Ns12 => 170_000,
- PulseWidth::Ns20 => 115_000,
- PulseWidth::Ns30 => 90_000,
- PulseWidth::Ns45 => 75_000,
- PulseWidth::Ns60 => 65_000,
- PulseWidth::Ns80 => 60_000,
- PulseWidth::Ns100 => 45_000,
- PulseWidth::Ns150 => 30_000,
- PulseWidth::Ns200 => 25_000,
- PulseWidth::Ns250 => 25_000,
- PulseWidth::Ns350 => 25_000,
- PulseWidth::Ns500 => 25_000,
- }
- }
- pub fn max_freq(&self) -> u32 {
- match self {
- PulseWidth::Ns2 => 4_000_000,
- PulseWidth::Ns4 => 4_000_000,
- PulseWidth::Ns6 => 4_000_000,
- PulseWidth::Ns8 => 4_000_000,
- PulseWidth::Ns12 => 3_000_000,
- PulseWidth::Ns20 => 3_000_000,
- PulseWidth::Ns30 => 3_000_000,
- PulseWidth::Ns45 => 2_000_000,
- PulseWidth::Ns60 => 2_000_000,
- PulseWidth::Ns80 => 2_000_000,
- PulseWidth::Ns100 => 1_000_000,
- PulseWidth::Ns150 => 1_000_000,
- PulseWidth::Ns200 => 1_000_000,
- PulseWidth::Ns250 => 900_000,
- PulseWidth::Ns350 => 600_000,
- PulseWidth::Ns500 => 500_000,
- }
- }
- }
|