#include <Adafruit_GFX.h>
#include <Adafruit_ILI9341.h>
#include <SPI.h>

// TFT parameters
#define TFT_CS 10
#define TFT_RST 9
#define TFT_DC 8

Adafruit_ILI9341 tft = Adafruit_ILI9341(TFT_CS, TFT_DC, TFT_RST);

#define TFT_WIDTH 320
#define TFT_HEIGHT 240

void setup() {
  tft.begin();
  tft.setRotation(3); // Adjust if necessary
  tft.fillScreen(ILI9341_BLACK);
}

void loop() {
  drawRainbowBackground();
  drawShapes();
  drawAttractiveDesign();
}

void drawShapes() {
  drawCircle();
  drawTriangle();
  drawRectangle();
  drawPentagon();
  drawHexagon();
  drawStar();
  drawEllipse();
  drawOctagon();
}

void drawCircle() {
  int x = TFT_WIDTH / 2;
  int y = TFT_HEIGHT / 2;
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    tft.drawCircle(x, y, size, ILI9341_RED);
    delay(500);
  }
}

void drawTriangle() {
  int x = TFT_WIDTH / 2;
  int y = TFT_HEIGHT / 2;
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    int halfSize = size / 2;
    tft.drawTriangle(x, y - halfSize, x - halfSize, y + halfSize, x + halfSize, y + halfSize, ILI9341_GREEN);
    delay(500);
  }
}

void drawRectangle() {
  int x = TFT_WIDTH / 2;
  int y = TFT_HEIGHT / 2;
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    tft.drawRect(x - size / 2, y - size / 2, size, size, ILI9341_BLUE);
    delay(500);
  }
}

void drawPentagon() {
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    drawPolygon(5, size, ILI9341_YELLOW);
    delay(500);
  }
}

void drawHexagon() {
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    drawPolygon(6, size, ILI9341_CYAN);
    delay(500);
  }
}

void drawStar() {
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    int x = TFT_WIDTH / 2;
    int y = TFT_HEIGHT / 2;
    int r = size / 2;
    for (int i = 0; i < 5; i++) {
      float angle1 = i * 72 * PI / 180;
      float angle2 = (i * 72 + 144) * PI / 180;
      float x1 = x + r * cos(angle1);
      float y1 = y - r * sin(angle1);
      float x2 = x + r * cos(angle2);
      float y2 = y - r * sin(angle2);
      tft.drawLine(x1, y1, x2, y2, ILI9341_MAGENTA);
    }
    delay(500);
  }
}

void drawEllipse() {
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    int x = TFT_WIDTH / 2;
    int y = TFT_HEIGHT / 2;
    int a = size / 2;
    int b = size / 4;
    for (int angle = 0; angle < 360; angle++) {
      float radian = angle * PI / 180;
      int x1 = x + a * cos(radian);
      int y1 = y + b * sin(radian);
      tft.drawPixel(x1, y1, ILI9341_ORANGE);
    }
    delay(500);
  }
}

void drawOctagon() {
  for (int size = 100; size > 0; size -= 10) {
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
    drawPolygon(8, size, ILI9341_PURPLE);
    delay(500);
  }
}

void drawPolygon(int sides, int size, uint16_t color) {
  int x = TFT_WIDTH / 2;
  int y = TFT_HEIGHT / 2;
  float angleStep = 360.0 / sides;
  float radianStep = angleStep * PI / 180.0;
  int prevX = x + size * cos(0);
  int prevY = y - size * sin(0);
  for (int i = 1; i <= sides; i++) {
    float angle = i * radianStep;
    int newX = x + size * cos(angle);
    int newY = y - size * sin(angle);
    tft.drawLine(prevX, prevY, newX, newY, color);
    prevX = newX;
    prevY = newY;
  }
}

void drawAttractiveDesign() {
  for (int i = 0; i < 35; i++) {
    uint16_t randomColor = tft.color565(random(255), random(255), random(255));
    int x = TFT_WIDTH / 2;
    int y = TFT_HEIGHT / 2;
    tft.drawCircle(x, y, 100, randomColor);
    delay(100);
    tft.fillScreen(ILI9341_BLACK);
    drawRainbowBackground();
  }

  tft.setCursor(TFT_WIDTH / 2 - 30, TFT_HEIGHT - 20);
  tft.setTextColor(ILI9341_WHITE);
  tft.setTextSize(2);
  tft.print("by Arvind");
}

void drawRainbowBackground() {
  for (int x = 0; x < TFT_WIDTH; x++) {
    float hue = (float)x / TFT_WIDTH;
    uint16_t color = HSVtoRGB(hue, 1.0, 1.0);
    tft.drawLine(x, 0, x, TFT_HEIGHT, color);
  }
}

uint16_t HSVtoRGB(float h, float s, float v) {
  int i = int(h * 6);
  float f = h * 6 - i;
  float p = v * (1 - s);
  float q = v * (1 - f * s);
  float t = v * (1 - (1 - f) * s);

  float r, g, b;
  switch (i % 6) {
    case 0: r = v, g = t, b = p; break;
    case 1: r = q, g = v, b = p; break;
    case 2: r = p, g = v, b = t; break;
    case 3: r = p, g = q, b = v; break;
    case 4: r = t, g = p, b = v; break;
    case 5: r = v, g = p, b = q; break;
  }

  uint8_t red = r * 255;
  uint8_t green = g * 255;
  uint8_t blue = b * 255;
  return tft.color565(red, green, blue);
}