/********* Pleasedontcode.com **********

	Pleasedontcode thanks you for automatic code generation! Enjoy your code!

	- Terms and Conditions:
	You have a non-exclusive, revocable, worldwide, royalty-free license
	for personal and commercial use. Attribution is optional; modifications
	are allowed, but you're responsible for code maintenance. We're not
	liable for any loss or damage. For full terms,
	please visit pleasedontcode.com/termsandconditions.

	- Project: "Methane Detector"
	- Source Code compiled for: Arduino Uno
	- Source Code created on: 2023-12-08 16:38:50

********* Pleasedontcode.com **********/

/****** SYSTEM REQUIREMENTS *****/
/****** SYSTEM REQUIREMENT 1 *****/
	/* Methane Level Detection:    Utilize the methane */
	/* gas sensor to detect the presence of methane, */
	/* which is released as fruits ripen or start to */
	/* decay.  Calibrate the sensor to distinguish */
	/* between safe levels and levels indicating */
	/* potential spoilage. */
/****** END SYSTEM REQUIREMENTS *****/

/****** DEFINITION OF LIBRARIES *****/
#include 
#include 
#include 

/****** FUNCTION PROTOTYPES *****/
void setup(void);
void loop(void);
void calibrateMQ4(void);
void detectMethaneLevel(void);

/***** DEFINITION OF DIGITAL INPUT PINS *****/
const uint8_t MQ4_PushButton_PIN_D2 = 2;
const uint8_t MQ4_PushButton_PIN_D3 = 3;
const uint8_t MQ4_Analog_PIN_A4 = A4;

/****** DEFINITION OF LIBRARIES CLASS INSTANCES*****/
EasyButton button1(MQ4_PushButton_PIN_D2, false, false);
EasyButton button2(MQ4_PushButton_PIN_D3, false, false);
MQUnifiedsensor MQ4("Arduino UNO", 5, 10, MQ4_Analog_PIN_A4, "MQ-4");

bool calibrateMode = false;

void setup(void)
{
  // Initialize Serial for debugging purposes.
  Serial.begin(9600);

  // Initialize the buttons.
  button1.begin();
  button2.begin();

  // Set up the calibration mode button callback.
  button1.onPressed(calibrateMQ4);

  // Set up the detection mode button callback.
  button2.onPressed(detectMethaneLevel);
  
  // Initialize the MQ-4 gas sensor.
  MQ4.setRegressionMethod(1);
  MQ4.init();

  // Check if the MQ-4 gas sensor calibration is required.
  float calcR0 = MQ4.getR0();
  if (calcR0 == 0) {
    // Perform calibration if R0 value is not set.
    calibrateMode = true;
    Serial.println("Calibration mode - Press button 1 to calibrate.");
  } else {
    // Start detection mode if R0 value is already set.
    Serial.println("Detection mode - Press button 2 to detect methane level.");
  }
}

void loop(void)
{
  // Check button states
  button1.read();
  button2.read();

  // Perform calibration or detection based on mode
  if (calibrateMode) {
    calibrateMQ4();
  } else {
    detectMethaneLevel();
  }
}

void calibrateMQ4()
{
  // Perform calibration of MQ-4 gas sensor
  MQ4.setR0(MQ4.calibrate(4.4));
  calibrateMode = false;
  Serial.println("Calibration complete - Press button 2 to detect methane level.");
}

void detectMethaneLevel()
{
  // Read methane level from MQ-4 gas sensor
  MQ4.setA(1012.7);
  MQ4.setB(-2.786);
  float methaneLevel = MQ4.readSensor();

  // Display methane level
  Serial.print("Methane Level: ");
  Serial.println(methaneLevel);

  // Add logic to determine spoilage based on methane level threshold
  // ...

  delay(1000);
}