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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define OLED_RESET     -1

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

void setup() {
  pinMode(10, OUTPUT);
  
  if(!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) { 
    Serial.println(F("SSD1306 allocation failed"));
    for(;;);
  }
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.display();
}

int bpmPotPin = A0;
int durationPotPin = A1;
int extraBeatsPotPin = A2;

void loop() {

  int bpm = map(analogRead(bpmPotPin), 0, 1023, 30, 240);
  int beatDuration = map(analogRead(durationPotPin), 0, 1023, 100, 1000);
  int extraBeats = map(analogRead(extraBeatsPotPin), 0, 1023, 0, 4);

  display.clearDisplay();
  display.setCursor(0,0);
  display.print("BPM: ");
  display.print(bpm);
  display.setCursor(0, 10);
  display.print("Beat Duration: ");
  display.print(beatDuration);
  display.setCursor(0, 20);
  display.print("Extra Beats: ");
  display.print(extraBeats);
  display.display();

  tone(10, 1000, beatDuration); 
  delay(60000 / bpm);

  for (int i = 0; i < extraBeats; i++) {
    tone(10, 800, beatDuration / 2);
    delay(beatDuration);
  }
}