#include <mechButton.h>

#define RED_PIN     12
#define YELLOW_PIN  11
#define GREEN_PIN   10
#define BTN_PIN     9

mechButton  theButton(BTN_PIN);
int         count;

void setup() {

  pinMode(RED_PIN, OUTPUT);
  pinMode(YELLOW_PIN, OUTPUT);
  pinMode(GREEN_PIN, OUTPUT);
  theButton.setCallback(butnClicked);
  count = 0;
  setLights();
}

void setLights(void) {

  if (count>2) count = 0;
  digitalWrite(RED_PIN,LOW);
  digitalWrite(YELLOW_PIN,LOW);
  digitalWrite(GREEN_PIN,LOW);
  switch(count) {
    case 0 : digitalWrite(RED_PIN,HIGH); break;
    case 1 : digitalWrite(YELLOW_PIN,HIGH); break;
    case 2 : digitalWrite(GREEN_PIN,HIGH); break;
  }
}


void butnClicked(void) {

  if (!theButton.trueFalse()) {
    count++;
    setLights();
  }
}


void loop() {

  idle();
}