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

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

#define OLED_RESET    -1
#define SCREEN_ADDRESS 0x3C
Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET);

int dropPositionX[10]; // Array to store X positions of water drops
int dropPositionY[10]; // Array to store Y positions of water drops

void setup() {
  if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
        Serial.println(F("SSD1306 allocation failed"));
                  for(;;);
                              }
                                            display.clearDisplay();
                                                          
                                                                            display.display();
                                                                                              }

                                                                                              void loop() {
                                                                                                animateWater();
                                                                                                }

                                                                                                void animateWater() {
                                                                                                  // Erase the previous water drops
                                                                                                    for (int i = 0; i < 10; i++) {
                                                                                                          display.fillRect(dropPositionX[i], dropPositionY[i], 3, 3, BLACK);
                                                                                                            }
                                                                                                              // Move water drops diagonally
                                                                                                                for (int i = 0; i < 10; i++) {
                                                                                                                      dropPositionX[i] -= 1; // Move the water drop towards left
                                                                                                                            dropPositionY[i] += 1; // Move the water drop downwards
                                                                                                                                  if (dropPositionX[i] < 100 || dropPositionY[i] > 63) { // If water drop reaches the end of the pipe or the bottom of the display
                                                                                                                                            dropPositionX[i] = random(113, 117); // Reset its X position to a random point between 113 and 117
                                                                                                                                                      dropPositionY[i] = random(0, 5); // Reset its Y position to a random point between 0 and 5
                                                                                                                                                            }
                                                                                                                                                                  // Draw the water drops at the new positions
                                                                                                                                                                        display.fillCircle(dropPositionX[i], dropPositionY[i], 1, WHITE); // Draw water droplets
                                                                                                                                                                          }
                                                                                                                                                                            // Display the changes
                                                                                                                                                                              display.display();
                                                                                                                                                                                // Introduce a short delay to control animation speed
                                                                                                                                                                                  delay(100); // Adjust the delay for desired animation speed
                                                                                                                                                                                  }