array_of.rs 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. use binrw::{binrw, BinRead, BinWrite};
  2. use std::ops::{Deref, DerefMut};
  3. /// Wrapper to prefix array of single type
  4. #[binrw]
  5. #[derive(Debug, PartialEq)]
  6. pub struct ArrayOf<T>
  7. where
  8. // Static bounds due to https://github.com/jam1garner/binrw/issues/199
  9. T: for<'a> BinRead<Args<'a> = ()> + 'static,
  10. T: for<'a> BinWrite<Args<'a> = ()> + 'static,
  11. {
  12. // Temporary variable that holds number of elements
  13. #[br(temp)]
  14. #[bw(try_calc(u32::try_from(value.len())))]
  15. count: u32,
  16. #[br(count = count)]
  17. pub value: Vec<T>,
  18. }
  19. impl<T> Deref for ArrayOf<T>
  20. where
  21. T: for<'a> BinRead<Args<'a> = ()> + 'static,
  22. T: for<'a> BinWrite<Args<'a> = ()> + 'static,
  23. {
  24. type Target = Vec<T>;
  25. fn deref(&self) -> &Self::Target {
  26. &self.value
  27. }
  28. }
  29. impl<T> DerefMut for ArrayOf<T>
  30. where
  31. T: for<'a> BinRead<Args<'a> = ()> + 'static,
  32. T: for<'a> BinWrite<Args<'a> = ()> + 'static,
  33. {
  34. fn deref_mut(&mut self) -> &mut Self::Target {
  35. &mut self.value
  36. }
  37. }
  38. impl<T> From<Vec<T>> for ArrayOf<T>
  39. where
  40. T: for<'a> BinRead<Args<'a> = ()> + 'static,
  41. T: for<'a> BinWrite<Args<'a> = ()> + 'static,
  42. {
  43. fn from(value: Vec<T>) -> Self {
  44. Self { value }
  45. }
  46. }