#ifndef POLE_H
#define POLE_H

#include "Light.h"
#include "Observer.h"
#include "Pelican.h"

class Pole : public Observer {
  public:
    Pole(int x, int y, Pelican&);
    ~Pole();

    void setTrafficLights(Pelican::PelicanState state);
    void setWaitLight(bool pressed);

    void update();

  private:
    enum LightName { RED_LIGHT, YELLOW_LIGHT, GREEN_LIGHT, GREEN_MAN, RED_MAN };

    Light *lights[RED_MAN + 1];
    Light *waitLight;
    Pelican& pelican;
};

#endif