VTX Table download

Downloads vtx tables if  supported and available. This is done as part of ui.lua's init state.

The vtx tables are saved as /BF/VTX/modelname.lua where "modelname" is the name of the currently selected model. Delete this file if the vtx tables need to be updated for some reason. The script will download them again.

Author: klutvott123

src/SCRIPTS/BF/ui.lua

@@ -247,13 +247,10 @@ end
 
 local function run_ui(event)
     if uiState == uiStatus.init then
-        local yMinLim = radio.yMinLimit
         lcd.clear()
         drawScreenTitle("Betaflight Config")
-        lcd.drawText(6, yMinLim, "Initialising")
-        if apiVersion == 0 then
-            init = init or assert(loadScript("data_init.lua"))()
-            init()
+        init = init or assert(loadScript("ui_init.lua"))()
+        if not init() then
             return 0
         end
         init = nil

src/SCRIPTS/BF/ui_init.lua

@@ -0,0 +1,28 @@
+local apiVersionReceived = false
+local vtxTablesReceived = loadScript("/BF/VTX/"..model.getInfo().name..".lua")
+local data_init, getVtxTables
+
+local function init()
+    if apiVersion == 0 then
+        lcd.drawText(6, radio.yMinLimit, "Initialising")
+        data_init = data_init or assert(loadScript("data_init.lua"))()
+        data_init()
+    elseif apiVersion > 0 and not apiVersionReceived then
+        data_init = nil
+        apiVersionReceived = true
+        collectgarbage()
+    elseif apiVersion >= 1.042 and not vtxTablesReceived then
+        lcd.drawText(6, radio.yMinLimit, "Downloading VTX Tables")
+        getVtxTables = getVtxTables or assert(loadScript("vtx_tables.lua"))()
+        vtxTablesReceived = getVtxTables()
+        if vtxTablesReceived then
+            getVtxTables = nil
+            collectgarbage()
+        end
+    else
+        return true
+    end
+    return apiVersionReceived and vtxTablesReceived
+end
+
+return init

src/SCRIPTS/BF/vtx_tables.lua

@@ -0,0 +1,119 @@
+local MSP_VTX_CONFIG = 88
+local MSP_VTXTABLE_BAND = 137
+local MSP_VTXTABLE_POWERLEVEL = 138
+
+local vtxTableAvailable = false
+local vtxConfigReceived = false
+local vtxFrequencyTableReceived = false
+local vtxPowerTableReceived = false
+local vtxTablesReceived = false
+local requestedBand = 1
+local requestedPowerLevel = 1
+local vtxTableConfig = {}
+local frequencyTable = {}
+local frequenciesPerBand = 0
+local bandTable = {}
+local powerTable = {}
+
+local lastRunTS = 0
+local INTERVAL = 100
+
+local function processMspReply(cmd, payload)
+    if cmd == MSP_VTX_CONFIG then
+        vtxConfigReceived = true
+        vtxTableAvailable = payload[12] ~= 0
+        vtxTableConfig.bands = payload[13]
+        vtxTableConfig.channels = payload[14]
+        vtxTableConfig.powerLevels = payload[15]
+    end
+    if cmd == MSP_VTXTABLE_BAND and payload[1] == requestedBand then
+        local i = 1
+        local receivedBand = payload[i]
+        i = i + 1
+        local bandNameLength = payload[i]
+        i = i + bandNameLength + 1
+        bandTable[receivedBand] = string.char(payload[i])
+        i = i + 2
+        local channels = payload[i]
+        i = i + 1
+        frequencyTable[receivedBand] = {}
+        for channel = 1, channels do
+            local frequency = 0
+            for idx=1, 2 do
+                local raw_val = payload[i]
+                raw_val = bit32.lshift(raw_val, (idx-1)*8)
+                frequency = bit32.bor(frequency, raw_val)
+                i = i + 1
+            end
+            frequencyTable[receivedBand][channel] = frequency
+        end
+        requestedBand = requestedBand + 1
+        vtxFrequencyTableReceived = requestedBand > vtxTableConfig.bands
+    end
+    if cmd == MSP_VTXTABLE_POWERLEVEL and payload[1] == requestedPowerLevel then
+        local i = 1
+        local powerLevel = payload[i]
+        i = i + 3
+        local powerLabelLength = payload[i]
+        i = i + 1
+        powerTable[powerLevel] = ''
+        for c = 1, powerLabelLength do
+            powerTable[powerLevel] = powerTable[powerLevel]..string.char(payload[i])
+            i = i + 1
+        end
+        requestedPowerLevel = requestedPowerLevel + 1
+        vtxPowerTableReceived = requestedPowerLevel > vtxTableConfig.powerLevels
+    end 
+end
+
+local function getVtxTables()
+    if lastRunTS + INTERVAL < getTime() then
+        lastRunTS = getTime()
+        if not vtxConfigReceived then
+            protocol.mspRead(MSP_VTX_CONFIG)
+        elseif vtxConfigReceived and not vtxTableAvailable then
+            return true
+        elseif not vtxFrequencyTableReceived then
+            protocol.mspWrite(MSP_VTXTABLE_BAND, { requestedBand })
+        elseif not vtxPowerTableReceived then
+            protocol.mspWrite(MSP_VTXTABLE_POWERLEVEL, { requestedPowerLevel })
+        else
+            vtxTablesReceived = true
+        end
+    end
+    if vtxTablesReceived then
+        local f = io.open("/BF/VTX/"..model.getInfo().name..".lua", 'w')
+        io.write(f, "return {", "\n")
+        io.write(f, "    frequencyTable = {", "\n")
+        for i = 1, #frequencyTable do
+            local frequencyString = "        { "
+            for k = 1, #frequencyTable[i] do
+                frequencyString = frequencyString..tostring(frequencyTable[i][k])..", "
+            end
+            frequencyString = frequencyString.."},"
+            io.write(f, frequencyString, "\n")
+        end
+        io.write(f, "    },", "\n")
+        io.write(f, "    frequenciesPerBand = "..tostring(vtxTableConfig.channels)..",", "\n")
+        local bandString = "    bandTable = { [0]=\"U\", "
+        for i = 1, #bandTable do
+            bandString = bandString.."\""..bandTable[i].."\", "
+        end
+        bandString = bandString.."},"
+        io.write(f, bandString, "\n")
+        local powerString = "    powerTable = { "
+        for i = 1, #powerTable do
+            powerString = powerString.."\""..powerTable[i].."\", "
+        end
+        powerString = powerString.."},"
+        io.write(f, powerString, "\n")
+        io.write(f, "}", "\n")
+        io.close(f)
+        assert(loadScript("/BF/VTX/"..model.getInfo().name..".lua", 'c'))
+    end
+    mspProcessTxQ()
+    processMspReply(mspPollReply())
+    return vtxTablesReceived
+end
+
+return getVtxTables