Colorlcd multirotor wizard (#8221)

radio/sdcard/horus/SCRIPTS/WIZARD/multirotor/wizard.lua

@@ -0,0 +1,411 @@
+---- #########################################################################
+---- #                                                                       #
+---- # Copyright (C) OpenTX                                                  #
+-----#                                                                       #
+-----# Credits: graphics by Radiomaster                                      #
+-----#                                                                       #
+---- # License GPLv2: http://www.gnu.org/licenses/gpl-2.0.html               #
+---- #                                                                       #
+---- # This program is free software; you can redistribute it and/or modify  #
+---- # it under the terms of the GNU General Public License version 2 as     #
+---- # published by the Free Software Foundation.                            #
+---- #                                                                       #
+---- # This program is distributed in the hope that it will be useful        #
+---- # but WITHOUT ANY WARRANTY; without even the implied warranty of        #
+---- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         #
+---- # GNU General Public License for more details.                          #
+---- #                                                                       #
+---- #########################################################################
+
+local VALUE = 0
+local COMBO = 1
+
+local edit = false
+local page = 1
+local current = 1
+local pages = {}
+local fields = {}
+local switches = {"SA", "SB", "SC", "SD", "SE", "SF", "SG"}
+
+-- load common Bitmaps
+local ImgMarkBg = Bitmap.open("img/mark_bg.png")
+local BackgroundImg = Bitmap.open("img/background.png")
+local ImgPageUp = Bitmap.open("img/pageup.png")
+local ImgPageDn = Bitmap.open("img/pagedn.png")
+
+
+-- Change display attribute to current field
+local function addField(step)
+  local field = fields[current]
+  local min, max
+  if field[3] == VALUE then
+    min = field[6]
+    max = field[7]
+  elseif field[3] == COMBO then
+    min = 0
+    max = #(field[6]) - 1
+  end
+  if (step < 0 and field[5] > min) or (step > 0 and field[5] < max) then
+    field[5] = field[5] + step
+  end
+end
+
+-- Select the next or previous page
+local function selectPage(step)
+  page = 1 + ((page + step - 1 + #pages) % #pages)
+  edit = false
+  current = 1
+end
+
+-- Select the next or previous editable field
+local function selectField(step)
+  repeat
+    current = 1 + ((current + step - 1 + #fields) % #fields)
+  until fields[current][4]==1
+end
+
+-- Redraw the current page
+local function redrawFieldsPage(event)
+
+  for index = 1, 10, 1 do
+    local field = fields[index]
+    if field == nil then
+      break
+    end
+
+    local attr = current == (index) and ((edit == true and BLINK or 0) + INVERS) or 0
+    attr = attr + TEXT_COLOR
+
+    if field[4] == 1 then
+      if field[3] == VALUE then
+        lcd.drawNumber(field[1], field[2], field[5], LEFT + attr)
+      elseif field[3] == COMBO then
+        if field[5] >= 0 and field[5] < #(field[6]) then
+          lcd.drawText(field[1],field[2], field[6][1+field[5]], attr)
+        end
+      end
+    end
+  end
+end
+
+local function updateField(field)
+  local value = field[5]
+end
+
+-- Main
+local function runFieldsPage(event)
+  if event == EVT_VIRTUAL_EXIT then -- exit script
+    return 2
+  elseif event == EVT_VIRTUAL_ENTER then -- toggle editing/selecting current field
+    if fields[current][5] ~= nil then
+      edit = not edit
+      if edit == false then
+        updateField(fields[current])
+      end
+    end
+  elseif edit then
+    if event == EVT_VIRTUAL_INC or event == EVT_VIRTUAL_INC_REPT then
+      addField(1)
+    elseif event == EVT_VIRTUAL_DEC or event == EVT_VIRTUAL_DEC_REPT then
+      addField(-1)
+    end
+  else
+    if event == EVT_VIRTUAL_NEXT then
+      selectField(1)
+    elseif event == EVT_VIRTUAL_PREV then
+      selectField(-1)
+    end
+  end
+  redrawFieldsPage(event)
+  return 0
+end
+
+-- set visibility flags starting with SECOND field of fields
+local function setFieldsVisible(...)
+  local arg={...}
+  local cnt = 2
+  for i,v in ipairs(arg) do
+    fields[cnt][4] = v
+    cnt = cnt + 1
+  end
+end
+
+-- draws one letter mark
+local function drawMark(x, y, name)
+  lcd.drawBitmap(ImgMarkBg, x, y)
+  lcd.drawText(x+8, y+3, name, TEXT_COLOR)
+end
+
+
+local ThrottleFields = {
+  {50, 50, COMBO, 1, 2, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local ThrottleBackground
+
+local function runThrottleConfig(event)
+  lcd.clear()
+  if ThrottleBackground == nil then
+    ThrottleBackground = Bitmap.open("img/throttle.png")
+  end
+  lcd.drawBitmap(ThrottleBackground, 0, 0)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = ThrottleFields
+  lcd.drawText(40, 20, "Assign Throttle channel", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local RollFields = {
+  {50, 50, COMBO, 1, 0, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local RollBackground
+
+local function runRollConfig(event)
+  lcd.clear()
+  if RollBackground == nil then
+    RollBackground = Bitmap.open("img/roll.png")
+  end
+  lcd.drawBitmap(RollBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = RollFields
+  lcd.drawText(40, 20, "Assign Roll channel", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local PitchFields = {
+  {50, 50, COMBO, 1, 1, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local PitchBackground
+
+local function runPitchConfig(event)
+  lcd.clear()
+  if PitchBackground == nil then
+    PitchBackground = Bitmap.open("img/pitch.png")
+  end
+  lcd.drawBitmap(PitchBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = PitchFields
+  lcd.drawText(40, 20, "Assign Pitch channel", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local YawFields = {
+  {50, 50, COMBO, 1, 3, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local YawBackground
+
+local function runYawConfig(event)
+  lcd.clear()
+  if YawBackground == nil then
+    YawBackground = Bitmap.open("img/yaw.png")
+  end
+  lcd.drawBitmap(YawBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = YawFields
+  lcd.drawText(40, 20, "Assign Yaw channel", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local ArmFields = {
+  {50, 50, COMBO, 1, 5, { "SA", "SB", "SC", "SD", "SE", "SF"} },
+}
+
+local ArmBackground
+
+local function runArmConfig(event)
+  lcd.clear()
+  if ArmBackground == nil then
+    ArmBackground = Bitmap.open("img/arm.png")
+  end
+  lcd.drawBitmap(ArmBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = ArmFields
+  lcd.drawText(40, 20, "Assign Arm switch", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local BeeperFields = {
+  {50, 50, COMBO, 1, 3, { "SA", "SB", "SC", "SD", "SE", "SF"} },
+}
+
+local BeeperBackground
+
+local function runBeeperConfig(event)
+  lcd.clear()
+  if BeeperBackground == nil then
+    BeeperBackground = Bitmap.open("img/beeper.png")
+  end
+  lcd.drawBitmap(BeeperBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = BeeperFields
+  lcd.drawText(40, 20, "Assign Beeper switch", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local ModeFields = {
+  {50, 50, COMBO, 1, 0, { "SA", "SB", "SC", "SD", "SE", "SF"} },
+}
+
+local ModeBackground
+
+local function runModeConfig(event)
+  lcd.clear()
+  if ModeBackground == nil then
+    ModeBackground = Bitmap.open("img/mode.png")
+  end
+  lcd.drawBitmap(ModeBackground, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgPageDn, 455, 95)
+  fields = ModeFields
+  lcd.drawText(40, 20, "Assign Mode switch", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 100, 30, TEXT_BGCOLOR)
+  fields[1][4]=1
+  local result = runFieldsPage(event)
+  return result
+end
+
+local lineIndex
+local function drawNextChanelLine(text, text2)
+  lcd.drawText(40, lineIndex, text, TEXT_COLOR)
+  lcd.drawText(242, lineIndex, ": CH" .. text2 + 1, TEXT_COLOR)
+  lineIndex = lineIndex + 20
+end
+
+local function drawNextSwitchLine(text, text2)
+  lcd.drawText(40, lineIndex, text, TEXT_COLOR)
+  lcd.drawText(242, lineIndex, ": " ..switches[text2 + 1], TEXT_COLOR)
+  lineIndex = lineIndex + 20
+end
+
+
+local ConfigSummaryFields = {
+  {110, 250, COMBO, 1, 0, { "No, I need to change something", "Yes, all is well, create the model !"} },
+}
+
+local ImgSummary
+
+local function runConfigSummary(event)
+  lcd.clear()
+  if ImgSummary == nil then
+    ImgSummary = Bitmap.open("img/summary.png")
+  end
+  fields = ConfigSummaryFields
+  lcd.drawBitmap(BackgroundImg, 0, 0)
+  lcd.drawBitmap(ImgPageUp, 0, 95)
+  lcd.drawBitmap(ImgSummary, 300, 60)
+  lineIndex = 40
+  -- throttle
+  drawNextChanelLine("Throttle channel", ThrottleFields[1][5])
+  -- roll
+  drawNextChanelLine("Roll channel",RollFields[1][5])
+  -- pitch
+  drawNextChanelLine("Pitch channel",PitchFields[1][5])
+  -- yaw
+  drawNextChanelLine("Yaw channel",YawFields[1][5])
+  -- arm
+  drawNextSwitchLine("Arm switch",ArmFields[1][5])
+  -- beeper
+  drawNextSwitchLine("Beeper switch",BeeperFields[1][5])
+  -- mode
+  drawNextSwitchLine("Mode switch",ModeFields[1][5])
+
+  local result = runFieldsPage(event)
+  if(fields[1][5] == 1 and edit == false) then
+    selectPage(1)
+  end
+  return result
+end
+
+local function addMix(channel, input, name, weight, index)
+  local mix = { source=input, name=name }
+  if weight ~= nil then
+    mix.weight = weight
+  end
+  if index == nil then
+    index = 0
+  end
+  model.insertMix(channel, index, mix)
+end
+
+local function createModel(event)
+  lcd.clear()
+  lcd.drawBitmap(BackgroundImg, 0, 0)
+  lcd.drawBitmap(ImgSummary, 300, 60)
+  model.defaultInputs()
+  model.deleteMixes()
+  -- throttle
+  addMix(ThrottleFields[1][5], MIXSRC_FIRST_INPUT+defaultChannel(2), "Thr")
+  -- roll
+  addMix(RollFields[1][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Roll")
+  -- pitch
+  addMix(PitchFields[1][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Roll")
+  -- yaw
+  addMix(YawFields[1][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Roll")
+  addMix(4, MIXSRC_SA + ArmFields[1][5], "Arm")
+  addMix(5, MIXSRC_SA + BeeperFields[1][5], "Beeper")
+  addMix(6, MIXSRC_SA + ModeFields[1][5], "Mode")
+
+  lcd.drawText(70, 90, "Model successfully created !", TEXT_COLOR)
+  lcd.drawText(100, 130, "Press RTN to exit", TEXT_COLOR)
+  return 2
+end
+
+-- Init
+local function init()
+  current, edit = 1, false
+  pages = {
+    runThrottleConfig,
+    runRollConfig,
+    runPitchConfig,
+    runYawConfig,
+    runArmConfig,
+    runBeeperConfig,
+    runModeConfig,
+    runConfigSummary,
+    createModel,
+  }
+end
+
+-- Main
+local function run(event)
+  if event == nil then
+    error("Cannot be run as a model script!")
+    return 2
+  elseif event == EVT_VIRTUAL_NEXT_PAGE and page < #pages-1 then
+    selectPage(1)
+  elseif event == EVT_VIRTUAL_PREV_PAGE and page > 1 then
+    killEvents(event);
+    selectPage(-1)
+  end
+
+  local result = pages[page](event)
+  return result
+end
+
+return { init=init, run=run }

radio/src/gui/480x272/model_select.cpp

@@ -191,7 +191,6 @@ bool menuModelWizard(event_t event)
           uint16_t x = WIZARD_LEFT_SPACING + (wizidx - first) * (WIZARD_SPACING + WIZARD_ICON_X);
           strcpy(&wizpath[sizeof(WIZARD_PATH)], fno.fname);
           strcpy(&wizpath[sizeof(WIZARD_PATH) + strlen(fno.fname)], "/icon.png");
-          lcdDrawText(x + 10, WIZARD_TEXT_Y, fno.fname);
           BitmapBuffer * background = BitmapBuffer::load(wizpath);
           lcd->drawBitmap(x, WIZARD_ICON_Y, background);
           if(wizidx == wizardSelected ) {