#include <Brain.h>

Brain brain(Serial);

void setup() {
  Serial.begin(9600);
}

void loop() {
  if (brain.update()) {
    Serial.println(brain.readErrors());
    Serial.println(brain.readCSV());
  }
}

import peasy.*;
import neurosky.*;
import org.json.*;

float rotX;
float rotY;
float rayon;
float rayonBase=100;
int hauteur=800;
float phi=60;
float facteur=0.7;

ThinkGearSocket neuroSocket;

int delta;
int theta;
int low_alpha;
int high_alpha;
int low_beta;
int high_beta;
int low_gamma;
int mid_gamma;
int BlinkStrength;
int attention;
int meditation;

void setup() {

  size(1280, 800, OPENGL);

  camera( 2000, -50, 20, 0, 0, 0, 0, 10, 0 );

  ThinkGearSocket neuroSocket = new ThinkGearSocket(this);
  try {
    neuroSocket.start();
  } 
  catch (ConnectException e) {
    //println("Is ThinkGear running??");
  }
  frameRate(60);
  strokeWeight(0.5);
  smooth();
}

void draw() {

  {
    rotateX(-.5);
    rotateY(-.5);
    background(0);
    fill(255, 0, 0);
    box(30);
    pushMatrix();
    translate(0, 0, 20);
    fill(0, 0, 255);
    box(5);
    popMatrix();
  }


  background(255);
  pushMatrix();


  translate(width/2, height/2, -300);

  rotateY(radians(rotY));
  rotateX(radians(rotX));
  beginShape(QUADS);
  fill(10, 237, 255, 34);
  stroke(0);
  for (int j=-hauteur/2 + 2 ;j<hauteur/2 + 2;j+=20) {
    for (int i=0;i<360;i+=20) {
      rayon=cos(radians(j+phi)*facteur)*meditation+rayonBase;
      vertex(cos(radians(i))*rayon, sin(radians(i))*rayon, j);
      vertex(cos(radians(i+20))*rayon, sin(radians(i+20))*rayon, j);
      rayon=cos(radians(j+20+phi)*facteur+attention)*attention+rayonBase;
      vertex(cos(radians(i+20))*rayon, sin(radians(i+20))*rayon, j+20);
      vertex(cos(radians(i))*rayon, sin(radians(i))*rayon, j+20);
    }
  }

  endShape();
  popMatrix();
}


void poorSignalEvent(int sig) {
  println("SignalEvent "+sig);
}

public void attentionEvent(int attentionLevel) {
  println("Attention Level: " + attentionLevel);
  attention = attentionLevel;
}


public void meditationEvent(int meditationLevel) {
  println("Meditation Level: " + meditationLevel);
  meditation = meditationLevel;
}

public void deltaEvent(int deltaLevel) {
  println("delta Level: " + deltaLevel);
  delta = deltaLevel;
}

public void thetaEvent(int thetaLevel) {
  println("theta Level: " + thetaLevel);
  theta = thetaLevel;
}

public void low_alphaEvent(int low_alphaLevel) {
  println("low_alpha Level: " + low_alphaLevel);
  low_alpha = low_alphaLevel;
}

public void high_alphaEvent(int high_alphaLevel) {
  println("high_alpha Level: " + high_alphaLevel);
  high_alpha = high_alphaLevel;
}

public void low_betaEvent(int low_betaLevel) {
  println("low_beta Level: " + low_betaLevel);
  low_beta = low_betaLevel;
}

public void high_betaEvent(int high_betaLevel) {
  println("high_beta Level: " + high_betaLevel);
  high_beta = high_betaLevel;
}

void low_gammaEvent(int low_gammaLevel) {
  println("low_gamma Level: " + low_gammaLevel);
  low_gamma = low_gammaLevel;
}

public void mid_gammaEvent(int mid_gammaLevel) {
  println("mid_gamma Level: " + mid_gammaLevel);
  mid_gamma = mid_gammaLevel;
}



public void blinkEvent(int blinkStrength) {
  println("blinkStrength: " + blinkStrength);
}


void rawEvent(int[] raw) {
  //println("rawEvent Level: " + raw);
}  

void stop() {
  neuroSocket.stop();
  super.stop();
}

void mouseDragged()
{
  rotX += (mouseX - pmouseX)  ;
  rotY -= (mouseY - pmouseY)  ;
}