//
// name: strange_attractors.pde
//
// desc: renders an assortment of strange attractors
//
// info: bugs, blame and beratement -> <ian@kiddphunk.com> <-
//
//       => www.kiddphunk.com/experiments/
//
//       remember: andre the giant has a posse.
//
// todo: * nap
//
///////////////////////////////////////////////////////////////////////////////

int dim           = 300;   // screen dimensions (square window)
int ndim          = dim;
int bailout       = 200;   // number of iterations before bail
int plots         = 500;   // number of plots to execute per frame (x30 = plots per second)
int time          = 0;

int num_strangers = 225;
int num_to_a_side = 1; // int(random(2,5));    // 10
int offset = int(dim/num_to_a_side);

Stranger strangers[] = new Stranger[num_strangers];

///////////////////////////////////////////////////////////////////////////////

void setup() {
  size(dim, dim);
  background(0);
  framerate(30);
  createStrangers();
}

void loop() {
  time++;
  boolean kp = keyPressed;
  for (int n=0; n<num_strangers; n++) {
    Stranger s = strangers[n];
    if (!s.done) {
      plotPlots(s);
      if (time%30==2) {
        s.done = finishedExposure(s);
        renderBrot(s);
      }
      if (kp) {
        s.randomizeVars();
      }
    } else {
      stroke(16);
      line(s.offx, s.offy, s.offx+offset, s.offy);
      line(s.offx, s.offy, s.offx, s.offy+offset);
      line(s.offx, s.offy+offset, s.offx+offset, s.offy+offset);
      line(s.offx+offset, s.offy, s.offx+offset, s.offy+offset);
    }
  }
}

void createStrangers() {
  ndim = offset;
  num_strangers = 0;
  for (int xxxx=0; xxxx<dim-10; xxxx+=offset) {
    for (int yyyy=0; yyyy<dim-10; yyyy+=offset) {
      num_strangers++;
      strangers[num_strangers-1] = new Stranger(offset, true);
      strangers[num_strangers-1].setOffset(xxxx, yyyy);
    }
  }
}

void plotPlots(Stranger s) {
  float x, y;
  for (int n=0;n<plots;n++) {
    color c = 255;
    String ret = s.doIter(c);
    if (ret != null) {
      int ret_array[] = splitInts(ret, ' ');
      int ix = ret_array[0];
      int iy = ret_array[1];
      s.exposure[ix*s.dim+iy]++;
      s.exposures++;
    }
  }
}

class Stranger {
  /*
  float a      =  2.24;
  float b      =   .43;
  float c      =  -.65;
  float d      = -2.43; 
  */
  float a      =  -.83017254; // -20
  float b      =   1.3246107; // .53
  float c      =  -3.6927686; // -.35
  float d      =  1.7803202;  // -1.43
  float e      =  0.03; // .03 .09 .27 //.95

  float xxmin  = -2;
  float xxmax  =  2;
  float yymin  = -2;
  float yymax  =  2;

  float x, y, z;
  float xx, yy, zz;
  float xinc, yinc;

  public int dim;
  public int offx = 0;
  public int offy = 0;

  public int[] exposure;
  public int[] exposureR;
  public int[] exposureG;
  public int[] exposureB;
  
  public int maxexposure; 
  public int exposures = 0;

  public int choiceR = 3;
  public int choiceG = 3;
  public int choiceB = 3;

  boolean randomize = false;

  public boolean done = false;

  Stranger(int pres, boolean randomize) {
    dim = pres;
    exposure = new int[dim*dim];
    exposureR = new int[dim*dim];
    exposureG = new int[dim*dim];
    exposureB = new int[dim*dim];
    x = y = z = 0;
    xx = yy = zz = 0;
    xinc = pres/(xxmax-xxmin);
    yinc = pres/(yymax-yymin);
    if (randomize) {
      randomizeVars();
    }
  }

  void randomizeVars() {
    a = random(-10, 1);
    b = random(-1, 1);
    c = random(-1, 1);
    d = random(-1, 10);
    e = random(-1, .1);

    choiceR = int(random(1, 4));
    choiceG = int(random(1, 4));
    choiceB = int(random(1, 4));
  }

  String doIter(color c) {
    xx = sin(a*y) - z*e*cos(b*x);
    yy = z*sin(c*x) - abs(cos(d*y));
    zz = e*sin(x);

    if (xx < xxmax && xx > xxmin & yy < yymax && yy > yymin) {
      int xxx = int((xx-xxmin)*xinc);
      int yyy = int((yy-yymin)*yinc);
      int ix = xxx = int(floor((xx+2)*(dim-16)/4+8));
      int iy = yyy = int(floor((-yy+2)*(dim-16)/4+8));

      if (this.exposureR[ix*this.dim+iy]+10 < 255) {
        switch (choiceR) {
          case 1:
          this.exposureR[ix*this.dim+iy]=int(ceil(this.exposureR[ix*this.dim+iy]+2*abs(yy-y)));
          break;

          case 2:
          this.exposureR[ix*this.dim+iy]=int(ceil(this.exposureR[ix*this.dim+iy]+2*abs(xx-x)));
          break;

          case 3:
          default:
          this.exposureR[ix*this.dim+iy]++;
        }
      }

      if (this.exposureG[ix*this.dim+iy]+10 < 255) {
        switch (choiceG) {
          case 1:
          this.exposureG[ix*this.dim+iy]=int(ceil(this.exposureG[ix*this.dim+iy]+2*abs(yy-y)));
          break;

          case 2:
          this.exposureG[ix*this.dim+iy]=int(ceil(this.exposureG[ix*this.dim+iy]+2*abs(xx-x)));
          break;

          case 3:
          default:
          this.exposureG[ix*this.dim+iy]++;
        }
      }

      if (this.exposureB[ix*this.dim+iy]+10 < 255) {
        switch (choiceB) {
          case 1:
          this.exposureB[ix*this.dim+iy]++;
          break;

          case 2:
          this.exposureB[ix*this.dim+iy]=int(ceil(this.exposureB[ix*this.dim+iy]+2*abs(xx-x)));
          break;

          case 3:
          default:
          this.exposureB[ix*this.dim+iy]=int(ceil(this.exposureB[ix*this.dim+iy]+2*abs(xx-x)));
          break;
        }
      }

      x = xx;
      y = yy;
      z = zz;
      return xxx+" "+yyy;
    }
    return null;
  }

  void setOffset(int x, int y) {
    offx = x;
    offy = y;
  }

}

int countMaxExposures(Stranger s) {
  int count = 0;
  for (int i=0;i<s.dim;i++) {
    for (int j=0;j<s.dim;j++) {
      if (s.exposureR[i*s.dim+j] >= 230 &&
          s.exposureG[i*s.dim+j] >= 230 &&
          s.exposureB[i*s.dim+j] >= 230) {
        count++;
      }
    }
  }
  return count;
}

boolean finishedExposure(Stranger s) {  
  return (((float)countMaxExposures(s)/(s.dim*s.dim)) > .012); //.015
}

void renderBrot(Stranger s) {
  // draw to screen
  for (int i=0;i<s.dim;i++) {
    for (int j=0;j<s.dim;j++) {
      color c = color(s.exposureR[i*s.dim+j], s.exposureG[i*s.dim+j], s.exposureB[i*s.dim+j]);
      set(j+s.offx,i+s.offy,c);
    }
  }
}

///////////////////////////////////////////////////////////////////////////////