windows/defdlg.c

   1 /*
   2  * Default dialog procedure
   3  *
   4  * Copyright 1993, 1996 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "wingdi.h"
  26 #include "wine/winuser16.h"
  27 #include "controls.h"
  28 #include "win.h"
  29 #include "winproc.h"
  30 #include "wine/debug.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(dialog);
  33
  34
  35 /***********************************************************************
  36  *           DEFDLG_GetDlgProc
  37  */
  38 static WNDPROC DEFDLG_GetDlgProc( HWND hwnd )
  39 {
  40     WNDPROC ret;
  41     WND *wndPtr = WIN_GetPtr( hwnd );
  42
  43     if (!wndPtr) return 0;
  44     if (wndPtr == WND_OTHER_PROCESS)
  45     {
  46         ERR( "cannot get dlg proc %p from other process\n", hwnd );
  47         return 0;
  48     }
  49     ret = *(WNDPROC *)((char *)wndPtr->wExtra + DWL_DLGPROC);
  50     WIN_ReleasePtr( wndPtr );
  51     return ret;
  52 }
  53
  54 /***********************************************************************
  55  *           DEFDLG_SetFocus
  56  *
  57  * Set the focus to a control of the dialog, selecting the text if
  58  * the control is an edit dialog.
  59  */
  60 static void DEFDLG_SetFocus( HWND hwndDlg, HWND hwndCtrl )
  61 {
  62     HWND hwndPrev = GetFocus();
  63
  64     if (IsChild( hwndDlg, hwndPrev ))
  65     {
  66         if (SendMessageW( hwndPrev, WM_GETDLGCODE, 0, 0 ) & DLGC_HASSETSEL)
  67             SendMessageW( hwndPrev, EM_SETSEL, -1, 0 );
  68     }
  69     if (SendMessageW( hwndCtrl, WM_GETDLGCODE, 0, 0 ) & DLGC_HASSETSEL)
  70         SendMessageW( hwndCtrl, EM_SETSEL, 0, -1 );
  71     SetFocus( hwndCtrl );
  72 }
  73
  74
  75 /***********************************************************************
  76  *           DEFDLG_SaveFocus
  77  */
  78 static void DEFDLG_SaveFocus( HWND hwnd )
  79 {
  80     DIALOGINFO *infoPtr;
  81     HWND hwndFocus = GetFocus();
  82
  83     if (!hwndFocus || !IsChild( hwnd, hwndFocus )) return;
  84     if (!(infoPtr = DIALOG_get_info( hwnd ))) return;
  85     infoPtr->hwndFocus = hwndFocus;
  86     /* Remove default button */
  87 }
  88
  89
  90 /***********************************************************************
  91  *           DEFDLG_RestoreFocus
  92  */
  93 static void DEFDLG_RestoreFocus( HWND hwnd )
  94 {
  95     DIALOGINFO *infoPtr;
  96
  97     if (IsIconic( hwnd )) return;
  98     if (!(infoPtr = DIALOG_get_info( hwnd ))) return;
  99     if (!IsWindow( infoPtr->hwndFocus )) return;
 100     /* Don't set the focus back to controls if EndDialog is already called.*/
 101     if (!(infoPtr->flags & DF_END))
 102     {
 103         DEFDLG_SetFocus( hwnd, infoPtr->hwndFocus );
 104         return;
 105     }
 106     /* This used to set infoPtr->hwndFocus to NULL for no apparent reason,
 107        sometimes losing focus when receiving WM_SETFOCUS messages. */
 108 }
 109
 110
 111 /***********************************************************************
 112  *           DEFDLG_FindDefButton
 113  *
 114  * Find the current default push-button.
 115  */
 116 static HWND DEFDLG_FindDefButton( HWND hwndDlg )
 117 {
 118     HWND hwndChild = GetWindow( hwndDlg, GW_CHILD );
 119     while (hwndChild)
 120     {
 121         if (SendMessageW( hwndChild, WM_GETDLGCODE, 0, 0 ) & DLGC_DEFPUSHBUTTON)
 122             break;
 123         hwndChild = GetWindow( hwndChild, GW_HWNDNEXT );
 124     }
 125     return hwndChild;
 126 }
 127
 128
 129 /***********************************************************************
 130  *           DEFDLG_SetDefButton
 131  *
 132  * Set the new default button to be hwndNew.
 133  */
 134 static BOOL DEFDLG_SetDefButton( HWND hwndDlg, DIALOGINFO *dlgInfo, WPARAM wParam )
 135 {
 136     DWORD dlgcode=0; /* initialize just to avoid a warning */
 137     HWND hwndNew = GetDlgItem(hwndDlg, wParam);
 138
 139     dlgInfo->idResult = wParam;
 140     if (hwndNew &&
 141         !((dlgcode=SendMessageW(hwndNew, WM_GETDLGCODE, 0, 0 ))
 142             & (DLGC_UNDEFPUSHBUTTON | DLGC_BUTTON)))
 143         return FALSE;  /* Destination is not a push button */
 144
 145     if (dlgInfo->idResult)  /* There's already a default pushbutton */
 146     {
 147         HWND hwndOld = GetDlgItem( hwndDlg, dlgInfo->idResult );
 148         if (hwndOld && (SendMessageA( hwndOld, WM_GETDLGCODE, 0, 0) & DLGC_DEFPUSHBUTTON))
 149             SendMessageA( hwndOld, BM_SETSTYLE, BS_PUSHBUTTON, TRUE );
 150     }
 151     if (hwndNew)
 152     {
 153         if(dlgcode==DLGC_UNDEFPUSHBUTTON)
 154             SendMessageA( hwndNew, BM_SETSTYLE, BS_DEFPUSHBUTTON, TRUE );
 155     }
 156     return TRUE;
 157 }
 158
 159
 160 /***********************************************************************
 161  *           DEFDLG_Proc
 162  *
 163  * Implementation of DefDlgProc(). Only handle messages that need special
 164  * handling for dialogs.
 165  */
 166 static LRESULT DEFDLG_Proc( HWND hwnd, UINT msg, WPARAM wParam,
 167                             LPARAM lParam, DIALOGINFO *dlgInfo )
 168 {
 169     switch(msg)
 170     {
 171         case WM_ERASEBKGND:
 172         {
 173             HBRUSH brush = (HBRUSH)SendMessageW( hwnd, WM_CTLCOLORDLG, wParam, (LPARAM)hwnd );
 174             if (!brush) brush = (HBRUSH)DefWindowProcW( hwnd, WM_CTLCOLORDLG, wParam, (LPARAM)hwnd );
 175             if (brush)
 176             {
 177                 RECT rect;
 178                 HDC hdc = (HDC)wParam;
 179                 GetClientRect( hwnd, &rect );
 180                 DPtoLP( hdc, (LPPOINT)&rect, 2 );
 181                 FillRect( hdc, &rect, brush );
 182             }
 183             return 1;
 184         }
 185         case WM_NCDESTROY:
 186             if ((dlgInfo = (DIALOGINFO *)SetWindowLongW( hwnd, DWL_WINE_DIALOGINFO, 0 )))
 187             {
 188                 /* Free dialog heap (if created) */
 189                 if (dlgInfo->hDialogHeap)
 190                 {
 191                     GlobalUnlock16(dlgInfo->hDialogHeap);
 192                     GlobalFree16(dlgInfo->hDialogHeap);
 193                 }
 194                 if (dlgInfo->hUserFont) DeleteObject( dlgInfo->hUserFont );
 195                 if (dlgInfo->hMenu) DestroyMenu( dlgInfo->hMenu );
 196                 WINPROC_FreeProc( DEFDLG_GetDlgProc( hwnd ), WIN_PROC_WINDOW );
 197                 HeapFree( GetProcessHeap(), 0, dlgInfo );
 198             }
 199               /* Window clean-up */
 200             return DefWindowProcA( hwnd, msg, wParam, lParam );
 201
 202         case WM_SHOWWINDOW:
 203             if (!wParam) DEFDLG_SaveFocus( hwnd );
 204             return DefWindowProcA( hwnd, msg, wParam, lParam );
 205
 206         case WM_ACTIVATE:
 207             if (wParam) DEFDLG_RestoreFocus( hwnd );
 208             else DEFDLG_SaveFocus( hwnd );
 209             return 0;
 210
 211         case WM_SETFOCUS:
 212             DEFDLG_RestoreFocus( hwnd );
 213             return 0;
 214
 215         case DM_SETDEFID:
 216             if (dlgInfo && !(dlgInfo->flags & DF_END))
 217                 DEFDLG_SetDefButton( hwnd, dlgInfo, wParam );
 218             return 1;
 219
 220         case DM_GETDEFID:
 221             if (dlgInfo && !(dlgInfo->flags & DF_END))
 222             {
 223                 HWND hwndDefId;
 224                 if (dlgInfo->idResult)
 225                     return MAKELONG( dlgInfo->idResult, DC_HASDEFID );
 226                 if ((hwndDefId = DEFDLG_FindDefButton( hwnd )))
 227                     return MAKELONG( GetDlgCtrlID( hwndDefId ), DC_HASDEFID);
 228             }
 229             return 0;
 230
 231         case WM_NEXTDLGCTL:
 232             if (dlgInfo)
 233             {
 234                 HWND hwndDest = (HWND)wParam;
 235                 if (!lParam)
 236                     hwndDest = GetNextDlgTabItem(hwnd, GetFocus(), wParam);
 237                 if (hwndDest) DEFDLG_SetFocus( hwnd, hwndDest );
 238                 DEFDLG_SetDefButton( hwnd, dlgInfo, GetDlgCtrlID(hwndDest) );
 239             }
 240             return 0;
 241
 242         case WM_ENTERMENULOOP:
 243         case WM_LBUTTONDOWN:
 244         case WM_NCLBUTTONDOWN:
 245             {
 246                 HWND hwndFocus = GetFocus();
 247                 if (hwndFocus)
 248                 {
 249                     /* always make combo box hide its listbox control */
 250                     if (!SendMessageA( hwndFocus, CB_SHOWDROPDOWN, FALSE, 0 ))
 251                         SendMessageA( GetParent(hwndFocus), CB_SHOWDROPDOWN, FALSE, 0 );
 252                 }
 253             }
 254             return DefWindowProcA( hwnd, msg, wParam, lParam );
 255
 256         case WM_GETFONT:
 257             return dlgInfo ? (LRESULT)dlgInfo->hUserFont : 0;
 258
 259         case WM_CLOSE:
 260             PostMessageA( hwnd, WM_COMMAND, MAKEWPARAM(IDCANCEL, BN_CLICKED),
 261                             (LPARAM)GetDlgItem( hwnd, IDCANCEL ) );
 262             return 0;
 263
 264         case WM_NOTIFYFORMAT:
 265             return DefWindowProcA( hwnd, msg, wParam, lParam );
 266     }
 267     return 0;
 268 }
 269
 270 /***********************************************************************
 271  *           DEFDLG_Epilog
 272  */
 273 static LRESULT DEFDLG_Epilog(HWND hwnd, UINT msg, BOOL fResult)
 274 {
 275     /* see SDK 3.1 */
 276
 277     if ((msg >= WM_CTLCOLORMSGBOX && msg <= WM_CTLCOLORSTATIC) ||
 278          msg == WM_CTLCOLOR || msg == WM_COMPAREITEM ||
 279          msg == WM_VKEYTOITEM || msg == WM_CHARTOITEM ||
 280          msg == WM_QUERYDRAGICON || msg == WM_INITDIALOG)
 281         return fResult;
 282
 283     return GetWindowLongA( hwnd, DWL_MSGRESULT );
 284 }
 285
 286 /***********************************************************************
 287 *               DEFDLG_InitDlgInfo
 288 *
 289 * Allocate memory for DIALOGINFO structure and store in DWL_DIALOGINFO
 290 * structure. Also flag the window as a dialog type.
 291 */
 292 static DIALOGINFO* DEFDLG_InitDlgInfo(HWND hwnd)
 293 {
 294     WND* wndPtr;
 295     DIALOGINFO* dlgInfo = DIALOG_get_info( hwnd );
 296     if(!dlgInfo)
 297     {
 298         if (!(dlgInfo = HeapAlloc( GetProcessHeap(), 0, sizeof(*dlgInfo) ))) return NULL;
 299         dlgInfo->hwndFocus   = 0;
 300         dlgInfo->hUserFont   = 0;
 301         dlgInfo->hMenu       = 0;
 302         dlgInfo->xBaseUnit   = 0;
 303         dlgInfo->yBaseUnit   = 0;
 304         dlgInfo->idResult    = 0;
 305         dlgInfo->flags       = 0;
 306         dlgInfo->hDialogHeap = 0;
 307         wndPtr = WIN_GetPtr( hwnd );
 308         if (wndPtr && wndPtr != WND_OTHER_PROCESS)
 309         {
 310             wndPtr->flags |= WIN_ISDIALOG;
 311             WIN_ReleasePtr( wndPtr );
 312             SetWindowLongW( hwnd, DWL_WINE_DIALOGINFO, (LONG)dlgInfo );
 313         }
 314         else
 315         {
 316             HeapFree( GetProcessHeap(), 0, dlgInfo );
 317             return NULL;
 318         }
 319     }
 320     return dlgInfo;
 321 }
 322
 323 /***********************************************************************
 324  *              DefDlgProc (USER.308)
 325  */
 326 LRESULT WINAPI DefDlgProc16( HWND16 hwnd, UINT16 msg, WPARAM16 wParam,
 327                              LPARAM lParam )
 328 {
 329     DIALOGINFO *dlgInfo;
 330     WNDPROC16 dlgproc;
 331     HWND hwnd32 = WIN_Handle32( hwnd );
 332     BOOL result = FALSE;
 333
 334     /* Perform DIALOGINFO intialization if not done */
 335     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd32))) return -1;
 336
 337     SetWindowLongW( hwnd32, DWL_MSGRESULT, 0 );
 338
 339     if ((dlgproc = (WNDPROC16)DEFDLG_GetDlgProc( hwnd32 )))
 340     {
 341         /* Call dialog procedure */
 342         result = CallWindowProc16( dlgproc, hwnd, msg, wParam, lParam );
 343         /* 16 bit dlg procs only return BOOL16 */
 344         if( WINPROC_GetProcType( (WNDPROC)dlgproc ) == WIN_PROC_16 )
 345             result = LOWORD(result);
 346     }
 347
 348     if (!result && IsWindow(hwnd32))
 349     {
 350         /* callback didn't process this message */
 351
 352         switch(msg)
 353         {
 354             case WM_ERASEBKGND:
 355             case WM_SHOWWINDOW:
 356             case WM_ACTIVATE:
 357             case WM_SETFOCUS:
 358             case DM_SETDEFID:
 359             case DM_GETDEFID:
 360             case WM_NEXTDLGCTL:
 361             case WM_GETFONT:
 362             case WM_CLOSE:
 363             case WM_NCDESTROY:
 364             case WM_ENTERMENULOOP:
 365             case WM_LBUTTONDOWN:
 366             case WM_NCLBUTTONDOWN:
 367                 return DEFDLG_Proc( hwnd32, msg, (WPARAM)wParam, lParam, dlgInfo );
 368             case WM_INITDIALOG:
 369             case WM_VKEYTOITEM:
 370             case WM_COMPAREITEM:
 371             case WM_CHARTOITEM:
 372                 break;
 373
 374             default:
 375                 return DefWindowProc16( hwnd, msg, wParam, lParam );
 376         }
 377     }
 378     return DEFDLG_Epilog( hwnd32, msg, result);
 379 }
 380
 381
 382 /***********************************************************************
 383  *              DefDlgProcA (USER32.@)
 384  */
 385 LRESULT WINAPI DefDlgProcA( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam )
 386 {
 387     DIALOGINFO *dlgInfo;
 388     WNDPROC dlgproc;
 389     BOOL result = FALSE;
 390
 391     /* Perform DIALOGINFO initialization if not done */
 392     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd))) return -1;
 393
 394     SetWindowLongW( hwnd, DWL_MSGRESULT, 0 );
 395
 396     if ((dlgproc = DEFDLG_GetDlgProc( hwnd )))
 397     {
 398         /* Call dialog procedure */
 399         result = CallWindowProcA( dlgproc, hwnd, msg, wParam, lParam );
 400         /* 16 bit dlg procs only return BOOL16 */
 401         if( WINPROC_GetProcType( dlgproc ) == WIN_PROC_16 )
 402             result = LOWORD(result);
 403     }
 404
 405     if (!result && IsWindow(hwnd))
 406     {
 407         /* callback didn't process this message */
 408
 409         switch(msg)
 410         {
 411             case WM_ERASEBKGND:
 412             case WM_SHOWWINDOW:
 413             case WM_ACTIVATE:
 414             case WM_SETFOCUS:
 415             case DM_SETDEFID:
 416             case DM_GETDEFID:
 417             case WM_NEXTDLGCTL:
 418             case WM_GETFONT:
 419             case WM_CLOSE:
 420             case WM_NCDESTROY:
 421             case WM_ENTERMENULOOP:
 422             case WM_LBUTTONDOWN:
 423             case WM_NCLBUTTONDOWN:
 424                  return DEFDLG_Proc( hwnd, msg, wParam, lParam, dlgInfo );
 425             case WM_INITDIALOG:
 426             case WM_VKEYTOITEM:
 427             case WM_COMPAREITEM:
 428             case WM_CHARTOITEM:
 429                  break;
 430
 431             default:
 432                  return DefWindowProcA( hwnd, msg, wParam, lParam );
 433         }
 434     }
 435     return DEFDLG_Epilog(hwnd, msg, result);
 436 }
 437
 438
 439 /***********************************************************************
 440  *              DefDlgProcW (USER32.@)
 441  */
 442 LRESULT WINAPI DefDlgProcW( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam )
 443 {
 444     DIALOGINFO *dlgInfo;
 445     BOOL result = FALSE;
 446     WNDPROC dlgproc;
 447
 448     /* Perform DIALOGINFO intialization if not done */
 449     if(!(dlgInfo = DEFDLG_InitDlgInfo(hwnd))) return -1;
 450
 451     SetWindowLongW( hwnd, DWL_MSGRESULT, 0 );
 452
 453     if ((dlgproc = DEFDLG_GetDlgProc( hwnd )))
 454     {
 455         /* Call dialog procedure */
 456         result = CallWindowProcW( dlgproc, hwnd, msg, wParam, lParam );
 457         /* 16 bit dlg procs only return BOOL16 */
 458         if( WINPROC_GetProcType( dlgproc ) == WIN_PROC_16 )
 459             result = LOWORD(result);
 460     }
 461
 462     if (!result && IsWindow(hwnd))
 463     {
 464         /* callback didn't process this message */
 465
 466         switch(msg)
 467         {
 468             case WM_ERASEBKGND:
 469             case WM_SHOWWINDOW:
 470             case WM_ACTIVATE:
 471             case WM_SETFOCUS:
 472             case DM_SETDEFID:
 473             case DM_GETDEFID:
 474             case WM_NEXTDLGCTL:
 475             case WM_GETFONT:
 476             case WM_CLOSE:
 477             case WM_NCDESTROY:
 478             case WM_ENTERMENULOOP:
 479             case WM_LBUTTONDOWN:
 480             case WM_NCLBUTTONDOWN:
 481                  return DEFDLG_Proc( hwnd, msg, wParam, lParam, dlgInfo );
 482             case WM_INITDIALOG:
 483             case WM_VKEYTOITEM:
 484             case WM_COMPAREITEM:
 485             case WM_CHARTOITEM:
 486                  break;
 487
 488             default:
 489                  return DefWindowProcW( hwnd, msg, wParam, lParam );
 490         }
 491     }
 492     return DEFDLG_Epilog(hwnd, msg, result);
 493 }