#include <helianthus.h>

#include <math.h>

#define CIRCLES_COUNT   20

#define CIRCLES_RADIUS  25

#define RING_RADIUS     50

#define ROTATION_SPEED 70

#define RINGS_COUNT 3

double angles[6];

double speeds[6];

void init() {

  background(colorByHSV(0, 0, 0.2));

}

void draw() {

  for(int j = 1; j <= RINGS_COUNT; j++) {

    saveState();

    strokeWidth(CIRCLES_RADIUS + j*10);

    translate(mouseX(), mouseY());

    if (mouseDown("left")) {

      angles[j] -= windowGetFrameTime() * (ROTATION_SPEED * (j/3.0));

      speeds[j] = 0;

    } else

    if (mouseDown("right")) {

      angles[j] += windowGetFrameTime() * (ROTATION_SPEED * (j/3.0));

      speeds[j] = 0;

    } else {

      speeds[j] -= angles[j];

      speeds[j] *= pow(0.1, windowGetFrameTime());

      angles[j] += speeds[j]*windowGetFrameTime();

    }

    rotate(windowGetSeconds() + angles[j]);

    for(int i = 0; i < CIRCLES_COUNT; i++) {

      double a = 0;

      //if (mouseDown("left"))

        a = 360.0*i/CIRCLES_COUNT;

      double x = 1 + 0.05*sin(windowGetSeconds()*10 + a/180*PI*13);

      stroke(colorByHSV(a, 0.7, 1));

      point(RING_RADIUS*j*x, 0);

      rotate(360.0/CIRCLES_COUNT);

    }

    restoreState();

  }

}

int main() {

  windowSetTitle("Iron Spring Simulator");

  windowSetVariableFrameRate();

  windowSetResizable(TRUE);

  windowSetInit(&init);

  windowSetDraw(&draw);

  windowRun();

  return 0;

}