#include "map.h"
#include "Button.h"

#define MODE_SWITCH_PIN A0 //一个控制模式   1-5
#define LENGTH_SWITCH_PIN A1 //一个控制循环长度  2-16
#define BUTTON_DIR_PIN 8 //一控制倒放  高电平倒放
#define BUTTON_REST_PIN 9 //一个控制复位  重置是高电平有效
#define BUTTON_PP_PIN 10 //脉冲输入的引脚
// 定义灯光模式，模式由CTorp建议
#define PAT_SINGLE 1  // 单一模式 (依次点亮 1-2-3-4-5-6-7-8-1-2-3-4-5-6-7-8-)
#define PAT_INCROT 2  // 包含旋转模式 (依次点亮 1-2-3-4-5-6-7-8-8-7-6-5-4-3-2-1-)
#define PAT_EXCROT 3  // 排除旋转模式 (依次点亮 1-2-3-4-5-6-7-8-7-6-5-4-3-2-)
#define PAT_DOUBLE 4  // 双重模式 (依次点亮 1-1-2-2-3-3-4-4-5-5-6-6-7-7-8-8-)
#define PAT_RANDOM 5  // 随机模式

Button Button_0(BUTTON_PP_PIN, 1);  //
Button Button_1(BUTTON_DIR_PIN, 10);  //
Button Button_2(BUTTON_REST_PIN, 10);  //

unsigned int display_mode = PAT_SINGLE;  // 当前模式，默认为单一模式
unsigned int display_length = 2;  // 显示的长度  默认2
unsigned int pat_dir = 1;            // 模式方向，1表示向前进行模式，-1表示向后
bool pat_first = true;                // 对于成对的灯光，true表示第一个，false表示第二个


void setup() {
  Serial.begin(115200);

  pinMode(BUTTON_DIR_PIN, INPUT_PULLUP);
  pinMode(BUTTON_REST_PIN, INPUT_PULLUP);

  pinMode(HC165_DATA_PIN, INPUT);
  pinMode(HC165_CLOCK_PIN, OUTPUT);
  pinMode( HC165_LATCH_PIN, OUTPUT);

  pinMode(HC595_SHCP_PIN, OUTPUT);                               //串行时钟针脚配置成输出模式
  pinMode(HC595_STCP_PIN, OUTPUT);                               //锁存针脚配置成输出模式
  pinMode(HC595_DS_PIN, OUTPUT);                                 //数据输入针脚配置成输出模式
  for (int i =  0; i < numOfRegisterPins; i++) { //初始化引脚为低电平
    registers[i] = 0;
  }
  Button_0.begin();               // 初始化
  Button_1.begin();               // 初始化

}
uint32_t old_time = 0;
uint32_t old_time_button = 0;
int step_num = -1;
void loop() {
  display_mode = map(analogRead(MODE_SWITCH_PIN), 0, 1023, 1, 5);
  display_length = map(analogRead(LENGTH_SWITCH_PIN), 0, 1023, 2, 16);
  read_button();
  if (Button_0.pressed()) {
    if (Button_0.repeat_count() == 0) {  //被单击

    }
  }
  if (Button_0.released()) {  //被松开
    for (int i =  0; i < numOfRegisterPins; i++) { //初始化引脚为低电平
      registers[i] = 0;
    }
    writeRegisters(registers);
  }

  if (Button_1.pressed()) {
    if (Button_1.repeat_count() == 0) {  //被单击

    }
  }
  if (millis() - old_time >= 100) {
    old_time = millis();
    for (int i =  0; i < 16; i++) {
      // Serial.print(button_nums[i]); Serial.print(" ");
    }
    //Serial.println();

  }
}
void mode_1() {
  step_num++;
  if (step_num >= display_length) step_num = 0;
  registers[step_num] = 1;
  if (step_num > 0) {
    registers[step_num - 1] = 0;
  }
  writeRegisters(registers);
}
void read_button() {
  digitalWrite( HC165_LATCH_PIN, LOW);
  digitalWrite( HC165_LATCH_PIN, HIGH);

  for (int i = 0; i < numOfRegisterPins; i++) {
    button_nums[i] = digitalRead(HC165_DATA_PIN);
    digitalWrite(HC165_CLOCK_PIN, HIGH); // Shift out the next bit
    digitalWrite(HC165_CLOCK_PIN, LOW);
  }
}



void writeRegisters(bool* _state) {
  uint16_t data = 0;
  for (int i = 0; i < numOfRegisterPins; i++) {
    data |= uint16_t((_state[i] ? 1 : 0)) << i;
  }
  digitalWrite(HC595_STCP_PIN, LOW);                        //拉低锁存针脚，为传输数据做准备
  for (int i = 0; i < NUM_SHIFT_REGS; i++) {
    byte byteToSend = (data >> (8 * i)) & 0xFF;
    shiftOut(HC595_DS_PIN, HC595_SHCP_PIN, LSBFIRST, byteToSend);
  }
  digitalWrite(HC595_STCP_PIN, HIGH);                       //拉高锁存针脚，结束传输
}