KohonenSet klist = new KohonenSet();
float colorList[][];
int numColors = 20;

void setup(){
  //  framerate(2);
  noStroke();
  size(950,500);
  createColorList();
  klist.kset = colorList;
  klist.initWeights(1, 100, 3);
  ellipseMode(CENTER);
}

void draw(){
  background(230);
  if(klist.mapMode==0)klist.map();
  if(klist.mapMode==1)klist.errorMap();
  if(klist.mapMode==2)klist.listMap();
  pushMatrix();
  translate(520,0);
  klist.listMapTwo();
  popMatrix();
  klist.legend();
  klist.bmuLabels();
  klist.train();
}

void createColorList(){
  colorList = new float[numColors][3];
  for (int i=0; i<numColors; i++){
    for (int j=0; j<3; j++){
      colorList[i][j] = i*(255/numColors);
//      colorList[i][j] = random(255);
    }
  }
}

void keyReleased(){
  if (key == '=') klist.mapMode = (klist.mapMode+1)%3;
  if (key == '-') klist.mapMode = (klist.mapMode+2)%3;
}