From a5acdf411311004cfdba046744d4c683a55179cb Mon Sep 17 00:00:00 2001
From: cornernote <brett@mrphp.com.au>
Date: Tue, 14 Jan 2020 08:40:45 +1030
Subject: [PATCH] add support for g815 on-board-mode

---
 src/classes/Keyboard.cpp | 27 ++++++++++++++++++++++++---
 src/classes/Keyboard.h   |  5 +++++
 src/helpers/help.cpp     |  8 ++++++++
 src/helpers/help.h       |  5 +++--
 src/helpers/utils.cpp    | 13 ++++++++++---
 src/helpers/utils.h      |  3 ++-
 src/main.cpp             | 11 +++++++++++
 7 files changed, 63 insertions(+), 9 deletions(-)

diff --git a/src/classes/Keyboard.cpp b/src/classes/Keyboard.cpp
index 7da2cfe..1c3b286 100644
--- a/src/classes/Keyboard.cpp
+++ b/src/classes/Keyboard.cpp
@@ -755,9 +755,6 @@ bool LedKeyboard::setStartupMode(StartupMode startupMode) {
 		case KeyboardModel::gpro:
 			data = { 0x11, 0xff, 0x0d, 0x5a, 0x00, 0x01 };
 			break;
-		case KeyboardModel::g815:
-			data = { 0x11, 0xff, 0x11, 0x1b, 0x02 };
-			break;
 		case KeyboardModel::g910:
 			data = { 0x11, 0xff, 0x10, 0x5e, 0x00, 0x01 };
 			break;
@@ -769,6 +766,30 @@ bool LedKeyboard::setStartupMode(StartupMode startupMode) {
 	return sendDataInternal(data);
 }
 
+bool LedKeyboard::setOnBoardMode(OnBoardMode onBoardMode) {
+	byte_buffer_t data;
+	switch (currentDevice.model) {
+		case KeyboardModel::g815:
+			switch (onBoardMode) {
+				case LedKeyboard::OnBoardMode::board:
+					data = { 0x11, 0xff, 0x11, 0x1a, 0x01 };
+					data.resize(20, 0x00);
+					return sendDataInternal(data);
+				case LedKeyboard::OnBoardMode::software:
+					data = { 0x11, 0xff, 0x11, 0x1b, 0x02 };
+					data.resize(20, 0x00);
+					return sendDataInternal(data);
+				default:
+					break;
+			}
+			break;
+		default:
+			return false;
+	}
+	data.push_back((unsigned char)onBoardMode);
+	data.resize(20, 0x00);
+	return sendDataInternal(data);
+}
 
 bool LedKeyboard::setNativeEffect(NativeEffect effect, NativeEffectPart part,
 				  std::chrono::duration<uint16_t, std::milli> period, Color color,
diff --git a/src/classes/Keyboard.h b/src/classes/Keyboard.h
index 74505e2..41dca80 100644
--- a/src/classes/Keyboard.h
+++ b/src/classes/Keyboard.h
@@ -78,6 +78,10 @@ class LedKeyboard {
 			wave = 0x01,
 			color
 		};
+		enum class OnBoardMode : uint8_t {
+			board = 0x01,
+			software
+		};
 		enum class NativeEffectGroup : uint8_t {
 			off,
 			color,
@@ -201,6 +205,7 @@ class LedKeyboard {
 		
 		bool setRegion(uint8_t region, Color color);
 		bool setStartupMode(StartupMode startupMode);
+		bool setOnBoardMode(OnBoardMode onBoardMode);
 
 		bool setNativeEffect(NativeEffect effect, NativeEffectPart part,
 				     std::chrono::duration<uint16_t, std::milli> period, Color color,
diff --git a/src/helpers/help.cpp b/src/helpers/help.cpp
index 30a7f45..f53a322 100644
--- a/src/helpers/help.cpp
+++ b/src/helpers/help.cpp
@@ -90,6 +90,10 @@ namespace help {
 			cout<<"  --startup-mode {startup mode}\t\tSet startup mode"<<endl;
 			cout<<endl;
 		}
+		if((features | KeyboardFeatures::onboardmode) == features) {
+			cout<<"  --on-board-mode {on board mode}\t\tSet on-board mode"<<endl;
+			cout<<endl;
+		}
 		cout<<"  --list-keyboards \t\t\tList connected keyboards"<<endl;
 		cout<<"  --print-device\t\t\tPrint device information for the keyboard"<<endl;
 		cout<<endl;
@@ -120,6 +124,7 @@ namespace help {
 		if((features | KeyboardFeatures::setgroup) == features)
 			cout<<"  group values :\t\t\tlogo, indicators, fkeys, ... (use --help-keys for more detail)"<<endl;
 		cout<<"  startup mode :\t\t\twave, color"<<endl;
+		cout<<"  on-board mode :\t\t\tboard, software"<<endl;
 		cout<<endl;
 	}
 	
@@ -324,6 +329,9 @@ namespace help {
 		if((features | KeyboardFeatures::poweronfx) == features)
 			cout<<cmdName<<" --startup-mode color          # Set keyboard power on effect"<<endl;
 		cout<<endl;
+		if((features | KeyboardFeatures::onboardmode) == features)
+			cout<<cmdName<<" --on-board-mode color          # Set keyboard power on effect"<<endl;
+		cout<<endl;
 		if((features | KeyboardFeatures::commit) == features) {
 			cout<<"Samples with no commit :"<<endl;
 			if((features | KeyboardFeatures::setall) == features)
diff --git a/src/helpers/help.h b/src/helpers/help.h
index e815794..f8ff15d 100644
--- a/src/helpers/help.h
+++ b/src/helpers/help.h
@@ -39,10 +39,11 @@ namespace help {
 		poweronfx = 8192,
 		// "user-stored lighting" can be recalled with backlight+7 on the G Pro
 		userstoredlighting = 16384,
+		onboardmode = 32768,
 		// fx features
 		
 		all = rgb | intensity | commit | logo1 | logo2 | numpad | multimedia | gkeys |
-		      setall | setgroup | setkey | setregion | setindicators | poweronfx | userstoredlighting,
+		      setall | setgroup | setkey | setregion | setindicators | poweronfx | userstoredlighting | onboardmode,
 		
 		g213 = rgb | logo1 | numpad | multimedia | setall | setregion | setindicators | poweronfx | userstoredlighting,
 		g410 = rgb | commit | setall | setgroup | setkey | poweronfx | userstoredlighting,
@@ -50,7 +51,7 @@ namespace help {
 		g513 = rgb | commit | numpad | setall | setgroup | setkey | setindicators | poweronfx | userstoredlighting,
 		g610 = intensity | commit | logo1 | numpad | multimedia | setall | setgroup | setkey | setindicators | poweronfx | userstoredlighting,
 		g810 = rgb | commit | logo1 | numpad | multimedia | setall | setgroup | setkey | setindicators | poweronfx,
-		g815 = rgb | commit | logo1 | numpad | multimedia | gkeys | setall | setgroup | setkey | setindicators | poweronfx,
+		g815 = rgb | commit | logo1 | numpad | multimedia | gkeys | setall | setgroup | setkey | setindicators | onboardmode,
 		g910 = rgb | commit | logo1 | logo2 | numpad | multimedia | gkeys | setall | setgroup | setkey | setindicators | poweronfx | userstoredlighting,
 		gpro = rgb | commit | logo1 | setall | setgroup | setkey | setindicators | poweronfx | userstoredlighting
 	};
diff --git a/src/helpers/utils.cpp b/src/helpers/utils.cpp
index ee2ad1f..6f97eba 100644
--- a/src/helpers/utils.cpp
+++ b/src/helpers/utils.cpp
@@ -27,15 +27,22 @@ namespace utils {
 	std::string getCmdName(std::string cmd) {
 		return cmd.substr(cmd.find_last_of("/\\") + 1);
 	}
-	
-	
-	
+
+
+
 	bool parseStartupMode(std::string val, LedKeyboard::StartupMode &startupMode) {
 		if (val == "wave") startupMode = LedKeyboard::StartupMode::wave;
 		else if (val == "color") startupMode = LedKeyboard::StartupMode::color;
 		else return false;
 		return true;
 	}
+
+	bool parseOnBoardMode(std::string val, LedKeyboard::OnBoardMode &onBoardMode) {
+		if (val == "software") onBoardMode = LedKeyboard::OnBoardMode ::software;
+		else if (val == "board") onBoardMode = LedKeyboard::OnBoardMode::board;
+		else return false;
+		return true;
+	}
 	
 	bool parseNativeEffect(std::string val, LedKeyboard::NativeEffect &nativeEffect) {
 		if (val == "color") nativeEffect = LedKeyboard::NativeEffect::color;
diff --git a/src/helpers/utils.h b/src/helpers/utils.h
index a5ec033..ced6165 100644
--- a/src/helpers/utils.h
+++ b/src/helpers/utils.h
@@ -24,8 +24,9 @@
 namespace utils {
 	
 	std::string getCmdName(std::string cmd);
-	
+
 	bool parseStartupMode(std::string val, LedKeyboard::StartupMode &startupMode);
+	bool parseOnBoardMode(std::string val, LedKeyboard::OnBoardMode &onBoardMode);
 	bool parseNativeEffect(std::string val, LedKeyboard::NativeEffect &nativeEffect);
 	bool parseNativeEffectPart(std::string val, LedKeyboard::NativeEffectPart &nativeEffectPart);
 	bool parseKey(std::string val, LedKeyboard::Key &key);
diff --git a/src/main.cpp b/src/main.cpp
index 4d7595e..0308c2f 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -171,6 +171,14 @@ int setStartupMode(LedKeyboard &kbd, std::string arg2) {
 	return 1;
 }
 
+int setOnBoardMode(LedKeyboard &kbd, std::string arg2) {
+	LedKeyboard::OnBoardMode onBoardMode;
+	if (! utils::parseOnBoardMode(arg2, onBoardMode)) return 1;
+	if (! kbd.open()) return 1;
+	if (kbd.setOnBoardMode(onBoardMode)) return 0;
+	return 1;
+}
+
 
 int parseProfile(LedKeyboard &kbd, std::istream &stream) {
 	std::string line;
@@ -219,6 +227,8 @@ int parseProfile(LedKeyboard &kbd, std::istream &stream) {
 				if (setGKeysMode(kbd, args[1]) == 1) retval = 1;
 			} else if (args[0] == "sm" && args.size() > 1) {
 				if (setStartupMode(kbd, args[1]) == 1) retval = 1;
+			} else if (args[0] == "obm" && args.size() > 1) {
+				if (setOnBoardMode(kbd, args[1]) == 1) retval = 1;
 			} else if (args[0] == "fx" && args.size() > 4) {
 				if (setFX(kbd, args[1], args[2], args[3], args[4]) == 1) retval = 1;
 			} else if (args[0] == "fx" && args.size() > 3) {
@@ -336,6 +346,7 @@ int main(int argc, char **argv) {
 		else if (argc > (argIndex + 3) && arg == "-fx-store")
 			return storeFX(kbd, argv[argIndex + 1], argv[argIndex + 2], argv[argIndex + 3]);
 		else if (argc > (argIndex + 1) && arg == "--startup-mode") return setStartupMode(kbd, argv[argIndex + 1]);
+		else if (argc > (argIndex + 1) && arg == "--on-board-mode") return setOnBoardMode(kbd, argv[argIndex + 1]);
 		else { help::usage(argv[0]); return 1; }
 	}