winamp/Src/Winamp/dwm.cpp
2024-09-24 14:54:57 +02:00

386 lines
12 KiB
C++

#include "main.h"
#include "config.h"
#include "wintheme.h"
#include <shobjidl.h>
#include <tataki/canvas/bltcanvas.h>
#include <tataki/bitmap/bitmap.h>
#include "../Agave/Language/api_language.h"
typedef HRESULT(WINAPI *DWMREGISTERTHUMBNAIL)(HWND hwndDestination, HWND hwndSource, void **phThumbnailId);
typedef HRESULT(WINAPI *DWMUPATETHUMBNAILPROPERTIES)(void* hThumbnailId, void* ptnProperties);
typedef HRESULT(WINAPI *DWMUNREGISTERTHUMBNAIL)(void *hThumbnailId);
typedef HRESULT(WINAPI *DWMSETWINDOWATTRIBUTE)(HWND hwnd, DWORD dwAttribute, LPCVOID pvAttribute, DWORD cbAttribute);
typedef HRESULT (WINAPI *DWMENABLEMMCSS)(BOOL fEnableMMCSS);
typedef HRESULT (WINAPI *DWMSETICONICTHUMBNAIL)(HWND hwnd, HBITMAP hbmp, DWORD dwSITFlags);
typedef HRESULT (WINAPI *DWMINVALIDATEICONICBITMAPS)( HWND hwnd);
typedef HRESULT (WINAPI *DWMSETICONICLIVEPREVIEWBITMAP )(HWND hwnd, HBITMAP hbmp, POINT *pptClient, DWORD dwSITFlags);
static HMODULE dwmapi;
static void *thumbnail;
static DWMREGISTERTHUMBNAIL regThumbnail;
static DWMUPATETHUMBNAILPROPERTIES updateProp;
static DWMUNREGISTERTHUMBNAIL unregThumbnail;
static DWMSETWINDOWATTRIBUTE setWindowAttribute;
static DWMENABLEMMCSS dwmEnableMMCSS;
static DWMSETICONICTHUMBNAIL dwmSetIconicThumbnail;
static DWMINVALIDATEICONICBITMAPS dwmInvalidateIconicBitmaps;
static DWMSETICONICLIVEPREVIEWBITMAP dwmSetIconicLivePreviewBitmap;
HIMAGELIST toolbarIcons = NULL;
BOOL atti_present=false;
static bool triedLoad=false;
static bool LoadDWMApi()
{
if (!triedLoad)
{
wchar_t gen_win7shell_path[MAX_PATH] = {0};
PathCombineW(gen_win7shell_path, PLUGINDIR, L"gen_win7shell.dll");
HMODULE gen_win7shell = LoadLibraryW(gen_win7shell_path);
if (gen_win7shell)
{
atti_present=true;
FreeLibrary(gen_win7shell);
}
dwmapi = LoadLibraryA("dwmapi.dll");
regThumbnail = (DWMREGISTERTHUMBNAIL)GetProcAddress(dwmapi, "DwmRegisterThumbnail");
updateProp = (DWMUPATETHUMBNAILPROPERTIES)GetProcAddress(dwmapi, "DwmUpdateThumbnailProperties");
unregThumbnail = (DWMUNREGISTERTHUMBNAIL)GetProcAddress(dwmapi, "DwmUnregisterThumbnail");
setWindowAttribute = (DWMSETWINDOWATTRIBUTE)GetProcAddress(dwmapi, "DwmSetWindowAttribute");
dwmEnableMMCSS = (DWMENABLEMMCSS)GetProcAddress(dwmapi, "DwmEnableMMCSS");
dwmSetIconicThumbnail = (DWMSETICONICTHUMBNAIL)GetProcAddress(dwmapi, "DwmSetIconicThumbnail");
dwmInvalidateIconicBitmaps = (DWMINVALIDATEICONICBITMAPS)GetProcAddress(dwmapi, "DwmInvalidateIconicBitmaps");
dwmSetIconicLivePreviewBitmap = (DWMSETICONICLIVEPREVIEWBITMAP)GetProcAddress(dwmapi, "DwmSetIconicLivePreviewBitmap");
triedLoad = true;
}
return dwmapi && regThumbnail && updateProp && unregThumbnail && setWindowAttribute && dwmEnableMMCSS;
}
bool LoadToolbarIcons()
{
//toolbarIcons already loaded
if (toolbarIcons != NULL)
{
return true;
}
//load toolbarIcons
toolbarIcons = ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), ILC_COLOR32, 5, 0);
if (toolbarIcons == NULL)
return false;
for (int i = 0; i < 5; i++)
{
HICON hIcon = LoadIconW(hMainInstance, MAKEINTRESOURCE(IDI_TBICON1+i));
if (hIcon != NULL)
ImageList_AddIcon(toolbarIcons, hIcon);
// no need to call DestroyIcon(..) if using LoadIcon(..)
// as the OS will free things anyway - might be cause of
// the random button disappearing issue...?
//DestroyIcon(hIcon);
}
return true;
}
static BOOL taskbar_inited = FALSE;
void OnTaskbarButtonCreated(BOOL force)
{
if (pTaskbar3 || (SUCCEEDED(CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARGS(&pTaskbar3))) && pTaskbar3))
{
if (force)
{
taskbar_inited = TRUE;
pTaskbar3->HrInit();
}
if (taskbar_inited)
{
RegisterThumbnailTab(IsWindow(g_dialog_box_parent) ? g_dialog_box_parent : hMainWindow);
}
}
}
void UnregisterThumbnailTab(HWND hWnd)
{
if (LoadDWMApi() && !atti_present)
{
if (!IsVistaOrLower() && IsWindow(hWnd) && pTaskbar3 != NULL)
{
if (taskbar_inited)
pTaskbar3->UnregisterTab(hWnd);
}
}
}
static void addToolbarButtons(HWND hWnd, BOOL update)
{
THUMBBUTTON thbButtons[5];
DWORD dwMask = THB_BITMAP | THB_TOOLTIP;
thbButtons[0].dwMask = (THUMBBUTTONMASK)dwMask;
thbButtons[0].iId = 0;
thbButtons[0].iBitmap = 0;
StringCbCopyW(thbButtons[0].szTip, sizeof(thbButtons[0].szTip), getStringW(IDS_WIN7TOOLBAR_TOOLTIP_PREVIOUS, NULL, 0));
thbButtons[1].dwMask = (THUMBBUTTONMASK)dwMask;
thbButtons[1].iId = 1;
thbButtons[1].iBitmap = 1;
StringCbCopyW(thbButtons[1].szTip, sizeof(thbButtons[1].szTip), getStringW(IDS_WIN7TOOLBAR_TOOLTIP_PLAY, NULL, 0));
thbButtons[2].dwMask = (THUMBBUTTONMASK)dwMask;
thbButtons[2].iId = 2;
thbButtons[2].iBitmap = 2;
StringCbCopyW(thbButtons[2].szTip, sizeof(thbButtons[2].szTip), getStringW(IDS_WIN7TOOLBAR_TOOLTIP_PAUSE, NULL, 0));
thbButtons[3].dwMask = (THUMBBUTTONMASK)dwMask;
thbButtons[3].iId = 3;
thbButtons[3].iBitmap = 3;
StringCbCopyW(thbButtons[3].szTip, sizeof(thbButtons[3].szTip), getStringW(IDS_WIN7TOOLBAR_TOOLTIP_STOP, NULL, 0));
thbButtons[4].dwMask = (THUMBBUTTONMASK)dwMask;
thbButtons[4].iId = 4;
thbButtons[4].iBitmap = 4;
StringCbCopyW(thbButtons[4].szTip, sizeof(thbButtons[4].szTip), getStringW(IDS_WIN7TOOLBAR_TOOLTIP_NEXT, NULL, 0));
if (update)
pTaskbar3->ThumbBarUpdateButtons(hWnd, ARRAYSIZE(thbButtons), thbButtons);
else
pTaskbar3->ThumbBarAddButtons(hWnd, ARRAYSIZE(thbButtons), thbButtons);
}
void RegisterThumbnailTab(HWND hWnd)
{
if (LoadDWMApi() && !atti_present)
{
if (dwmInvalidateIconicBitmaps) // even if DWM is loaded, this only exists on win7 (NT6.1)
dwmInvalidateIconicBitmaps(hMainWindow);
if (!IsWindow(hWnd))
{
hWnd = hMainWindow;
}
if (hWnd != hMainWindow)
{
wchar_t title[512] = {0};
GetWindowTextW(hMainWindow, title, sizeof(title));
SetWindowTextW(hWnd, title);
#ifdef WIN64
HICON hIcon = (HICON)GetClassLongPtr(hMainWindow, GCLP_HICONSM);
#else
HICON hIcon = (HICON)GetClassLong(hMainWindow, GCL_HICONSM);
#endif
SendMessageW(hWnd, WM_SETICON, (WPARAM)ICON_SMALL, (LPARAM)hIcon);
}
if (IsWindow(hWnd) && IsWindow(hMainWindow))
{
BOOL dwm_setting = FALSE;
setWindowAttribute(hWnd, DWMWA_FORCE_ICONIC_REPRESENTATION, &dwm_setting, sizeof(dwm_setting));
setWindowAttribute(hWnd, DWMWA_HAS_ICONIC_BITMAP, &dwm_setting, sizeof(dwm_setting));
if (!IsVistaOrLower() && pTaskbar3 != NULL)
{
// shouldn't fail, but there's a case on loading where it can due to timing quirks
// so for that we then allow the regsistation to still happen (hence the dup code)
HRESULT hr = pTaskbar3->RegisterTab(hWnd, hMainWindow);
if (SUCCEEDED(hr))
{
if (SUCCEEDED(pTaskbar3->SetTabOrder(hWnd, NULL)))
{
pTaskbar3->SetTabActive(hWnd, hMainWindow, 0);
if (LoadToolbarIcons())
{
HRESULT hr = pTaskbar3->ThumbBarSetImageList(hWnd, toolbarIcons);
if (SUCCEEDED(hr))
{
addToolbarButtons(hWnd, false);
}
}
}
}
else
{
if (SUCCEEDED(pTaskbar3->SetTabOrder(hWnd, NULL)))
{
pTaskbar3->SetTabActive(hWnd, hMainWindow, 0);
if (LoadToolbarIcons())
{
HRESULT hr = pTaskbar3->ThumbBarSetImageList(hWnd, toolbarIcons);
if (SUCCEEDED(hr))
{
addToolbarButtons(hWnd, false);
}
}
}
}
}
if (hWnd != hMainWindow)
{
dwm_setting = TRUE;
setWindowAttribute(hMainWindow, DWMWA_FORCE_ICONIC_REPRESENTATION, &dwm_setting, sizeof(dwm_setting));
setWindowAttribute(hMainWindow, DWMWA_HAS_ICONIC_BITMAP, &dwm_setting, sizeof(dwm_setting));
}
}
}
}
void DisableVistaPreview()
{
if (LoadDWMApi())
{
BOOL blah=TRUE;
setWindowAttribute(hMainWindow, DWMWA_FORCE_ICONIC_REPRESENTATION, &blah, sizeof(blah));
setWindowAttribute(hMainWindow, DWMWA_HAS_ICONIC_BITMAP, &blah, sizeof(blah));
}
}
static bool done_the_dance=false;
void DoTheVistaVideoDance()
{
if (!done_the_dance && LoadDWMApi())
{
dwmEnableMMCSS(TRUE); // the magic "make my program not suck" function
/* TODO:
DWM_PRESENT_PARAMETERS dpp;
memset(&dpp, 0, sizeof(dpp));
dpp.cbSize = sizeof(dpp);
dpp.fQueue = true;
dpp.cBuffer = 8;
dpp.fUseSourceRate = false;
dpp.cRefreshesPerFrame = 1;
dpp.eSampling = DWM_SOURCE_FRAME_SAMPLING_POINT;
HRESULT hr = DwmSetPresentParameters(hWnd, &dpp);
*/
// TODO also, unrelated, but check out AvSetMmThreadCharacteristics sometime.
}
done_the_dance = true;
}
static void Adjust(int bmpw, int bmph, RECT &r)
{
// maintain 'square' stretching
int w = r.right - r.left;
int h = r.bottom - r.top;
double aspX = (double)(w)/(double)bmpw;
double aspY = (double)(h)/(double)bmph;
double asp = min(aspX, aspY);
int newW = (int)(bmpw*asp);
int newH = (int)(bmph*asp);
r.left = (w - newW)/2;
r.top = (h - newH)/2;
r.right = r.left + newW;
r.bottom = r.top + newH;
}
void RefreshIconicThumbnail()
{
if (LoadDWMApi() && !atti_present)
{
if (dwmInvalidateIconicBitmaps) // even if DWM is loaded, this only exists on win7 (NT6.1)
dwmInvalidateIconicBitmaps(hMainWindow);
}
}
void OnIconicThumbnail(int width, int height)
{
static BltCanvas *iconic_thumbnail_bitmap=0;
HWND hWnd = g_dialog_box_parent?g_dialog_box_parent:hMainWindow;
RECT client_size;
GetClientRect(hWnd, &client_size);
if (!iconic_thumbnail_bitmap)
{
iconic_thumbnail_bitmap = new BltCanvas(client_size.right, client_size.bottom, hMainWindow);
}
else
{
iconic_thumbnail_bitmap->DestructiveResize(client_size.right, client_size.bottom);
}
SendMessageW(hWnd, WM_PRINTCLIENT, (WPARAM) iconic_thumbnail_bitmap->getHDC(), PRF_CHILDREN | PRF_CLIENT | /*PRF_ERASEBKGND |*/ PRF_NONCLIENT /*| PRF_OWNED*/);
void *bits=0;
BITMAPINFO bmi = {0};
bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
bmi.bmiHeader.biWidth = width;
bmi.bmiHeader.biHeight = -height;
bmi.bmiHeader.biPlanes = 1;
bmi.bmiHeader.biBitCount = 32;
bmi.bmiHeader.biCompression = BI_RGB;
HBITMAP hbmp = CreateDIBSection(iconic_thumbnail_bitmap->getHDC(), &bmi, DIB_RGB_COLORS, &bits, NULL, 0);
if (hbmp)
{
BltCanvas resizedBitmap(hbmp);
int x=0, y=0;
RECT dest;
dest.left = x;
dest.top = y;
dest.right = width;
dest.bottom = height;
resizedBitmap.drawRect(&dest, 1, 0);
Adjust(client_size.right, client_size.bottom, dest);
iconic_thumbnail_bitmap->stretchToRectAlpha(&resizedBitmap, &client_size, &dest);
dwmSetIconicThumbnail(hMainWindow, hbmp, 0);
}
}
void OnThumbnailPreview()
{
static BltCanvas *thumbnail_preview_bitmap=0;
HWND hWnd = g_dialog_box_parent?g_dialog_box_parent:hMainWindow;
RECT client_size;
GetClientRect(hWnd, &client_size);
if (!thumbnail_preview_bitmap)
{
thumbnail_preview_bitmap = new BltCanvas(client_size.right, client_size.bottom, hWnd);
}
else
{
thumbnail_preview_bitmap->DestructiveResize(client_size.right, client_size.bottom);
}
SendMessageW(hWnd, WM_PRINT, (WPARAM) thumbnail_preview_bitmap->getHDC(), PRF_CHILDREN | PRF_CLIENT | PRF_ERASEBKGND | PRF_NONCLIENT /*| PRF_OWNED*/);
void *bits=0;
BITMAPINFO bmi = {0};
bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
bmi.bmiHeader.biWidth = client_size.right - client_size.left;
bmi.bmiHeader.biHeight = client_size.top - client_size.bottom;
bmi.bmiHeader.biPlanes = 1;
bmi.bmiHeader.biBitCount = 32;
bmi.bmiHeader.biCompression = BI_RGB;
HBITMAP hbmp = CreateDIBSection(thumbnail_preview_bitmap->getHDC(), &bmi, DIB_RGB_COLORS, &bits, NULL, 0);
if (hbmp)
{
POINT offset;
offset.x = client_size.left;
offset.y = client_size.top;
if (dwmSetIconicLivePreviewBitmap(hWnd, hbmp, &offset, 1) == S_OK)
{
MessageBoxA(NULL, "winamp/live", "winamp/live", MB_OK);
}
else
{
MessageBoxA(NULL, "winamp/no live", "winamp/no live", MB_OK);
}
}
}