NSIS/Contrib/BgImage/BgImage.cpp
wizou 752d7d239a Jim Park's Unicode NSIS merging - Step 1 : switch to TCHARs where relevant.
Compiler output is identical before & after this step

git-svn-id: https://svn.code.sf.net/p/nsis/code/NSIS/branches/wizou@6036 212acab6-be3b-0410-9dea-997c60f758d6
2010-03-24 17:22:56 +00:00

624 lines
16 KiB
C++

// Unicode support by Jim Park -- 08/22/2007
#include <windows.h>
#include <mmsystem.h>
#include <nsis/pluginapi.h> // nsis plugin
#undef EXDLL_INIT
#define EXDLL_INIT() { \
g_stringsize=string_size; \
g_stacktop=stacktop; }
#define NSISFunc(name) extern "C" void __declspec(dllexport) name(HWND hwndParent, int string_size, TCHAR *variables, stack_t **stacktop, extra_parameters *extra)
TCHAR szTemp[2048];
HWND hWndImage, hWndParent;
HINSTANCE g_hInstance;
CRITICAL_SECTION CriticalSection;
void ECS() {
EnterCriticalSection(&CriticalSection);
}
void LCS() {
LeaveCriticalSection(&CriticalSection);
}
enum {
MIL_DUMMY,
MIL_GRADIENT,
MIL_BITMAP,
MIL_TRANSPARENT_BITMAP,
MIL_TEXT
};
struct myImageList {
BYTE iType;
union {
HBITMAP hBitmap;
TCHAR *szText;
COLORREF cGradientFrom;
};
RECT rPos;
union {
COLORREF cTransparent;
COLORREF cTextColor;
COLORREF cGradientTo;
};
HFONT hFont;
BOOL bReady;
myImageList *next;
} bgBitmap;
unsigned int uWndWidth, uWndHeight;
void *oldProc;
LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam);
HBITMAP __stdcall LoadBitmapFile(long right, long bottom, BITMAP *bBitmap);
COLORREF GetColor();
void __stdcall GetXY(LPPOINT lpPoint);
NSISFunc(Destroy);
static UINT_PTR PluginCallback(enum NSPIM msg)
{
if (msg == NSPIM_GUIUNLOAD)
{
Destroy(0, 0, 0, 0, 0);
}
return 0;
}
BOOL bReturn;
NSISFunc(SetReturn) {
EXDLL_INIT();
extra->RegisterPluginCallback(g_hInstance, PluginCallback);
popstring(szTemp);
bReturn = !lstrcmpi(szTemp, _T("on"));
}
static void __stdcall my_pushstring(TCHAR *str)
{
stack_t *th;
if (!g_stacktop || !bReturn) return;
th=(stack_t*)GlobalAlloc(GPTR,sizeof(stack_t)+(g_stringsize*sizeof(TCHAR)));
lstrcpyn(th->text,str,g_stringsize);
th->next=*g_stacktop;
*g_stacktop=th;
}
NSISFunc(SetBg) {
EXDLL_INIT();
extra->RegisterPluginCallback(g_hInstance, PluginCallback);
ECS();
if (!hWndImage) {
hWndParent = hwndParent;
if (!hwndParent) {
my_pushstring(_T("can't find parent window"));
LCS();
return;
}
WNDCLASSEX wc = {
sizeof(WNDCLASSEX),
CS_VREDRAW|CS_HREDRAW,
WndProc,
0,
0,
g_hInstance,
0,
LoadCursor(0, IDC_ARROW),
0,
0,
_T("NSISBGImage"),
0
};
ATOM atomClass = RegisterClassEx(&wc);
if (!atomClass) {
my_pushstring(_T("can't create window"));
return;
}
hWndImage = CreateWindowEx(
WS_EX_TOOLWINDOW,
(LPTSTR)(DWORD)atomClass,
0,
WS_CLIPSIBLINGS|WS_POPUP,
0,
0,
0,
0,
0,
0,
g_hInstance,
0
);
if (!hWndImage) {
my_pushstring(_T("can't create window"));
LCS();
return;
}
oldProc = (void *)SetWindowLong(hwndParent, GWL_WNDPROC, (long)WndProc);
}
bgBitmap.bReady = FALSE;
if (bgBitmap.iType == MIL_BITMAP)
DeleteObject(bgBitmap.hBitmap);
unsigned int uScrWidth = GetSystemMetrics(SM_CXSCREEN);
unsigned int uScrHeight = GetSystemMetrics(SM_CYSCREEN);
bgBitmap.iType = MIL_BITMAP;
bgBitmap.rPos.right = 0;
bgBitmap.rPos.bottom = 0;
uWndWidth = uScrWidth;
uWndHeight = uScrHeight;
char szGradient[] = {'/', 'G', 'R', 'A', 'D', 'I', 'E', 'N', 'T', 0};
char szFillScreen[] = {'/', 'F', 'I' ,'L', 'L', 'S', 'C', 'R', 'E', 'E', 'N', 0};
char szTiled[] = {'/', 'T', 'I', 'L', 'E', 'D', 0};
popstring(szTemp);
if (!lstrcmpi(szTemp, szGradient)) {
bgBitmap.cGradientFrom = GetColor();
bgBitmap.cGradientTo = GetColor();
bgBitmap.iType = MIL_GRADIENT;
goto done;
}
if (!lstrcmpi(szTemp, szFillScreen)) {
bgBitmap.rPos.right = uScrWidth;
bgBitmap.rPos.bottom = uScrHeight;
popstring(szTemp);
}
else if (!lstrcmpi(szTemp, szTiled)) {
popstring(szTemp);
}
else {
uWndWidth = 0;
uWndHeight = 0;
}
BITMAP bBitmap;
bgBitmap.hBitmap = LoadBitmapFile(bgBitmap.rPos.right, bgBitmap.rPos.bottom, &bBitmap);
if (!bgBitmap.hBitmap)
return;
if (!bgBitmap.rPos.right) {
bgBitmap.rPos.right = bBitmap.bmWidth;
bgBitmap.rPos.bottom = bBitmap.bmHeight;
}
if (!uWndWidth) {
uWndWidth = bBitmap.bmWidth;
uWndHeight = bBitmap.bmHeight;
}
done:
bgBitmap.bReady = TRUE;
LCS();
if (hWndImage) {
SetWindowPos(
hWndImage,
hWndParent,
(uScrWidth-uWndWidth)/2,
(uScrHeight-uWndHeight)/2,
uWndWidth,
uWndHeight,
SWP_NOACTIVATE
);
}
my_pushstring(_T("success"));
}
NSISFunc(AddImage) {
ECS();
myImageList *newImg = (myImageList *)GlobalAlloc(GPTR, sizeof(myImageList));
if (!newImg) {
my_pushstring(_T("memory allocation error"));
LCS();
return;
}
newImg->iType = MIL_BITMAP;
newImg->cTransparent = (COLORREF)-1;
popstring(szTemp);
if (!lstrcmpi(szTemp, _T("/TRANSPARENT"))) {
newImg->iType = MIL_TRANSPARENT_BITMAP;
newImg->cTransparent = GetColor();
popstring(szTemp);
}
BITMAP bBitmap;
newImg->hBitmap = LoadBitmapFile(0, 0, &bBitmap);
if (!newImg->hBitmap) {
GlobalFree(newImg);
return;
}
GetXY(LPPOINT(&newImg->rPos));
newImg->rPos.right = newImg->rPos.left + bBitmap.bmWidth;
newImg->rPos.bottom = newImg->rPos.top + bBitmap.bmHeight;
myImageList *img = &bgBitmap;
while (img->next) img = img->next;
img->next = newImg;
my_pushstring(_T("success"));
LCS();
}
NSISFunc(AddText) {
ECS();
myImageList *newImg = (myImageList *)GlobalAlloc(GPTR, sizeof(myImageList));
if (!newImg) {
my_pushstring(_T("memory allocation error"));
LCS();
return;
}
newImg->iType = MIL_TEXT;
popstring(szTemp);
newImg->szText = (TCHAR *)GlobalAlloc(GPTR, (lstrlen(szTemp)+1)*sizeof(TCHAR));
if (!newImg->szText) {
my_pushstring(_T("memory allocation error"));
GlobalFree(newImg);
LCS();
return;
}
lstrcpy(newImg->szText, szTemp);
popstring(szTemp);
newImg->hFont = (HFONT)myatoi(szTemp);
newImg->cTextColor = GetColor();
GetXY(LPPOINT(&newImg->rPos));
GetXY(LPPOINT(&newImg->rPos) + 1);
myImageList *img = &bgBitmap;
while (img->next) img = img->next;
img->next = newImg;
my_pushstring(_T("success"));
LCS();
}
NSISFunc(Redraw) {
RedrawWindow(hWndImage, 0, 0, RDW_INVALIDATE | RDW_UPDATENOW);
ShowWindow(hWndImage, SW_SHOWNA);
}
NSISFunc(Clear) {
ECS();
myImageList *img = &bgBitmap;
while (img) {
switch (img->iType) {
case MIL_BITMAP:
case MIL_TRANSPARENT_BITMAP:
DeleteObject(img->hBitmap);
break;
case MIL_TEXT:
GlobalFree(img->szText);
break;
}
myImageList *thisImg = img;
img = img->next;
thisImg->next = NULL;
if (thisImg != &bgBitmap)
GlobalFree(thisImg);
}
bgBitmap.bReady = FALSE;
LCS();
}
NSISFunc(Destroy) {
bgBitmap.bReady = FALSE;
if (IsWindow(hwndParent) && oldProc)
SetWindowLong(hwndParent, GWL_WNDPROC, (long)oldProc);
if (IsWindow(hWndImage))
SendMessage(hWndImage, WM_CLOSE, 0, 0);
hWndImage = 0;
oldProc = NULL;
Clear(0, 0, 0, 0, 0);
UnregisterClass(_T("NSISBGImage"), g_hInstance);
}
NSISFunc(Sound) {
char szLoop[] = {'/', 'L', 'O', 'O', 'P', 0};
char szWait[] = {'/', 'W', 'A', 'I', 'T', 0};
char szStop[] = {'/', 'S', 'T', 'O', 'P', 0};
DWORD flags = SND_FILENAME | SND_NODEFAULT;
g_stacktop = stacktop;
popstring(szTemp);
if (lstrcmpi(szTemp, szWait))
flags |= SND_ASYNC;
else
popstring(szTemp);
if (!lstrcmpi(szTemp, szLoop)) {
flags |= SND_LOOP;
popstring(szTemp);
}
PlaySound(lstrcmpi(szTemp, szStop) ? szTemp : 0, 0, flags);
}
LRESULT CALLBACK WndProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam) {
HWND hwndParent = hWndParent;
HWND hwndImage = hWndImage;
if (hwnd == hwndParent) {
if (message == WM_SIZE) {
ShowWindow(hwndImage, wParam == SIZE_MINIMIZED ? SW_HIDE : SW_SHOW);
}
if (message == WM_WINDOWPOSCHANGED) {
SetWindowPos(hwndImage, hwndParent, 0, 0, 0, 0, SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOSIZE);
}
return CallWindowProc(
(long (__stdcall *)(HWND,unsigned int,unsigned int,long))oldProc,
hwnd,
message,
wParam,
lParam
);
}
switch (message) {
case WM_PAINT:
if (bgBitmap.bReady) {
ECS();
PAINTSTRUCT ps;
HDC hdc = BeginPaint(hwnd, &ps);
if (bgBitmap.iType == MIL_BITMAP) {
HDC cdc = CreateCompatibleDC(hdc);
SelectObject(cdc, bgBitmap.hBitmap);
for (unsigned int x = 0; x < uWndWidth; x += bgBitmap.rPos.right) {
for (unsigned int y = 0; y < uWndHeight; y += bgBitmap.rPos.bottom) {
BitBlt(hdc, x, y, bgBitmap.rPos.right, bgBitmap.rPos.bottom, cdc, 0, 0, SRCCOPY);
}
}
DeleteDC(cdc);
}
else {
int r = GetRValue(bgBitmap.cGradientFrom) << 10;
int g = GetGValue(bgBitmap.cGradientFrom) << 10;
int b = GetBValue(bgBitmap.cGradientFrom) << 10;
int dr = ((GetRValue(bgBitmap.cGradientTo) << 10) - r) / (int)uWndHeight * 4;
int dg = ((GetGValue(bgBitmap.cGradientTo) << 10) - g) / (int)uWndHeight * 4;
int db = ((GetBValue(bgBitmap.cGradientTo) << 10) - b) / (int)uWndHeight * 4;
RECT rect;
rect.left = 0;
rect.top = 0;
rect.right = uWndWidth;
rect.bottom = 4;
while (rect.top < (int)uWndHeight)
{
HBRUSH brush = CreateSolidBrush(RGB(r>>10,g>>10,b>>10));
FillRect(hdc, &rect, brush);
DeleteObject(brush);
rect.top+=4;
rect.bottom+=4;
r+=dr;
g+=dg;
b+=db;
}
}
myImageList *img = bgBitmap.next;
while (img) {
if (img->iType == MIL_TEXT) {
SetBkMode(hdc, TRANSPARENT);
SetTextColor(hdc, img->cTextColor);
SelectObject(hdc, img->hFont);
DrawText(hdc, img->szText, -1, &img->rPos, DT_TOP | DT_LEFT | DT_NOPREFIX | DT_WORDBREAK);
}
else if (img->iType == MIL_BITMAP) {
HDC cdc = CreateCompatibleDC(hdc);
SelectObject(cdc, img->hBitmap);
BitBlt(hdc, img->rPos.left, img->rPos.top, img->rPos.right - img->rPos.left, img->rPos.bottom - img->rPos.top, cdc, 0, 0, SRCCOPY);
DeleteDC(cdc);
}
else {
COLORREF cColor;
HBITMAP bmAndBack, bmAndObject, bmAndMem, bmSave;
HBITMAP bmBackOld, bmObjectOld, bmMemOld, bmSaveOld;
HDC hdcMem, hdcBack, hdcObject, hdcTemp, hdcSave;
POINT ptSize;
HBITMAP hBitmap = img->hBitmap;
hdcTemp = CreateCompatibleDC(hdc);
SelectObject(hdcTemp, hBitmap); // Select the bitmap
ptSize.x = img->rPos.right - img->rPos.left;
ptSize.y = img->rPos.bottom - img->rPos.top;
DPtoLP(hdcTemp, &ptSize, 1); // Convert from device to logical points
// Create some DCs to hold temporary data.
hdcBack = CreateCompatibleDC(hdc);
hdcObject = CreateCompatibleDC(hdc);
hdcMem = CreateCompatibleDC(hdc);
hdcSave = CreateCompatibleDC(hdc);
// Create a bitmap for each DC. DCs are required for a number of
// GDI functions.
// Monochrome DC
bmAndBack = CreateBitmap(ptSize.x, ptSize.y, 1, 1, NULL);
// Monochrome DC
bmAndObject = CreateBitmap(ptSize.x, ptSize.y, 1, 1, NULL);
bmAndMem = CreateCompatibleBitmap(hdc, ptSize.x, ptSize.y);
bmSave = CreateCompatibleBitmap(hdc, ptSize.x, ptSize.y);
// Each DC must select a bitmap object to store pixel data.
bmBackOld = (HBITMAP)SelectObject(hdcBack, bmAndBack);
bmObjectOld = (HBITMAP)SelectObject(hdcObject, bmAndObject);
bmMemOld = (HBITMAP)SelectObject(hdcMem, bmAndMem);
bmSaveOld = (HBITMAP)SelectObject(hdcSave, bmSave);
// Set proper mapping mode.
SetMapMode(hdcTemp, GetMapMode(hdc));
// Save the bitmap sent here, because it will be overwritten.
BitBlt(hdcSave, 0, 0, ptSize.x, ptSize.y, hdcTemp, 0, 0, SRCCOPY);
// Set the background color of the source DC to the color.
// contained in the parts of the bitmap that should be transparent
cColor = SetBkColor(hdcTemp, img->cTransparent);
// Create the object mask for the bitmap by performing a BitBlt
// from the source bitmap to a monochrome bitmap.
BitBlt(hdcObject, 0, 0, ptSize.x, ptSize.y, hdcTemp, 0, 0,
SRCCOPY);
// Set the background color of the source DC back to the original
// color.
SetBkColor(hdcTemp, cColor);
// Create the inverse of the object mask.
BitBlt(hdcBack, 0, 0, ptSize.x, ptSize.y, hdcObject, 0, 0,
NOTSRCCOPY);
// Copy the background of the main DC to the destination.
BitBlt(hdcMem, 0, 0, ptSize.x, ptSize.y, hdc, img->rPos.left, img->rPos.top,
SRCCOPY);
// Mask out the places where the bitmap will be placed.
BitBlt(hdcMem, 0, 0, ptSize.x, ptSize.y, hdcObject, 0, 0, SRCAND);
// Mask out the transparent colored pixels on the bitmap.
BitBlt(hdcTemp, 0, 0, ptSize.x, ptSize.y, hdcBack, 0, 0, SRCAND);
// XOR the bitmap with the background on the destination DC.
BitBlt(hdcMem, 0, 0, ptSize.x, ptSize.y, hdcTemp, 0, 0, SRCPAINT);
// Copy the destination to the screen.
BitBlt(hdc, img->rPos.left, img->rPos.top, ptSize.x, ptSize.y, hdcMem, 0, 0,
SRCCOPY);
// Place the original bitmap back into the bitmap sent here.
BitBlt(hdcTemp, 0, 0, ptSize.x, ptSize.y, hdcSave, 0, 0, SRCCOPY);
// Delete the memory bitmaps.
DeleteObject(SelectObject(hdcBack, bmBackOld));
DeleteObject(SelectObject(hdcObject, bmObjectOld));
DeleteObject(SelectObject(hdcMem, bmMemOld));
DeleteObject(SelectObject(hdcSave, bmSaveOld));
// Delete the memory DCs.
DeleteDC(hdcMem);
DeleteDC(hdcBack);
DeleteDC(hdcObject);
DeleteDC(hdcSave);
DeleteDC(hdcTemp);
}
img = img->next;
}
LCS();
EndPaint(hwnd, &ps);
}
break;
case WM_WINDOWPOSCHANGING:
if (IsWindow(hwndParent))
{
LPWINDOWPOS wp = (LPWINDOWPOS) lParam;
wp->flags |= SWP_NOACTIVATE;
wp->hwndInsertAfter = hwndParent;
}
break;
case WM_CLOSE:
DestroyWindow(hwnd);
break;
default:
return DefWindowProc(hwnd, message, wParam, lParam);
}
return 0;
}
HBITMAP __stdcall LoadBitmapFile(long right, long bottom, BITMAP *bBitmap)
{
HBITMAP hBitmap = (HBITMAP)LoadImage(0, szTemp, IMAGE_BITMAP, right, bottom, LR_LOADFROMFILE);
if (!hBitmap || !GetObject(hBitmap, sizeof(BITMAP), (void *)bBitmap)) {
my_pushstring(_T("can't load bitmap"));
if (hBitmap)
DeleteObject(hBitmap);
LCS();
return 0;
}
return hBitmap;
}
COLORREF GetColor() {
COLORREF cColor = 0;
popstring(szTemp);
cColor |= (BYTE) myatoi(szTemp);
popstring(szTemp);
cColor |= ((BYTE) myatoi(szTemp)) << 8;
popstring(szTemp);
cColor |= ((BYTE) myatoi(szTemp)) << 16;
return cColor;
}
void __stdcall GetXY(LPPOINT lpPoint) {
popstring(szTemp);
int iPosTemp = myatoi(szTemp);
if (iPosTemp < 0) iPosTemp = iPosTemp + (int)uWndWidth;
lpPoint->x = (unsigned int)iPosTemp;
popstring(szTemp);
iPosTemp = myatoi(szTemp);
if (iPosTemp < 0) iPosTemp = iPosTemp + (int)uWndHeight;
lpPoint->y = (unsigned int)iPosTemp;
}
extern "C" BOOL WINAPI DllMain(HINSTANCE hInst, ULONG ul_reason_for_call, LPVOID lpReserved) {
g_hInstance=hInst;
switch (ul_reason_for_call) {
case DLL_PROCESS_ATTACH:
InitializeCriticalSection(&CriticalSection);
break;
case DLL_PROCESS_DETACH:
DeleteCriticalSection(&CriticalSection);
break;
}
return TRUE;
}