Water water[] = new Water[10];
int waterCount = 0;
Water watermap[];
int waterpixels[];

int waterColor = 0xFF0000FF;
int alphaColor = 0xFF000000;
int bgColor = 0x00C8C8FF;
int bgColor2 = 0xFFC8C8FF;
int toolbarColor = 0xFF8080A0;

Random rand;

PImage waterImage;
final int WATER = 0;
final int LAND = 1;
final int ERASE = 2;
int inputMode = WATER;

void setup() {
  Dimension d = getSize();
  size(400,400,P3D);
  
  framerate(100);

  hint(DISABLE_TEXT_SMOOTH);
  textFont(loadFont("Verdana-9.vlw"));

  loadPixels();
  waterpixels = pixels;
  //  waterpixels = new int[width*height];
  //  waterImage = new PImage(waterpixels,width,height,RGBA);
  for (int i=0; i<waterpixels.length; i++)
      waterpixels[i]=i<10*width ? toolbarColor : bgColor;

  watermap = new Water[waterpixels.length];
  
  rand = new Random();
  drawColor();
  drawMode();
  drawParticleCount();
  
}

void textCenter(String s, int x, int y) {
  textAlign(LEFT); 
  text(s, x, y);
}


void drawColor() {
  fill(waterColor);
  stroke(bgColor);
  rect(1,1,7,7);
}
String modeText = "";
void drawMode() {
  textAlign(LEFT); 
  fill(toolbarColor);
  text(modeText,10,9);
  fill(0xFFFFFFFF);
  switch (inputMode) {
  case WATER:
    modeText = "Add Water";
    break;
  case LAND:
    modeText = "Add Land";
    break;
  case ERASE:
    modeText = "Erase";
    break;
  }
  if (liquid(mouseX,mouseY)) {
    Water w = watermap[mouseX+mouseY*width];
    modeText+=" ["+w+"]";
  }
  text(modeText,10,9);
}
String particleCount = "";
void drawParticleCount() {
  textAlign(RIGHT); 
  fill(toolbarColor);
  noStroke();
  int w=(particleCount.length())*7;
  rect(width-w-2,0,w,9);
  particleCount = waterCount+"/"+water.length+" - "+sleepCount+" sleeping - "+fps.fps+"fps";
  fill(0xFFFFFFFF);
  text(particleCount,width-2,8);
}

void keyReleased() {
  switch (key) {
  case '1': 
    inputMode = WATER; 
    drawMode();
    break;
  case '2': 
    inputMode = LAND; 
    drawMode();
    break;
  case '3': 
    inputMode = ERASE; 
    drawMode();
    break;
  case '4': 
    waterColor = alphaColor|rand.nextInt(); 
    drawColor();
    break;
  case '5': 
    for (int i=10*width; i<waterpixels.length; i++) {
      int c = waterpixels[i];
      if ((0xFF000000&c)!=0 && watermap[i]==null)
        addWater(i,0,0,c);
    }
    break;
  case '6':
    int x,y;
    for (int i=waterCount; --i>=0;) {
      Water w = water[i];
      waterpixels[w.pos]=0xFFFFFFFF;
    }
    break;
  case '7':
    for (int i=waterpixels.length; --i>=10*width;) {
      waterpixels[i] = bgColor;
      watermap[i] = null;
    }
    sleepCount = 0;
    waterCount = 0;
    break;
  }
}


void draw() {
  //  background(200);
  if (mousePressed) {

    int dx=mouseX-pmouseX;
    int dy=mouseY-pmouseY;
    if (dx==0&&dy==0)
      drawPoint(mouseX,mouseY,0,0);
    else {
      double x=mouseX,y=mouseY;
      double units=max(abs(dx),abs(dy));
      double hunit=-dx/units;
      double vunit=-dy/units;
      for (int ii=(int)units; --ii>=0;) {
        x+=hunit;
        y+=vunit;
        drawPoint((int)x,(int)y,dx*0.5f,dy*0.5f);
      }
    }
  }
  moveWater();
  fps.frame();
  drawParticleCount();
  //  image(waterImage,0,0);
}

final private void drawPoint(int xx, int yy, float dx, float dy) {
  for (int y=-2;y<=2;y++)
    for (int x=-2;x<=2;x++) {
      switch (inputMode) {
      case WATER:
        addWater(xx+x,yy+y,dx,dy,waterColor);
        break;
      case LAND:
        addLand(xx+x,yy+y,0xFFB77E39);
        break;
      case ERASE:
        erase(xx+x,yy+y);
        break;
      }
    }
}