Merge pull request #690 from jw098/record-keypress

Record Keyboard Controller

Author: Mysticial

Common/Cpp/Json/JsonObject.h

@@ -18,6 +18,13 @@ class JsonObject{
     JsonObject() = default;
     JsonObject(JsonObject&& x) = default;
     JsonObject& operator=(JsonObject&& x) = default;
+
+    bool operator==(const JsonObject& x){ // TODO: implement == properly. For JsonValue as well.
+        return dump() == x.dump();
+    }
+    bool operator!=(const JsonObject& x){
+        return !(*this == *this);
+    }
 private:
     //  Private to avoid accidental copying.
     JsonObject(const JsonObject& x);

Common/Cpp/StringTools.cpp

@@ -6,6 +6,7 @@
 
 #include <stdint.h>
 #include <stdexcept>
+#include <vector>
 #include "StringTools.h"
 
 namespace PokemonAutomation{
@@ -48,6 +49,21 @@ size_t to_size_t(const std::string& str){
     }
 }
 
+std::vector<std::string> split(const std::string& str, const std::string& delimiter) {
+    std::vector<std::string> tokens;
+    size_t start = 0;
+    size_t end = str.find(delimiter);
+    
+    while (end != std::string::npos) {
+        tokens.push_back(str.substr(start, end - start));
+        start = end + delimiter.length();
+        end = str.find(delimiter, start);
+    }
+    
+    tokens.push_back(str.substr(start));
+    return tokens;
+}
+
 
 }
 }

Common/Cpp/StringTools.h

@@ -21,6 +21,8 @@ std::string strip(const std::string& str);
 // Parse str to size_t. Return SIZE_MAX if parsing fails
 size_t to_size_t(const std::string& str);
 
+std::vector<std::string> split(const std::string& str, const std::string& delimiter);
+
 }
 }
 #endif

Common/Qt/Options/BoxFloatWidget.cpp

@@ -7,6 +7,7 @@
 #include <QHBoxLayout>
 #include <QVBoxLayout>
 #include <QLabel>
+#include "Common/Cpp/StringTools.h"
 #include "BoxFloatWidget.h"
 
 //#include <iostream>
@@ -24,21 +25,6 @@ ConfigWidget* BoxFloatOption::make_QtWidget(QWidget& parent){
 
 
 
-std::vector<std::string> split(const std::string& str, const std::string& delimiter) {
-    std::vector<std::string> tokens;
-    size_t start = 0;
-    size_t end = str.find(delimiter);
-
-    while (end != std::string::npos) {
-        tokens.push_back(str.substr(start, end - start));
-        start = end + delimiter.length();
-        end = str.find(delimiter, start);
-    }
-
-    tokens.push_back(str.substr(start));
-    return tokens;
-}
-
 
 
 BoxFloatWidget::~BoxFloatWidget(){
@@ -167,7 +153,7 @@ BoxFloatWidget::BoxFloatWidget(QWidget& parent, BoxFloatOption& value)
     connect(
         m_array, &QLineEdit::editingFinished,
         this, [this](){
-            std::vector<std::string> all_coords = split(m_array->text().toStdString(), ", ");
+            std::vector<std::string> all_coords = StringTools::split(m_array->text().toStdString(), ", ");
             if (all_coords.size() != 4){
                 return;
             }

SerialPrograms/CMakeLists.txt

@@ -701,6 +701,8 @@ file(GLOB MAIN_SOURCES
     Source/Controllers/JoystickTools.h
     Source/Controllers/KeyboardInput/GlobalQtKeyMap.cpp
     Source/Controllers/KeyboardInput/GlobalQtKeyMap.h
+    Source/Controllers/KeyboardInput/KeyboardEventHandler.cpp
+    Source/Controllers/KeyboardInput/KeyboardEventHandler.h
     Source/Controllers/KeyboardInput/KeyboardInput.cpp
     Source/Controllers/KeyboardInput/KeyboardInput.h
     Source/Controllers/KeyboardInput/KeyboardStateTracker.cpp
@@ -1119,6 +1121,8 @@ file(GLOB MAIN_SOURCES
     Source/NintendoSwitch/Programs/NintendoSwitch_PreventSleep.h
     Source/NintendoSwitch/Programs/NintendoSwitch_PushJoySticks.cpp
     Source/NintendoSwitch/Programs/NintendoSwitch_PushJoySticks.h
+    Source/NintendoSwitch/Programs/NintendoSwitch_RecordKeyboardController.cpp
+    Source/NintendoSwitch/Programs/NintendoSwitch_RecordKeyboardController.h
     Source/NintendoSwitch/Programs/NintendoSwitch_SnapshotDumper.cpp
     Source/NintendoSwitch/Programs/NintendoSwitch_SnapshotDumper.h
     Source/NintendoSwitch/Programs/NintendoSwitch_SwitchViewer.cpp

SerialPrograms/Source/Controllers/Controller.h

@@ -10,6 +10,7 @@
 #include "Common/Compiler.h"
 #include "Common/Cpp/AbstractLogger.h"
 #include "Common/Cpp/Time.h"
+#include "Controllers/KeyboardInput/KeyboardEventHandler.h"
 #include "Common/Cpp/CancellableScope.h"
 
 class QKeyEvent;
@@ -19,6 +20,7 @@ namespace PokemonAutomation{
 class RecursiveThrottler;
 enum class ControllerType;
 enum class ControllerPerformanceClass;
+enum class ControllerClass;
 
 
 
@@ -64,6 +66,7 @@ class AbstractController{
 
     virtual const char* name() = 0;
     virtual ControllerType controller_type() const = 0;
+    virtual ControllerClass controller_class() const = 0;
     virtual ControllerPerformanceClass performance_class() const = 0;
 
     //  If the controller is polled at a fixed interval, this is that interval.
@@ -203,6 +206,9 @@ class AbstractController{
     virtual void keyboard_release_all(){}
     virtual void keyboard_press(const QKeyEvent& event){}
     virtual void keyboard_release(const QKeyEvent& event){}
+
+    virtual void add_keyboard_listener(KeyboardEventHandler::KeyboardListener& keyboard_listener){};
+    virtual void remove_keyboard_listener(KeyboardEventHandler::KeyboardListener& keyboard_listener){};
 };
 
 
@@ -294,6 +300,7 @@ class ControllerContext final : public CancellableScope{
 };
 
 
+using AbstractControllerContext = ControllerContext<AbstractController>;
 
 using AbstractControllerContext = ControllerContext<AbstractController>;
 

SerialPrograms/Source/Controllers/ControllerTypeStrings.cpp

@@ -32,6 +32,16 @@ const EnumStringMap<ControllerType> CONTROLLER_TYPE_STRINGS{
     {ControllerType::NintendoSwitch2_RightJoycon,           "NS2: Right Joycon"},
 };
 
+const EnumStringMap<ControllerClass>& CONTROLLER_CLASS_STRINGS(){
+    static EnumStringMap<ControllerClass> database{
+        {ControllerClass::NONE,                                  "none"},
+        {ControllerClass::LEFT_JOYCON,                           "left-joycon"},
+        {ControllerClass::RIGHT_JOYCON,                          "right-joycon"},
+        {ControllerClass::PRO_CONTROLLER,                        "pro-controller"},
+    };
+    return database;
+}
+
 
 
 

SerialPrograms/Source/Controllers/ControllerTypeStrings.h

@@ -16,6 +16,7 @@ namespace PokemonAutomation{
 
 extern const EnumStringMap<ControllerInterface> CONTROLLER_INTERFACE_STRINGS;
 extern const EnumStringMap<ControllerType> CONTROLLER_TYPE_STRINGS;
+const EnumStringMap<ControllerClass>& CONTROLLER_CLASS_STRINGS();
 
 
 

SerialPrograms/Source/Controllers/ControllerTypes.h

@@ -44,6 +44,14 @@ enum class ControllerPerformanceClass{
     SysbotBase,
 };
 
+enum class ControllerClass{
+    NONE,
+    LEFT_JOYCON,
+    RIGHT_JOYCON,
+    PRO_CONTROLLER,
+    KEYBOARD,
+};
+
 
 
 

SerialPrograms/Source/Controllers/KeyboardInput/KeyboardEventHandler.cpp

@@ -0,0 +1,47 @@
+/*  Keyboard Event Handler
+ *
+ */
+#include "Common/Cpp/ListenerSet.h"
+#include "Common/Cpp/Containers/Pimpl.tpp"
+#include "Controllers/KeyboardInput/KeyboardInput.h"
+#include "KeyboardEventHandler.h"
+
+namespace PokemonAutomation{
+
+
+
+struct KeyboardEventHandler::Data{
+    ListenerSet<KeyboardListener> m_listeners;
+};
+
+KeyboardEventHandler::KeyboardEventHandler()
+    : m_data(CONSTRUCT_TOKEN)
+{}
+
+KeyboardEventHandler::~KeyboardEventHandler() = default;
+
+void KeyboardEventHandler::add_listener(KeyboardListener& listener){
+    auto scope = m_lifetime_sanitizer.check_scope();
+    Data& data = *m_data;
+    data.m_listeners.add(listener);
+}
+void KeyboardEventHandler::remove_listener(KeyboardListener& listener){
+    auto scope = m_lifetime_sanitizer.check_scope();
+    Data& data = *m_data;
+    data.m_listeners.remove(listener);
+}
+
+void KeyboardEventHandler::report_keyboard_command_sent(WallClock time_stamp, const ControllerState& state){
+    auto scope = m_lifetime_sanitizer.check_scope();
+    Data& data = *m_data;
+    data.m_listeners.run_method_unique(&KeyboardListener::on_keyboard_command_sent, time_stamp, state);
+}
+
+void KeyboardEventHandler::report_keyboard_command_stopped(WallClock time_stamp){
+    auto scope = m_lifetime_sanitizer.check_scope();
+    Data& data = *m_data;
+    data.m_listeners.run_method_unique(&KeyboardListener::on_keyboard_command_stopped, time_stamp);
+}
+
+
+}