449 lines
17 KiB
C++
449 lines
17 KiB
C++
/*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
* MA 02110-1301, USA.
|
|
*
|
|
* g++ mouse_m908.cpp -o mouse_m908 -lusb-1.0
|
|
*
|
|
*/
|
|
|
|
#include <map>
|
|
#include <array>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <exception>
|
|
#include <regex>
|
|
#include <getopt.h>
|
|
|
|
#include "include/mouse_m908.h"
|
|
#include "include/load_config.h"
|
|
#include "include/print_help.cpp"
|
|
|
|
// this is the default version string
|
|
// the version string gets overwritten by the makefile
|
|
#ifndef VERSION_STRING
|
|
#define VERSION_STRING "no version defined during compilation"
|
|
#endif
|
|
|
|
// this function checks its arguments and opens the mouse accordingly
|
|
// (with vid and pid or with bus and device)
|
|
int open_mouse_wrapper( mouse_m908 &m, const bool flag_bus, const bool flag_device,
|
|
const std::string &string_bus, const std::string &string_device ){
|
|
|
|
int open_return = 0; // open_mouse() return value
|
|
|
|
if( flag_bus != flag_device ){ // improper arguments
|
|
|
|
std::cout << "Missing argument, --bus and --device must be used together.\n";
|
|
return 1;
|
|
|
|
} else if( flag_bus && flag_device ){ // open with bus and device
|
|
|
|
if( !std::regex_match( string_bus, std::regex("[0-9]+") ) ||
|
|
!std::regex_match( string_device, std::regex("[0-9]+") ) ){
|
|
|
|
std::cout << "Wrong argument, expected number.\n";
|
|
return 1;
|
|
}
|
|
|
|
open_return = m.open_mouse_bus_device( stoi(string_bus), stoi(string_device) );
|
|
|
|
} else{ // open with vid and pid
|
|
|
|
open_return = m.open_mouse();
|
|
|
|
}
|
|
|
|
// Could not open → print message
|
|
if( open_return != 0 ){
|
|
std::cout << "Couldn't open mouse.\n";
|
|
std::cout << "- Check hardware and permissions (maybe you need to be root?)\n";
|
|
std::cout << "- Try with or without the --kernel-driver option\n";
|
|
std::cout << "- Try with the --bus and --device options\n";
|
|
std::cout << "If nothing works please report this as a bug.\n";
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int main( int argc, char **argv ){
|
|
|
|
mouse_m908 m;
|
|
|
|
// if no arguments: print help
|
|
if( argc == 1 ){
|
|
print_help();
|
|
return 0;
|
|
}
|
|
|
|
//command line options
|
|
static struct option long_options[] = {
|
|
{"help", no_argument, 0, 'h'},
|
|
{"config", required_argument, 0, 'c'},
|
|
{"profile", required_argument, 0, 'p'},
|
|
{"macro", required_argument, 0, 'm'},
|
|
{"number", required_argument, 0, 'n'},
|
|
//{"repeat", required_argument, 0, 'r'},
|
|
{"bus", required_argument, 0, 'b'},
|
|
{"device", required_argument, 0, 'd'},
|
|
{"kernel-driver", no_argument, 0, 'k'},
|
|
{"version", no_argument, 0, 'v'},
|
|
{"dump", required_argument, 0, 'D'},
|
|
{"read", required_argument, 0, 'R'},
|
|
{0, 0, 0, 0}
|
|
};
|
|
|
|
bool flag_config = false, flag_profile = false;
|
|
bool flag_macro = false, flag_number = false;
|
|
bool flag_bus = false, flag_device = false;
|
|
bool flag_kernel_driver = false;
|
|
bool flag_version = false;
|
|
bool flag_dump_settings = false;
|
|
bool flag_read_settings = false;
|
|
//bool flag_repeat;
|
|
|
|
std::string string_config, string_profile;
|
|
std::string string_macro, string_number;
|
|
//std::string string_repeat;
|
|
std::string string_bus, string_device;
|
|
std::string string_dump, string_read;
|
|
|
|
//parse command line options
|
|
int c, option_index = 0;
|
|
//while( (c = getopt_long( argc, argv, "hc:p:m:n:b:d:kvr:",
|
|
while( (c = getopt_long( argc, argv, "hc:p:m:n:b:d:kvD:R:",
|
|
long_options, &option_index ) ) != -1 ){
|
|
|
|
switch( c ){
|
|
case 'h':
|
|
print_help();
|
|
return 0;
|
|
break;
|
|
case 'c':
|
|
flag_config = true;
|
|
string_config = optarg;
|
|
break;
|
|
case 'p':
|
|
flag_profile = true;
|
|
string_profile = optarg;
|
|
break;
|
|
case 'm':
|
|
flag_macro = true;
|
|
string_macro = optarg;
|
|
break;
|
|
case 'n':
|
|
flag_number = true;
|
|
string_number = optarg;
|
|
break;
|
|
case 'b':
|
|
flag_bus = true;
|
|
string_bus = optarg;
|
|
break;
|
|
case 'd':
|
|
flag_device = true;
|
|
string_device = optarg;
|
|
break;
|
|
//case 'r':
|
|
// flag_repeat = true;
|
|
// string_repeat = optarg;
|
|
// break;
|
|
case 'k':
|
|
flag_kernel_driver = true;
|
|
break;
|
|
case 'v':
|
|
flag_version = true;
|
|
break;
|
|
case 'D':
|
|
flag_dump_settings = true;
|
|
string_dump = optarg;
|
|
break;
|
|
case 'R':
|
|
flag_read_settings = true;
|
|
string_read = optarg;
|
|
break;
|
|
case '?':
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
// set whether to detach kernel driver
|
|
m.set_detach_kernel_driver( !flag_kernel_driver );
|
|
|
|
// print version if requested
|
|
if( flag_version )
|
|
std::cout << "Version: " << VERSION_STRING << "\n";
|
|
|
|
// read settings and dump raw data
|
|
if( flag_dump_settings ){
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// dump to file or cout
|
|
if( string_dump != "-" ){
|
|
std::ofstream out( string_dump );
|
|
|
|
if( out.is_open() ){
|
|
// dump settings
|
|
m.dump_settings( out );
|
|
|
|
out.close();
|
|
} else{
|
|
std::cout << "Couldn't open " << string_dump << "\n";
|
|
}
|
|
} else{
|
|
m.dump_settings( std::cout );
|
|
}
|
|
|
|
|
|
m.close_mouse();
|
|
|
|
}
|
|
|
|
// read settings and print in .ini format
|
|
if( flag_read_settings ){
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// dump to file or cout
|
|
if( string_read != "-" ){
|
|
std::ofstream out( string_read );
|
|
|
|
if( out.is_open() ){
|
|
// read settings
|
|
m.read_and_print_settings( out );
|
|
|
|
out.close();
|
|
} else{
|
|
std::cout << "Couldn't open " << string_read << "\n";
|
|
}
|
|
} else{
|
|
m.read_and_print_settings( std::cout );
|
|
}
|
|
|
|
m.close_mouse();
|
|
|
|
}
|
|
|
|
// load and write config
|
|
if( flag_config ){
|
|
try{
|
|
|
|
simple_ini_parser pt;
|
|
if( pt.read_ini( string_config ) != 0 ){
|
|
std::cout << "Could not open configuration file.\n";
|
|
return 1;
|
|
}
|
|
|
|
// lookup table int → profile enum
|
|
std::array< mouse_m908::m908_profile, 5 > profile_lut = { mouse_m908::profile_1, mouse_m908::profile_2, mouse_m908::profile_3, mouse_m908::profile_4, mouse_m908::profile_5 };
|
|
|
|
//parse config file
|
|
for( int i = 1; i < 6; i++ ){
|
|
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "breathing" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_breathing ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "rainbow" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_rainbow ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "static" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_static ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "wave" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_wave ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "alternating" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_alternating ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "reactive" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_reactive ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "flashing" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_flashing ); }
|
|
if( pt.get("profile"+std::to_string(i)+".lightmode", "") == "off" ){ m.set_lightmode( profile_lut[i-1], mouse_m908::lightmode_off ); }
|
|
if( pt.get("profile"+std::to_string(i)+".color", "").length() == 6 ){
|
|
m.set_color( profile_lut[i-1],
|
|
{(uint8_t)stoi( pt.get("profile"+std::to_string(i)+".color", "").substr(0,2), 0, 16),
|
|
(uint8_t)stoi( pt.get("profile"+std::to_string(i)+".color", "").substr(2,2), 0, 16),
|
|
(uint8_t)stoi( pt.get("profile"+std::to_string(i)+".color", "").substr(4,2), 0, 16)} );
|
|
}
|
|
if( pt.get("profile"+std::to_string(i)+".brightness", "").length() != 0 ){
|
|
m.set_brightness( profile_lut[i-1], (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".brightness", ""), 0, 16) );
|
|
}
|
|
if( pt.get("profile"+std::to_string(i)+".speed", "").length() != 0 ){
|
|
m.set_speed( profile_lut[i-1], (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".speed", ""), 0, 16) );
|
|
}
|
|
if( pt.get("profile"+std::to_string(i)+".scrollspeed", "").length() != 0 ){
|
|
m.set_scrollspeed( profile_lut[i-1], (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".scrollspeed", ""), 0, 16) );
|
|
}
|
|
if( pt.get("profile"+std::to_string(i)+".dpi1_enable", "") == "0" ){ m.set_dpi_enable( profile_lut[i-1], 0, false ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi2_enable", "") == "0" ){ m.set_dpi_enable( profile_lut[i-1], 1, false ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi3_enable", "") == "0" ){ m.set_dpi_enable( profile_lut[i-1], 2, false ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi4_enable", "") == "0" ){ m.set_dpi_enable( profile_lut[i-1], 3, false ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi5_enable", "") == "0" ){ m.set_dpi_enable( profile_lut[i-1], 4, false ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi1", "").length() != 0 ){ m.set_dpi( profile_lut[i-1], 0, (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".dpi1", ""), 0, 16) ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi2", "").length() != 0 ){ m.set_dpi( profile_lut[i-1], 1, (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".dpi2", ""), 0, 16) ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi3", "").length() != 0 ){ m.set_dpi( profile_lut[i-1], 2, (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".dpi3", ""), 0, 16) ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi4", "").length() != 0 ){ m.set_dpi( profile_lut[i-1], 3, (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".dpi4", ""), 0, 16) ); }
|
|
if( pt.get("profile"+std::to_string(i)+".dpi5", "").length() != 0 ){ m.set_dpi( profile_lut[i-1], 4, (uint8_t)stoi( pt.get("profile"+std::to_string(i)+".dpi5", ""), 0, 16) ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_left", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 0, pt.get("profile"+std::to_string(i)+".button_left", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_right", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 1, pt.get("profile"+std::to_string(i)+".button_right", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_middle", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 2, pt.get("profile"+std::to_string(i)+".button_middle", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_fire", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 3, pt.get("profile"+std::to_string(i)+".button_fire", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_dpi_up", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 4, pt.get("profile"+std::to_string(i)+".button_dpi_up", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_dpi_down", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 5, pt.get("profile"+std::to_string(i)+".button_dpi_down", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_1", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 6, pt.get("profile"+std::to_string(i)+".button_1", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_2", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 7, pt.get("profile"+std::to_string(i)+".button_2", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_3", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 8, pt.get("profile"+std::to_string(i)+".button_3", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_4", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 9, pt.get("profile"+std::to_string(i)+".button_4", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_5", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 10, pt.get("profile"+std::to_string(i)+".button_5", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_6", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 11, pt.get("profile"+std::to_string(i)+".button_6", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_7", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 12, pt.get("profile"+std::to_string(i)+".button_7", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_8", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 13, pt.get("profile"+std::to_string(i)+".button_8", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_9", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 14, pt.get("profile"+std::to_string(i)+".button_9", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_10", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 15, pt.get("profile"+std::to_string(i)+".button_10", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_11", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 16, pt.get("profile"+std::to_string(i)+".button_11", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".button_12", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 17, pt.get("profile"+std::to_string(i)+".button_12", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".scroll_up", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 18, pt.get("profile"+std::to_string(i)+".scroll_up", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".scroll_down", "").length() != 0 ){ m.set_key_mapping( profile_lut[i-1], 19, pt.get("profile"+std::to_string(i)+".scroll_down", "") ); }
|
|
if( pt.get("profile"+std::to_string(i)+".report_rate", "") == "125" ){ m.set_report_rate( profile_lut[i-1], mouse_m908::r_125Hz ); }
|
|
if( pt.get("profile"+std::to_string(i)+".report_rate", "") == "250" ){ m.set_report_rate( profile_lut[i-1], mouse_m908::r_250Hz ); }
|
|
if( pt.get("profile"+std::to_string(i)+".report_rate", "") == "500" ){ m.set_report_rate( profile_lut[i-1], mouse_m908::r_500Hz ); }
|
|
if( pt.get("profile"+std::to_string(i)+".report_rate", "") == "1000" ){ m.set_report_rate( profile_lut[i-1], mouse_m908::r_1000Hz ); }
|
|
|
|
}
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// write settings
|
|
m.write_settings();
|
|
|
|
m.close_mouse();
|
|
} catch( std::exception& e ){
|
|
std::cerr << "Error: " << e.what() << "\n";
|
|
return 1;
|
|
}
|
|
|
|
}
|
|
|
|
// change active profile
|
|
if( flag_profile ){
|
|
|
|
|
|
// set profile
|
|
if( !std::regex_match( string_profile, std::regex("[1-5]") ) ){
|
|
|
|
std::cout << "Wrong argument, expected 1-5.\n";
|
|
return 1;
|
|
}
|
|
|
|
if( string_profile == "1" ){ m.set_profile( mouse_m908::profile_1 ); }
|
|
if( string_profile == "2" ){ m.set_profile( mouse_m908::profile_2 ); }
|
|
if( string_profile == "3" ){ m.set_profile( mouse_m908::profile_3 ); }
|
|
if( string_profile == "4" ){ m.set_profile( mouse_m908::profile_4 ); }
|
|
if( string_profile == "5" ){ m.set_profile( mouse_m908::profile_5 ); }
|
|
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// write profile
|
|
m.write_profile();
|
|
|
|
m.close_mouse();
|
|
|
|
}
|
|
|
|
// send all macros
|
|
if( flag_macro && !flag_number ){
|
|
|
|
// load macros
|
|
int r = m.set_all_macros( string_macro );
|
|
if( r != 0 ){
|
|
std::cout << "Couldn't load macros\n";
|
|
return 1;
|
|
}
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// write macros
|
|
for( int i = 1; i < 16; i++ )
|
|
m.write_macro(i);
|
|
|
|
|
|
m.close_mouse();
|
|
|
|
}
|
|
|
|
// send individual macro
|
|
if( flag_macro && flag_number ){
|
|
|
|
|
|
// set macro and macro slot (number)
|
|
int r;
|
|
int number;
|
|
|
|
if( std::regex_match( string_number, std::regex("[0-9]+") ) ){
|
|
number = (int)stoi(string_number);
|
|
} else{
|
|
std::cout << "Wrong argument, expected 1-15.\n";
|
|
return 1;
|
|
}
|
|
|
|
if( number < 1 || number > 15 ){
|
|
std::cout << "Wrong argument, expected 1-15.\n";
|
|
return 1;
|
|
}
|
|
|
|
r = m.set_macro( number, string_macro );
|
|
if( r != 0 ){
|
|
std::cout << "Couldn't load macro\n";
|
|
return 1;
|
|
}
|
|
|
|
// open mouse
|
|
if( open_mouse_wrapper( m, flag_bus, flag_device, string_bus, string_device ) != 0 ){
|
|
return 1;
|
|
}
|
|
|
|
// write macro
|
|
m.write_macro(number);
|
|
|
|
// Repeating macros appear to be broken (in the official software as well),
|
|
// so this option is disabled for now.
|
|
// If you want to try it out, uncomment the next section and the
|
|
// appropriate lines in the commandline options section
|
|
/*if( flag_repeat ){
|
|
int repeat = (int)stoi(string_repeat);
|
|
r = m.set_macro_repeat( number, repeat );
|
|
if( r != 0 ){
|
|
std::cout << "Invalid repeats\n";
|
|
return 1;
|
|
}
|
|
m.write_macro_repeat( number );
|
|
}*/
|
|
|
|
m.close_mouse();
|
|
|
|
} else if( !flag_macro && flag_number ){
|
|
std::cout << "Misssing option, --macro and --number must be used together.\n";
|
|
}
|
|
|
|
return 0;
|
|
}
|