/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* vim:set ts=2 sw=2 sts=2 et cindent: */ /* ***** BEGIN LICENSE BLOCK ***** * Version: MPL 1.1/GPL 2.0/LGPL 2.1 * * The contents of this file are subject to the Mozilla Public License Version * 1.1 (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * http://www.mozilla.org/MPL/ * * Software distributed under the License is distributed on an "AS IS" basis, * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License * for the specific language governing rights and limitations under the * License. * * The Original Code is mozilla.org code. * * The Initial Developer of the Original Code is Google Inc. * Portions created by the Initial Developer are Copyright (C) 2005 * the Initial Developer. All Rights Reserved. * * Contributor(s): * Darin Fisher * Masayuki Nakano * * Alternatively, the contents of this file may be used under the terms of * either the GNU General Public License Version 2 or later (the "GPL"), or * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), * in which case the provisions of the GPL or the LGPL are applicable instead * of those above. If you wish to allow use of your version of this file only * under the terms of either the GPL or the LGPL, and not to allow others to * use your version of this file under the terms of the MPL, indicate your * decision by deleting the provisions above and replace them with the notice * and other provisions required by the GPL or the LGPL. If you do not delete * the provisions above, a recipient may use your version of this file under * the terms of any one of the MPL, the GPL or the LGPL. * * ***** END LICENSE BLOCK ***** */ #include #include #include #include #include #include "resource.h" #include "progressui.h" #define TIMER_ID 1 #define TIMER_INTERVAL 100 #define MAX_INFO_LENGTH 512 #define RESIZE_WINDOW(hwnd, extrax, extray) \ { \ RECT windowSize; \ GetWindowRect(hwnd, &windowSize); \ SetWindowPos(hwnd, 0, 0, 0, windowSize.right - windowSize.left + extrax, \ windowSize.bottom - windowSize.top + extray, \ SWP_NOMOVE | SWP_NOZORDER); \ } #define MOVE_WINDOW(hwnd, dx, dy) \ { \ WINDOWPLACEMENT windowPos; \ windowPos.length = sizeof(windowPos); \ GetWindowPlacement(hwnd, &windowPos); \ SetWindowPos(hwnd, 0, windowPos.rcNormalPosition.left + dx, windowPos.rcNormalPosition.top + dy, 0, 0, \ SWP_NOSIZE | SWP_NOZORDER); \ } static float sProgress; // between 0 and 100 static BOOL sQuit = FALSE; static HFONT sSystemFont = 0; static BOOL GetStringsFile(WCHAR filename[MAX_PATH]) { if (!GetModuleFileNameW(NULL, filename, MAX_PATH)) return FALSE; WCHAR *dot = wcsrchr(filename, '.'); if (!dot || wcsicmp(dot + 1, L"exe")) return FALSE; wcscpy(dot + 1, L"ini"); return TRUE; } static void UpdateDialog(HWND hDlg) { int pos = int(sProgress + 0.5f); SendDlgItemMessage(hDlg, IDC_PROGRESS, PBM_SETPOS, pos, 0L); } static void ResizeDialogToFit(HWND hDlg) { WCHAR text[MAX_INFO_LENGTH]; RECT infoSize, textSize; HFONT hInfoFont, hOldFont; HWND hWndInfo = GetDlgItem(hDlg, IDC_INFO); HWND hWndPro = GetDlgItem(hDlg, IDC_PROGRESS); // Get the text that is displayed - this is what we're going to make fit. if (!GetWindowText(hWndInfo, text, sizeof(text))) return; // We need the current size and font to calculate the adjustment. GetClientRect(hWndInfo, &infoSize); HDC hDCInfo = GetDC(hWndInfo); hInfoFont = (HFONT)SendMessage(hWndInfo, WM_GETFONT, 0, 0); if (hInfoFont) hOldFont = (HFONT)SelectObject(hDCInfo, hInfoFont); // Measure the space needed for the text - DT_CALCRECT means nothing is drawn. if (DrawText(hDCInfo, text, -1, &textSize, DT_CALCRECT | DT_NOCLIP | DT_SINGLELINE)) { SIZE extra; extra.cx = (textSize.right - textSize.left) - (infoSize.right - infoSize.left); extra.cy = (textSize.bottom - textSize.top) - (infoSize.bottom - infoSize.top); if (extra.cx < 0) extra.cx = 0; if (extra.cy < 0) extra.cy = 0; if ((extra.cx > 0) || (extra.cy > 0)) { RESIZE_WINDOW(hDlg, extra.cx, extra.cy); RESIZE_WINDOW(hWndInfo, extra.cx, extra.cy); RESIZE_WINDOW(hWndPro, extra.cx, 0); MOVE_WINDOW(hWndPro, 0, extra.cy); } } if (hOldFont) SelectObject(hDCInfo, hOldFont); ReleaseDC(hWndInfo, hDCInfo); } // The code in this function is from MSDN: // http://msdn.microsoft.com/library/default.asp?url=/library/en-us/winui/winui/windowsuserinterface/windowing/dialogboxes/usingdialogboxes.asp static void CenterDialog(HWND hDlg) { RECT rc, rcOwner, rcDlg; // Get the owner window and dialog box rectangles. HWND desktop = GetDesktopWindow(); GetWindowRect(desktop, &rcOwner); GetWindowRect(hDlg, &rcDlg); CopyRect(&rc, &rcOwner); // Offset the owner and dialog box rectangles so that // right and bottom values represent the width and // height, and then offset the owner again to discard // space taken up by the dialog box. OffsetRect(&rcDlg, -rcDlg.left, -rcDlg.top); OffsetRect(&rc, -rc.left, -rc.top); OffsetRect(&rc, -rcDlg.right, -rcDlg.bottom); // The new position is the sum of half the remaining // space and the owner's original position. SetWindowPos(hDlg, HWND_TOP, rcOwner.left + (rc.right / 2), rcOwner.top + (rc.bottom / 2), 0, 0, // ignores size arguments SWP_NOSIZE); } static void SetItemText(HWND hwnd, const WCHAR *key, const WCHAR *ini) { WCHAR text[MAX_INFO_LENGTH]; if (!GetPrivateProfileStringW(L"Strings", key, NULL, text, sizeof(text), ini)) return; SetWindowTextW(hwnd, text); } static void InitDialog(HWND hDlg) { WCHAR filename[MAX_PATH]; if (!GetStringsFile(filename)) return; SetItemText(hDlg, L"Title", filename); SetItemText(GetDlgItem(hDlg, IDC_INFO), L"Info", filename); // On Win9x, we need to send WM_SETFONT for l10n builds. Yes, we shouldn't // use the system font. For example, if the text has Japanese characters on // Win98-en, then the text may not be displayed correctly. We should perhaps // support loading a font named in updater.ini; however, even then there are // cases where it might not work properly. if (!sSystemFont) { NONCLIENTMETRICS ncm; memset(&ncm, 0, sizeof(ncm)); ncm.cbSize = sizeof(ncm); SystemParametersInfo(SPI_GETNONCLIENTMETRICS, 0, &ncm, 0); sSystemFont = CreateFontIndirect(&ncm.lfMessageFont); } if (sSystemFont) SendDlgItemMessage(hDlg, IDC_INFO, WM_SETFONT, (WPARAM)sSystemFont, 0L); // Set dialog icon HICON hIcon = LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDI_DIALOG)); if (hIcon) SendMessage(hDlg, WM_SETICON, ICON_BIG, (LPARAM) hIcon); SendDlgItemMessage(hDlg, IDC_PROGRESS, PBM_SETRANGE, 0, MAKELPARAM(0, 100)); // Resize dialog to fit all the text. ResizeDialogToFit(hDlg); CenterDialog(hDlg); // make dialog appear in the center of the screen SetTimer(hDlg, TIMER_ID, TIMER_INTERVAL, NULL); } // Message handler for about box. static LRESULT CALLBACK DialogProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_INITDIALOG: InitDialog(hDlg); return TRUE; case WM_TIMER: if (sQuit) { EndDialog(hDlg, 0); if (sSystemFont) { DeleteObject(sSystemFont); sSystemFont = 0; } } else { UpdateDialog(hDlg); } return TRUE; case WM_COMMAND: return TRUE; } return FALSE; } int InitProgressUI(int *argc, NS_tchar ***argv) { return 0; } int ShowProgressUI() { // Only show the Progress UI if the process is taking significant time. // Here we measure significant time as taking more than one second. Sleep(500); if (sQuit || sProgress > 50.0f) return 0; // If we do not have updater.ini, then we should not bother showing UI. WCHAR filename[MAX_PATH]; if (!GetStringsFile(filename)) return -1; if (_waccess(filename, 04)) return -1; INITCOMMONCONTROLSEX icc = { sizeof(INITCOMMONCONTROLSEX), ICC_PROGRESS_CLASS }; InitCommonControlsEx(&icc); DialogBox(GetModuleHandle(NULL), MAKEINTRESOURCE(IDD_DIALOG), NULL, (DLGPROC) DialogProc); return 0; } void QuitProgressUI() { sQuit = TRUE; } void UpdateProgressUI(float progress) { sProgress = progress; // 32-bit writes are atomic }