src/main/cms/cms_menu_vtx.c

   1 /*
   2  * This file is part of Cleanflight.
   3  *
   4  * Cleanflight is free software: you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation, either version 3 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * Cleanflight is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with Cleanflight.  If not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #include <ctype.h>
  19 #include <stdbool.h>
  20 #include <stdint.h>
  21 #include <string.h>
  22
  23 #include "platform.h"
  24
  25 #include "build/version.h"
  26
  27 #if defined(USE_CMS) && defined(USE_VTX_CONTROL)
  28
  29 #include "common/printf.h"
  30 #include "common/utils.h"
  31
  32 #include "cms/cms.h"
  33 #include "cms/cms_types.h"
  34 #include "cms/cms_menu_vtx.h"
  35
  36 #include "drivers/vtx_common.h"
  37
  38 #include "fc/config.h"
  39
  40 #include "io/vtx_string.h"
  41 #include "io/vtx.h"
  42
  43
  44 // Config-time settings
  45 static uint8_t  vtxBand = 0;
  46 static uint8_t  vtxChan = 0;
  47 static uint8_t  vtxPower = 0;
  48 static uint8_t  vtxPitMode = 0;
  49
  50 static const char * const vtxCmsPitModeNames[] = {
  51     "---", "OFF", "ON "
  52 };
  53
  54 // Menus (these are not const because we update them at run-time )
  55 static OSD_TAB_t cms_Vtx_EntBand = { &vtxBand, VTX_SETTINGS_BAND_COUNT, vtx58BandNames };
  56 static OSD_TAB_t cms_Vtx_EntChan = { &vtxChan, VTX_SETTINGS_CHANNEL_COUNT, vtx58ChannelNames };
  57 static OSD_TAB_t cms_Vtx_EntPower = { &vtxPower, VTX_SETTINGS_POWER_COUNT, vtx58DefaultPowerNames };
  58 static const OSD_TAB_t cms_Vtx_EntPitMode = { &vtxPitMode, 2, vtxCmsPitModeNames };
  59
  60 static long cms_Vtx_configPitMode(displayPort_t *pDisp, const void *self)
  61 {
  62     UNUSED(pDisp);
  63     UNUSED(self);
  64
  65     if (vtxPitMode == 0) {
  66         vtxPitMode = 1;
  67     }
  68
  69     // Pit mode changes are immediate, without saving
  70     vtxCommonSetPitMode(vtxCommonDevice(), vtxPitMode >= 2 ? 1 : 0);
  71
  72     return 0;
  73 }
  74
  75 static long cms_Vtx_configBand(displayPort_t *pDisp, const void *self)
  76 {
  77     UNUSED(pDisp);
  78     UNUSED(self);
  79
  80     if (vtxBand == 0) {
  81         vtxBand = 1;
  82     }
  83     return 0;
  84 }
  85
  86 static long cms_Vtx_configChan(displayPort_t *pDisp, const void *self)
  87 {
  88     UNUSED(pDisp);
  89     UNUSED(self);
  90
  91     if (vtxChan == 0) {
  92         vtxChan = 1;
  93     }
  94     return 0;
  95 }
  96
  97 static long cms_Vtx_configPower(displayPort_t *pDisp, const void *self)
  98 {
  99     UNUSED(pDisp);
 100     UNUSED(self);
 101
 102     if (vtxPower == 0) {
 103         vtxPower = 1;
 104     }
 105     return 0;
 106 }
 107
 108 static void cms_Vtx_initSettings(void)
 109 {
 110     vtxDevice_t * vtxDevice = vtxCommonDevice();
 111     vtxDeviceCapability_t vtxDeviceCapability;
 112
 113     if (vtxCommonGetDeviceCapability(vtxDevice, &vtxDeviceCapability)) {
 114         cms_Vtx_EntBand.max = vtxDeviceCapability.bandCount;
 115         cms_Vtx_EntBand.names = (const char * const *)vtxDeviceCapability.bandNames;
 116
 117         cms_Vtx_EntChan.max = vtxDeviceCapability.channelCount;
 118         cms_Vtx_EntChan.names = (const char * const *)vtxDeviceCapability.channelNames;
 119
 120         cms_Vtx_EntPower.max = vtxDeviceCapability.powerCount;
 121         cms_Vtx_EntPower.names = (const char * const *)vtxDeviceCapability.powerNames;
 122     }
 123     else {
 124         cms_Vtx_EntBand.max = VTX_SETTINGS_BAND_COUNT;
 125         cms_Vtx_EntBand.names = vtx58BandNames;
 126
 127         cms_Vtx_EntChan.max = VTX_SETTINGS_CHANNEL_COUNT;
 128         cms_Vtx_EntChan.names = vtx58ChannelNames;
 129
 130         cms_Vtx_EntPower.max = VTX_SETTINGS_POWER_COUNT;
 131         cms_Vtx_EntPower.names = vtx58DefaultPowerNames;
 132     }
 133
 134     vtxBand = vtxSettingsConfig()->band;
 135     vtxChan = vtxSettingsConfig()->channel;
 136     vtxPower = vtxSettingsConfig()->power;
 137
 138     // If device is ready - read actual PIT mode
 139     if (vtxCommonDeviceIsReady(vtxDevice)) {
 140         uint8_t onoff;
 141         vtxCommonGetPitMode(vtxDevice, &onoff);
 142         vtxPitMode = onoff ? 2 : 1;
 143     }
 144     else {
 145         vtxPitMode = 0;
 146     }
 147 }
 148
 149 static long cms_Vtx_onEnter(const OSD_Entry *self)
 150 {
 151     UNUSED(self);
 152     cms_Vtx_initSettings();
 153     return 0;
 154 }
 155
 156 static long cms_Vtx_Commence(displayPort_t *pDisp, const void *self)
 157 {
 158     UNUSED(pDisp);
 159     UNUSED(self);
 160
 161     vtxCommonSetBandAndChannel(vtxCommonDevice(), vtxBand, vtxChan);
 162     vtxCommonSetPowerByIndex(vtxCommonDevice(), vtxPower);
 163     vtxCommonSetPitMode(vtxCommonDevice(), vtxPitMode == 2 ? 1 : 0);
 164
 165     vtxSettingsConfigMutable()->band = vtxBand;
 166     vtxSettingsConfigMutable()->channel = vtxChan;
 167     vtxSettingsConfigMutable()->power = vtxPower;
 168
 169     saveConfigAndNotify();
 170
 171     return MENU_CHAIN_BACK;
 172 }
 173
 174 static bool cms_Vtx_drawStatusString(char *buf, unsigned bufsize)
 175 {
 176     const char *defaultString = "-- ---- ----";
 177 //                               bc ffff pppp
 178 //                               012345678901
 179
 180     if (bufsize < strlen(defaultString) + 1) {
 181         return false;
 182     }
 183
 184     strcpy(buf, defaultString);
 185
 186     vtxDevice_t * vtxDevice = vtxCommonDevice();
 187     vtxDeviceOsdInfo_t osdInfo;
 188
 189     if (!vtxDevice || !vtxCommonGetOsdInfo(vtxDevice, &osdInfo) || !vtxCommonDeviceIsReady(vtxDevice)) {
 190         return true;
 191     }
 192
 193     buf[0] = osdInfo.bandLetter;
 194     buf[1] = osdInfo.channelName[0];
 195     buf[2] = ' ';
 196
 197     if (osdInfo.frequency)
 198         tfp_sprintf(&buf[3], "%4d", osdInfo.frequency);
 199     else
 200         tfp_sprintf(&buf[3], "----");
 201
 202     if (osdInfo.powerIndex) {
 203         // If OSD driver provides power in milliwatt - display MW, otherwise - power level
 204         if (osdInfo.powerMilliwatt) {
 205             tfp_sprintf(&buf[7], " %4d", osdInfo.powerMilliwatt);
 206         }
 207         else {
 208             tfp_sprintf(&buf[7], " PL=%c", osdInfo.powerIndex);
 209         }
 210     } else {
 211         tfp_sprintf(&buf[7], " ----");
 212     }
 213
 214     return true;
 215 }
 216
 217 static const OSD_Entry cms_menuCommenceEntries[] =
 218 {
 219     OSD_LABEL_ENTRY("CONFIRM"),
 220     OSD_FUNC_CALL_ENTRY("YES", cms_Vtx_Commence),
 221
 222     OSD_BACK_AND_END_ENTRY,
 223 };
 224
 225 static const CMS_Menu cms_menuCommence = {
 226 #ifdef CMS_MENU_DEBUG
 227     .GUARD_text = "XVTXTRC",
 228     .GUARD_type = OME_MENU,
 229 #endif
 230     .onEnter = NULL,
 231     .onExit = NULL,
 232     .onGlobalExit = NULL,
 233     .entries = cms_menuCommenceEntries,
 234 };
 235
 236 static const OSD_Entry cms_menuVtxEntries[] =
 237 {
 238     OSD_LABEL_ENTRY("--- VTX ---"),
 239     OSD_LABEL_FUNC_DYN_ENTRY("", cms_Vtx_drawStatusString),
 240     OSD_TAB_CALLBACK_ENTRY("PIT",   cms_Vtx_configPitMode, &cms_Vtx_EntPitMode),
 241     OSD_TAB_CALLBACK_ENTRY("BAND",  cms_Vtx_configBand,    &cms_Vtx_EntBand),
 242     OSD_TAB_CALLBACK_ENTRY("CHAN",  cms_Vtx_configChan,    &cms_Vtx_EntChan),
 243     OSD_TAB_CALLBACK_ENTRY("POWER", cms_Vtx_configPower,   &cms_Vtx_EntPower),
 244
 245     OSD_SUBMENU_ENTRY("SET", &cms_menuCommence),
 246     OSD_BACK_AND_END_ENTRY,
 247 };
 248
 249 const CMS_Menu cmsx_menuVtxControl = {
 250 #ifdef CMS_MENU_DEBUG
 251     .GUARD_text = "MENUVTX",
 252     .GUARD_type = OME_MENU,
 253 #endif
 254     .onEnter = cms_Vtx_onEnter,
 255     .onExit = NULL,
 256     .onGlobalExit = NULL,
 257     .entries = cms_menuVtxEntries
 258 };
 259
 260 #endif // CMS