Initial community commit

1 files changed, 386 insertions, 0 deletions

diff --git a/Src/Winamp/dwm.cpp b/Src/Winamp/dwm.cpp
new file mode 100644
index 00000000..03d94c52
--- /dev/null
+++ b/Src/Winamp/dwm.cpp
@@ -0,0 +1,386 @@
+#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);
+		}
+	}
+}
\ No newline at end of file