diff options
Diffstat (limited to 'facetracknoir/process_detector.cpp')
| -rw-r--r-- | facetracknoir/process_detector.cpp | 243 |
1 files changed, 243 insertions, 0 deletions
diff --git a/facetracknoir/process_detector.cpp b/facetracknoir/process_detector.cpp new file mode 100644 index 00000000..3ac6329b --- /dev/null +++ b/facetracknoir/process_detector.cpp @@ -0,0 +1,243 @@ +#include "process_detector.h" +#include "facetracknoir/ui.h" +#include <QList> +#include <QFileDialog> +#include <QComboBox> +#include <QString> +#include <QHash> +#include <QPushButton> + +void settings::set_game_list(const QString &game_list) +{ + QSettings settings(group::org); + settings.setValue("executable-list", game_list); +} + +QString settings::get_game_list() +{ + QSettings settings(group::org); + return settings.value("executable-list").toString(); +} + +bool settings::is_enabled() +{ + QSettings settings(group::org); + return settings.value("executable-detector-enabled", false).toBool(); +} + +void settings::set_is_enabled(bool enabled) +{ + QSettings settings(group::org); + settings.setValue("executable-detector-enabled", enabled); +} + +QHash<QString, QString> settings::split_process_names() +{ + QHash<QString, QString> ret; + QString str = get_game_list(); + QStringList pairs = str.split('|'); + for (auto pair : pairs) + { + QList<QString> tmp = pair.split(':'); + if (tmp.count() != 2) + continue; + ret[tmp[0]] = tmp[1]; + } + return ret; +} + +void BrowseButton::browse() +{ + QFileDialog dialog(this); + dialog.setFileMode(QFileDialog::ExistingFile); + QString dir_path = QFileInfo(group::ini_pathname()).absolutePath(); + QString filename = dialog.getOpenFileName( + this, + tr("Set executable name"), + dir_path, + tr("Executable (*.exe);;All Files (*)")); + MainWindow::set_working_directory(); + filename = QFileInfo(filename).fileName(); + twi->setText(filename); +} + +process_detector::add_row(QString exe_name, QString profile) +{ + int i = ui.tableWidget->rowCount(); + ui.tableWidget->insertRow(i); + + QComboBox* cb = new QComboBox(); + cb->setSizePolicy(QSizePolicy::Minimum, QSizePolicy::Maximum); + cb->addItems(group::ini_list()); + ui.tableWidget->setCellWidget(i, 1, cb); + + QTableWidgetItem* twi = new QTableWidgetItem(exe_name); + ui.tableWidget->setItem(i, 0, twi); + + { + BrowseButton* b = new BrowseButton(twi); + b->setText("..."); + QObject::connect(b, SIGNAL(pressed()), b, SLOT(browse())); + ui.tableWidget->setCellWidget(i, 2, b); + } + + cb->setCurrentText(profile); + + return i; +} + +void process_detector::add_items() +{ + auto names = s.split_process_names(); + ui.tableWidget->clearContents(); + auto keys = names.keys(); + qSort(keys); + for (auto n : keys) + add_row(n, names[n]); +} + +process_detector::process_detector(QWidget* parent) : QWidget(parent) +{ + ui.setupUi(this); + connect(ui.add, SIGNAL(pressed()), this, SLOT(add())); + connect(ui.remove, SIGNAL(pressed()), this, SLOT(remove())); + + add_items(); + + QResizeEvent e(ui.tableWidget->size(), ui.tableWidget->size()); + ui.tableWidget->resizeEvent(&e); + + settings s; + + ui.enabled->setChecked(s.is_enabled()); +} + +void process_detector::save() +{ + QString str; + + for (int i = 0; i < ui.tableWidget->rowCount(); i++) + { + auto exe = ui.tableWidget->item(i, 0)->text(); + auto profile = reinterpret_cast<QComboBox*>(ui.tableWidget->cellWidget(i, 1))->currentText(); + str += "|" + exe + ":" + profile; + } + + s.set_game_list(str); + s.set_is_enabled(ui.enabled->isChecked()); +} + +void process_detector::revert() +{ +} + +void process_detector::add() +{ + add_row(); +} + +void process_detector::remove() +{ + int r = ui.tableWidget->currentRow(); + if (r != -1) + ui.tableWidget->removeRow(r); +} + +#ifdef _WIN32 + +#include <windows.h> +#include <TlHelp32.h> + +bool process_detector_worker::should_stop() +{ + if (last_exe_name == "") + return false; + + settings s; + + if (!s.is_enabled()) + { + last_exe_name = ""; + return false; + } + + HANDLE h = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); + if (h == INVALID_HANDLE_VALUE) + return false; + + PROCESSENTRY32 e; + + if (Process32First(h, &e) == TRUE) + { + QString exe_name(e.szExeFile); + if (exe_name == last_exe_name) + { + CloseHandle(h); + return false; + } + } + + while (Process32Next(h, &e) == TRUE) + { + QString exe_name(e.szExeFile); + if (exe_name == last_exe_name) + { + CloseHandle(h); + return false; + } + } + + last_exe_name = ""; + CloseHandle(h); + + return true; +} + +bool process_detector_worker::config_to_start(QString& str) +{ + settings s; + if (!s.is_enabled()) + { + last_exe_name = ""; + return false; + } + + auto filenames = s.split_process_names(); + + HANDLE h = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0); + if (h == INVALID_HANDLE_VALUE) + return false; + + PROCESSENTRY32 e; + + if (Process32First(h, &e) == TRUE) + { + QString exe_name(e.szExeFile); + if (filenames.contains(exe_name)) + { + str = filenames[exe_name]; + last_exe_name = str; + CloseHandle(h); + return true; + } + } + + while (Process32Next(h, &e) == TRUE) + { + QString exe_name(e.szExeFile); + if (filenames.contains(exe_name)) + { + str = filenames[exe_name]; + last_exe_name = str; + CloseHandle(h); + return true; + } + } + + CloseHandle(h); + + return false; +} + + +#endif |