File: rle.hpp

package info (click to toggle)
ares 126-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 32,600 kB
  • sloc: cpp: 356,508; ansic: 20,394; makefile: 16; sh: 2
file content (56 lines) | stat: -rw-r--r-- 1,214 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#pragma once

namespace nall::Encode {

template<u32 S = 1, u32 M = 4 / S>  //S = word size; M = match length
inline auto RLE(array_view<u8> input) -> vector<u8> {
  vector<u8> output;
  for(u32 byte : range(8)) output.append(input.size() >> byte * 8);

  u32 base = 0;
  u32 skip = 0;

  auto load = [&](u32 offset) -> u8 {
    return input(offset);
  };

  auto read = [&](u32 offset) -> u64 {
    u64 value = 0;
    for(u32 byte : range(S)) value |= load(offset + byte) << byte * 8;
    return value;
  };

  auto write = [&](u64 value) -> void {
    for(u32 byte : range(S)) output.append(value >> byte * 8);
  };

  auto flush = [&] {
    output.append(skip - 1);
    do {
      write(read(base));
      base += S;
    } while(--skip);
  };

  while(base + S * skip < input.size()) {
    u32 same = 1;
    for(u32 offset = base + S * (skip + 1); offset < input.size(); offset += S) {
      if(read(offset) != read(base + S * skip)) break;
      if(++same == 127 + M) break;
    }

    if(same < M) {
      if(++skip == 128) flush();
    } else {
      if(skip) flush();
      output.append(128 | same - M);
      write(read(base));
      base += S * same;
    }
  }
  if(skip) flush();

  return output;
}

}