PowerToys/src/modules/fancyzones/FancyZonesLib/MonitorWorkAreaMap.cpp

#include "pch.h"
#include "MonitorWorkAreaMap.h"

#include <FancyZonesLib/WorkArea.h>

WorkArea* const MonitorWorkAreaMap::GetWorkArea(HMONITOR monitor) const
{
    auto iter = m_workAreaMap.find(monitor);
    if (iter != m_workAreaMap.end())
    {
        return iter->second.get();
    }

    return nullptr;
}

WorkArea* const MonitorWorkAreaMap::GetWorkAreaFromCursor() const
{
    const auto allMonitorsWorkArea = GetWorkArea(nullptr);
    if (allMonitorsWorkArea)
    {
        // First, check if there's a work area spanning all monitors (signalled by the NULL monitor handle)
        return allMonitorsWorkArea;
    }
    else
    {
        // Otherwise, look for the work area based on cursor position
        POINT cursorPoint;
        if (!GetCursorPos(&cursorPoint))
        {
            return nullptr;
        }

        return GetWorkArea(MonitorFromPoint(cursorPoint, MONITOR_DEFAULTTONULL));
    }
}

WorkArea* const MonitorWorkAreaMap::GetWorkAreaFromWindow(HWND window) const
{
    const auto allMonitorsWorkArea = GetWorkArea(nullptr);
    if (allMonitorsWorkArea)
    {
        // First, check if there's a work area spanning all monitors (signalled by the NULL monitor handle)
        return allMonitorsWorkArea;
    }
    else
    {
        // Otherwise, look for the work area based on the window's position
        HMONITOR monitor = MonitorFromWindow(window, MONITOR_DEFAULTTONULL);
        return GetWorkArea(monitor);
    }
}

const std::unordered_map<HMONITOR, std::unique_ptr<WorkArea>>& MonitorWorkAreaMap::GetAllWorkAreas() const noexcept
{
    return m_workAreaMap;
}

void MonitorWorkAreaMap::AddWorkArea(HMONITOR monitor, std::unique_ptr<WorkArea> workArea)
{
    m_workAreaMap.insert({ monitor, std::move(workArea) });
}

FancyZonesDataTypes::WorkAreaId MonitorWorkAreaMap::GetParent(HMONITOR monitor) const
{
    if (m_workAreaParents.contains(monitor))
    {
        return m_workAreaParents.at(monitor);
    }
    
    return FancyZonesDataTypes::WorkAreaId{};
}

void MonitorWorkAreaMap::SaveParentIds()
{
    m_workAreaParents.clear();
    for (const auto& [monitor, workArea] : m_workAreaMap)
    {
        if (workArea)
        {
            m_workAreaParents.insert({ monitor, workArea->UniqueId() });
        }
    }
}

void MonitorWorkAreaMap::Clear() noexcept
{
    m_workAreaMap.clear();
}