src/lib/BUTTON/devButton.cpp

   1 #include "devButton.h"
   2
   3 #include "logging.h"
   4 #include "button.h"
   5 #include "config.h"
   6 #include "helpers.h"
   7 #include "handset.h"
   8
   9 static Button button1;
  10 static Button button2;
  11
  12 // only check every second if the device is in-use, i.e. RX connected, or TX is armed
  13 static constexpr int MS_IN_USE = 1000;
  14
  15 #if defined(TARGET_RX)
  16 static constexpr struct {
  17     bool pressType;
  18     uint8_t count;
  19     action_e action;
  20 } button_actions[] = {
  21     // half second durations + 1 (i.e. 2=1.5s)
  22     {true, 2, ACTION_BIND},             // 1.5s
  23     {true, 9, ACTION_START_WIFI},       // 5.0s
  24     {true, 23, ACTION_RESET_REBOOT}     // 12.0s
  25 };
  26 #endif
  27
  28 static ButtonAction_fn actions[ACTION_LAST] = { nullptr };
  29
  30 void registerButtonFunction(action_e action, ButtonAction_fn function)
  31 {
  32     actions[action] = function;
  33 }
  34
  35 size_t button_GetActionCnt()
  36 {
  37 #if defined(TARGET_RX)
  38     return ARRAY_SIZE(button_actions);
  39 #else
  40     return CONFIG_TX_BUTTON_ACTION_CNT;
  41 #endif
  42 }
  43
  44 static void handlePress(uint8_t button, bool longPress, uint8_t count)
  45 {
  46     DBGLN("handlePress(%u, %u, %u)", button, (uint8_t)longPress, count);
  47 #if defined(TARGET_TX)
  48     const button_action_t *button_actions = config.GetButtonActions(button)->val.actions;
  49 #endif
  50     for (unsigned i=0 ; i<button_GetActionCnt() ; i++)
  51     {
  52         if (button_actions[i].action != ACTION_NONE && button_actions[i].pressType == longPress && button_actions[i].count == count-1)
  53         {
  54             if (actions[button_actions[i].action])
  55             {
  56                 actions[button_actions[i].action]();
  57             }
  58         }
  59     }
  60 }
  61
  62 static int start()
  63 {
  64     if (GPIO_PIN_BUTTON == UNDEF_PIN && GPIO_PIN_BUTTON2 == UNDEF_PIN)
  65     {
  66         return DURATION_NEVER;
  67     }
  68
  69     if (GPIO_PIN_BUTTON != UNDEF_PIN)
  70     {
  71         button1.init(GPIO_PIN_BUTTON);
  72         button1.OnShortPress = [](){ handlePress(0, false, button1.getCount()); };
  73         button1.OnLongPress = [](){ handlePress(0, true, button1.getLongCount()+1); };
  74     }
  75     if (GPIO_PIN_BUTTON2 != UNDEF_PIN)
  76     {
  77         button2.init(GPIO_PIN_BUTTON2);
  78         button2.OnShortPress = [](){ handlePress(1, false, button2.getCount()); };
  79         button2.OnLongPress = [](){ handlePress(1, true, button2.getLongCount()+1); };
  80     }
  81
  82     return DURATION_IMMEDIATELY;
  83 }
  84
  85 static int event()
  86 {
  87     if (GPIO_PIN_BUTTON == UNDEF_PIN && GPIO_PIN_BUTTON2 == UNDEF_PIN)
  88     {
  89         return DURATION_NEVER;
  90     }
  91 #if defined(TARGET_TX)
  92     if (handset->IsArmed())
  93     {
  94         return DURATION_NEVER;
  95     }
  96 #else
  97     if (connectionState == connected)
  98     {
  99         return DURATION_NEVER;
 100     }
 101 #endif
 102     return DURATION_IMMEDIATELY;
 103 }
 104
 105 static int timeout()
 106 {
 107     int timeout = DURATION_NEVER;
 108     if (GPIO_PIN_BUTTON != UNDEF_PIN)
 109     {
 110         timeout = button1.update();
 111     }
 112     if (GPIO_PIN_BUTTON2 != UNDEF_PIN)
 113     {
 114         timeout = button2.update();
 115     }
 116     return timeout;
 117 }
 118
 119 device_t Button_device = {
 120     .initialize = nullptr,
 121     .start = start,
 122     .event = event,
 123     .timeout = timeout
 124 };