01
02
03
04
05
06
07
08
09
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package algs55; // section 5.5
import stdlib.*;
/* ***********************************************************************
 *  Compilation:  javac RunLength.java
 *  Execution:    java RunLength - < input.txt   (compress)
 *  Execution:    java RunLength + < input.txt   (expand)
 *  Dependencies: BinaryIn.java BinaryOut.java
 *
 *  Compress or expand binary input from standard input using
 *  run-length encoding.
 *
 *  % java BinaryDump 40 < 4runs.bin
 *  0000000000000001111111000000011111111111
 *  40 bits
 *
 *  This has runs of 15 0s, 7 1s, 7 0s, and 11 1s.
 *
 *  % java RunLength - < 4runs.bin | java HexDump
 *  0f 07 07 0b
 *  4 bytes
 *
 *************************************************************************/

public class RunLength {
  private static BinaryIn binaryIn;
  private static BinaryOut binaryOut;
  private static final int R   = 256;
  private static final int lgR = 8;

  public static void expand() {
    boolean b = false;
    while (!binaryIn.isEmpty()) {
      int run = binaryIn.readInt(lgR);
      for (int i = 0; i < run; i++)
        binaryOut.write(b);
      b = !b;
    }
    binaryOut.close();
  }

  public static void compress() {
    char run = 0;
    boolean old = false;
    while (!binaryIn.isEmpty()) {
      boolean b = binaryIn.readBoolean();
      if (b != old) {
        binaryOut.write(run, lgR);
        run = 1;
        old = !old;
      }
      else {
        if (run == R-1) {
          binaryOut.write(run, lgR);
          run = 0;
          binaryOut.write(run, lgR);
        }
        run++;
      }
    }
    binaryOut.write(run, lgR);
    binaryOut.close();
  }


  public static void main(String[] args) {
    String txtFile = "data/4runs.bin";
    String binFile = "4runsOut.bin";
    //args = new String[] { "+" }; binaryIn = new BinaryIn(binFile); binaryOut = new BinaryOut();
    args = new String[] { "-" }; binaryIn = new BinaryIn(txtFile); binaryOut = new BinaryOut(binFile);
    if      (args[0].equals("-")) compress();
    else if (args[0].equals("+")) expand();
    else throw new Error("Illegal command line argument");
  }

}