#include <Wire.h>
#include "MAX30105.h"
#include "spo2_algorithm.h"
#include <LiquidCrystal_I2C.h> 

MAX30105 particleSensor;
LiquidCrystal_I2C lcd(0x27, 16, 2);

#define IR_THRESHOLD 50000
#define TIMEOUT 1000  // میلی‌ثانیه

#if defined(__AVR_ATmega328P__) || defined(__AVR_ATmega168__)
uint16_t irBuffer[100];
uint16_t redBuffer[100];
#else
uint32_t irBuffer[100];
uint32_t redBuffer[100];
#endif

int32_t bufferLength = 100;
int32_t spo2, heartRate;
int8_t validSPO2, validHeartRate;

byte readLED = 13;

// فیلتر EMA برای smooth کردن داده‌ها
float emaHR = 0, emaSpO2 = 0;
float alpha = 0.3; // ضریب فیلتر نمایی

#define MAX_HR 150
#define MIN_HR 40
#define MAX_SPO2 99
#define MIN_SPO2 70

// --- تنظیمات آلارم ---
#define ALARM_PIN A0

#define HR_THRESHOLD 150           // اگر ضربان قلب بالاتر از این شد
#define HR_ALARM_DURATION 500     // میلی‌ثانیه (5 ثانیه)

#define SPO2_THRESHOLD 93          // اگر SpO2 کمتر از این شد
#define SPO2_ALARM_DURATION 500   // میلی‌ثانیه (4 ثانیه)

unsigned long hrAlarmStart = 0;
unsigned long spo2AlarmStart = 0;
bool hrAlarmActive = false;
bool spo2AlarmActive = false;

void setup() {
  Serial.begin(9600);
  pinMode(readLED, OUTPUT);
  pinMode(ALARM_PIN, OUTPUT);

  if (!particleSensor.begin(Wire, I2C_SPEED_FAST)) {
    Serial.println(F("MAX30105 not found."));
    while (1);
  }

  particleSensor.setup(60, 4, 2, 100, 411, 4096);

  lcd.init();
  lcd.backlight();
  lcd.clear();
  lcd.setCursor(0,0);
  lcd.print("Place finger");
}

void readInitialBuffer() {
  for (byte i = 0; i < bufferLength; i++) { unsigned long start = millis(); while (!particleSensor.available()) { particleSensor.check(); if (millis() - start > TIMEOUT) break;
    }

    redBuffer[i] = particleSensor.getRed();
    irBuffer[i] = particleSensor.getIR();
    particleSensor.nextSample();
  }
}

void loop() {
  if (particleSensor.check() && particleSensor.getIR() > IR_THRESHOLD) {
    lcd.clear();
    lcd.setCursor(0,0);
    lcd.print("Reading...");
    readInitialBuffer();

    maxim_heart_rate_and_oxygen_saturation(irBuffer, bufferLength, redBuffer, &spo2, &validSPO2, &heartRate, &validHeartRate);

    emaHR = validHeartRate ? heartRate : 0;
    emaSpO2 = validSPO2 ? spo2 : 0;

    while (particleSensor.getIR() > IR_THRESHOLD) {
      // شیفت داده‌ها برای buffer
      for (byte i = 25; i < 100; i++) {
        redBuffer[i - 25] = redBuffer[i];
        irBuffer[i - 25] = irBuffer[i];
      }

      for (byte i = 75; i < 100; i++) { unsigned long start = millis(); while (!particleSensor.available()) { particleSensor.check(); if (millis() - start > TIMEOUT) break;
        }

        digitalWrite(readLED, !digitalRead(readLED));

        redBuffer[i] = particleSensor.getRed();
        irBuffer[i] = particleSensor.getIR();
        particleSensor.nextSample();

        maxim_heart_rate_and_oxygen_saturation(irBuffer, bufferLength, redBuffer, &spo2, &validSPO2, &heartRate, &validHeartRate);

        // محدود کردن منطقی
        if(validHeartRate){
          if(heartRate > MAX_HR) heartRate = MAX_HR;
          if(heartRate < MIN_HR) heartRate = MIN_HR; } if(validSPO2){ if(spo2 > MAX_SPO2) spo2 = MAX_SPO2;
          if(spo2 < MIN_SPO2) spo2 = MIN_SPO2; } // فیلتر نمایی EMA if(validHeartRate) emaHR = alpha * heartRate + (1 - alpha) * emaHR; if(validSPO2) emaSpO2 = alpha * spo2 + (1 - alpha) * emaSpO2; // نمایش روی LCD lcd.setCursor(0,0); lcd.print("HR:"); if(emaHR > 0){
              if(emaHR < 100) lcd.print(" "); // اضافه کردن فضای خالی برای دو رقمی lcd.print((int)emaHR); } else lcd.print("--"); lcd.print(" SpO2:"); if(emaSpO2 > 0){
              if(emaSpO2 < 100) lcd.print(" "); // فضای خالی برای تک رقمی یا دو رقمی lcd.print((int)emaSpO2); } else lcd.print("--"); // --- بررسی آلارم‌ها --- unsigned long currentMillis = millis(); // ضربان قلب if(emaHR > HR_THRESHOLD){
          if(!hrAlarmActive){
            hrAlarmActive = true;
            hrAlarmStart = currentMillis;
            digitalWrite(ALARM_PIN, HIGH);
          }
        }

        if(hrAlarmActive && (currentMillis - hrAlarmStart >= HR_ALARM_DURATION)){
          hrAlarmActive = false;
          digitalWrite(ALARM_PIN, LOW);
        }

        // SpO2
        if(emaSpO2 < SPO2_THRESHOLD){ if(!spo2AlarmActive){ spo2AlarmActive = true; spo2AlarmStart = currentMillis; digitalWrite(ALARM_PIN, HIGH); } } if(spo2AlarmActive && (currentMillis - spo2AlarmStart >= SPO2_ALARM_DURATION)){
          spo2AlarmActive = false;
          digitalWrite(ALARM_PIN, LOW);
        }

        delay(300); // به‌روزرسانی آرام برای LCD
      }
    }

    lcd.clear();
    lcd.setCursor(0,0);
    lcd.print("Place finger");
  }
}