From dac2a9401bbb72dd3f95b8cdd191066d50301d16 Mon Sep 17 00:00:00 2001
From: elecpower <neilh713@tpg.com.au>
Date: Sat, 18 Dec 2021 06:25:29 +1100
Subject: [PATCH] Add potswarningmode

---
 companion/src/firmwares/edgetx/yaml_modeldata.cpp | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/companion/src/firmwares/edgetx/yaml_modeldata.cpp b/companion/src/firmwares/edgetx/yaml_modeldata.cpp
index 7c82c0540..b96957770 100644
--- a/companion/src/firmwares/edgetx/yaml_modeldata.cpp
+++ b/companion/src/firmwares/edgetx/yaml_modeldata.cpp
@@ -63,6 +63,12 @@ static const YamlLookupTable swashTypeLut = {
   {  4, "TYPE_90"  },
 };
 
+static const YamlLookupTable potsWarningModeLut = {
+  {  0, "WARN_OFF"  },
+  {  1, "WARN_MANUAL"  },
+  {  2, "WARN_AUTO"  },
+};
+
 struct YamlTrim {
   int mode = 0;
   int ref = 0;
@@ -403,7 +409,7 @@ Node convert<ModelData>::encode(const ModelData& rhs)
 
   // switchWarningStates
   node["thrTrimSw"] = rhs.thrTrimSwitch;
-  // potsWarningMode
+  node["potsWarnMode"] = potsWarningModeLut << rhs.potsWarningMode;
   // potsWarnEnabled[]
   // potsWarnPosition[]
 
@@ -499,13 +505,12 @@ bool convert<ModelData>::decode(const Node& node, ModelData& rhs)
 
   // switchWarningStates
   node["thrTrimSw"] >> rhs.thrTrimSwitch;
-  // potsWarningMode
+  node["potsWarnMode"] >> potsWarningModeLut >> rhs.potsWarningMode;
   // potsWarnEnabled[]
   // potsWarnPosition[]
 
   node["displayChecklist"] >> rhs.displayChecklist;
 
-  // gvarData[]
   node["gvars"] >> rhs.gvarData;
   // mavlink ???
   // telemetryProtocol ???