26 lines
413 B
C++
26 lines
413 B
C++
|
|
|
|
#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
|