public class AMPEL
{
    private SPIELWALZE oben;
    private SPIELWALZE mitte;
    private SPIELWALZE unten;

    enum Zustand { aus, gruen, rot, gelb, rotgelb };
    private Zustand z;

    public AMPEL() {
        z = Zustand.aus;
        oben = new SPIELWALZE(50,50,50,0);
        mitte = new SPIELWALZE(50,120,50,0);
        unten = new SPIELWALZE(50,190,50,0);

        oben.zeichne();
        mitte.zeichne();
        unten.zeichne();
    }

    // Methoden

    private void setzeZustand(Zustand neuerZustand) {
        z = neuerZustand;
        if (neuerZustand == Zustand.gruen) {
            oben.faerbeLampe(0);
            mitte.faerbeLampe(0);
            unten.faerbeLampe(2);
            neuZeichnen();
        } else if (neuerZustand == Zustand.rot) {
            oben.faerbeLampe(4);
            mitte.faerbeLampe(0);
            unten.faerbeLampe(0);
            neuZeichnen();
        } else if (neuerZustand == Zustand.gelb) {
            oben.faerbeLampe(0);
            mitte.faerbeLampe(6);
            unten.faerbeLampe(0);
            neuZeichnen();
        } else if (neuerZustand == Zustand.rotgelb) {
            oben.faerbeLampe(4);
            mitte.faerbeLampe(6);
            unten.faerbeLampe(0);
            neuZeichnen();
        } else if (neuerZustand == Zustand.aus) {
            oben.faerbeLampe(0);
            mitte.faerbeLampe(0);
            unten.faerbeLampe(0);
            neuZeichnen();
        }
    }


    public void ein() {
        if (z == Zustand.aus) {
            setzeZustand(Zustand.gruen);
        }
    }

    public void aus() {
        if (z == Zustand.gruen) {
            setzeZustand(Zustand.aus);
        }
    }

    public void weiter() {
        if (z == Zustand.rot) {
            setzeZustand(Zustand.rotgelb);
        } else if (z == Zustand.gruen) {
            setzeZustand(Zustand.gelb);
        } else if (z == Zustand.gelb) {
            setzeZustand(Zustand.rot);
        } else if (z == Zustand.rotgelb) {
            setzeZustand(Zustand.gruen);
        }
    }

    private void neuZeichnen() {
        oben.zeichne();
        mitte.zeichne();
        unten.zeichne();
    }


}