#include "mainwindow.h"
#include "ui_mainwindow.h"
#include <QApplication>
#include <QWidget>
#include <QLabel>
#include <QPushButton>
#include <QGridLayout>
#include <QTimer>

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);
}

MainWindow::~MainWindow()
{
    delete ui;
}

class TrafficLightWidget : public QWidget {
public:
    TrafficLightWidget(QWidget *parent = nullptr) : QWidget(parent) {
        // Set up the layout
        QGridLayout *layout = new QGridLayout(this);

        // Create the labels to display the traffic light colors
        greenLabel = new QLabel(this);
        yellowLabel = new QLabel(this);
        redLabel = new QLabel(this);

        // Create the button to start the traffic light
        startButton = new QPushButton("Start", this);

        // Add the labels and button to the layout
        layout->addWidget(greenLabel, 0, 0);
        layout->addWidget(yellowLabel, 0, 1);
        layout->addWidget(redLabel, 0, 2);
        layout->addWidget(startButton, 1, 0, 1, 3);

        // Connect the start button to the startTrafficLight() slot
        connect(startButton, &QPushButton::clicked, this, &TrafficLightWidget::startTrafficLight);
    }

private slots:
    void startTrafficLight() {
        // Disable the button to prevent changing the time interval during execution
        startButton->setEnabled(false);

        // Set the default time interval in milliseconds (change this to desired interval)
        int timeInterval = 2000;

        // Load the image files for green, yellow, and red lights
        QPixmap greenPixmap("green.jpg");
        QPixmap yellowPixmap("yellow.jpg");
        QPixmap redPixmap("red.jpg");

        // Set the images for the labels
        greenLabel->setPixmap(greenPixmap);
        yellowLabel->setPixmap(yellowPixmap);
        redLabel->setPixmap(redPixmap);

        // Hide the yellow and red lights at the beginning
        yellowLabel->hide();
        redLabel->hide();

        // Create a QTimer to handle the time intervals
        QTimer *timer = new QTimer(this);

        // Connect the timer's timeout signal to the updateTrafficLight() slot
        connect(timer, &QTimer::timeout, this, &TrafficLightWidget::updateTrafficLight);

        // Start the timer with the specified time interval
        timer->start(timeInterval);
    }

    void updateTrafficLight() {
        // Toggle the visibility of the lights
        if (greenLabel->isVisible()) {
            greenLabel->hide();
            yellowLabel->show();
        } else if (yellowLabel->isVisible()) {
            yellowLabel->hide();
            redLabel->show();
        } else if (redLabel->isVisible()) {
            redLabel->hide();
            greenLabel->show();
        }
    }

private:
    QLabel *greenLabel;
    QLabel *yellowLabel;
    QLabel *redLabel;
    QPushButton *startButton;
};

int main(int argc, char **argv)
{
    QApplication app(argc, argv);

    widget.resize(110, 300);
    widget.show();


    return app.exec();
}

#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>

namespace Ui {
class MainWindow;
}

class MainWindow : public QMainWindow
{
    Q_OBJECT

public:
    explicit MainWindow(QWidget *parent = 0);
    ~MainWindow();

private:
    Ui::MainWindow *ui;
};

#endif // MAINWINDOW_H