#include <U8g2lib.h>

U8G2_SSD1306_128X64_NONAME_F_HW_I2C u8g2(U8G2_R0, /* reset=*/ U8X8_PIN_NONE);   //这是比较常用的0.9寸的OLED显示器的驱动

typedef struct 
{
  char *str;      // 字符串
  int len;        // 用于设置选择框的长度
}Ui_list_t;

// 按键值结构体
typedef struct
{
  byte val;
  byte last_val;
}KEY_T;

// 按键属性
typedef struct
{
  byte id;
  byte press;             // 按键按下标志
  byte update_flag;       // 按键更新标志
  byte res;
}KEY_MSG;

Ui_list_t list[] = 
{
  {"test1",7},
  {"test251",9},
  {"4tet", 6},
  {"tet4", 6}
};




KEY_T key[3] = {0};
KEY_MSG key_msg = {0};
int16_t x ,x_trg ;
int16_t y = 10, y_trg = 10;
int16_t menu_x, menu_x_trg;
int16_t menu_y = 40, menu_y_trg = 40;
int16_t frame_len = 5, frame_len_trg = 5;     // 选择框的宽度
int16_t frame_y = 0, frame_y_trg = 0;         // 选择框的高度
int16_t ui_select_index = 0;             // 选择索引
 

/* res为当前值 , res_trg为目标值, step为步长 */
int ui_run(int16_t *res, int16_t *res_trg, int16_t step)
{
  if (*res < *res_trg) 
  {
    *res += step;
    if (*res > *res_trg) *res = *res_trg;         // 加完溢出
  } 

  else if (*res > *res_trg) 
  {
    *res -= step;
    if (*res < *res_trg) *res = *res_trg;         // 减完溢出
  } 

  else 
  {
    return 0;
  }
  
  return 1;
}



/* 按键读取函数 */
bool button_read(uint8_t ch)
{
    if(ch == 0)
  {
    return digitalRead(2);
  }
  if (ch == 1)
  {
    return digitalRead(3);
  }
  if (ch == 2)
  {
    return digitalRead(4);
  }
}


/* 按键初始化函数 */
void button_init(void)
{
    /* 按键初始化 */
  pinMode(2, INPUT_PULLUP);
  pinMode(3, INPUT_PULLUP);
  pinMode(4, INPUT_PULLUP);

  /* 按键赋初值 */
  for (uint8_t i=0; i < 3; i++)
  {
    key[i].val = key[i].last_val = button_read(i);
  }

}

/* 按键处理函数 */
void button_scan(void)
{
    for(uint8_t i = 0;i<3;i++)
  {
    key[i].val =  button_read(i);

    /* 按键按下 */
    if(key[i].val != key[i].last_val)
    {
      key[i].last_val = key[i].val;
      if(key[i].val == 0)
      {
        key_msg.id = i;
        key_msg.press = 1;                  // 将按下标志位置1
        key_msg.update_flag = 1;            // 将更新标志位置1 
        Serial.println(key_msg.id);
      }
    }
  }
}

/* 选择框UI绘制 */
void select_ui_show(void)
{
  u8g2.setDrawColor(2);                             // 异或,颜色反转
  u8g2.drawRBox(x,frame_y,frame_len,12,1);        // 绘制选择框
  u8g2.setDrawColor(1);                             // 异或,颜色恢复

  ui_run(&frame_y, &frame_y_trg, 2);                // y轴移动
  ui_run(&frame_len, &frame_len_trg, 10);           // x轴移动
} 

/* 菜单页UI绘制 */
void menu_ui_show(void)
{
    // u8g2.clearBuffer();             // 清除缓存

    for (int i = 0; i < 3; i++ )
    {
      u8g2.drawStr(menu_x+50+i*90, menu_y, "test");
    }
        
    ui_run(&menu_x, &menu_x_trg,10);          // ui移动
    ui_run(&menu_y, &menu_y_trg,5);          // ui移动

    // u8g2.sendBuffer();              // 将缓存发送并显示

}




void select_test_ui_show(void)
{
  
}

/* ui显示函数 */
void Ui_show(void)
{
    int list_len = sizeof(list) / sizeof(Ui_list_t);
     u8g2.clearBuffer();             // 清除缓存

    for (int i = 0; i < list_len; i++ )
    {
      u8g2.drawStr(x+5, y+i*10, list[i].str);
    }

    select_ui_show();               // 选择框UI绘制
    // ui_run(&y, &y_trg,10);
     u8g2.sendBuffer();              // 将缓存发送并显示
}

/* ui处理函数 */
void ui_proc(void)
{
  int list_len = sizeof(list) / sizeof(Ui_list_t);      // 选项列表的个数

  if(key_msg.update_flag && key_msg.press)     // 当更新标志位和按下标志位都为1
  {
    key_msg.update_flag = 0;
    if(key_msg.id == 1)
    {
      y_trg += 10;
      Serial.println("y_trg:");
      Serial.println(y_trg);

        ui_select_index ++;
        if (ui_select_index >= list_len) 
        {
          ui_select_index = list_len - 1;
        } 
        Serial.println("ui_select_index:");
        Serial.println(ui_select_index);      
    }
    else
    {
      y_trg -= 10;
      Serial.println("y_trg:");
      Serial.println(y_trg);
      
        ui_select_index --;
        if (ui_select_index <= 0) 
        {
          ui_select_index = 0;
        }      
        Serial.println("ui_select_index:");
        Serial.println(ui_select_index);


      // /* 防止溢出 */
      // if(y_trg < 10 )
      // {
      //   y_trg = 10;
      // }
    }
  }

  frame_y_trg = ui_select_index * 10;
  frame_len_trg = list[ui_select_index].len *5;
  Ui_show();
}

void setup(void) 
{

  Serial.begin(115200);
  button_init();                                    // 按键初始化函数
  u8g2.begin();
  u8g2.setFont(u8g2_font_ncenB08_tr);                 // 设置字体
}

void loop(void) 
{
    button_scan();                  // 按键处理函数
    ui_proc();                      // ui 处理函数
}