mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-21 15:27:55 +08:00
106 lines
3.1 KiB
C++
106 lines
3.1 KiB
C++
#include "pch.h"
|
|
|
|
struct Zone : winrt::implements<Zone, IZone>
|
|
{
|
|
public:
|
|
Zone(RECT zoneRect) :
|
|
m_zoneRect(zoneRect)
|
|
{
|
|
}
|
|
|
|
IFACEMETHODIMP_(RECT) GetZoneRect() noexcept { return m_zoneRect; }
|
|
IFACEMETHODIMP_(bool) IsEmpty() noexcept { return m_windows.empty(); };
|
|
IFACEMETHODIMP_(bool) ContainsWindow(HWND window) noexcept;
|
|
IFACEMETHODIMP_(void) AddWindowToZone(HWND window, HWND zoneWindow, bool stampZone) noexcept;
|
|
IFACEMETHODIMP_(void) RemoveWindowFromZone(HWND window, bool restoreSize) noexcept;
|
|
IFACEMETHODIMP_(void) SetId(size_t id) noexcept { m_id = id; }
|
|
IFACEMETHODIMP_(size_t) Id() noexcept { return m_id; }
|
|
|
|
private:
|
|
void SizeWindowToZone(HWND window, HWND zoneWindow) noexcept;
|
|
void StampZone(HWND window, bool stamp) noexcept;
|
|
|
|
RECT m_zoneRect{};
|
|
size_t m_id{};
|
|
std::map<HWND, RECT> m_windows{};
|
|
};
|
|
|
|
IFACEMETHODIMP_(bool) Zone::ContainsWindow(HWND window) noexcept
|
|
{
|
|
return (m_windows.find(window) != m_windows.end());
|
|
}
|
|
|
|
IFACEMETHODIMP_(void) Zone::AddWindowToZone(HWND window, HWND zoneWindow, bool stampZone) noexcept
|
|
{
|
|
WINDOWPLACEMENT placement;
|
|
::GetWindowPlacement(window, &placement);
|
|
::GetWindowRect(window, &placement.rcNormalPosition);
|
|
m_windows.emplace(std::pair<HWND, RECT>(window, placement.rcNormalPosition));
|
|
|
|
SizeWindowToZone(window, zoneWindow);
|
|
if (stampZone)
|
|
{
|
|
StampZone(window, true);
|
|
}
|
|
}
|
|
|
|
IFACEMETHODIMP_(void) Zone::RemoveWindowFromZone(HWND window, bool restoreSize) noexcept
|
|
{
|
|
auto iter = m_windows.find(window);
|
|
if (iter != m_windows.end())
|
|
{
|
|
m_windows.erase(iter);
|
|
StampZone(window, false);
|
|
}
|
|
}
|
|
|
|
void Zone::SizeWindowToZone(HWND window, HWND zoneWindow) noexcept
|
|
{
|
|
// Take care of 1px border
|
|
RECT zoneRect = m_zoneRect;
|
|
|
|
RECT windowRect{};
|
|
::GetWindowRect(window, &windowRect);
|
|
|
|
RECT frameRect{};
|
|
// Failure is expected on down level systems.
|
|
if (SUCCEEDED(DwmGetWindowAttribute(window, DWMWA_EXTENDED_FRAME_BOUNDS, &frameRect, sizeof(frameRect))))
|
|
{
|
|
zoneRect.bottom -= (frameRect.bottom - windowRect.bottom);
|
|
zoneRect.right -= (frameRect.right - windowRect.right);
|
|
zoneRect.left -= (frameRect.left - windowRect.left);
|
|
}
|
|
|
|
// Map to screen coords
|
|
MapWindowRect(zoneWindow, nullptr, &zoneRect);
|
|
|
|
MONITORINFO mi{sizeof(mi)};
|
|
if (GetMonitorInfoW(MonitorFromWindow(zoneWindow, MONITOR_DEFAULTTONEAREST), &mi))
|
|
{
|
|
OffsetRect(&zoneRect, mi.rcMonitor.left - mi.rcWork.left, mi.rcMonitor.top - mi.rcWork.top);
|
|
}
|
|
|
|
WINDOWPLACEMENT placement;
|
|
::GetWindowPlacement(window, &placement);
|
|
placement.rcNormalPosition = zoneRect;
|
|
placement.flags |= WPF_ASYNCWINDOWPLACEMENT;
|
|
placement.showCmd = SW_RESTORE | SW_SHOWNA;
|
|
::SetWindowPlacement(window, &placement);
|
|
}
|
|
|
|
void Zone::StampZone(HWND window, bool stamp) noexcept
|
|
{
|
|
if (stamp)
|
|
{
|
|
SetProp(window, ZONE_STAMP, reinterpret_cast<HANDLE>(m_id));
|
|
}
|
|
else
|
|
{
|
|
RemoveProp(window, ZONE_STAMP);
|
|
}
|
|
}
|
|
|
|
winrt::com_ptr<IZone> MakeZone(RECT zoneRect) noexcept
|
|
{
|
|
return winrt::make_self<Zone>(zoneRect);
|
|
} |