// Semafor.hpp
#ifndef SEMAFOR_H_INCLUDED
#define SEMAFOR_H_INCLUDED

#include <iostream>
#include <cstdlib>
#include <math.h>

using namespace std;

enum stanje
{
    sON, sOFF, sOUT, sBLINK
};

enum boje
{
    cRED, cGREEN, cYELLOW, cYELLOWRED, cBLINK, cNONE
};

class Semafor
{
    private:
        stanje state;
        boje color;

    public:
        Semafor();
        Semafor(const Semafor&);

        stanje getStanje() const;
        boje getBoja() const;
        bool turnOn();
        bool turnOff();
        bool turnOut();
        bool turnBlink();
        bool repair();
        bool changeColor();
};

#endif // SEMAFOR_H_INCLUDED


// Semafor.cpp
#include "semafor.h"

Semafor :: Semafor()
{
    state = sOFF;
    color = cNONE;
}

Semafor :: Semafor(Semafor const &rf)
{
    state = rf.state;
    color = rf.color;
}

stanje Semafor :: getStanje() const
{
    return state;
}

boje Semafor :: getBoja() const
{
    return color;
}

bool Semafor :: turnOn()
{
    if (state == sOFF)
    {
        state = sON;
        color = cRED;
        return true;
    }
    else
        return false;
}

bool Semafor :: turnOff()
{
    if (state == sON || state == sBLINK)
    {
        state = sOFF;
        color = cNONE;
        return true;
    }
    else
        return false;
}

bool Semafor :: turnOut()
{
    if (state != sOUT)
    {
        state = sOUT;
        color = cNONE;
        return true;
    }
    else
        return false;
}

bool Semafor :: turnBlink()
{
    if (state == sOFF)
    {
        state = sBLINK;
        color = cBLINK;
        return true;
    }
    else
        return false;
}

bool Semafor :: repair()
{
    if (state == sOUT)
    {
        state = sOFF;
        color = cNONE;
        return true;
    }
    else
        return false;
}

bool Semafor :: changeColor()
{
    if (state == sON)
    {
        switch(color)
        {
            case cRED:
                color = cYELLOWRED;
                break;

            case cYELLOWRED:
                color = cGREEN;
                break;

            case cGREEN:
                color = cYELLOW;
                break;

            case cYELLOW:
                color = cRED;
                break;

            default:
                color = cNONE;
        }
        return true;
    }
    else
        return false;
}



// Main.cpp
#include "semafor.h"

void printSemafor(const Semafor &rf)
{
    cout << "*** Stanje semafora: ";
    switch(rf.getStanje())
    {
        case sON:
            cout << "ON! ***" << endl;
            break;

        case sOFF:
            cout << "OFF! ***" << endl;
            break;

        case sOUT:
            cout << "OUT! ***" << endl;
            break;

        case sBLINK:
            cout << "BLINK! ***" << endl;
            break;

        default:
            cout << "N/A ***" << endl;
    }

    cout << "*** Boja: ";
    switch(rf.getBoja())
    {
        case cNONE:
            cout << "NONE! ***" << endl;
            break;

        case cBLINK:
            cout << "BLINK! ***" << endl;
            break;

        case cRED:
            cout << "RED! ***" << endl;
            break;

        case cYELLOW:
        case cYELLOWRED:
            cout << "YELLOW! ***" << endl;
            break;

        case cGREEN:
            cout << "GREEN! ***" << endl;
            break;

        default:
            cout << "N/A ***" << endl;
    }
    cout << endl;
}

char meni()
{
    char odg;

    do
    {
        cout << "Izaberite opciju: " << endl;
        cout << "1. Ukljuci semafor" << endl;
        cout << "2. Iskljuci semafor" << endl;
        cout << "3. Ukljuci trepcuce zuto" << endl;
        cout << "4. Pokvari semafor" << endl;
        cout << "5. Popravi semafor" << endl;
        cout << "6. Promjeni boju" << endl;
        cout << "7. Kraj rada" << endl;
        cin >> odg;
    }
    while (odg < '1' || odg > '7');

    return odg;
}

int main()
{
    Semafor s;
    char ch;

    do
    {
        ch = meni();
        switch(ch)
        {
            case '1':
                if (s.turnOn())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

            case '2':
                if (s.turnOff())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

            case '3':
                if (s.turnBlink())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

            case '4':
                if (s.turnOut())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

            case '5':
                if (s.repair())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

            case '6':
                if (s.changeColor())
                    cout << "Operacija izvrsena!" << endl;
                else
                    cout << "Opercaija nije izvrsena!" << endl;
                    printSemafor(s);
                    break;

        }
    }
    while (ch != '7');

    return 0;

}