src/lib/SCREEN/devScreen.cpp

   1 #include "targets.h"
   2
   3 #if defined(PLATFORM_ESP32)
   4
   5 #include "devScreen.h"
   6
   7 #include "common.h"
   8 #include "logging.h"
   9
  10 #include "OLED/oleddisplay.h"
  11 #include "TFT/tftdisplay.h"
  12
  13 #include "devButton.h"
  14 #include "handset.h"
  15
  16 FiniteStateMachine state_machine(entry_fsm);
  17
  18 Display *display;
  19
  20 #include "FiveWayButton/FiveWayButton.h"
  21 FiveWayButton fivewaybutton;
  22
  23 #include "gsensor.h"
  24 extern Gsensor gsensor;
  25 static bool is_screen_flipped = false;
  26 static bool is_pre_screen_flipped = false;
  27
  28 #define SCREEN_DURATION 20
  29
  30 extern void jumpToWifiRunning();
  31 extern void jumpToBleRunning();
  32
  33 static bool jumpToBandSelect = false;
  34 static bool jumpToChannelSelect = false;
  35
  36 static int handle(void)
  37 {
  38     is_screen_flipped = gsensor.isFlipped();
  39
  40     if ((is_screen_flipped == true) && (is_pre_screen_flipped == false))
  41     {
  42         display->doScreenBackLight(SCREEN_BACKLIGHT_OFF);
  43     }
  44     else if ((is_screen_flipped == false) && (is_pre_screen_flipped == true))
  45     {
  46         display->doScreenBackLight(SCREEN_BACKLIGHT_ON);
  47         state_machine.start(millis(), STATE_IDLE);
  48     }
  49     is_pre_screen_flipped = is_screen_flipped;
  50     if (is_screen_flipped)
  51     {
  52         return 100; // no need to check as often if the screen is off!
  53     }
  54     uint32_t now = millis();
  55
  56     if (state_machine.getParentState() != STATE_WIFI_TX && connectionState == wifiUpdate)
  57     {
  58         jumpToWifiRunning();
  59     }
  60
  61     if (state_machine.getParentState() != STATE_JOYSTICK && connectionState == bleJoystick)
  62     {
  63         jumpToBleRunning();
  64     }
  65
  66     if (!handset->IsArmed())
  67     {
  68         int key;
  69         bool isLongPressed;
  70         // if we are using analog joystick then we can't cancel because WiFi is using the ADC2 (i.e. channel >= 8)!
  71         if (connectionState == wifiUpdate && digitalPinToAnalogChannel(GPIO_PIN_JOYSTICK) >= 8)
  72         {
  73             key = INPUT_KEY_NO_PRESS;
  74         }
  75         else
  76         {
  77             fivewaybutton.update(&key, &isLongPressed);
  78         }
  79         fsm_event_t fsm_event;
  80         switch (key)
  81         {
  82         case INPUT_KEY_DOWN_PRESS:
  83             fsm_event = EVENT_DOWN;
  84             break;
  85         case INPUT_KEY_UP_PRESS:
  86             fsm_event = EVENT_UP;
  87             break;
  88         case INPUT_KEY_LEFT_PRESS:
  89             fsm_event = EVENT_LEFT;
  90             break;
  91         case INPUT_KEY_RIGHT_PRESS:
  92             fsm_event = EVENT_RIGHT;
  93             break;
  94         case INPUT_KEY_OK_PRESS:
  95             fsm_event = EVENT_ENTER;
  96             break;
  97         default: // INPUT_KEY_NO_PRESS
  98             fsm_event = EVENT_TIMEOUT;
  99         }
 100         if (fsm_event != EVENT_TIMEOUT && isLongPressed)
 101         {
 102             fsm_event = (fsm_event | LONG_PRESSED);
 103         }
 104 #if defined(DEBUG_SCREENSHOT)
 105         if (key == INPUT_KEY_DOWN_PRESS && isLongPressed)
 106         {
 107             DBGLN("state_%d", state_machine.getCurrentState());
 108             Display::printScreenshot();
 109         }
 110 #endif
 111         if (jumpToBandSelect)
 112         {
 113             state_machine.jumpTo(vtx_menu_fsm, STATE_VTX_BAND);
 114             state_machine.jumpTo(value_select_fsm, STATE_VALUE_INIT);
 115             jumpToBandSelect = false;
 116         }
 117         else if (jumpToChannelSelect)
 118         {
 119             state_machine.jumpTo(vtx_menu_fsm, STATE_VTX_CHANNEL);
 120             state_machine.jumpTo(value_select_fsm, STATE_VALUE_INIT);
 121             jumpToChannelSelect = false;
 122         }
 123         else
 124         {
 125             state_machine.handleEvent(now, fsm_event);
 126         }
 127     }
 128     else
 129     {
 130         state_machine.handleEvent(now, EVENT_TIMEOUT);
 131     }
 132
 133     return SCREEN_DURATION;
 134 }
 135
 136 static void initialize()
 137 {
 138     if (OPT_HAS_SCREEN)
 139     {
 140         fivewaybutton.init();
 141         if (OPT_HAS_TFT_SCREEN)
 142         {
 143             display = new TFTDisplay();
 144         }
 145         else
 146         {
 147             display = new OLEDDisplay();
 148         }
 149         display->init();
 150         state_machine.start(millis(), getInitialState());
 151
 152         registerButtonFunction(ACTION_GOTO_VTX_BAND, [](){
 153             jumpToBandSelect = true;
 154             devicesTriggerEvent();
 155         });
 156         registerButtonFunction(ACTION_GOTO_VTX_CHANNEL, [](){
 157             jumpToChannelSelect = true;
 158             devicesTriggerEvent();
 159         });
 160     }
 161 }
 162
 163 static int start()
 164 {
 165     if (OPT_HAS_SCREEN)
 166     {
 167         return DURATION_IMMEDIATELY;
 168     }
 169     return DURATION_NEVER;
 170 }
 171
 172 static int event()
 173 {
 174     if (OPT_HAS_SCREEN)
 175     {
 176         return handle();
 177     }
 178     return DURATION_NEVER;
 179 }
 180
 181 static int timeout()
 182 {
 183     if (OPT_HAS_SCREEN)
 184     {
 185         return handle();
 186     }
 187     return DURATION_NEVER;
 188 }
 189
 190 device_t Screen_device = {
 191     .initialize = initialize,
 192     .start = start,
 193     .event = event,
 194     .timeout = timeout};
 195 #endif