From bb33c65f11752a676f9c4a24378758a709b58264 Mon Sep 17 00:00:00 2001
From: Ivan Efimov <gendalf44@yandex.ru>
Date: Thu, 26 May 2022 22:50:45 -0500
Subject: [PATCH] Changes in crash flip OSD warnings:

1. Show "ARMED"<new line>"> CRASH FLIP <" when arming in crash flip
2. Show "CRASH FLIP SWITCH" when disarmed, but in crash flip.
3. Hide the disarmed stats screen if "CRASH FLIP SWITCH" is activated.
---
 src/main/fc/core.c          |  2 +-
 src/main/osd/osd.c          | 11 +++++++++--
 src/main/osd/osd_warnings.c | 16 ++++++++++++----
 src/main/osd/osd_warnings.h |  2 ++
 4 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/src/main/fc/core.c b/src/main/fc/core.c
index c6b36d749b..9e542d6726 100644
--- a/src/main/fc/core.c
+++ b/src/main/fc/core.c
@@ -440,7 +440,7 @@ void disarm(flightLogDisarmReason_e reason)
         lastDisarmTimeUs = micros();
 
 #ifdef USE_OSD
-        if (flipOverAfterCrashActive || isLaunchControlActive()) {
+        if (IS_RC_MODE_ACTIVE(BOXFLIPOVERAFTERCRASH) || isLaunchControlActive()) {
             osdSuppressStats(true);
         }
 #endif
diff --git a/src/main/osd/osd.c b/src/main/osd/osd.c
index 9c45c5cc75..63f8846998 100644
--- a/src/main/osd/osd.c
+++ b/src/main/osd/osd.c
@@ -60,6 +60,7 @@
 #include "drivers/sdcard.h"
 #include "drivers/time.h"
 
+#include "fc/core.h"
 #include "fc/rc_controls.h"
 #include "fc/rc_modes.h"
 #include "fc/runtime_config.h"
@@ -79,6 +80,7 @@
 
 #include "osd/osd.h"
 #include "osd/osd_elements.h"
+#include "osd/osd_warnings.h"
 
 #include "pg/motor.h"
 #include "pg/pg.h"
@@ -1016,6 +1018,10 @@ static timeDelta_t osdShowArmed(void)
     }
     displayWrite(osdDisplayPort, 12, 7, DISPLAYPORT_ATTR_NONE, "ARMED");
 
+    if (isFlipOverAfterCrashActive()) {
+        displayWrite(osdDisplayPort, 8, 8, DISPLAYPORT_ATTR_NONE, CRASH_FLIP_WARNING);
+    }
+
     return ret;
 }
 
@@ -1087,8 +1093,9 @@ void osdProcessStats2(timeUs_t currentTimeUs)
 
     if (resumeRefreshAt) {
         if (cmp32(currentTimeUs, resumeRefreshAt) < 0) {
-            // in timeout period, check sticks for activity to resume display.
-            if (IS_HI(THROTTLE) || IS_HI(PITCH)) {
+            // in timeout period, check sticks for activity or CRASH FLIP switch to resume display.
+            if (!ARMING_FLAG(ARMED) &&
+                (IS_HI(THROTTLE) || IS_HI(PITCH) || IS_RC_MODE_ACTIVE(BOXFLIPOVERAFTERCRASH))) {
                 resumeRefreshAt = currentTimeUs;
             }
             return;
diff --git a/src/main/osd/osd_warnings.c b/src/main/osd/osd_warnings.c
index 05c056e7f7..d0f823c5e1 100644
--- a/src/main/osd/osd_warnings.c
+++ b/src/main/osd/osd_warnings.c
@@ -63,6 +63,8 @@
 #include "sensors/battery.h"
 #include "sensors/sensors.h"
 
+const char CRASH_FLIP_WARNING[] = "> CRASH FLIP <";
+
 void renderOsdWarning(char *warningText, bool *blinking, uint8_t *displayAttr)
 {
     const batteryState_e batteryState = getBatteryState();
@@ -132,10 +134,16 @@ void renderOsdWarning(char *warningText, bool *blinking, uint8_t *displayAttr)
     }
 
     // Warn when in flip over after crash mode
-    if (osdWarnGetState(OSD_WARNING_CRASH_FLIP) && isFlipOverAfterCrashActive()) {
-        tfp_sprintf(warningText, "CRASH FLIP");
-        *displayAttr = DISPLAYPORT_ATTR_INFO;
-        return;
+    if (osdWarnGetState(OSD_WARNING_CRASH_FLIP) && IS_RC_MODE_ACTIVE(BOXFLIPOVERAFTERCRASH)) {
+        if (isFlipOverAfterCrashActive()) { // if was armed in crash flip mode
+            tfp_sprintf(warningText, CRASH_FLIP_WARNING);
+            *displayAttr = DISPLAYPORT_ATTR_INFO;
+            return;
+        } else if (!ARMING_FLAG(ARMED)) { // if disarmed, but crash flip mode is activated
+            tfp_sprintf(warningText, "CRASH FLIP SWITCH");
+            *displayAttr = DISPLAYPORT_ATTR_INFO;
+            return;
+        }
     }
 
 #ifdef USE_LAUNCH_CONTROL
diff --git a/src/main/osd/osd_warnings.h b/src/main/osd/osd_warnings.h
index 58e0062faa..393ea8b26a 100644
--- a/src/main/osd/osd_warnings.h
+++ b/src/main/osd/osd_warnings.h
@@ -23,6 +23,8 @@
 #define OSD_WARNINGS_MAX_SIZE 12
 #define OSD_FORMAT_MESSAGE_BUFFER_SIZE (OSD_WARNINGS_MAX_SIZE + 1)
 
+extern const char CRASH_FLIP_WARNING[];
+
 STATIC_ASSERT(OSD_FORMAT_MESSAGE_BUFFER_SIZE <= OSD_ELEMENT_BUFFER_LENGTH, osd_warnings_size_exceeds_buffer_size);
 
 void renderOsdWarning(char *warningText, bool *blinking, uint8_t *displayAttr);