3djc/horus wizard plane (#4082)

* Introduce plane Wizard * Simplify code
2016-11-30 11:51:23 +01:00
parent 781ce1ea34
commit 9f0f79469a
20 changed files with 443 additions and 35 deletions
@@ -1,17 +0,0 @@
-
-
-- Init
-local function init()
-
-end
-
-- Main
-local function run(event)
-  lcd.clear()
-  lcd.drawFilledRectangle(0, 0, LCD_W, LCD_H, TEXT_BGCOLOR)
-  lcd.drawText(50, 50, "You are running dummy glider wizard 1", TEXT_COLOR)
-
-  return 1
-end
-
-return { init=init, run=run }
@@ -1,17 +1,409 @@
-
-
-- Init
-local function init()
-
-end
-
-- Main
-local function run(event)
-  lcd.clear()
-  lcd.drawFilledRectangle(0, 0, LCD_W, LCD_H, TEXT_BGCOLOR)
-  lcd.drawText(50, 50, "You are running dummy plane wizard", TEXT_COLOR)
-
-  return 1
-end
-
-return { init=init, run=run }
+local VALUE = 0
+local COMBO = 1
+
+local edit = false
+local page = 1
+local current = 1
+local pages = {}
+local fields = {}
+local AilCfgBg = {}
+local FlapsCfgBg = {}
+local TAilCfgBg = {}
+local SummaryBackground
+
+-- Release memory
+local function clean()
+  print("FREE MEMORY CALLED")
+  MotorConfigBackground = nil
+  AilCfgBg = nil
+  FlapsCfgBg = nil
+  TAilCfgBg = nil
+  SummaryBackground = nil
+end
+
+-- 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_EXIT_BREAK then -- exit script
+    return 2
+  elseif event == EVT_ENTER_BREAK or event == EVT_ROT_BREAK 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_PLUS_FIRST or event == EVT_ROT_RIGHT or event == EVT_PLUS_REPT then
+      addField(1)
+    elseif event == EVT_MINUS_FIRST or event == EVT_ROT_LEFT or event == EVT_MINUS_REPT then
+      addField(-1)
+    end
+  else
+    if event == EVT_MINUS_FIRST or event == EVT_ROT_RIGHT then
+      selectField(1)
+    elseif event == EVT_PLUS_FIRST or event == EVT_ROT_LEFT 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
+
+-- on demand load of bitmaps
+local  function loadIndexedBitmap(bitmap, index, name)
+  if bitmap[index] == nil then
+    bitmap[index] = Bitmap.open(name..index..".gif")
+  end
+end
+
+local MotorConfigBackground = Bitmap.open("img/bg_engine.gif")
+local MotorFields = {
+  {50, 50, COMBO, 1, 1, { "No", "Yes"} },
+  {50, 127, COMBO, 1, 2, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local function runMotorConfig(event)
+  lcd.clear()
+  lcd.drawBitmap(MotorConfigBackground, 0, 0)
+  lcd.setColor(CUSTOM_COLOR, lcd.RGB(255, 255, 255))
+  fields = MotorFields
+  lcd.drawText(40, 20, "Does your model have a motor ?", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 200, 30, CUSTOM_COLOR)
+  fields[2][4]=0
+  if fields[1][5] == 1 then
+    lcd.drawText(40, 100, "What channel is it on ?", TEXT_COLOR)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    fields[2][4]=1
+  end
+  loadIndexedBitmap(AilCfgBg, 2, "img/bg_ail")
+  local result = runFieldsPage(event)
+  return result
+end
+
+-- fields format : {[1]x, [2]y, [3]COMBO, [4]visible, [5]default, [6]{values}}
+-- fields format : {[1]x, [2]y, [3]VALUE, [4]visible, [5]default, [6]min, [7]max}
+local AilFields = {
+  {50, 50, COMBO, 1, 2, { "None", "One, or two with Y cable", "Two"} },
+  {50, 127, COMBO, 1, 0, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } }, -- Ail1 chan
+  {50, 167, COMBO, 1, 4, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } }, -- Ail2 chan
+}
+
+local function runAilConfig(event)
+  lcd.clear()
+  fields = AilFields
+  if fields[1][5] == 1 then
+    loadIndexedBitmap(AilCfgBg, 1, "img/bg_ail")
+    lcd.drawBitmap(AilCfgBg[1], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 0)
+  elseif fields[1][5] == 2 then
+    loadIndexedBitmap(AilCfgBg, 2, "img/bg_ail")
+    lcd.drawBitmap(AilCfgBg[2], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 162, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 1)
+  else
+    loadIndexedBitmap(AilCfgBg, 0, "img/bg_ail")
+    lcd.drawBitmap(AilCfgBg[0], 0, 0)
+    setFieldsVisible(0, 0)
+  end
+  lcd.drawText(40, 20, "Number of ailerons on your model ?", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 400, 30, CUSTOM_COLOR)
+  loadIndexedBitmap(FlapsCfgBg, 1, "img/bg_flaps")
+  local result = runFieldsPage(event)
+  return result
+end
+
+local FlapsFields = {
+  {50, 50, COMBO, 1, 0, { "No", "Yes, on one channel", "Yes, on two channels"} },
+  {50, 127, COMBO, 1, 6, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+  {50, 167, COMBO, 1, 7, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } },
+}
+
+local function runFlapsConfig(event)
+  lcd.clear()
+  fields = FlapsFields
+  if fields[1][5] == 1 then
+    loadIndexedBitmap(FlapsCfgBg, 1, "img/bg_flaps")
+    lcd.drawBitmap(FlapsCfgBg[1], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 0)
+  elseif fields[1][5] == 2 then
+    loadIndexedBitmap(FlapsCfgBg, 2, "img/bg_flaps")
+    lcd.drawBitmap(FlapsCfgBg[2], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 162, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 1)
+  else
+    loadIndexedBitmap(FlapsCfgBg, 0, "img/bg_flaps")
+    lcd.drawBitmap(FlapsCfgBg[0], 0, 0)
+    setFieldsVisible(0, 0)
+  end
+  lcd.drawText(40, 20, "Does your model have flaps ?", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 400, 30, CUSTOM_COLOR)
+  loadIndexedBitmap(TAilCfgBg, 1, "img/bg_tail")
+  local result = runFieldsPage(event)
+  return result
+end
+
+local TailFields = {
+  {50, 50, COMBO, 1, 1, { "1 channel for Elevator, no Rudder", "One chan for Elevator, one for Rudder", "Two chans for Elevator, one for Rudder", "V Tail"} },
+  {50, 127, COMBO, 1, 1, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } }, --ele
+  {50, 167, COMBO, 1, 3, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } }, --rud
+  {50, 207, COMBO, 0, 5, { "CH1", "CH2", "CH3", "CH4", "CH5", "CH6", "CH7", "CH8" } }, --ele2
+}
+
+local function runTailConfig(event)
+  lcd.clear()
+  fields = TailFields
+  if fields[1][5] == 0 then
+    loadIndexedBitmap(TAilCfgBg, 0, "img/bg_tail")
+    lcd.drawBitmap(TAilCfgBg[0], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 0, 0)
+  end
+  if fields[1][5] == 1 then
+    loadIndexedBitmap(TAilCfgBg, 1, "img/bg_tail")
+    lcd.drawBitmap(TAilCfgBg[1], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 162, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 1, 0)
+  end
+  if fields[1][5] == 2 then
+    loadIndexedBitmap(TAilCfgBg, 2, "img/bg_tail")
+    lcd.drawBitmap(TAilCfgBg[2], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 162, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 202, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 1, 1)
+  end
+  if fields[1][5] == 3 then
+    loadIndexedBitmap(TAilCfgBg, 3, "img/bg_tail")
+    lcd.drawBitmap(TailCfgBg[3], 0, 0)
+    lcd.drawFilledRectangle(40, 122, 100, 30, CUSTOM_COLOR)
+    lcd.drawFilledRectangle(40, 162, 100, 30, CUSTOM_COLOR)
+    setFieldsVisible(1, 1, 0)
+  end
+  lcd.drawText(40, 20, "Pick the tail config of your model", TEXT_COLOR)
+  lcd.drawFilledRectangle(40, 45, 400, 30, CUSTOM_COLOR)
+  if SummaryBackground == nil then
+    SummaryBackground = Bitmap.open("img/bg_sum.gif")
+  end
+  local result = runFieldsPage(event)
+  return result
+end
+
+local lineIndex
+local function drawNextLine(text, text2)
+  lcd.drawText(40, lineIndex, text, TEXT_COLOR)
+  lcd.drawText(250, lineIndex, 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 plane !"} },
+}
+
+local function runConfigSummary(event)
+  lcd.clear()
+  fields = ConfigSummaryFields
+  lcd.drawBitmap(SummaryBackground, 0, 0)
+  lineIndex = 40
+  -- motors
+  if(MotorFields[1][5] == 1) then
+    drawNextLine("Motor chan :", MotorFields[2][5])
+  elseif (MotorFields[2][5] == 2) then
+    drawNextLine("Motor 1 chan :", MotorFields[2][5])
+    drawNextLine("Motor 2 chan :", MotorFields[3][5])
+  end
+  -- ail
+  if(AilFields[1][5] == 1) then
+    drawNextLine("Aileron chan :",AilFields[2][5])
+  elseif (AilFields[1][5] == 2) then
+    drawNextLine("Aileron 1 chan :",AilFields[2][5])
+    drawNextLine("Aileron 2 chan :",AilFields[3][5])
+  end
+  -- flaps
+  if(FlapsFields[1][5] == 1) then
+    drawNextLine("Flaps chan :",FlapsFields[2][5])
+  elseif (FlapsFields[1][5] == 2) then
+    drawNextLine("Flaps 1 chan :",FlapsFields[2][5])
+    drawNextLine("Flaps 2 chan :",FlapsFields[3][5])
+  end
+  -- tail
+  if(TailFields[1][5] == 0) then
+    drawNextLine("Elevator chan :",TailFields[2][5])
+  elseif (TailFields[1][5] == 1) then
+    drawNextLine("Elevator chan :",TailFields[2][5])
+    drawNextLine("Rudder chan :",TailFields[3][5])
+  elseif (TailFields[1][5] == 2) then
+    drawNextLine("Elevator 1 chan :",TailFields[2][5])
+    drawNextLine("Rudder chan :",TailFields[3][5])
+    drawNextLine("Elevator 2 chan :",TailFields[4][5])
+  elseif (TailFields[1][5] == 3) then
+    drawNextLine("V-Tail elevator :", TailFields[2][5])
+    drawNextLine("V-Tail rudder :", TailFields[2][5])
+  end
+  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(SummaryBackground, 0, 0)
+  model.defaultInputs()
+  model.deleteMixes()
+  -- motor
+  if(MotorFields[1][5] == 1) then
+    addMix(MotorFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(2), "Motor")
+  elseif (MotorFields[2][5] == 2) then
+    addMix(MotorFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(2), "Motor1")
+    addMix(MotorFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(2), "Motor2")
+  end
+  -- Ailerons
+  if(AilFields[1][5] == 1) then
+    addMix(AilFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Ail")
+  elseif (AilFields[1][5] == 2) then
+    addMix(AilFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Ail1")
+    addMix(AilFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(3), "Ail2", -100)
+  end
+  -- Flaps
+  if(FlapsFields[1][5] == 1) then
+    addMix(FlapsFields[2][5], MIXSRC_SA, "Flaps")
+  elseif (FlapsFields[1][5] == 2) then
+    addMix(FlapsFields[2][5], MIXSRC_SA, "Flaps1")
+    addMix(FlapsFields[3][5], MIXSRC_SA, "Flaps2")
+  end
+  -- Tail
+  if(TailFields[1][5] == 0) then
+    addMix(TailFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "Elev")
+  elseif (TailFields[1][5] == 1) then
+    addMix(TailFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "Elev")
+    addMix(TailFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(0), "Rudder")
+  elseif (TailFields[1][5] == 2) then
+    addMix(TailFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "Elev1")
+    addMix(TailFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(0), "Rudder")
+    addMix(TailFields[4][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "Elev2")
+  elseif (TailFields[1][5] == 3) then
+    addMix(TailFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "VTailE", 50)
+    addMix(TailFields[2][5], MIXSRC_FIRST_INPUT+defaultChannel(0), "VTailR", 50, 1)
+    addMix(TailFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(1), "VTailE", 50)
+    addMix(TailFields[3][5], MIXSRC_FIRST_INPUT+defaultChannel(0), "VTailR", -50, 1)
+  end
+  lcd.drawText(70, 140, "Model successfully created !", TEXT_COLOR)
+  lcd.drawText(100, 180, "Press RTN to exit", TEXT_COLOR)
+  return 2
+end
+
+-- Init
+local function init()
+  current, edit = 1, false
+  pages = {
+    runMotorConfig,
+    runAilConfig,
+    runFlapsConfig,
+    runTailConfig,
+    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_PAGE_BREAK or event == EVT_PAGEDN_FIRST) and page < #pages-1 then
+    selectPage(1)
+  elseif (event == EVT_PAGE_LONG or event == EVT_PAGEUP_FIRST) and page > 1 then
+    killEvents(event);
+    selectPage(-1)
+  end
+
+  local result = pages[page](event)
+  return result
+end
+
+return { init=init, run=run }
@@ -190,6 +190,7 @@ BitmapBuffer * modelselModelQtyBitmap = NULL;
 BitmapBuffer * modelselModelNameBitmap = NULL;
 BitmapBuffer * modelselModelMoveBackground = NULL;
 BitmapBuffer * modelselModelMoveIcon = NULL;
+BitmapBuffer * modelselWizardBackground = NULL;
 BitmapBuffer * chanMonLockedBitmap = NULL;
 BitmapBuffer * chanMonInvertedBitmap = NULL;
 BitmapBuffer * mixerSetupMixerBitmap = NULL;
@@ -77,6 +77,7 @@ extern BitmapBuffer * modelselModelQtyBitmap;
 extern BitmapBuffer * modelselModelNameBitmap;
 extern BitmapBuffer * modelselModelMoveBackground;
 extern BitmapBuffer * modelselModelMoveIcon;
+extern BitmapBuffer * modelselWizardBackground;

 // calibration bitmaps
 extern BitmapBuffer * calibStick;
@@ -177,7 +177,7 @@ bool menuModelWizard(event_t event)
  strncpy(wizpath, WIZARD_PATH, sizeof(WIZARD_PATH));
  strcpy(&wizpath[sizeof(WIZARD_PATH)-1], "/");
  lcdDrawSolidFilledRect(0, 0, LCD_W, LCD_H, TEXT_BGCOLOR);
-  lcd->drawBitmap(0, 0, BitmapBuffer::load(getThemePath("wizard/background.png")));
+  lcd->drawBitmap(0, 0,modelselWizardBackground);
  FRESULT res = f_opendir(&dir, WIZARD_PATH);
  if (res == FR_OK) {
    for (uint8_t wizidx=0;;wizidx++) {
@@ -140,6 +140,34 @@ class DarkblueTheme: public Theme
      delete chanMonInvertedBitmap;
      chanMonInvertedBitmap = BitmapBuffer::loadMaskOnBackground("mask_monitor_inver.png", TEXT_COLOR, TEXT_BGCOLOR);

+     // Model Selection screen
+      delete modelselIconBitmap;
+      modelselIconBitmap = BitmapBuffer::loadMaskOnBackground("modelsel/mask_iconback.png", TITLE_BGCOLOR, TEXT_BGCOLOR);
+      if (modelselIconBitmap) {
+        BitmapBuffer * bitmap = BitmapBuffer::load(getThemePath("modelsel/icon_default.png"));
+        modelselIconBitmap->drawBitmap(25, 8, bitmap);
+        delete bitmap;
+      }
+
+      delete modelselSdFreeBitmap;
+      modelselSdFreeBitmap = BitmapBuffer::loadMaskOnBackground("modelsel/mask_sdfree.png", TEXT_COLOR, TEXT_BGCOLOR);
+
+      delete modelselModelQtyBitmap;
+      modelselModelQtyBitmap = BitmapBuffer::loadMaskOnBackground("modelsel/mask_modelqty.png", TEXT_COLOR, TEXT_BGCOLOR);
+
+      delete modelselModelNameBitmap;
+      modelselModelNameBitmap = BitmapBuffer::loadMaskOnBackground("modelsel/mask_modelname.png", TEXT_COLOR, TEXT_BGCOLOR);
+
+      delete modelselModelMoveBackground;
+      modelselModelMoveBackground = BitmapBuffer::loadMask(getThemePath("modelsel/mask_moveback.png"));
+
+      delete modelselModelMoveIcon;
+      modelselModelMoveIcon = BitmapBuffer::loadMask(getThemePath("modelsel/mask_moveico.png"));
+
+      delete modelselWizardBackground;
+      modelselWizardBackground = BitmapBuffer::load(getThemePath("wizard/background.png"));
+
+
      // Mixer setup screen
      delete mixerSetupMixerBitmap;
      mixerSetupMixerBitmap = BitmapBuffer::loadMaskOnBackground("mask_sbar_mixer.png", MENU_TITLE_COLOR, HEADER_BGCOLOR);
@@ -191,6 +191,9 @@ class DefaultTheme: public Theme
      delete modelselModelMoveIcon;
      modelselModelMoveIcon = BitmapBuffer::loadMask(getThemePath("modelsel/mask_moveico.png"));

+      delete modelselWizardBackground;
+      modelselWizardBackground = BitmapBuffer::load(getThemePath("wizard/background.png"));
+
      // Channels monitor screen
      delete chanMonLockedBitmap;
      chanMonLockedBitmap = BitmapBuffer::loadMaskOnBackground("mask_monitor_lockch.png", TEXT_COLOR, TEXT_BGCOLOR);