#include "OpenRGBDialog2.h" #include "OpenRGBDevicePage.h" #include "OpenRGBDeviceInfoPage.h" #include "OpenRGBServerInfoPage.h" #include "OpenRGBProfileSaveDialog.h" #include "ResourceManager.h" #include #include #include #include using namespace Ui; static QString GetIconString(device_type type, bool dark) { /*-----------------------------------------------------*\ | Return the icon filename string for the given device | | type value | \*-----------------------------------------------------*/ QString filename; switch(type) { case DEVICE_TYPE_MOTHERBOARD: filename = "motherboard"; break; case DEVICE_TYPE_DRAM: filename = "dram"; break; case DEVICE_TYPE_GPU: filename = "gpu"; break; case DEVICE_TYPE_COOLER: filename = "fan"; break; case DEVICE_TYPE_LEDSTRIP: filename = "ledstrip"; break; case DEVICE_TYPE_KEYBOARD: filename = "keyboard"; break; case DEVICE_TYPE_MOUSE: filename = "mouse"; break; case DEVICE_TYPE_MOUSEMAT: filename = "mousemat"; break; case DEVICE_TYPE_HEADSET: filename = "headset"; break; case DEVICE_TYPE_HEADSET_STAND: filename = "headsetstand"; break; case DEVICE_TYPE_GAMEPAD: filename = "gamepad"; break; case DEVICE_TYPE_LIGHT: filename = "light"; break; default: filename = "unknown"; break; } if(dark) { filename += "_dark"; } filename += ".png"; return filename; } static void UpdateDeviceListCallback(void * this_ptr) { OpenRGBDialog2 * this_obj = (OpenRGBDialog2 *)this_ptr; QMetaObject::invokeMethod(this_obj, "onDeviceListUpdated", Qt::QueuedConnection); } static void UpdateDetectionProgressCallback(void * this_ptr) { OpenRGBDialog2 * this_obj = (OpenRGBDialog2 *)this_ptr; QMetaObject::invokeMethod(this_obj, "onDetectionProgressUpdated", Qt::QueuedConnection); } OpenRGBDialog2::OpenRGBDialog2(std::vector& bus, std::vector& control, QWidget *parent) : QMainWindow(parent), busses(bus), controllers(control), ui(new OpenRGBDialog2Ui) { ui->setupUi(this); /*-----------------------------------------------------*\ | Set window icon | \*-----------------------------------------------------*/ QIcon logo(":OpenRGB.png"); setWindowIcon(logo); /*-----------------------------------------------------*\ | Register detection progress callback with resource | | manager | \*-----------------------------------------------------*/ ResourceManager::get()->RegisterDetectionProgressCallback(UpdateDetectionProgressCallback, this); ResourceManager::get()->RegisterDeviceListChangeCallback(UpdateDeviceListCallback, this); /*-----------------------------------------------------*\ | Initialize page pointers | \*-----------------------------------------------------*/ ClientInfoPage = NULL; SMBusToolsPage = NULL; SoftInfoPage = NULL; onDetectionProgressUpdated(); ui->DetectionProgressBar->setRange(0, 100); ui->DetectionProgressBar->setValue(0); ui->DetectionProgressBar->setTextVisible(true); ui->DetectionProgressBar->setFormat(""); ui->DetectionProgressBar->setAlignment(Qt::AlignCenter); /*-----------------------------------------------------*\ | Set up tray icon menu | \*-----------------------------------------------------*/ QMenu* trayIconMenu = new QMenu( this ); trayIcon = new QSystemTrayIcon(this); QAction* actionShowHide = new QAction("Show/Hide", this); connect(actionShowHide, SIGNAL(triggered()), this, SLOT(on_ShowHide())); trayIconMenu->addAction(actionShowHide); profileMenu = new QMenu("Profiles", this); trayIconMenu->addMenu(profileMenu); QMenu* quickColorsMenu = new QMenu("Quick Colors", this); QAction* actionQuickRed = new QAction("Red", this); connect(actionQuickRed, SIGNAL(triggered()), this, SLOT(on_QuickRed())); quickColorsMenu->addAction(actionQuickRed); QAction* actionQuickYellow = new QAction("Yellow", this); connect(actionQuickYellow, SIGNAL(triggered()), this, SLOT(on_QuickYellow())); quickColorsMenu->addAction(actionQuickYellow); QAction* actionQuickGreen = new QAction("Green", this); connect(actionQuickGreen, SIGNAL(triggered()), this, SLOT(on_QuickGreen())); quickColorsMenu->addAction(actionQuickGreen); QAction* actionQuickCyan = new QAction("Cyan", this); connect(actionQuickCyan, SIGNAL(triggered()), this, SLOT(on_QuickCyan())); quickColorsMenu->addAction(actionQuickCyan); QAction* actionQuickBlue = new QAction("Blue", this); connect(actionQuickBlue, SIGNAL(triggered()), this, SLOT(on_QuickBlue())); quickColorsMenu->addAction(actionQuickBlue); QAction* actionQuickMagenta = new QAction("Magenta", this); connect(actionQuickMagenta, SIGNAL(triggered()), this, SLOT(on_QuickMagenta())); quickColorsMenu->addAction(actionQuickMagenta); QAction* actionQuickWhite = new QAction("White", this); connect(actionQuickWhite, SIGNAL(triggered()), this, SLOT(on_QuickWhite())); quickColorsMenu->addAction(actionQuickWhite); trayIconMenu->addMenu(quickColorsMenu); QAction* actionLightsOff = new QAction("Lights Off", this); connect(actionLightsOff, SIGNAL(triggered()), this, SLOT(on_LightsOff())); trayIconMenu->addAction(actionLightsOff); QAction* actionExit = new QAction( "Exit", this ); connect( actionExit, SIGNAL( triggered() ), this, SLOT( on_Exit() )); trayIconMenu->addAction(actionExit); trayIcon->setIcon(logo); trayIcon->setToolTip("OpenRGB"); trayIcon->setContextMenu(trayIconMenu); trayIcon->show(); darkTheme = palette().window().color().value() < 127; // Adjust /*-----------------------------------------------------*\ | Update the profile list | \*-----------------------------------------------------*/ UpdateProfileList(); /*-----------------------------------------------------*\ | Update the device list and make sure the | | ProgressBar gets a proper value | \*-----------------------------------------------------*/ UpdateDevicesList(); /*-----------------------------------------------------*\ | Add Server Tab | \*-----------------------------------------------------*/ AddServerTab(); /*-----------------------------------------------------*\ | Add Client Tab | \*-----------------------------------------------------*/ AddClientTab(); /*-----------------------------------------------------*\ | Add the Software Info page | \*-----------------------------------------------------*/ AddSoftwareInfoPage(); /*-----------------------------------------------------*\ | Add the SMBus Tools page if enabled | \*-----------------------------------------------------*/ if(ShowI2CTools) { AddI2CToolsPage(); } } OpenRGBDialog2::~OpenRGBDialog2() { delete ui; } void OpenRGBDialog2::closeEvent(QCloseEvent *event) { ResourceManager::get()->WaitForDeviceDetection(); event->accept(); } void OpenRGBDialog2::AddSoftwareInfoPage() { /*-----------------------------------------------------*\ | Create the Software Information page | \*-----------------------------------------------------*/ SoftInfoPage = new OpenRGBSoftwareInfoPage(); ui->InformationTabBar->addTab(SoftInfoPage, ""); QString SoftwareLabelString = "
Software
"; QLabel *SoftwareTabLabel = new QLabel(); SoftwareTabLabel->setText(SoftwareLabelString); SoftwareTabLabel->setIndent(20); SoftwareTabLabel->setGeometry(0, 0, 200, 20); ui->InformationTabBar->tabBar()->setTabButton(ui->InformationTabBar->tabBar()->count() - 1, QTabBar::LeftSide, SoftwareTabLabel); } void OpenRGBDialog2::AddI2CToolsPage() { ShowI2CTools = true; /*-----------------------------------------------------*\ | Create the I2C Tools page if it doesn't exist yet | \*-----------------------------------------------------*/ SMBusToolsPage = new OpenRGBSystemInfoPage(busses); /*-----------------------------------------------------*\ | Create the I2C Tools tab in the Information bar | \*-----------------------------------------------------*/ ui->InformationTabBar->addTab(SMBusToolsPage, ""); QString SMBusToolsLabelString = "
SMBus Tools
"; QLabel *SMBusToolsTabLabel = new QLabel(); SMBusToolsTabLabel->setText(SMBusToolsLabelString); SMBusToolsTabLabel->setIndent(20); SMBusToolsTabLabel->setGeometry(0, 0, 200, 20); ui->InformationTabBar->tabBar()->setTabButton(ui->InformationTabBar->tabBar()->count() - 1, QTabBar::LeftSide, SMBusToolsTabLabel); } void OpenRGBDialog2::AddClientTab() { /*-----------------------------------------------------*\ | Add client information tab if it doesn't exist yet | \*-----------------------------------------------------*/ if(ClientInfoPage == NULL) { ClientInfoPage = new OpenRGBClientInfoPage(controllers); ui->MainTabBar->addTab(ClientInfoPage, "SDK Client"); /*-----------------------------------------------------*\ | Connect the page's Set All button to the Set All slot | \*-----------------------------------------------------*/ connect(ClientInfoPage, SIGNAL(ClientListUpdated()), this, SLOT(on_ClientListUpdated())); } } void OpenRGBDialog2::AddClient(NetworkClient* new_client) { /*-----------------------------------------------------*\ | Add a client to the client information page | \*-----------------------------------------------------*/ if(ClientInfoPage != NULL) { ClientInfoPage->AddClient(new_client); } } void OpenRGBDialog2::AddServerTab() { /*-----------------------------------------------------*\ | Add server information tab if there is a server | \*-----------------------------------------------------*/ OpenRGBServerInfoPage *ServerInfoPage = new OpenRGBServerInfoPage(ResourceManager::get()->GetServer()); ui->MainTabBar->addTab(ServerInfoPage, "SDK Server"); } void OpenRGBDialog2::ClearDevicesList() { for(int tab_idx = 0; tab_idx < ui->InformationTabBar->count(); tab_idx++) { delete ui->DevicesTabBar->widget(tab_idx); } ui->DevicesTabBar->clear(); for(int tab_idx = 0; tab_idx < ui->InformationTabBar->count(); tab_idx++) { delete ui->InformationTabBar->widget(tab_idx); } ui->InformationTabBar->clear(); } void OpenRGBDialog2::UpdateDevicesList() { /*-----------------------------------------------------*\ | Loop through each controller in the list. | \*-----------------------------------------------------*/ for(unsigned int controller_idx = 0; controller_idx < controllers.size(); controller_idx++) { /*-----------------------------------------------------*\ | Loop through each tab in the devices tab bar | \*-----------------------------------------------------*/ bool found = false; for(unsigned int tab_idx = 0; tab_idx < ui->DevicesTabBar->count(); tab_idx++) { OpenRGBDevicePage* page = (OpenRGBDevicePage*) ui->DevicesTabBar->widget(tab_idx); /*-----------------------------------------------------*\ | If the current tab matches the current controller, | | move the tab to the correct position | \*-----------------------------------------------------*/ if(controllers[controller_idx] == page->GetController()) { found = true; ui->DevicesTabBar->tabBar()->moveTab(tab_idx, controller_idx); break; } } if(!found) { /*-----------------------------------------------------*\ | The controller does not have a tab already created | | Create a new tab and move it to the correct position | \*-----------------------------------------------------*/ OpenRGBDevicePage *NewPage = new OpenRGBDevicePage(controllers[controller_idx]); ui->DevicesTabBar->addTab(NewPage, ""); /*-----------------------------------------------------*\ | Connect the page's Set All button to the Set All slot | \*-----------------------------------------------------*/ connect(NewPage, SIGNAL(SetAllDevices(unsigned char, unsigned char, unsigned char)), this, SLOT(on_SetAllDevices(unsigned char, unsigned char, unsigned char))); /*-----------------------------------------------------*\ | Connect the page's Resize signal to the Save Size slot| \*-----------------------------------------------------*/ connect(NewPage, SIGNAL(SaveSizeProfile()), this, SLOT(on_SaveSizeProfile())); /*-----------------------------------------------------*\ | Use Qt's HTML capabilities to display both icon and | | text in the tab label. Choose icon based on device | | type and append device name string. | \*-----------------------------------------------------*/ QString NewLabelString = "
" + QString::fromStdString(controllers[controller_idx]->name) + "
"; QLabel *NewTabLabel = new QLabel(); NewTabLabel->setText(NewLabelString); NewTabLabel->setIndent(20); NewTabLabel->setGeometry(0, 0, 200, 20); ui->DevicesTabBar->tabBar()->setTabButton(ui->DevicesTabBar->count() - 1, QTabBar::LeftSide, NewTabLabel); /*-----------------------------------------------------*\ | Now move the new tab to the correct position | \*-----------------------------------------------------*/ ui->DevicesTabBar->tabBar()->moveTab(ui->DevicesTabBar->count() - 1, controller_idx); } /*-----------------------------------------------------*\ | Loop through each tab in the information tab bar | \*-----------------------------------------------------*/ found = false; for(unsigned int tab_idx = 0; tab_idx < ui->InformationTabBar->count(); tab_idx++) { /*-----------------------------------------------------*\ | If type is a device info page, check it | \*-----------------------------------------------------*/ std::string type_str = ui->InformationTabBar->widget(tab_idx)->metaObject()->className(); if(type_str == "Ui::OpenRGBDeviceInfoPage") { OpenRGBDeviceInfoPage* page = (OpenRGBDeviceInfoPage*) ui->InformationTabBar->widget(tab_idx); /*-----------------------------------------------------*\ | If the current tab matches the current controller, | | move the tab to the correct position | \*-----------------------------------------------------*/ if(controllers[controller_idx] == page->GetController()) { found = true; ui->InformationTabBar->tabBar()->moveTab(tab_idx, controller_idx); break; } } } if(!found) { /*-----------------------------------------------------*\ | The controller does not have a tab already created | | Create a new tab and move it to the correct position | \*-----------------------------------------------------*/ OpenRGBDeviceInfoPage *NewPage = new OpenRGBDeviceInfoPage(controllers[controller_idx]); ui->InformationTabBar->addTab(NewPage, ""); /*-----------------------------------------------------*\ | Use Qt's HTML capabilities to display both icon and | | text in the tab label. Choose icon based on device | | type and append device name string. | \*-----------------------------------------------------*/ QString NewLabelString = "
" + QString::fromStdString(controllers[controller_idx]->name) + "
"; QLabel *NewTabLabel = new QLabel(); NewTabLabel->setText(NewLabelString); NewTabLabel->setIndent(20); NewTabLabel->setGeometry(0, 0, 200, 20); ui->InformationTabBar->tabBar()->setTabButton(ui->InformationTabBar->count() - 1, QTabBar::LeftSide, NewTabLabel); /*-----------------------------------------------------*\ | Now move the new tab to the correct position | \*-----------------------------------------------------*/ ui->InformationTabBar->tabBar()->moveTab(ui->InformationTabBar->count() - 1, controller_idx); } } /*-----------------------------------------------------*\ | Remove all remaining device tabs | \*-----------------------------------------------------*/ unsigned int tab_count = ui->DevicesTabBar->count(); for(unsigned int tab_idx = controllers.size(); tab_idx < tab_count; tab_idx++) { ui->DevicesTabBar->removeTab(ui->DevicesTabBar->count() - 1); } bool found = true; while(found) { found = false; /*-----------------------------------------------------*\ | Remove all remaining device information tabs, leaving | | other information tabs alone | \*-----------------------------------------------------*/ for(unsigned int tab_idx = controllers.size(); tab_idx < ui->InformationTabBar->count(); tab_idx++) { std::string type_str = ui->InformationTabBar->widget(tab_idx)->metaObject()->className(); if(type_str == "Ui::OpenRGBDeviceInfoPage") { found = true; ui->InformationTabBar->removeTab(tab_idx); break; } } } } void OpenRGBDialog2::UpdateProfileList() { ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*-----------------------------------------------------*\ | Clear profile combo box and tray icon menu | \*-----------------------------------------------------*/ ui->ProfileBox->clear(); profileMenu->clear(); for(std::size_t profile_index = 0; profile_index < profile_manager->profile_list.size(); profile_index++) { /*-----------------------------------------------------*\ | Fill in profile combo box | \*-----------------------------------------------------*/ ui->ProfileBox->addItem(profile_manager->profile_list[profile_index].c_str()); /*-----------------------------------------------------*\ | Fill in profile tray icon menu | \*-----------------------------------------------------*/ QAction* actionProfileSelected = new QAction(profile_manager->profile_list[profile_index].c_str(), this); actionProfileSelected->setObjectName(profile_manager->profile_list[profile_index].c_str()); connect(actionProfileSelected, SIGNAL(triggered()), this, SLOT(on_ProfileSelected())); profileMenu->addAction(actionProfileSelected); } } } void OpenRGBDialog2::on_Exit() { trayIcon->hide(); close(); } void OpenRGBDialog2::on_LightsOff() { on_SetAllDevices(0x00, 0x00, 0x00); } void OpenRGBDialog2::on_QuickRed() { on_SetAllDevices(0xFF, 0x00, 0x00); } void OpenRGBDialog2::on_QuickYellow() { on_SetAllDevices(0xFF, 0xFF, 0x00); } void OpenRGBDialog2::on_QuickGreen() { on_SetAllDevices(0x00, 0xFF, 0x00); } void OpenRGBDialog2::on_QuickCyan() { on_SetAllDevices(0x00, 0xFF, 0xFF); } void OpenRGBDialog2::on_QuickBlue() { on_SetAllDevices(0x00, 0x00, 0xFF); } void OpenRGBDialog2::on_QuickMagenta() { on_SetAllDevices(0xFF, 0x00, 0xFF); } void OpenRGBDialog2::on_QuickWhite() { on_SetAllDevices(0xFF, 0xFF, 0xFF); } void OpenRGBDialog2::on_ClientListUpdated() { UpdateDevicesList(); } void OpenRGBDialog2::onDeviceListUpdated() { UpdateDevicesList(); } void OpenRGBDialog2::onDetectionProgressUpdated() { ui->DetectionProgressBar->setValue(ResourceManager::get()->GetDetectionPercent()); ui->DetectionProgressBar->setFormat(QString::fromStdString(ResourceManager::get()->GetDetectionString())); if(ResourceManager::get()->GetDetectionPercent() == 100) { SetDetectionViewState(false); } else { SetDetectionViewState(true); } } void OpenRGBDialog2::on_SetAllDevices(unsigned char red, unsigned char green, unsigned char blue) { for(int device = 0; device < ui->DevicesTabBar->count(); device++) { qobject_cast(ui->DevicesTabBar->widget(device))->SetCustomMode(red, green, blue); } } void OpenRGBDialog2::on_SaveSizeProfile() { ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*---------------------------------------------------------*\ | Save the profile | \*---------------------------------------------------------*/ profile_manager->SaveProfile("sizes.ors"); } } void OpenRGBDialog2::on_ShowHide() { if(isHidden()) { show(); } else { hide(); } } void Ui::OpenRGBDialog2::on_ProfileSelected() { ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*---------------------------------------------------------*\ | Get the profile filename from the selected object | \*---------------------------------------------------------*/ std::string profile_name = QObject::sender()->objectName().toStdString(); /*---------------------------------------------------------*\ | Load the profile | \*---------------------------------------------------------*/ if(profile_manager->LoadProfile(profile_name)) { for(int device = 0; device < ui->DevicesTabBar->count(); device++) { qobject_cast(ui->DevicesTabBar->widget(device))->UpdateDevice(); } } } } void Ui::OpenRGBDialog2::on_ButtonSaveProfile_clicked() { OpenRGBProfileSaveDialog dialog; ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*---------------------------------------------------------*\ | Open Profile Name Dialog | \*---------------------------------------------------------*/ std::string profile_name = dialog.show(); /*---------------------------------------------------------*\ | Extension .orp - OpenRgb Profile | \*---------------------------------------------------------*/ std::string filename = profile_name + ".orp"; /*---------------------------------------------------------*\ | Save the profile | \*---------------------------------------------------------*/ if(profile_manager->SaveProfile(filename)) { UpdateProfileList(); } } } void Ui::OpenRGBDialog2::on_ButtonLoadProfile_clicked() { ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*---------------------------------------------------------*\ | Get the profile filename from the profiles list | \*---------------------------------------------------------*/ std::string profile_name = ui->ProfileBox->currentText().toStdString(); /*---------------------------------------------------------*\ | Load the profile | \*---------------------------------------------------------*/ if(profile_manager->LoadProfile(profile_name)) { for(int device = 0; device < ui->DevicesTabBar->count(); device++) { qobject_cast(ui->DevicesTabBar->widget(device))->UpdateDevice(); } } } } void Ui::OpenRGBDialog2::on_ButtonDeleteProfile_clicked() { ProfileManager* profile_manager = ResourceManager::get()->GetProfileManager(); if(profile_manager != NULL) { /*---------------------------------------------------------*\ | Get the profile filename from the profiles list | \*---------------------------------------------------------*/ std::string profile_name = ui->ProfileBox->currentText().toStdString(); /*---------------------------------------------------------*\ | Confirm we want to delete the profile | \*---------------------------------------------------------*/ QMessageBox::StandardButton reply; reply = QMessageBox::question(this, "Delete Profile", "Do you really want to delete this profile?", QMessageBox::Yes|QMessageBox::No); /*---------------------------------------------------------*\ | Load the profile | \*---------------------------------------------------------*/ if(reply == QMessageBox::Yes) { profile_manager->DeleteProfile(profile_name); UpdateProfileList(); } } } void Ui::OpenRGBDialog2::on_ButtonToggleDeviceView_clicked() { if(device_view_showing) { for(int device = 0; device < ui->DevicesTabBar->count(); device++) { qobject_cast(ui->DevicesTabBar->widget(device))->HideDeviceView(); } device_view_showing = false; } else { for(int device = 0; device < ui->DevicesTabBar->count(); device++) { qobject_cast(ui->DevicesTabBar->widget(device))->ShowDeviceView(); } device_view_showing = true; } } void Ui::OpenRGBDialog2::on_ButtonStopDetection_clicked() { /*---------------------------------------------------------*\ | Notify the detection thread that it has to die | \*---------------------------------------------------------*/ ResourceManager::get()->StopDeviceDetection(); /*---------------------------------------------------------*\ | Pretend we're done already by hiding the progress bar | \*---------------------------------------------------------*/ SetDetectionViewState(false); } void Ui::OpenRGBDialog2::SetDetectionViewState(bool detection_showing) { if(detection_showing) { /*---------------------------------------------------------*\ | Show the detection progress and hide the normal buttons | \*---------------------------------------------------------*/ ui->ButtonToggleDeviceView->setVisible(false); ui->ButtonRescan->setVisible(false); ui->ButtonLoadProfile->setVisible(false); ui->ButtonSaveProfile->setVisible(false); ui->ButtonDeleteProfile->setVisible(false); ui->ProfileBox->setVisible(false); ui->DetectionProgressBar->setVisible(true); ui->DetectionProgressLabel->setVisible(true); ui->ButtonStopDetection->setVisible(true); } else { /*---------------------------------------------------------*\ | Hide the detection progress and show the normal buttons | \*---------------------------------------------------------*/ ui->DetectionProgressBar->setVisible(false); ui->DetectionProgressLabel->setVisible(false); ui->ButtonStopDetection->setVisible(false); ui->ButtonToggleDeviceView->setVisible(true); ui->ButtonRescan->setVisible(true); ui->ButtonLoadProfile->setVisible(true); ui->ButtonSaveProfile->setVisible(true); ui->ButtonDeleteProfile->setVisible(true); ui->ProfileBox->setVisible(true); } } void Ui::OpenRGBDialog2::on_ButtonRescan_clicked() { SetDetectionViewState(true); ResourceManager::get()->DetectDevices(); }