Advertisement
wswartzendruber

Untitled

Jun 24th, 2021
1,685
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Rust 3.73 KB | None | 0 0
  1. fn rle_decompress(input: &[u8]) -> SegmentReadResult<Vec<Vec<u8>>> {
  2.  
  3.     let mut output = Vec::<Vec<u8>>::new();
  4.     let mut line = vec![];
  5.     let mut iter = input.iter();
  6.  
  7.     loop {
  8.         match iter.next() {
  9.             Some(byte_1) => {
  10.                 if *byte_1 == 0x00 {
  11.                     match iter.next() {
  12.                         Some(byte_2) => {
  13.                             if *byte_2 >> 6 == 0 {
  14.                                 for _ in 0..(*byte_2 & 0x3F) {
  15.                                     line.push(0);
  16.                                 }
  17.                             } else if *byte_2 >> 6 == 1 {
  18.                                 match iter.next() {
  19.                                     Some(byte_3) => {
  20.                                         for _ in 0..(
  21.                                             (*byte_2 as u16 & 0x3F) << 8
  22.                                             | *byte_3 as u16
  23.                                         ) {
  24.                                             line.push(0);
  25.                                         }
  26.                                     }
  27.                                     None => {
  28.                                         return Err(ReadError::IncompleteRleSequence)
  29.                                     }
  30.                                 }
  31.                             } else if *byte_2 >> 6 == 2 {
  32.                                 match iter.next() {
  33.                                     Some(byte_3) => {
  34.                                         for _ in 0..(*byte_2 & 0x3F) {
  35.                                             line.push(*byte_3);
  36.                                         }
  37.                                     }
  38.                                     None => {
  39.                                         return Err(ReadError::IncompleteRleSequence)
  40.                                     }
  41.                                 }
  42.                             } else if *byte_2 >> 6 == 3 {
  43.                                 match iter.next() {
  44.                                     Some(byte_3) => {
  45.                                         match iter.next() {
  46.                                             Some(byte_4) => {
  47.                                                 for _ in 0..(
  48.                                                     (*byte_2 as u16 & 0x3F) << 8
  49.                                                     | *byte_3 as u16
  50.                                                 ) {
  51.                                                     line.push(*byte_4);
  52.                                                 }
  53.                                             }
  54.                                             None => {
  55.                                                 return Err(ReadError::IncompleteRleSequence)
  56.                                             }
  57.                                         }
  58.                                     }
  59.                                     None => {
  60.                                         return Err(ReadError::IncompleteRleSequence)
  61.                                     }
  62.                                 }
  63.                             } else if *byte_2 == 0x00 {
  64.                                 output.push(line);
  65.                                 line = vec![];
  66.                             } else {
  67.                                 return Err(ReadError::InvalidRleSequence)
  68.                             }
  69.                         }
  70.                         None => {
  71.                             return Err(ReadError::IncompleteRleSequence)
  72.                         }
  73.                     }
  74.                 } else {
  75.                     line.push(*byte_1);
  76.                 }
  77.             }
  78.             None => {
  79.                 break
  80.             }
  81.         }
  82.     }
  83.  
  84.     Ok(output)
  85. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement