Transmitter.pde

import hypermedia.net.*;

public class Transmitter {
  private UDP udp;
  private byte buffer[];

  public Transmitter(PApplet parent) {
    this.udp = new UDP(parent);
    buffer = new byte[Config.STRIPS * Config.LEDS * 3 + 1];
    buffer[0] = 1;
  }

  public void sendData(PGraphics pg, MapEntry[] map) {
    int w = pg.width;
    pg.loadPixels();

    for (MapEntry entry : map) {
      color c = pg.pixels[entry.y * w + entry.x];
      int r = (c >> 16) & 0xff;
      int g = (c >> 8) & 0xff;
      int b = c & 0xff;

      if (Config.ENABLE_GAMMA) {
        r = gammaTable[r];
        g = gammaTable[g];
        b = gammaTable[b];
      }

      int idx = (entry.strip + entry.led * Config.STRIPS) * 3 + 1;
      buffer[idx] = (byte)r;

      if (Config.SWAP_LOOKUP[entry.strip]) {
        buffer[idx + 2] = (byte)g;
        buffer[idx + 1] = (byte)b;
      } else {
        buffer[idx + 1] = (byte)g;
        buffer[idx + 2] = (byte)b;
      }
    }

    udp.send(buffer, Config.HOST, Config.PORT);
  }

  public void sendData(PGraphics pg) {
    pg.loadPixels();
    int length = pg.width * pg.height;

    for (int i = 0; i < length; i++) {
      color c = pg.pixels[i];
      int r = (c >> 16) & 0xff;
      int g = (c >> 8) & 0xff;
      int b = c & 0xff;

      if (Config.ENABLE_GAMMA) {
        r = gammaTable[r];
        g = gammaTable[g];
        b = gammaTable[b];
      }
      int idx = i * 3 + 1;
      buffer[idx] = (byte)r;

      // TODO: Double check the lookup index
      if (Config.SWAP_LOOKUP[i / Config.LEDS]) {
        buffer[idx + 2] = (byte)g;
        buffer[idx + 1] = (byte)b;
      } else {
        buffer[idx + 1] = (byte)g;
        buffer[idx + 2] = (byte)b;
      }
    }

    udp.send(buffer, Config.HOST, Config.PORT);
  }
}