#ifndef POLE_H
#define POLE_H

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

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

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

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

    Light *lights[RED_MAN + 1];

    Light *waitLight;
};

#endif