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

const int trigPin = 5;
const int echoPin = 18;

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

#define SOUND_SPEED 0.034
#define CM_TO_INCH 0.393701

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, -1);

Adafruit_MPU6050 mpu;

void setup() {
  Serial.begin(115200);

  pinMode(trigPin, OUTPUT);
  pinMode(echoPin, INPUT);

  if (!display.begin(SSD1306_SWITCHCAPVCC, 0x3C)) {
    Serial.println(F("SSD1306 allocation failed"));
    for (;;);
  }

  while (!mpu.begin()) {
    Serial.println("MPU6050 not connected!");
    delay(1000);
  }    
  Serial.println("MPU6050 ready!");
  delay(500);

  display.clearDisplay();

  display.setTextSize(1);
  display.setTextColor(WHITE);
}

sensors_event_t event;

void loop() {
  long duration;
  float distance;

  digitalWrite(trigPin, LOW);
  delayMicroseconds(2);
  digitalWrite(trigPin, HIGH);
  delayMicroseconds(10);
  digitalWrite(trigPin, LOW);

  duration = pulseIn(echoPin, HIGH);

  distance = duration * SOUND_SPEED / 2;

  mpu.getAccelerometerSensor()->getEvent(&event);

  display.clearDisplay();
  display.setCursor(0, 0);
  display.print("Distance: ");
  display.print(distance);
  display.println(" cm");
  display.setCursor(0, 20);
  display.print("Accelerometer:");
  display.setCursor(0, 30);
  display.print("X: ");
  display.print(event.acceleration.x);
  display.print(" m/s^2");
  display.setCursor(0, 40);
  display.print("Y: ");
  display.print(event.acceleration.y);
  display.print(" m/s^2");
  display.setCursor(0, 50);
  display.print("Z: ");
  display.print(event.acceleration.z);
  display.println(" m/s^2");

  display.display();

  delay(500);
}