Monitor on/off event listener (Win)

awawa-dev · Feb 2, 2024 · 4740755 · 4740755
1 parent 611e5ae
commit 4740755
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 8 deletions.
diff --git a/include/utils/Components.h b/include/utils/Components.h
@@ -97,5 +97,5 @@ namespace hyperhdr
 		return COMP_INVALID;
 	}
 
-	enum SystemComponent { SUSPEND, LOCKER };
+	enum SystemComponent { SUSPEND, LOCKER, MONITOR };
 }
diff --git a/sources/base/HyperHdrManager.cpp b/sources/base/HyperHdrManager.cpp
@@ -176,8 +176,9 @@ void HyperHdrManager::toggleGrabbersAllInstances(bool pause)
 
 void HyperHdrManager::hibernate(bool wakeUp, hyperhdr::SystemComponent source)
 {
-	if (source == hyperhdr::SystemComponent::LOCKER)
+	if (source == hyperhdr::SystemComponent::LOCKER || source == hyperhdr::SystemComponent::MONITOR)
 	{
+		Debug(_log, "OS event: %s", (source == hyperhdr::SystemComponent::LOCKER) ? ((wakeUp) ? "OS unlocked" : "OS locked") : ((wakeUp) ? "Monitor On" : "Monitor Off"));
 		bool _hasEffect = false;
 		for (const auto& instance : _runningInstances)
 		{

diff --git a/sources/hyperhdr/SuspendHandlerWindows.cpp b/sources/hyperhdr/SuspendHandlerWindows.cpp
@@ -41,7 +41,10 @@
 
 #pragma comment (lib, "WtsApi32.Lib")
 
-SuspendHandler::SuspendHandler(bool sessionLocker)
+SuspendHandler::SuspendHandler(bool sessionLocker):
+	_notifyHandle(NULL),
+	_notifyMonitorHandle(NULL),
+	_sessionLocker(sessionLocker)
 {
 	auto handle = reinterpret_cast<HWND> (_widget.winId());
 	_notifyHandle = RegisterSuspendResumeNotification(handle, DEVICE_NOTIFY_WINDOW_HANDLE);
@@ -50,15 +53,19 @@ SuspendHandler::SuspendHandler(bool sessionLocker)
 		std::cout << "COULD NOT REGISTER SLEEP HANDLER!" << std::endl;
 	else
 		std::cout << "Sleep handler registered!" << std::endl;
-
-	_sessionLocker = sessionLocker;
+
 	if (_sessionLocker)
 	{
-		if (WTSRegisterSessionNotification(handle, NOTIFY_FOR_THIS_SESSION))
+		_notifyMonitorHandle = RegisterPowerSettingNotification(handle, &GUID_SESSION_DISPLAY_STATUS, DEVICE_NOTIFY_WINDOW_HANDLE);
+
+		if (_notifyMonitorHandle != NULL && WTSRegisterSessionNotification(handle, NOTIFY_FOR_THIS_SESSION))
 			std::cout << "Session handler registered!" << std::endl;
 		else
 		{
-			std::cout << "COULD NOT REGISTER SESSION HANDLER!" << std::endl;
+			std::cout << "COULD NOT REGISTER SESSION HANDLER!" << std::endl;			
+			if (_notifyMonitorHandle != NULL)
+				UnregisterSuspendResumeNotification(_notifyMonitorHandle);
+			_notifyMonitorHandle = NULL;
 			_sessionLocker = false;
 		}
 	}
@@ -75,6 +82,13 @@ SuspendHandler::~SuspendHandler()
 
 	if (_sessionLocker)
 	{
+		if (_notifyMonitorHandle != NULL)
+		{
+			UnregisterSuspendResumeNotification(_notifyMonitorHandle);
+			std::cout << "Monitor state handler deregistered!" << std::endl;
+		}
+		_notifyMonitorHandle = NULL;
+
 		auto handle = reinterpret_cast<HWND> (_widget.winId());
 		WTSUnRegisterSessionNotification(handle);
 	}
@@ -101,6 +115,18 @@ bool SuspendHandler::nativeEventFilter(const QByteArray& eventType, void* messag
 				return true;
 				break;
 		}
+
+		if (_sessionLocker && msg->wParam == PBT_POWERSETTINGCHANGE && msg->lParam != 0)
+		{
+			POWERBROADCAST_SETTING* s = reinterpret_cast<POWERBROADCAST_SETTING*>(msg->lParam);			
+			if (s != nullptr && s->PowerSetting == GUID_SESSION_DISPLAY_STATUS && s->DataLength > 0)
+			{
+				if (s->Data[0] == 1)
+					emit SignalHibernate(true, hyperhdr::SystemComponent::MONITOR);
+				else if (s->Data[0] == 0)
+					emit SignalHibernate(false, hyperhdr::SystemComponent::MONITOR);
+			}
+		}
 	}
 
 	if (_sessionLocker)

diff --git a/sources/hyperhdr/SuspendHandlerWindows.h b/sources/hyperhdr/SuspendHandlerWindows.h
@@ -41,7 +41,7 @@ class SuspendHandler : public QObject, public QAbstractNativeEventFilter {
 	Q_OBJECT
 
 	QWidget			_widget;
-	HPOWERNOTIFY	_notifyHandle;
+	HPOWERNOTIFY	_notifyHandle, _notifyMonitorHandle;
 	bool			_sessionLocker;
 
 signals: