374 lines
9.8 KiB
C++
374 lines
9.8 KiB
C++
#include "taskbar_manager.h"
|
||
#include "dwm_util.h"
|
||
#include <shobjidl.h>
|
||
|
||
using namespace ui;
|
||
|
||
TaskbarTabItem::TaskbarTabItem(ui::Control *bind_control)
|
||
{
|
||
ASSERT(NULL != bind_control);
|
||
bind_control_ = bind_control;
|
||
is_win7_or_greater_ = IsWindows7OrGreater();
|
||
taskbar_manager_ = NULL;
|
||
}
|
||
|
||
ui::Control* TaskbarTabItem::GetBindControl()
|
||
{
|
||
return bind_control_;
|
||
}
|
||
|
||
std::string& TaskbarTabItem::GetId()
|
||
{
|
||
return id_;
|
||
}
|
||
|
||
void TaskbarTabItem::Init(const std::wstring &taskbar_title, const std::string &id)
|
||
{
|
||
id_ = id;
|
||
if (!is_win7_or_greater_)
|
||
return;
|
||
|
||
Create(NULL, taskbar_title.c_str(), WS_OVERLAPPED, 0, false);
|
||
|
||
HRESULT ret = S_OK;
|
||
BOOL truth = TRUE;
|
||
ret |= DwmSetWindowAttribute(m_hWnd, DWMWA_HAS_ICONIC_BITMAP, &truth, sizeof(truth));
|
||
ret |= DwmSetWindowAttribute(m_hWnd, DWMWA_FORCE_ICONIC_REPRESENTATION, &truth, sizeof(truth));
|
||
if (ret != S_OK)
|
||
{
|
||
is_win7_or_greater_ = false;
|
||
}
|
||
}
|
||
|
||
void TaskbarTabItem::UnInit()
|
||
{
|
||
if (NULL != m_hWnd)
|
||
DestroyWindow(m_hWnd);
|
||
}
|
||
|
||
void TaskbarTabItem::SetTaskbarTitle(const std::wstring &title)
|
||
{
|
||
::SetWindowTextW(m_hWnd, title.c_str());
|
||
}
|
||
|
||
void TaskbarTabItem::SetTaskbarManager(TaskbarManager *taskbar_manager)
|
||
{
|
||
taskbar_manager_ = taskbar_manager;
|
||
}
|
||
|
||
TaskbarManager* TaskbarTabItem::GetTaskbarManager()
|
||
{
|
||
return taskbar_manager_;
|
||
}
|
||
|
||
bool TaskbarTabItem::InvalidateTab()
|
||
{
|
||
if (!is_win7_or_greater_ || NULL == taskbar_manager_)
|
||
return false;
|
||
|
||
return (S_OK == DwmInvalidateIconicBitmaps(this->GetHWND()));
|
||
}
|
||
|
||
void TaskbarTabItem::OnSendThumbnail(int width, int height)
|
||
{
|
||
if (!is_win7_or_greater_ || NULL == taskbar_manager_)
|
||
return;
|
||
|
||
HBITMAP bitmap = taskbar_manager_->GenerateBindControlBitmap(bind_control_, width, height);
|
||
DwmSetIconicThumbnail(m_hWnd, bitmap, 0);
|
||
|
||
DeleteObject(bitmap);
|
||
}
|
||
|
||
void TaskbarTabItem::OnSendPreview()
|
||
{
|
||
if (!is_win7_or_greater_ || NULL == taskbar_manager_)
|
||
return;
|
||
|
||
HBITMAP bitmap = taskbar_manager_->GenerateBindControlBitmapWithForm(bind_control_);
|
||
DwmSetIconicLivePreviewBitmap(m_hWnd, bitmap, NULL, 0);
|
||
|
||
DeleteObject(bitmap);
|
||
}
|
||
|
||
std::wstring TaskbarTabItem::GetWindowClassName() const
|
||
{
|
||
return L"Nim.TaskbarItem";
|
||
}
|
||
|
||
LRESULT TaskbarTabItem::HandleMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
|
||
{
|
||
if (uMsg == WM_DWMSENDICONICTHUMBNAIL)
|
||
{
|
||
OnSendThumbnail(HIWORD(lParam), LOWORD(lParam));
|
||
return 0;
|
||
}
|
||
else if (uMsg == WM_DWMSENDICONICLIVEPREVIEWBITMAP)
|
||
{
|
||
OnSendPreview();
|
||
return 0;
|
||
}
|
||
else if (uMsg == WM_GETICON)
|
||
{
|
||
InvalidateTab();
|
||
}
|
||
else if (uMsg == WM_CLOSE)
|
||
{
|
||
if (NULL != taskbar_manager_)
|
||
taskbar_manager_->OnTabItemClose(*this);
|
||
|
||
return 0;
|
||
}
|
||
else if (uMsg == WM_ACTIVATE)
|
||
{
|
||
if (NULL != taskbar_manager_)
|
||
{
|
||
if (wParam != WA_INACTIVE)
|
||
{
|
||
taskbar_manager_->OnTabItemClicked(*this);
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
return __super::HandleMessage(uMsg, wParam, lParam);
|
||
}
|
||
|
||
void TaskbarTabItem::OnFinalMessage(HWND hWnd)
|
||
{
|
||
__super::OnFinalMessage(hWnd);
|
||
delete this;
|
||
}
|
||
|
||
////////////////////////////////////////////////////////////////////////////////////////////
|
||
////////////////////////////////////////////////////////////////////////////////////////////
|
||
|
||
TaskbarManager::TaskbarManager()
|
||
{
|
||
taskbar_delegate_ = NULL;
|
||
taskbar_list_ = NULL;
|
||
}
|
||
|
||
void TaskbarManager::Init(ITaskbarDelegate *taskbar_delegate)
|
||
{
|
||
ASSERT(NULL != taskbar_delegate);
|
||
taskbar_delegate_ = taskbar_delegate;
|
||
|
||
::CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&taskbar_list_));
|
||
if (taskbar_list_)
|
||
{
|
||
taskbar_list_->HrInit();
|
||
|
||
BOOL truth = FALSE;
|
||
DwmSetWindowAttribute(taskbar_delegate_->GetHandle(), DWMWA_HAS_ICONIC_BITMAP, &truth, sizeof(truth));
|
||
DwmSetWindowAttribute(taskbar_delegate_->GetHandle(), DWMWA_FORCE_ICONIC_REPRESENTATION, &truth, sizeof(truth));
|
||
}
|
||
|
||
}
|
||
|
||
bool TaskbarManager::RegisterTab(TaskbarTabItem &tab_item)
|
||
{
|
||
if (taskbar_list_ && NULL == tab_item.GetTaskbarManager())
|
||
{
|
||
if (S_OK == taskbar_list_->RegisterTab(tab_item.GetHWND(), taskbar_delegate_->GetHandle()))
|
||
{
|
||
if (S_OK == taskbar_list_->SetTabOrder(tab_item.GetHWND(), NULL))
|
||
{
|
||
tab_item.SetTaskbarManager(this);
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
bool TaskbarManager::UnregisterTab(TaskbarTabItem &tab_item)
|
||
{
|
||
if (taskbar_list_)
|
||
{
|
||
tab_item.SetTaskbarManager(NULL);
|
||
return (S_OK == taskbar_list_->UnregisterTab(tab_item.GetHWND()));
|
||
}
|
||
else
|
||
return false;
|
||
}
|
||
|
||
bool TaskbarManager::SetTabOrder(const TaskbarTabItem &tab_item, const TaskbarTabItem &tab_item_insert_before)
|
||
{
|
||
if (taskbar_list_)
|
||
{
|
||
return (S_OK == taskbar_list_->SetTabOrder(tab_item.GetHWND(), tab_item_insert_before.GetHWND()));
|
||
}
|
||
else
|
||
return false;
|
||
}
|
||
|
||
bool TaskbarManager::SetTabActive(const TaskbarTabItem &tab_item)
|
||
{
|
||
if (taskbar_list_)
|
||
{
|
||
return (S_OK == taskbar_list_->SetTabActive(tab_item.GetHWND(), taskbar_delegate_->GetHandle(), 0));
|
||
}
|
||
else
|
||
return false;
|
||
}
|
||
|
||
HBITMAP TaskbarManager::GenerateBindControlBitmapWithForm(ui::Control *control)
|
||
{
|
||
ASSERT( NULL != control);
|
||
if ( NULL == control)
|
||
return NULL;
|
||
|
||
int window_width = 0, window_height = 0;
|
||
RECT rc_wnd;
|
||
bool check_wnd_size = false;
|
||
if (::IsIconic(taskbar_delegate_->GetHandle())) //当前是最小化状态
|
||
{
|
||
WINDOWPLACEMENT placement{ sizeof(WINDOWPLACEMENT) };
|
||
::GetWindowPlacement(taskbar_delegate_->GetHandle(), &placement);
|
||
if (placement.flags == WPF_RESTORETOMAXIMIZED) //最小化前是最大化状态
|
||
{
|
||
MONITORINFO oMonitor = { sizeof(MONITORINFO) };
|
||
::GetMonitorInfo(::MonitorFromWindow(taskbar_delegate_->GetHandle(), MONITOR_DEFAULTTONEAREST), &oMonitor);
|
||
rc_wnd = oMonitor.rcWork;
|
||
}
|
||
else
|
||
{
|
||
rc_wnd = placement.rcNormalPosition;
|
||
check_wnd_size = true; //少数情况下，WINDOWPLACEMENT::rcNormalPosition不正确
|
||
}
|
||
}
|
||
else
|
||
::GetWindowRect(taskbar_delegate_->GetHandle(), &rc_wnd);
|
||
window_width = rc_wnd.right - rc_wnd.left;
|
||
window_height = rc_wnd.bottom - rc_wnd.top;
|
||
if (window_width == 0 || window_height == 0)
|
||
return nullptr;
|
||
|
||
// 1.创建内存dc
|
||
auto render = GlobalManager::CreateRenderContext();
|
||
render->Resize(window_width, window_height);
|
||
|
||
// 2.把窗口双缓冲的位图画到内存dc
|
||
render->BitBlt(0, 0, window_width, window_height, taskbar_delegate_->GetRenderDC());
|
||
|
||
// 3.把某个会话盒子的位图画到内存dc，覆盖原窗口对应位置的位图
|
||
UiRect rcPaint = control->GetPos();
|
||
if (rcPaint.IsRectEmpty())
|
||
return NULL;
|
||
rcPaint.Intersect(UiRect(0, 0, window_width, window_height));
|
||
|
||
// 这里不设置剪裁区域，就无法正常绘制
|
||
{
|
||
AutoClip rectClip(render.get(), rcPaint);
|
||
|
||
bool visible = control->IsInternVisible();
|
||
control->SetInternVisible(true);
|
||
control->Paint(render.get(), rcPaint);
|
||
control->SetInternVisible(visible);
|
||
}
|
||
|
||
// 4.修复绘制区域的alpha通道
|
||
render->RestoreAlpha(rcPaint);
|
||
|
||
return render->DetachBitmap();
|
||
}
|
||
|
||
HBITMAP TaskbarManager::GenerateBindControlBitmap(ui::Control *control, const int dest_width, const int dest_height)
|
||
{
|
||
ASSERT(dest_width > 0 && dest_height > 0 && NULL != control);
|
||
if (dest_width <= 0 || dest_height <= 0 || NULL == control)
|
||
return NULL;
|
||
|
||
int window_width = 0, window_height = 0;
|
||
RECT rc_wnd;
|
||
bool check_wnd_size = false;
|
||
if (::IsIconic(taskbar_delegate_->GetHandle())) //当前是最小化状态
|
||
{
|
||
WINDOWPLACEMENT placement{ sizeof(WINDOWPLACEMENT) };
|
||
::GetWindowPlacement(taskbar_delegate_->GetHandle(), &placement);
|
||
if (placement.flags == WPF_RESTORETOMAXIMIZED) //最小化前是最大化状态
|
||
{
|
||
MONITORINFO oMonitor = { sizeof(MONITORINFO) };
|
||
::GetMonitorInfo(::MonitorFromWindow(taskbar_delegate_->GetHandle(), MONITOR_DEFAULTTONEAREST), &oMonitor);
|
||
rc_wnd = oMonitor.rcWork;
|
||
}
|
||
else
|
||
{
|
||
rc_wnd = placement.rcNormalPosition;
|
||
check_wnd_size = true; //少数情况下，WINDOWPLACEMENT::rcNormalPosition不正确
|
||
}
|
||
}
|
||
else
|
||
::GetWindowRect(taskbar_delegate_->GetHandle(), &rc_wnd);
|
||
window_width = rc_wnd.right - rc_wnd.left;
|
||
window_height = rc_wnd.bottom - rc_wnd.top;
|
||
if (window_width == 0 || window_height == 0)
|
||
return nullptr;
|
||
|
||
// 1.创建内存dc
|
||
auto render = GlobalManager::CreateRenderContext();
|
||
render->Resize(window_width, window_height);
|
||
|
||
// 2.把某个会话盒子的位图画到内存dc，覆盖原窗口对应位置的位图
|
||
UiRect rcPaint = control->GetPos();
|
||
if (rcPaint.IsRectEmpty())
|
||
return NULL;
|
||
rcPaint.Intersect(UiRect(0, 0, window_width, window_height));
|
||
|
||
// 这里不设置剪裁区域，就无法正常绘制
|
||
{
|
||
AutoClip rectClip(render.get(), rcPaint);
|
||
|
||
bool visible = control->IsInternVisible();
|
||
control->SetInternVisible(true);
|
||
control->Paint(render.get(), rcPaint);
|
||
control->SetInternVisible(visible);
|
||
}
|
||
|
||
// 3.修复绘制区域的alpha通道
|
||
render->RestoreAlpha(rcPaint);
|
||
|
||
// 4.缩放到目标尺寸
|
||
UiRect rcControl = control->GetPos();
|
||
return ResizeBitmap(dest_width, dest_height, render->GetDC(), rcControl.left, rcControl.top, rcControl.GetWidth(), rcControl.GetHeight());
|
||
}
|
||
|
||
HBITMAP TaskbarManager::ResizeBitmap(int dest_width, int dest_height, HDC src_dc, int src_x, int src_y, int src_width, int src_height)
|
||
{
|
||
auto render = GlobalManager::CreateRenderContext();
|
||
if (render->Resize(dest_width, dest_height))
|
||
{
|
||
int scale_width = 0;
|
||
int scale_height = 0;
|
||
|
||
float src_scale = (float)src_width / (float)src_height;
|
||
float dest_scale = (float)dest_width / (float)dest_height;
|
||
if (src_scale >= dest_scale)
|
||
{
|
||
scale_width = dest_width;
|
||
scale_height = (int)(dest_width * (float)src_height / (float)src_width);
|
||
}
|
||
else
|
||
{
|
||
scale_height = dest_height;
|
||
scale_width = (int)(dest_height * (float)src_width / (float)src_height);
|
||
}
|
||
|
||
render->AlphaBlend((dest_width - scale_width) / 2, (dest_height - scale_height) / 2, scale_width, scale_height, src_dc, src_x, src_y, src_width, src_height);
|
||
}
|
||
|
||
return render->DetachBitmap();
|
||
}
|
||
|
||
void TaskbarManager::OnTabItemClose(TaskbarTabItem &tab_item)
|
||
{
|
||
taskbar_delegate_->CloseTaskbarItem(tab_item.GetId());
|
||
}
|
||
|
||
void TaskbarManager::OnTabItemClicked(TaskbarTabItem &tab_item)
|
||
{
|
||
taskbar_delegate_->SetActiveTaskbarItem(tab_item.GetId());
|
||
} |