#include  
#include 
#include 
#include   // I2C LCD library

// Basic pump speed
const double baseSpeed = 2000;

// PID parameters
double Kp_Pressure = 2.0;  
double Ki_Pressure = 1.0;  
double Kd_Pressure = 0.5;  

double Kp_Temperature = 20.0;  

// Target pressure 10-21mmHg,target temperature 35°C
double targetPressure = 1500.0;  // Target pressure set to 1500 Pa
double targetTemperature = 35;  // Target temperature set to 35°C

// Current values
double currentPressure = 0.0;    // Current pressure (Pa)
double currentTemperature = 0.0;  // Current temperature

// PID variables
double error_Pressure, previousError_Pressure = 0.0;
double I_Pressure = 0.0;
double D_Pressure = 0.0;
double output = 0.0;

// Stepper motor setup (AccelStepper configuration)
AccelStepper pumpIn(AccelStepper::DRIVER, 3, 4);    // First pump group (In)
AccelStepper pumpOut(AccelStepper::DRIVER, 7, 8); // Second pump group (Out)
AccelStepper pumpIn_tear(AccelStepper::DRIVER, 11, 12);    // First tear pump group (In)
AccelStepper pumpOut_tear(AccelStepper::DRIVER, 25, 27); // Second tear pump group (Out)

// Initialize MS5837 sensor
MS5837 sensor;

// Initialize I2C LCD screen (address: 0x27, 16 cols 2 rows)
LiquidCrystal_I2C lcd(0x27, 16, 2);

// Variables for time-based pulse speed increase
unsigned long lastPulseTime = 0;  // Last time a pulse was added
const unsigned long pulseInterval = 5000;  // 5 seconds interval for increasing speed
const unsigned long pulseDuration = 1000;  // Duration to hold increased speed (1 second)
int pulseIncrement = 100;  // Amount to increase speed by each pulse
double tearSpeedIncrement = 0;  // Tear pump speed increment

unsigned long pulseEndTime = 0;  // Time when the high-speed duration ends

const int pwmPin = 2;

void setup() {
    // Initialize stepper motors
    pumpIn.setMaxSpeed(1000);  
    pumpIn.setAcceleration(500);  
    
    pumpOut.setMaxSpeed(1000);  
    pumpOut.setAcceleration(500);  
    
    pumpIn_tear.setMaxSpeed(1000);  
    pumpIn_tear.setAcceleration(500);  
    
    pumpOut_tear.setMaxSpeed(1000);  
    pumpOut_tear.setAcceleration(500);  
    
    // Initialize sensor
    Wire.begin();

    // Initialize LCD screen
    lcd.begin();
    lcd.backlight(); 
    lcd.setCursor(0, 0);
    delay(2000); 

    // heating module
    pinMode(A2, OUTPUT);
}

void loop() {
    // Read sensor data
    sensor.read();
    currentPressure = sensor.getPressure() * 100.0;  // Convert mbar to Pa
    currentTemperature = sensor.getTemperature(CELSIUS);

    // Calculate error
    error_Pressure = targetPressure - currentPressure;
    
    // Calculate I and D terms
    I_Pressure += error_Pressure;
    D_Pressure = error_Pressure - previousError_Pressure;
    
    // Calculate output
    output = Kp_Pressure * error_Pressure + 
            Ki_Pressure * I_Pressure + 
            Kd_Pressure * D_Pressure;
    
    // Control pumps based on output
    controlPumps(output);
    
    // Sync tear pumps and increase speed every 5 seconds
    syncTearPumpsAndIncreaseSpeed();

    // Save current error for next derivative calculation
    previousError_Pressure = error_Pressure;

    // Display pressure and temperature on LCD
    displayDataOnLCD(currentPressure, currentTemperature);

    delay(100);
}

// Pump control function
void controlPumps(double output) {
    // Adjust pump speeds based on PID output
    double pumpInSpeed = constrain(output + baseSpeed, 0, 1000); 
    double pumpOutSpeed = constrain(-output + baseSpeed, -1000, 0);
    
    // Set stepper motor speeds
    pumpIn.setSpeed(pumpInSpeed);  
    pumpOut.setSpeed(abs(pumpOutSpeed));
    
    // Run motors
    for(int i = 0; i < 10; i++) {
        pumpIn.runSpeed();
        pumpOut.runSpeed();
        pumpIn_tear.runSpeed();
        pumpOut_tear.runSpeed();
    }
}

// Function to sync tear pumps and increase speed every 5 seconds
void syncTearPumpsAndIncreaseSpeed() {
    unsigned long currentMillis = millis();
    
    // Check if it's time to increase speed
    if (currentMillis - lastPulseTime >= pulseInterval) {
        // Increase speed for tear pumps
        tearSpeedIncrement += pulseIncrement;
        
        // Set the new speed for both tear pumps
        pumpIn_tear.setSpeed(tearSpeedIncrement);
        pumpOut_tear.setSpeed(tearSpeedIncrement);
        
        // Update the last pulse time and set the pulse end time
        lastPulseTime = currentMillis;
        pulseEndTime = currentMillis + pulseDuration;  // Set the end time for the high-speed phase
    }

    // Check if it's time to restore the speed to base speed
    if (currentMillis >= pulseEndTime) {
        // Restore speed to original value
        pumpIn_tear.setSpeed(tearSpeedIncrement - pulseIncrement);
        pumpOut_tear.setSpeed(tearSpeedIncrement - pulseIncrement);
    }
    //heating
    if(temperature < targetTemperature){
        analogWrite(pwmPin, Kp_Temperature*(targetTemperature-temperature));
    }else{
        analogWrite(pwmPin, 0));
    }
}

// Display pressure and temperature data on LCD screen
void displayDataOnLCD(double pressure, double temperature) {
    lcd.clear();
    
    // Display pressure
    lcd.setCursor(0, 0);
    lcd.print("P: ");
    lcd.print(pressure);
    lcd.print(" Pa");
    
    // Display temperature
    lcd.setCursor(0, 1);
    lcd.print("T: ");
    lcd.print(temperature);
    lcd.print(" C");
}

#include 
#include 
#include 

const int bottom = 27;     // Button pin
const int pwmPin = 5;      // PWM output to PT4115 DIM pin

//threshold
const double Safe_threshold = 200;
const double Replace_Recommended_threshold = 100;
const double Not_Usable_threshold = 50;

LiquidCrystal_I2C lcd(0x27, 16, 2);  // I2C address 0x27, 16x2 LCD
BH1750 lightMeter;

int brightness = 128;      // Initial brightness (0-255)
bool ledOn = false;        // LED switch state
const int threshold = 200; // Light intensity threshold (lux)

void setup() {
    pinMode(encoderSW, INPUT_PULLUP); // Use internal pull-up resistor
    pinMode(pwmPin, OUTPUT);
    analogWrite(pwmPin, brightness);  // Initialize brightness
    
    // Initialize LCD and BH1750
    Wire.begin();
    lcd.begin();
    lcd.backlight();
    lightMeter.begin();
    lcd.setCursor(0, 0);
    lcd.print("Initialized successfully!");
    delay(1000);
    lcd.clear();
}

void loop() {
    handleButton();     // Button controls LED and light intensity detection
}

void handleButton() {
    if (digitalRead(bottom) == LOW) { // Detect button press
        delay(50); // Debounce
        lcd.clear();

        ledOn = true;
        analogWrite(pwmPin, brightness);
        lcd.setCursor(0, 0);
        lcd.print("LED ON: 5 sec");
        delay(5000);

        float lux = lightMeter.readLightLevel(); 
        lcd.setCursor(0, 1);
        lcd.print("Light: ");
        lcd.print(lux);
        lcd.print(" lux");

        lcd.setCursor(12, 1);
        if (lux >= Safe_threshold) {
            lcd.print("Safe");  
        } else if (lux >= Replace_Recommended_threshold){
            lcd.print("Replace_Recommended"); 
        } else {
            lcd.print("Not_Usable"); 
        }
        
        analogWrite(ledPin, 0);
        
        delay(2000); 
        lcd.clear();
    }
}