#include <SPI.h>
#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#include "ArduinoTrace.h"

#include "Square.h"
#include "Circle.h"

#define SCREEN_WIDTH        128 // OLED display width, in pixels
#define SCREEN_HEIGHT        64 // OLED display height, in pixels

// Declaration for an SSD1306 display connected to I2C (SDA, SCL pins)
#define OLED_RESET            4 // Reset pin # (or -1 if sharing Arduino reset pin)
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

int x,y;  //posicion en x, en y

//======================== SETUP ===========================//

void setup() {
  // put your setup code here, to run once:
  Serial.begin(115200);
  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }
  display.clearDisplay();
  display.display();
}

//======================== LOOP ===========================//

void loop() {
  // put your main code here, to run repeatedly:
  static unsigned long pm=0;
  if((millis()-pm)>=50)
  {
    pm=millis();
    display.clearDisplay();
    Task1();
    display.display();
  }
}

//======================== TASK1 ===========================//
void Task1()
{
  static int step=1;

  display.drawRect(x,y,20,20,WHITE);

  switch (step)
  {
    case 1:
      x=0;
      y=0;
      step=2;
    break;

    case 2:
      mRight(); 
      if(x>=107)
      {
        step=3;
      }
    break;

    case 3:
      mDown(); 
      if(y>=43)
      {
        step=4;
      }
    break;

    case 4:
      mLeft();
      if(x<=0)
      {
        step=5;
      }
    break;

    case 5:
      mUp(); 
      if(y<=0)
      {
        step=2;
      }
    break;
  }
}

void mRight()
{
  x=x+2;
}

void mLeft()
{
  x=x-2;
}

void mDown()
{
  y=y+2;
}

void mUp()
{
  y=y-2;
}
//============================================================//