From 6a2096c41977d6c33c2ca20494d9f623bed221ff Mon Sep 17 00:00:00 2001
From: Jordan Woyak <jordan.woyak@gmail.com>
Date: Sat, 26 Jan 2019 13:18:20 -0600
Subject: ExpressionParser: Add optional 2nd argument to toggle function which
 clears state.

---
 .../InputCommon/ControlReference/FunctionExpression.cpp     | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'Source/Core/InputCommon/ControlReference/FunctionExpression.cpp')

diff --git a/Source/Core/InputCommon/ControlReference/FunctionExpression.cpp b/Source/Core/InputCommon/ControlReference/FunctionExpression.cpp
index 15a66ec9d1..84f3ac907e 100644
--- a/Source/Core/InputCommon/ControlReference/FunctionExpression.cpp
+++ b/Source/Core/InputCommon/ControlReference/FunctionExpression.cpp
@@ -32,7 +32,8 @@ class ToggleExpression : public FunctionExpression
 private:
   virtual bool ValidateArguments(const std::vector<std::unique_ptr<Expression>>& args) override
   {
-    return 1 == args.size();
+    // Optional 2nd argument for clearing state:
+    return 1 == args.size() || 2 == args.size();
   }
 
   ControlState GetValue() const override
@@ -49,6 +50,11 @@ private:
       m_state ^= true;
     }
 
+    if (2 == GetArgCount() && GetArg(1).GetValue() > CONDITION_THRESHOLD)
+    {
+      m_state = false;
+    }
+
     return m_state;
   }
 
@@ -257,5 +263,10 @@ const Expression& FunctionExpression::GetArg(u32 number) const
   return *m_args[number];
 }
 
+u32 FunctionExpression::GetArgCount() const
+{
+  return u32(m_args.size());
+}
+
 }  // namespace ExpressionParser
 }  // namespace ciface
-- 
cgit v1.2.3