src/main/io/vtx_control.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
  19 // Get target build configuration
  20 #include "platform.h"
  21
  22 #include "common/maths.h"
  23
  24 #include "config/config_eeprom.h"
  25 #include "config/parameter_group.h"
  26 #include "config/parameter_group_ids.h"
  27
  28 #include "drivers/vtx_common.h"
  29 #include "drivers/light_led.h"
  30 #include "drivers/system.h"
  31
  32 #include "fc/config.h"
  33 #include "fc/runtime_config.h"
  34 #include "fc/settings.h"
  35
  36 #include "io/beeper.h"
  37 #include "io/osd.h"
  38 #include "io/vtx_control.h"
  39
  40
  41 #if defined(USE_VTX_CONTROL)
  42
  43 PG_REGISTER_WITH_RESET_TEMPLATE(vtxConfig_t, vtxConfig, PG_VTX_CONFIG, 4);
  44
  45 PG_RESET_TEMPLATE(vtxConfig_t, vtxConfig,
  46       .halfDuplex = SETTING_VTX_HALFDUPLEX_DEFAULT,
  47       .smartAudioEarlyAkkWorkaroundEnable = SETTING_VTX_SMARTAUDIO_EARLY_AKK_WORKAROUND_DEFAULT,
  48       .smartAudioAltSoftSerialMethod = SETTING_VTX_SMARTAUDIO_ALTERNATE_SOFTSERIAL_METHOD_DEFAULT,
  49       .softSerialShortStop = SETTING_VTX_SOFTSERIAL_SHORTSTOP_DEFAULT,
  50       .smartAudioStopBits = SETTING_VTX_SMARTAUDIO_STOPBITS_DEFAULT,
  51 );
  52
  53 static uint8_t locked = 0;
  54
  55 void vtxControlInit(void)
  56 {
  57     // NOTHING TO DO
  58 }
  59
  60 void vtxControlInputPoll(void)
  61 {
  62   // Check variuos input sources for VTX config updates
  63
  64   // XXX: None supported in INAV
  65 }
  66
  67 static void vtxUpdateBandAndChannel(uint8_t bandStep, uint8_t channelStep)
  68 {
  69     if (ARMING_FLAG(ARMED)) {
  70         locked = 1;
  71     }
  72
  73     if (!locked) {
  74         vtxDevice_t *vtxDevice = vtxCommonDevice();
  75         if (vtxDevice) {
  76             uint8_t band = 0, channel = 0;
  77             vtxCommonGetBandAndChannel(vtxDevice, &band, &channel);
  78             vtxCommonSetBandAndChannel(vtxDevice, band + bandStep, channel + channelStep);
  79         }
  80     }
  81 }
  82
  83 void vtxIncrementBand(void)
  84 {
  85     vtxUpdateBandAndChannel(+1, 0);
  86 }
  87
  88 void vtxDecrementBand(void)
  89 {
  90     vtxUpdateBandAndChannel(-1, 0);
  91 }
  92
  93 void vtxIncrementChannel(void)
  94 {
  95     vtxUpdateBandAndChannel(0, +1);
  96 }
  97
  98 void vtxDecrementChannel(void)
  99 {
 100     vtxUpdateBandAndChannel(0, -1);
 101 }
 102
 103 void vtxUpdateActivatedChannel(void)
 104 {
 105     if (ARMING_FLAG(ARMED)) {
 106         locked = 1;
 107     }
 108
 109     if (!locked) {
 110         vtxDevice_t *vtxDevice = vtxCommonDevice();
 111         if (vtxDevice) {
 112             static uint8_t lastIndex = -1;
 113
 114             for (uint8_t index = 0; index < MAX_CHANNEL_ACTIVATION_CONDITION_COUNT; index++) {
 115                 const vtxChannelActivationCondition_t *vtxChannelActivationCondition = &vtxConfig()->vtxChannelActivationConditions[index];
 116
 117                 if (isRangeActive(vtxChannelActivationCondition->auxChannelIndex, &vtxChannelActivationCondition->range)
 118                     && index != lastIndex) {
 119                     lastIndex = index;
 120
 121                     vtxCommonSetBandAndChannel(vtxDevice, vtxChannelActivationCondition->band, vtxChannelActivationCondition->channel);
 122                     break;
 123                 }
 124             }
 125         }
 126     }
 127 }
 128
 129 void vtxCycleBandOrChannel(const uint8_t bandStep, const uint8_t channelStep)
 130 {
 131     vtxDevice_t *vtxDevice = vtxCommonDevice();
 132
 133     if (!vtxDevice) {
 134         return;
 135     }
 136
 137     uint8_t band = 0, channel = 0;
 138     vtxDeviceCapability_t capability;
 139
 140     bool haveAllNeededInfo = vtxCommonGetBandAndChannel(vtxDevice, &band, &channel) && vtxCommonGetDeviceCapability(vtxDevice, &capability);
 141     if (!haveAllNeededInfo) {
 142         return;
 143     }
 144
 145     int newChannel = channel + channelStep;
 146     if (newChannel > capability.channelCount) {
 147         newChannel = 1;
 148     } else if (newChannel < 1) {
 149         newChannel = capability.channelCount;
 150     }
 151
 152     int newBand = band + bandStep;
 153     if (newBand > capability.bandCount) {
 154         newBand = 1;
 155     } else if (newBand < 1) {
 156         newBand = capability.bandCount;
 157     }
 158
 159     vtxCommonSetBandAndChannel(vtxDevice, newBand, newChannel);
 160 }
 161
 162 void vtxCyclePower(const uint8_t powerStep)
 163 {
 164     vtxDevice_t *vtxDevice = vtxCommonDevice();
 165
 166     if (!vtxDevice) {
 167         return;
 168     }
 169
 170     uint8_t power = 0;
 171     vtxDeviceCapability_t capability;
 172
 173     bool haveAllNeededInfo = vtxCommonGetPowerIndex(vtxDevice, &power) && vtxCommonGetDeviceCapability(vtxDevice, &capability);
 174     if (!haveAllNeededInfo) {
 175         return;
 176     }
 177
 178     int newPower = power + powerStep;
 179     if (newPower >= capability.powerCount) {
 180         newPower = 0;
 181     } else if (newPower < 0) {
 182         newPower = capability.powerCount;
 183     }
 184
 185     vtxCommonSetPowerByIndex(vtxDevice, newPower);
 186 }
 187
 188 #endif