mfreadwrite/reader: Add missing allocation check (Coverity).
[wine/zf.git] / dlls / comctl32 / updown.c
bloba804018d5f8f9406baef47152d790cfd8d914a6c
1 /*
2 * Updown control
4 * Copyright 1997, 2002 Dimitrie O. Paun
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.
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.
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include <assert.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <stdarg.h>
25 #include <stdio.h>
27 #include "windef.h"
28 #include "winbase.h"
29 #include "wingdi.h"
30 #include "winuser.h"
31 #include "winnls.h"
32 #include "commctrl.h"
33 #include "comctl32.h"
34 #include "uxtheme.h"
35 #include "vssym32.h"
36 #include "wine/heap.h"
37 #include "wine/debug.h"
39 WINE_DEFAULT_DEBUG_CHANNEL(updown);
41 typedef struct
43 HWND Self; /* Handle to this up-down control */
44 HWND Notify; /* Handle to the parent window */
45 DWORD dwStyle; /* The GWL_STYLE for this window */
46 UINT AccelCount; /* Number of elements in AccelVect */
47 UDACCEL* AccelVect; /* Vector containing AccelCount elements */
48 INT AccelIndex; /* Current accel index, -1 if not accel'ing */
49 INT Base; /* Base to display nr in the buddy window */
50 INT CurVal; /* Current up-down value */
51 INT MinVal; /* Minimum up-down value */
52 INT MaxVal; /* Maximum up-down value */
53 HWND Buddy; /* Handle to the buddy window */
54 INT BuddyType; /* Remembers the buddy type BUDDY_TYPE_* */
55 INT Flags; /* Internal Flags FLAG_* */
56 BOOL UnicodeFormat; /* Marks the use of Unicode internally */
57 } UPDOWN_INFO;
59 /* Control configuration constants */
61 #define INITIAL_DELAY 500 /* initial timer until auto-inc kicks in */
62 #define AUTOPRESS_DELAY 250 /* time to keep arrow pressed on KEY_DOWN */
63 #define REPEAT_DELAY 50 /* delay between auto-increments */
65 #define DEFAULT_WIDTH 16 /* default width of the ctrl */
66 #define DEFAULT_XSEP 0 /* default separation between buddy and ctrl */
67 #define DEFAULT_ADDTOP 0 /* amount to extend above the buddy window */
68 #define DEFAULT_ADDBOT 0 /* amount to extend below the buddy window */
69 #define DEFAULT_BUDDYBORDER 2 /* Width/height of the buddy border */
70 #define DEFAULT_BUDDYSPACER 2 /* Spacer between the buddy and the ctrl */
71 #define DEFAULT_BUDDYBORDER_THEMED 1 /* buddy border when theming is enabled */
72 #define DEFAULT_BUDDYSPACER_THEMED 0 /* buddy spacer when theming is enabled */
74 /* Work constants */
76 #define FLAG_INCR 0x01
77 #define FLAG_DECR 0x02
78 #define FLAG_MOUSEIN 0x04
79 #define FLAG_PRESSED 0x08
80 #define FLAG_BUDDYINT 0x10 /* UDS_SETBUDDYINT was set on creation */
81 #define FLAG_ARROW (FLAG_INCR | FLAG_DECR)
83 #define BUDDY_TYPE_UNKNOWN 0
84 #define BUDDY_TYPE_LISTBOX 1
85 #define BUDDY_TYPE_EDIT 2
87 #define TIMER_AUTOREPEAT 1
88 #define TIMER_ACCEL 2
89 #define TIMER_AUTOPRESS 3
91 #define UPDOWN_GetInfoPtr(hwnd) ((UPDOWN_INFO *)GetWindowLongPtrW (hwnd,0))
93 /* id used for SetWindowSubclass */
94 #define BUDDY_SUBCLASSID 1
96 static void UPDOWN_DoAction (UPDOWN_INFO *infoPtr, int delta, int action);
98 /***********************************************************************
99 * UPDOWN_IsBuddyEdit
100 * Tests if our buddy is an edit control.
102 static inline BOOL UPDOWN_IsBuddyEdit(const UPDOWN_INFO *infoPtr)
104 return infoPtr->BuddyType == BUDDY_TYPE_EDIT;
107 /***********************************************************************
108 * UPDOWN_IsBuddyListbox
109 * Tests if our buddy is a listbox control.
111 static inline BOOL UPDOWN_IsBuddyListbox(const UPDOWN_INFO *infoPtr)
113 return infoPtr->BuddyType == BUDDY_TYPE_LISTBOX;
116 /***********************************************************************
117 * UPDOWN_InBounds
118 * Tests if a given value 'val' is between the Min&Max limits
120 static BOOL UPDOWN_InBounds(const UPDOWN_INFO *infoPtr, int val)
122 if(infoPtr->MaxVal > infoPtr->MinVal)
123 return (infoPtr->MinVal <= val) && (val <= infoPtr->MaxVal);
124 else
125 return (infoPtr->MaxVal <= val) && (val <= infoPtr->MinVal);
128 /***********************************************************************
129 * UPDOWN_OffsetVal
130 * Change the current value by delta.
131 * It returns TRUE is the value was changed successfully, or FALSE
132 * if the value was not changed, as it would go out of bounds.
134 static BOOL UPDOWN_OffsetVal(UPDOWN_INFO *infoPtr, int delta)
136 /* check if we can do the modification first */
137 if(!UPDOWN_InBounds (infoPtr, infoPtr->CurVal+delta)) {
138 if (infoPtr->dwStyle & UDS_WRAP) {
139 delta += (delta < 0 ? -1 : 1) *
140 (infoPtr->MaxVal < infoPtr->MinVal ? -1 : 1) *
141 (infoPtr->MinVal - infoPtr->MaxVal) +
142 (delta < 0 ? 1 : -1);
143 } else if ((infoPtr->MaxVal > infoPtr->MinVal && infoPtr->CurVal+delta > infoPtr->MaxVal)
144 || (infoPtr->MaxVal < infoPtr->MinVal && infoPtr->CurVal+delta < infoPtr->MaxVal)) {
145 delta = infoPtr->MaxVal - infoPtr->CurVal;
146 } else {
147 delta = infoPtr->MinVal - infoPtr->CurVal;
151 infoPtr->CurVal += delta;
152 return delta != 0;
155 /***********************************************************************
156 * UPDOWN_HasBuddyBorder
158 * When we have a buddy set and that we are aligned on our buddy, we
159 * want to draw a sunken edge to make like we are part of that control.
161 static BOOL UPDOWN_HasBuddyBorder(const UPDOWN_INFO *infoPtr)
163 return ( ((infoPtr->dwStyle & (UDS_ALIGNLEFT | UDS_ALIGNRIGHT)) != 0) &&
164 UPDOWN_IsBuddyEdit(infoPtr) );
167 /***********************************************************************
168 * UPDOWN_GetArrowRect
169 * wndPtr - pointer to the up-down wnd
170 * rect - will hold the rectangle
171 * arrow - FLAG_INCR to get the "increment" rect (up or right)
172 * FLAG_DECR to get the "decrement" rect (down or left)
174 static void UPDOWN_GetArrowRect (const UPDOWN_INFO* infoPtr, RECT *rect, unsigned int arrow)
176 HTHEME theme = GetWindowTheme (infoPtr->Self);
177 const int border = theme ? DEFAULT_BUDDYBORDER_THEMED : DEFAULT_BUDDYBORDER;
178 const int spacer = theme ? DEFAULT_BUDDYSPACER_THEMED : DEFAULT_BUDDYSPACER;
179 int size;
181 assert(arrow && (arrow & (FLAG_INCR | FLAG_DECR)) != (FLAG_INCR | FLAG_DECR));
183 GetClientRect (infoPtr->Self, rect);
186 * Make sure we calculate the rectangle to fit even if we draw the
187 * border.
189 if (UPDOWN_HasBuddyBorder(infoPtr)) {
190 if (infoPtr->dwStyle & UDS_ALIGNLEFT)
191 rect->left += border;
192 else
193 rect->right -= border;
195 InflateRect(rect, 0, -border);
198 /* now figure out if we need a space away from the buddy */
199 if (IsWindow(infoPtr->Buddy) ) {
200 if (infoPtr->dwStyle & UDS_ALIGNLEFT) rect->right -= spacer;
201 else if (infoPtr->dwStyle & UDS_ALIGNRIGHT) rect->left += spacer;
205 * We're calculating the midpoint to figure-out where the
206 * separation between the buttons will lay.
208 if (infoPtr->dwStyle & UDS_HORZ) {
209 size = (rect->right - rect->left) / 2;
210 if (arrow & FLAG_INCR)
211 rect->left = rect->right - size;
212 else if (arrow & FLAG_DECR)
213 rect->right = rect->left + size;
214 } else {
215 size = (rect->bottom - rect->top) / 2;
216 if (arrow & FLAG_INCR)
217 rect->bottom = rect->top + size;
218 else if (arrow & FLAG_DECR)
219 rect->top = rect->bottom - size;
223 /***********************************************************************
224 * UPDOWN_GetArrowFromPoint
225 * Returns the rectagle (for the up or down arrow) that contains pt.
226 * If it returns the up rect, it returns FLAG_INCR.
227 * If it returns the down rect, it returns FLAG_DECR.
229 static INT UPDOWN_GetArrowFromPoint (const UPDOWN_INFO *infoPtr, RECT *rect, POINT pt)
231 UPDOWN_GetArrowRect (infoPtr, rect, FLAG_INCR);
232 if(PtInRect(rect, pt)) return FLAG_INCR;
234 UPDOWN_GetArrowRect (infoPtr, rect, FLAG_DECR);
235 if(PtInRect(rect, pt)) return FLAG_DECR;
237 return 0;
241 /***********************************************************************
242 * UPDOWN_GetThousandSep
243 * Returns the thousand sep. If an error occurs, it returns ','.
245 static WCHAR UPDOWN_GetThousandSep(void)
247 WCHAR sep[2];
249 if(GetLocaleInfoW(LOCALE_USER_DEFAULT, LOCALE_STHOUSAND, sep, 2) != 1)
250 sep[0] = ',';
252 return sep[0];
255 /***********************************************************************
256 * UPDOWN_GetBuddyInt
257 * Tries to read the pos from the buddy window and if it succeeds,
258 * it stores it in the control's CurVal
259 * returns:
260 * TRUE - if it read the integer from the buddy successfully
261 * FALSE - if an error occurred
263 static BOOL UPDOWN_GetBuddyInt (UPDOWN_INFO *infoPtr)
265 WCHAR txt[20], sep, *src, *dst;
266 int newVal;
268 if (!((infoPtr->Flags & FLAG_BUDDYINT) && IsWindow(infoPtr->Buddy)))
269 return FALSE;
271 /*if the buddy is a list window, we must set curr index */
272 if (UPDOWN_IsBuddyListbox(infoPtr)) {
273 newVal = SendMessageW(infoPtr->Buddy, LB_GETCARETINDEX, 0, 0);
274 if(newVal < 0) return FALSE;
275 } else {
276 /* we have a regular window, so will get the text */
277 /* note that a zero-length string is a legitimate value for 'txt',
278 * and ought to result in a successful conversion to '0'. */
279 if (GetWindowTextW(infoPtr->Buddy, txt, ARRAY_SIZE(txt)) < 0)
280 return FALSE;
282 sep = UPDOWN_GetThousandSep();
284 /* now get rid of the separators */
285 for(src = dst = txt; *src; src++)
286 if(*src != sep) *dst++ = *src;
287 *dst = 0;
289 /* try to convert the number and validate it */
290 newVal = wcstol(txt, &src, infoPtr->Base);
291 if(*src || !UPDOWN_InBounds (infoPtr, newVal)) return FALSE;
294 TRACE("new value(%d) from buddy (old=%d)\n", newVal, infoPtr->CurVal);
295 infoPtr->CurVal = newVal;
296 return TRUE;
300 /***********************************************************************
301 * UPDOWN_SetBuddyInt
302 * Tries to set the pos to the buddy window based on current pos
303 * returns:
304 * TRUE - if it set the caption of the buddy successfully
305 * FALSE - if an error occurred
307 static BOOL UPDOWN_SetBuddyInt (const UPDOWN_INFO *infoPtr)
309 const WCHAR *fmt;
310 WCHAR txt[20], txt_old[20] = { 0 };
311 int len;
313 if (!((infoPtr->Flags & FLAG_BUDDYINT) && IsWindow(infoPtr->Buddy)))
314 return FALSE;
316 TRACE("set new value(%d) to buddy.\n", infoPtr->CurVal);
318 /*if the buddy is a list window, we must set curr index */
319 if (UPDOWN_IsBuddyListbox(infoPtr)) {
320 return SendMessageW(infoPtr->Buddy, LB_SETCURSEL, infoPtr->CurVal, 0) != LB_ERR;
323 /* Regular window, so set caption to the number */
324 fmt = (infoPtr->Base == 16) ? L"0x%04X" : L"%d";
325 len = wsprintfW(txt, fmt, infoPtr->CurVal);
328 /* Do thousands separation if necessary */
329 if ((infoPtr->Base == 10) && !(infoPtr->dwStyle & UDS_NOTHOUSANDS) && (len > 3)) {
330 WCHAR tmp[ARRAY_SIZE(txt)], *src = tmp, *dst = txt;
331 WCHAR sep = UPDOWN_GetThousandSep();
332 int start = len % 3;
334 memcpy(tmp, txt, sizeof(txt));
335 if (start == 0) start = 3;
336 dst += start;
337 src += start;
338 for (len=0; *src; len++) {
339 if (len % 3 == 0) *dst++ = sep;
340 *dst++ = *src++;
342 *dst = 0;
345 /* if nothing changed exit earlier */
346 GetWindowTextW(infoPtr->Buddy, txt_old, ARRAY_SIZE(txt_old));
347 if (lstrcmpiW(txt_old, txt) == 0) return FALSE;
349 return SetWindowTextW(infoPtr->Buddy, txt);
352 /***********************************************************************
353 * UPDOWN_DrawBuddyBackground
355 * Draw buddy background for visual integration.
357 static BOOL UPDOWN_DrawBuddyBackground (const UPDOWN_INFO *infoPtr, HDC hdc)
359 RECT br, r;
360 HTHEME buddyTheme = GetWindowTheme (infoPtr->Buddy);
361 if (!buddyTheme) return FALSE;
363 GetWindowRect (infoPtr->Buddy, &br);
364 MapWindowPoints (NULL, infoPtr->Self, (POINT*)&br, 2);
365 GetClientRect (infoPtr->Self, &r);
367 if (infoPtr->dwStyle & UDS_ALIGNLEFT)
368 br.left = r.left;
369 else if (infoPtr->dwStyle & UDS_ALIGNRIGHT)
370 br.right = r.right;
371 /* FIXME: take disabled etc. into account */
372 DrawThemeBackground (buddyTheme, hdc, 0, 0, &br, NULL);
373 return TRUE;
376 /***********************************************************************
377 * UPDOWN_Draw
379 * Draw the arrows. The background need not be erased.
381 static LRESULT UPDOWN_Draw (const UPDOWN_INFO *infoPtr, HDC hdc)
383 BOOL uPressed, uHot, dPressed, dHot;
384 RECT rect;
385 HTHEME theme = GetWindowTheme (infoPtr->Self);
386 int uPart = 0, uState = 0, dPart = 0, dState = 0;
387 BOOL needBuddyBg = FALSE;
389 uPressed = (infoPtr->Flags & FLAG_PRESSED) && (infoPtr->Flags & FLAG_INCR);
390 uHot = (infoPtr->Flags & FLAG_INCR) && (infoPtr->Flags & FLAG_MOUSEIN);
391 dPressed = (infoPtr->Flags & FLAG_PRESSED) && (infoPtr->Flags & FLAG_DECR);
392 dHot = (infoPtr->Flags & FLAG_DECR) && (infoPtr->Flags & FLAG_MOUSEIN);
393 if (theme) {
394 uPart = (infoPtr->dwStyle & UDS_HORZ) ? SPNP_UPHORZ : SPNP_UP;
395 uState = (infoPtr->dwStyle & WS_DISABLED) ? DNS_DISABLED
396 : (uPressed ? DNS_PRESSED : (uHot ? DNS_HOT : DNS_NORMAL));
397 dPart = (infoPtr->dwStyle & UDS_HORZ) ? SPNP_DOWNHORZ : SPNP_DOWN;
398 dState = (infoPtr->dwStyle & WS_DISABLED) ? DNS_DISABLED
399 : (dPressed ? DNS_PRESSED : (dHot ? DNS_HOT : DNS_NORMAL));
400 needBuddyBg = IsWindow (infoPtr->Buddy)
401 && (IsThemeBackgroundPartiallyTransparent (theme, uPart, uState)
402 || IsThemeBackgroundPartiallyTransparent (theme, dPart, dState));
405 /* Draw the common border between ourselves and our buddy */
406 if (UPDOWN_HasBuddyBorder(infoPtr) || needBuddyBg) {
407 if (!theme || !UPDOWN_DrawBuddyBackground (infoPtr, hdc)) {
408 GetClientRect(infoPtr->Self, &rect);
409 DrawEdge(hdc, &rect, EDGE_SUNKEN,
410 BF_BOTTOM | BF_TOP |
411 (infoPtr->dwStyle & UDS_ALIGNLEFT ? BF_LEFT : BF_RIGHT));
415 /* Draw the incr button */
416 UPDOWN_GetArrowRect (infoPtr, &rect, FLAG_INCR);
417 if (theme) {
418 DrawThemeBackground(theme, hdc, uPart, uState, &rect, NULL);
419 } else {
420 DrawFrameControl(hdc, &rect, DFC_SCROLL,
421 (infoPtr->dwStyle & UDS_HORZ ? DFCS_SCROLLRIGHT : DFCS_SCROLLUP) |
422 ((infoPtr->dwStyle & UDS_HOTTRACK) && uHot ? DFCS_HOT : 0) |
423 (uPressed ? DFCS_PUSHED : 0) |
424 (infoPtr->dwStyle & WS_DISABLED ? DFCS_INACTIVE : 0) );
427 /* Draw the decr button */
428 UPDOWN_GetArrowRect(infoPtr, &rect, FLAG_DECR);
429 if (theme) {
430 DrawThemeBackground(theme, hdc, dPart, dState, &rect, NULL);
431 } else {
432 DrawFrameControl(hdc, &rect, DFC_SCROLL,
433 (infoPtr->dwStyle & UDS_HORZ ? DFCS_SCROLLLEFT : DFCS_SCROLLDOWN) |
434 ((infoPtr->dwStyle & UDS_HOTTRACK) && dHot ? DFCS_HOT : 0) |
435 (dPressed ? DFCS_PUSHED : 0) |
436 (infoPtr->dwStyle & WS_DISABLED ? DFCS_INACTIVE : 0) );
439 return 0;
442 /***********************************************************************
443 * UPDOWN_Paint
445 * Asynchronous drawing (must ONLY be used in WM_PAINT).
446 * Calls UPDOWN_Draw.
448 static LRESULT UPDOWN_Paint (const UPDOWN_INFO *infoPtr, HDC hdc)
450 PAINTSTRUCT ps;
451 if (hdc) return UPDOWN_Draw (infoPtr, hdc);
452 hdc = BeginPaint (infoPtr->Self, &ps);
453 UPDOWN_Draw (infoPtr, hdc);
454 EndPaint (infoPtr->Self, &ps);
455 return 0;
458 /***********************************************************************
459 * UPDOWN_KeyPressed
461 * Handle key presses (up & down) when we have to do so
463 static LRESULT UPDOWN_KeyPressed(UPDOWN_INFO *infoPtr, int key)
465 int arrow, accel;
467 if (key == VK_UP) arrow = FLAG_INCR;
468 else if (key == VK_DOWN) arrow = FLAG_DECR;
469 else return 1;
471 UPDOWN_GetBuddyInt (infoPtr);
472 infoPtr->Flags &= ~FLAG_ARROW;
473 infoPtr->Flags |= FLAG_PRESSED | arrow;
474 InvalidateRect (infoPtr->Self, NULL, FALSE);
475 SetTimer(infoPtr->Self, TIMER_AUTOPRESS, AUTOPRESS_DELAY, 0);
476 accel = (infoPtr->AccelCount && infoPtr->AccelVect) ? infoPtr->AccelVect[0].nInc : 1;
477 UPDOWN_DoAction (infoPtr, accel, arrow);
478 return 0;
481 static int UPDOWN_GetPos(UPDOWN_INFO *infoPtr, BOOL *err)
483 BOOL succ = UPDOWN_GetBuddyInt(infoPtr);
484 int val = infoPtr->CurVal;
486 if(!UPDOWN_InBounds(infoPtr, val)) {
487 if((infoPtr->MinVal < infoPtr->MaxVal && val < infoPtr->MinVal)
488 || (infoPtr->MinVal > infoPtr->MaxVal && val > infoPtr->MinVal))
489 val = infoPtr->MinVal;
490 else
491 val = infoPtr->MaxVal;
493 succ = FALSE;
496 if(err) *err = !succ;
497 return val;
500 static int UPDOWN_SetPos(UPDOWN_INFO *infoPtr, int pos)
502 int ret = infoPtr->CurVal;
504 if(!UPDOWN_InBounds(infoPtr, pos)) {
505 if((infoPtr->MinVal < infoPtr->MaxVal && pos < infoPtr->MinVal)
506 || (infoPtr->MinVal > infoPtr->MaxVal && pos > infoPtr->MinVal))
507 pos = infoPtr->MinVal;
508 else
509 pos = infoPtr->MaxVal;
512 infoPtr->CurVal = pos;
513 UPDOWN_SetBuddyInt(infoPtr);
515 if(!UPDOWN_InBounds(infoPtr, ret)) {
516 if((infoPtr->MinVal < infoPtr->MaxVal && ret < infoPtr->MinVal)
517 || (infoPtr->MinVal > infoPtr->MaxVal && ret > infoPtr->MinVal))
518 ret = infoPtr->MinVal;
519 else
520 ret = infoPtr->MaxVal;
522 return ret;
526 /***********************************************************************
527 * UPDOWN_SetRange
529 * Handle UDM_SETRANGE, UDM_SETRANGE32
531 * FIXME: handle Max == Min properly:
532 * - arrows should be disabled (without WS_DISABLED set),
533 * visually they can't be pressed and don't respond;
534 * - all input messages should still pass in.
536 static LRESULT UPDOWN_SetRange(UPDOWN_INFO *infoPtr, INT Max, INT Min)
538 infoPtr->MaxVal = Max;
539 infoPtr->MinVal = Min;
541 TRACE("UpDown Ctrl new range(%d to %d), hwnd=%p\n",
542 infoPtr->MinVal, infoPtr->MaxVal, infoPtr->Self);
544 return 0;
547 /***********************************************************************
548 * UPDOWN_MouseWheel
550 * Handle mouse wheel scrolling
552 static LRESULT UPDOWN_MouseWheel(UPDOWN_INFO *infoPtr, WPARAM wParam)
554 int iWheelDelta = GET_WHEEL_DELTA_WPARAM(wParam) / WHEEL_DELTA;
556 if (wParam & (MK_SHIFT | MK_CONTROL))
557 return 0;
559 if (iWheelDelta != 0)
561 UPDOWN_GetBuddyInt(infoPtr);
562 UPDOWN_DoAction(infoPtr, abs(iWheelDelta), iWheelDelta > 0 ? FLAG_INCR : FLAG_DECR);
565 return 1;
569 /***********************************************************************
570 * UPDOWN_Buddy_SubclassProc used to handle messages sent to the buddy
571 * control.
573 static LRESULT CALLBACK
574 UPDOWN_Buddy_SubclassProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam,
575 UINT_PTR uId, DWORD_PTR ref_data)
577 UPDOWN_INFO *infoPtr = UPDOWN_GetInfoPtr((HWND)ref_data);
579 TRACE("hwnd=%p, uMsg=%04x, wParam=%08lx, lParam=%08lx\n",
580 hwnd, uMsg, wParam, lParam);
582 switch(uMsg)
584 case WM_KEYDOWN:
585 if (infoPtr)
587 UPDOWN_KeyPressed(infoPtr, (int)wParam);
588 if (wParam == VK_UP || wParam == VK_DOWN)
589 return 0;
591 break;
593 case WM_MOUSEWHEEL:
594 if (infoPtr)
595 UPDOWN_MouseWheel(infoPtr, (int)wParam);
596 break;
598 case WM_NCDESTROY:
599 RemoveWindowSubclass(hwnd, UPDOWN_Buddy_SubclassProc, BUDDY_SUBCLASSID);
600 break;
601 default:
602 break;
605 return DefSubclassProc(hwnd, uMsg, wParam, lParam);
608 static void UPDOWN_ResetSubclass (UPDOWN_INFO *infoPtr)
610 SetWindowSubclass(infoPtr->Buddy, UPDOWN_Buddy_SubclassProc, BUDDY_SUBCLASSID, 0);
613 /***********************************************************************
614 * UPDOWN_SetBuddy
616 * Sets bud as a new Buddy.
617 * Then, it should subclass the buddy
618 * If window has the UDS_ARROWKEYS, it subclasses the buddy window to
619 * process the UP/DOWN arrow keys.
620 * If window has the UDS_ALIGNLEFT or UDS_ALIGNRIGHT style
621 * the size/pos of the buddy and the control are adjusted accordingly.
623 static HWND UPDOWN_SetBuddy (UPDOWN_INFO* infoPtr, HWND bud)
625 RECT budRect; /* new coord for the buddy */
626 int x, width; /* new x position and width for the up-down */
627 WCHAR buddyClass[40];
628 HWND old_buddy;
630 TRACE("(hwnd=%p, bud=%p)\n", infoPtr->Self, bud);
632 old_buddy = infoPtr->Buddy;
634 UPDOWN_ResetSubclass (infoPtr);
636 if (!IsWindow(bud)) bud = NULL;
638 /* Store buddy window handle */
639 infoPtr->Buddy = bud;
641 if(bud) {
642 /* Store buddy window class type */
643 infoPtr->BuddyType = BUDDY_TYPE_UNKNOWN;
644 if (GetClassNameW(bud, buddyClass, ARRAY_SIZE(buddyClass))) {
645 if (lstrcmpiW(buddyClass, WC_EDITW) == 0)
646 infoPtr->BuddyType = BUDDY_TYPE_EDIT;
647 else if (lstrcmpiW(buddyClass, WC_LISTBOXW) == 0)
648 infoPtr->BuddyType = BUDDY_TYPE_LISTBOX;
651 if (infoPtr->dwStyle & UDS_ARROWKEYS)
652 SetWindowSubclass(bud, UPDOWN_Buddy_SubclassProc, BUDDY_SUBCLASSID,
653 (DWORD_PTR)infoPtr->Self);
655 /* Get the rect of the buddy relative to its parent */
656 GetWindowRect(infoPtr->Buddy, &budRect);
657 MapWindowPoints(HWND_DESKTOP, GetParent(infoPtr->Buddy), (POINT *)(&budRect.left), 2);
659 /* now do the positioning */
660 if (infoPtr->dwStyle & UDS_ALIGNLEFT) {
661 x = budRect.left;
662 budRect.left += DEFAULT_WIDTH + DEFAULT_XSEP;
663 } else if (infoPtr->dwStyle & UDS_ALIGNRIGHT) {
664 budRect.right -= DEFAULT_WIDTH + DEFAULT_XSEP;
665 x = budRect.right+DEFAULT_XSEP;
666 } else {
667 /* nothing to do */
668 return old_buddy;
671 /* first adjust the buddy to accommodate the up/down */
672 SetWindowPos(infoPtr->Buddy, 0, budRect.left, budRect.top,
673 budRect.right - budRect.left, budRect.bottom - budRect.top,
674 SWP_NOACTIVATE|SWP_NOZORDER);
676 /* now position the up/down */
677 /* Since the UDS_ALIGN* flags were used, */
678 /* we will pick the position and size of the window. */
679 width = DEFAULT_WIDTH;
682 * If the updown has a buddy border, it has to overlap with the buddy
683 * to look as if it is integrated with the buddy control.
684 * We nudge the control or change its size to overlap.
686 if (UPDOWN_HasBuddyBorder(infoPtr)) {
687 if(infoPtr->dwStyle & UDS_ALIGNLEFT)
688 width += DEFAULT_BUDDYBORDER;
689 else
690 x -= DEFAULT_BUDDYBORDER;
693 SetWindowPos(infoPtr->Self, 0, x,
694 budRect.top - DEFAULT_ADDTOP, width,
695 budRect.bottom - budRect.top + DEFAULT_ADDTOP + DEFAULT_ADDBOT,
696 SWP_NOACTIVATE|SWP_FRAMECHANGED|SWP_NOZORDER);
697 } else if (!(infoPtr->dwStyle & UDS_HORZ) && old_buddy != NULL) {
698 RECT rect;
699 GetWindowRect(infoPtr->Self, &rect);
700 MapWindowPoints(HWND_DESKTOP, GetParent(infoPtr->Self), (POINT *)&rect, 2);
701 SetWindowPos(infoPtr->Self, 0, rect.left, rect.top, DEFAULT_WIDTH, rect.bottom - rect.top,
702 SWP_NOACTIVATE|SWP_FRAMECHANGED|SWP_NOZORDER);
705 return old_buddy;
708 /***********************************************************************
709 * UPDOWN_DoAction
711 * This function increments/decrements the CurVal by the
712 * 'delta' amount according to the 'action' flag which can be a
713 * combination of FLAG_INCR and FLAG_DECR
714 * It notifies the parent as required.
715 * It handles wrapping and non-wrapping correctly.
716 * It is assumed that delta>0
718 static void UPDOWN_DoAction (UPDOWN_INFO *infoPtr, int delta, int action)
720 NM_UPDOWN ni;
722 TRACE("%d by %d\n", action, delta);
724 /* check if we can do the modification first */
725 delta *= (action & FLAG_INCR ? 1 : -1) * (infoPtr->MaxVal < infoPtr->MinVal ? -1 : 1);
726 if ( (action & FLAG_INCR) && (action & FLAG_DECR) ) delta = 0;
728 TRACE("current %d, delta: %d\n", infoPtr->CurVal, delta);
730 /* We must notify parent now to obtain permission */
731 ni.iPos = infoPtr->CurVal;
732 ni.iDelta = delta;
733 ni.hdr.hwndFrom = infoPtr->Self;
734 ni.hdr.idFrom = GetWindowLongPtrW (infoPtr->Self, GWLP_ID);
735 ni.hdr.code = UDN_DELTAPOS;
736 if (!SendMessageW(infoPtr->Notify, WM_NOTIFY, ni.hdr.idFrom, (LPARAM)&ni)) {
737 /* Parent said: OK to adjust */
739 /* Now adjust value with (maybe new) delta */
740 if (UPDOWN_OffsetVal (infoPtr, ni.iDelta)) {
741 TRACE("new %d, delta: %d\n", infoPtr->CurVal, ni.iDelta);
743 /* Now take care about our buddy */
744 UPDOWN_SetBuddyInt (infoPtr);
748 /* Also, notify it. This message is sent in any case. */
749 SendMessageW( infoPtr->Notify, (infoPtr->dwStyle & UDS_HORZ) ? WM_HSCROLL : WM_VSCROLL,
750 MAKELONG(SB_THUMBPOSITION, infoPtr->CurVal), (LPARAM)infoPtr->Self);
753 /***********************************************************************
754 * UPDOWN_IsEnabled
756 * Returns TRUE if it is enabled as well as its buddy (if any)
757 * FALSE otherwise
759 static BOOL UPDOWN_IsEnabled (const UPDOWN_INFO *infoPtr)
761 if (!IsWindowEnabled(infoPtr->Self))
762 return FALSE;
763 if(infoPtr->Buddy)
764 return IsWindowEnabled(infoPtr->Buddy);
765 return TRUE;
768 /***********************************************************************
769 * UPDOWN_CancelMode
771 * Deletes any timers, releases the mouse and does redraw if necessary.
772 * If the control is not in "capture" mode, it does nothing.
773 * If the control was not in cancel mode, it returns FALSE.
774 * If the control was in cancel mode, it returns TRUE.
776 static BOOL UPDOWN_CancelMode (UPDOWN_INFO *infoPtr)
778 if (!(infoPtr->Flags & FLAG_PRESSED)) return FALSE;
780 KillTimer (infoPtr->Self, TIMER_AUTOREPEAT);
781 KillTimer (infoPtr->Self, TIMER_ACCEL);
782 KillTimer (infoPtr->Self, TIMER_AUTOPRESS);
784 if (GetCapture() == infoPtr->Self)
785 ReleaseCapture();
787 infoPtr->Flags &= ~FLAG_PRESSED;
788 InvalidateRect (infoPtr->Self, NULL, FALSE);
790 return TRUE;
793 /***********************************************************************
794 * UPDOWN_HandleMouseEvent
796 * Handle a mouse event for the updown.
797 * 'pt' is the location of the mouse event in client or
798 * windows coordinates.
800 static void UPDOWN_HandleMouseEvent (UPDOWN_INFO *infoPtr, UINT msg, INT x, INT y)
802 POINT pt = { x, y };
803 RECT rect;
804 int temp, arrow;
805 TRACKMOUSEEVENT tme;
807 TRACE("msg %04x point %s\n", msg, wine_dbgstr_point(&pt));
809 switch(msg)
811 case WM_LBUTTONDOWN: /* Initialise mouse tracking */
813 /* If the buddy is an edit, will set focus to it */
814 if (UPDOWN_IsBuddyEdit(infoPtr)) SetFocus(infoPtr->Buddy);
816 /* Now see which one is the 'active' arrow */
817 arrow = UPDOWN_GetArrowFromPoint (infoPtr, &rect, pt);
819 /* Update the flags if we are in/out */
820 infoPtr->Flags &= ~(FLAG_MOUSEIN | FLAG_ARROW);
821 if (arrow)
822 infoPtr->Flags |= FLAG_MOUSEIN | arrow;
823 else
824 if (infoPtr->AccelIndex != -1) infoPtr->AccelIndex = 0;
826 if (infoPtr->Flags & FLAG_ARROW) {
828 /* Update the CurVal if necessary */
829 UPDOWN_GetBuddyInt (infoPtr);
831 /* Set up the correct flags */
832 infoPtr->Flags |= FLAG_PRESSED;
834 /* repaint the control */
835 InvalidateRect (infoPtr->Self, NULL, FALSE);
837 /* process the click */
838 temp = (infoPtr->AccelCount && infoPtr->AccelVect) ? infoPtr->AccelVect[0].nInc : 1;
839 UPDOWN_DoAction (infoPtr, temp, infoPtr->Flags & FLAG_ARROW);
841 /* now capture all mouse messages */
842 SetCapture (infoPtr->Self);
844 /* and startup the first timer */
845 SetTimer(infoPtr->Self, TIMER_AUTOREPEAT, INITIAL_DELAY, 0);
847 break;
849 case WM_MOUSEMOVE:
850 /* save the flags to see if any got modified */
851 temp = infoPtr->Flags;
853 /* Now see which one is the 'active' arrow */
854 arrow = UPDOWN_GetArrowFromPoint (infoPtr, &rect, pt);
856 /* Update the flags if we are in/out */
857 infoPtr->Flags &= ~(FLAG_MOUSEIN | FLAG_ARROW);
858 if(arrow) {
859 infoPtr->Flags |= FLAG_MOUSEIN | arrow;
860 } else {
861 if(infoPtr->AccelIndex != -1) infoPtr->AccelIndex = 0;
864 /* If state changed, redraw the control */
865 if(temp != infoPtr->Flags)
866 InvalidateRect (infoPtr->Self, NULL, FALSE);
868 /* Set up tracking so the mousein flags can be reset when the
869 * mouse leaves the control */
870 tme.cbSize = sizeof( tme );
871 tme.dwFlags = TME_LEAVE;
872 tme.hwndTrack = infoPtr->Self;
873 TrackMouseEvent (&tme);
875 break;
876 case WM_MOUSELEAVE:
877 infoPtr->Flags &= ~(FLAG_MOUSEIN | FLAG_ARROW);
878 InvalidateRect (infoPtr->Self, NULL, FALSE);
879 break;
881 default:
882 ERR("Impossible case (msg=%x)!\n", msg);
887 /***********************************************************************
888 * UpDownWndProc
890 static LRESULT WINAPI UpDownWindowProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
892 UPDOWN_INFO *infoPtr = UPDOWN_GetInfoPtr (hwnd);
893 HTHEME theme;
895 TRACE("hwnd=%p msg=%04x wparam=%08lx lparam=%08lx\n", hwnd, message, wParam, lParam);
897 if (!infoPtr && (message != WM_CREATE))
898 return DefWindowProcW (hwnd, message, wParam, lParam);
900 switch(message)
902 case WM_CREATE:
904 CREATESTRUCTW *pcs = (CREATESTRUCTW*)lParam;
906 infoPtr = heap_alloc_zero(sizeof(*infoPtr));
907 SetWindowLongPtrW (hwnd, 0, (DWORD_PTR)infoPtr);
909 /* initialize the info struct */
910 infoPtr->Self = hwnd;
911 infoPtr->Notify = pcs->hwndParent;
912 infoPtr->dwStyle = pcs->style;
913 infoPtr->AccelCount = 0;
914 infoPtr->AccelVect = 0;
915 infoPtr->AccelIndex = -1;
916 infoPtr->CurVal = 0;
917 infoPtr->MinVal = 100;
918 infoPtr->MaxVal = 0;
919 infoPtr->Base = 10; /* Default to base 10 */
920 infoPtr->Buddy = 0; /* No buddy window yet */
921 infoPtr->Flags = (infoPtr->dwStyle & UDS_SETBUDDYINT) ? FLAG_BUDDYINT : 0;
923 SetWindowLongW (hwnd, GWL_STYLE, infoPtr->dwStyle & ~WS_BORDER);
924 if (!(infoPtr->dwStyle & UDS_HORZ))
925 SetWindowPos (hwnd, NULL, 0, 0, DEFAULT_WIDTH, pcs->cy,
926 SWP_NOOWNERZORDER | SWP_NOZORDER | SWP_NOMOVE);
928 /* Do we pick the buddy win ourselves? */
929 if (infoPtr->dwStyle & UDS_AUTOBUDDY)
930 UPDOWN_SetBuddy (infoPtr, GetWindow (hwnd, GW_HWNDPREV));
932 OpenThemeData (hwnd, L"Spin");
934 TRACE("UpDown Ctrl creation, hwnd=%p\n", hwnd);
936 break;
938 case WM_DESTROY:
939 heap_free (infoPtr->AccelVect);
940 UPDOWN_ResetSubclass (infoPtr);
941 heap_free (infoPtr);
942 SetWindowLongPtrW (hwnd, 0, 0);
943 theme = GetWindowTheme (hwnd);
944 CloseThemeData (theme);
945 TRACE("UpDown Ctrl destruction, hwnd=%p\n", hwnd);
946 break;
948 case WM_ENABLE:
949 if (wParam) {
950 infoPtr->dwStyle &= ~WS_DISABLED;
951 } else {
952 infoPtr->dwStyle |= WS_DISABLED;
953 UPDOWN_CancelMode (infoPtr);
955 InvalidateRect (infoPtr->Self, NULL, FALSE);
956 break;
958 case WM_STYLECHANGED:
959 if (wParam == GWL_STYLE) {
960 infoPtr->dwStyle = ((LPSTYLESTRUCT)lParam)->styleNew;
961 InvalidateRect (infoPtr->Self, NULL, FALSE);
963 break;
965 case WM_THEMECHANGED:
966 theme = GetWindowTheme (hwnd);
967 CloseThemeData (theme);
968 OpenThemeData (hwnd, L"Spin");
969 InvalidateRect (hwnd, NULL, FALSE);
970 break;
972 case WM_TIMER:
973 /* is this the auto-press timer? */
974 if(wParam == TIMER_AUTOPRESS) {
975 KillTimer(hwnd, TIMER_AUTOPRESS);
976 infoPtr->Flags &= ~(FLAG_PRESSED | FLAG_ARROW);
977 InvalidateRect(infoPtr->Self, NULL, FALSE);
980 /* if initial timer, kill it and start the repeat timer */
981 if(wParam == TIMER_AUTOREPEAT) {
982 INT delay;
984 KillTimer(hwnd, TIMER_AUTOREPEAT);
985 /* if no accel info given, used default timer */
986 if(infoPtr->AccelCount==0 || infoPtr->AccelVect==0) {
987 infoPtr->AccelIndex = -1;
988 delay = REPEAT_DELAY;
989 } else {
990 infoPtr->AccelIndex = 0; /* otherwise, use it */
991 delay = infoPtr->AccelVect[infoPtr->AccelIndex].nSec * 1000 + 1;
993 SetTimer(hwnd, TIMER_ACCEL, delay, 0);
996 /* now, if the mouse is above us, do the thing...*/
997 if(infoPtr->Flags & FLAG_MOUSEIN) {
998 int temp;
1000 temp = infoPtr->AccelIndex == -1 ? 1 : infoPtr->AccelVect[infoPtr->AccelIndex].nInc;
1001 UPDOWN_DoAction(infoPtr, temp, infoPtr->Flags & FLAG_ARROW);
1003 if(infoPtr->AccelIndex != -1 && infoPtr->AccelIndex < infoPtr->AccelCount-1) {
1004 KillTimer(hwnd, TIMER_ACCEL);
1005 infoPtr->AccelIndex++; /* move to the next accel info */
1006 temp = infoPtr->AccelVect[infoPtr->AccelIndex].nSec * 1000 + 1;
1007 /* make sure we have at least 1ms intervals */
1008 SetTimer(hwnd, TIMER_ACCEL, temp, 0);
1011 break;
1013 case WM_CANCELMODE:
1014 return UPDOWN_CancelMode (infoPtr);
1016 case WM_LBUTTONUP:
1017 if (GetCapture() != infoPtr->Self) break;
1019 if ( (infoPtr->Flags & FLAG_MOUSEIN) &&
1020 (infoPtr->Flags & FLAG_ARROW) ) {
1022 SendMessageW( infoPtr->Notify,
1023 (infoPtr->dwStyle & UDS_HORZ) ? WM_HSCROLL : WM_VSCROLL,
1024 MAKELONG(SB_ENDSCROLL, infoPtr->CurVal),
1025 (LPARAM)hwnd);
1026 if (UPDOWN_IsBuddyEdit(infoPtr))
1027 SendMessageW(infoPtr->Buddy, EM_SETSEL, 0, MAKELONG(0, -1));
1029 UPDOWN_CancelMode(infoPtr);
1030 break;
1032 case WM_LBUTTONDOWN:
1033 case WM_MOUSEMOVE:
1034 case WM_MOUSELEAVE:
1035 if(UPDOWN_IsEnabled(infoPtr))
1036 UPDOWN_HandleMouseEvent (infoPtr, message, (SHORT)LOWORD(lParam), (SHORT)HIWORD(lParam));
1037 break;
1039 case WM_MOUSEWHEEL:
1040 UPDOWN_MouseWheel(infoPtr, wParam);
1041 break;
1043 case WM_KEYDOWN:
1044 if((infoPtr->dwStyle & UDS_ARROWKEYS) && UPDOWN_IsEnabled(infoPtr))
1045 return UPDOWN_KeyPressed(infoPtr, (int)wParam);
1046 break;
1048 case WM_PRINTCLIENT:
1049 case WM_PAINT:
1050 return UPDOWN_Paint (infoPtr, (HDC)wParam);
1052 case UDM_GETACCEL:
1053 if (wParam==0 && lParam==0) return infoPtr->AccelCount;
1054 if (wParam && lParam) {
1055 int temp = min(infoPtr->AccelCount, wParam);
1056 memcpy((void *)lParam, infoPtr->AccelVect, temp*sizeof(UDACCEL));
1057 return temp;
1059 return 0;
1061 case UDM_SETACCEL:
1063 TRACE("UDM_SETACCEL\n");
1065 if(infoPtr->AccelVect) {
1066 heap_free (infoPtr->AccelVect);
1067 infoPtr->AccelCount = 0;
1068 infoPtr->AccelVect = 0;
1070 if(wParam==0) return TRUE;
1071 infoPtr->AccelVect = heap_alloc(wParam*sizeof(UDACCEL));
1072 if(!infoPtr->AccelVect) return FALSE;
1073 memcpy(infoPtr->AccelVect, (void*)lParam, wParam*sizeof(UDACCEL));
1074 infoPtr->AccelCount = wParam;
1076 if (TRACE_ON(updown))
1078 UINT i;
1080 for (i = 0; i < wParam; i++)
1081 TRACE("%u: nSec %u nInc %u\n", i,
1082 infoPtr->AccelVect[i].nSec, infoPtr->AccelVect[i].nInc);
1085 return TRUE;
1087 case UDM_GETBASE:
1088 return infoPtr->Base;
1090 case UDM_SETBASE:
1091 TRACE("UpDown Ctrl new base(%ld), hwnd=%p\n", wParam, hwnd);
1092 if (wParam==10 || wParam==16) {
1093 WPARAM old_base = infoPtr->Base;
1094 infoPtr->Base = wParam;
1096 if (old_base != infoPtr->Base)
1097 UPDOWN_SetBuddyInt(infoPtr);
1099 return old_base;
1101 break;
1103 case UDM_GETBUDDY:
1104 return (LRESULT)infoPtr->Buddy;
1106 case UDM_SETBUDDY:
1107 return (LRESULT)UPDOWN_SetBuddy (infoPtr, (HWND)wParam);
1109 case UDM_GETPOS:
1111 BOOL err;
1112 int pos;
1114 pos = UPDOWN_GetPos(infoPtr, &err);
1115 return MAKELONG(pos, err);
1117 case UDM_SETPOS:
1119 return UPDOWN_SetPos(infoPtr, (short)LOWORD(lParam));
1121 case UDM_GETRANGE:
1122 return MAKELONG(infoPtr->MaxVal, infoPtr->MinVal);
1124 case UDM_SETRANGE:
1125 /* we must have:
1126 UD_MINVAL <= Max <= UD_MAXVAL
1127 UD_MINVAL <= Min <= UD_MAXVAL
1128 |Max-Min| <= UD_MAXVAL */
1129 UPDOWN_SetRange(infoPtr, (short)lParam, (short)HIWORD(lParam));
1130 break;
1132 case UDM_GETRANGE32:
1133 if (wParam) *(LPINT)wParam = infoPtr->MinVal;
1134 if (lParam) *(LPINT)lParam = infoPtr->MaxVal;
1135 break;
1137 case UDM_SETRANGE32:
1138 UPDOWN_SetRange(infoPtr, (INT)lParam, (INT)wParam);
1139 break;
1141 case UDM_GETPOS32:
1143 return UPDOWN_GetPos(infoPtr, (BOOL*)lParam);
1145 case UDM_SETPOS32:
1147 return UPDOWN_SetPos(infoPtr, (int)lParam);
1149 case UDM_GETUNICODEFORMAT:
1150 /* we lie a bit here, we're always using Unicode internally */
1151 return infoPtr->UnicodeFormat;
1153 case UDM_SETUNICODEFORMAT:
1155 /* do we really need to honour this flag? */
1156 int temp = infoPtr->UnicodeFormat;
1157 infoPtr->UnicodeFormat = (BOOL)wParam;
1158 return temp;
1160 default:
1161 if ((message >= WM_USER) && (message < WM_APP) && !COMCTL32_IsReflectedMessage(message))
1162 ERR("unknown msg %04x wp=%04lx lp=%08lx\n", message, wParam, lParam);
1163 return DefWindowProcW (hwnd, message, wParam, lParam);
1166 return 0;
1169 /***********************************************************************
1170 * UPDOWN_Register [Internal]
1172 * Registers the updown window class.
1174 void UPDOWN_Register(void)
1176 WNDCLASSW wndClass;
1178 ZeroMemory( &wndClass, sizeof( WNDCLASSW ) );
1179 wndClass.style = CS_GLOBALCLASS | CS_VREDRAW | CS_HREDRAW;
1180 wndClass.lpfnWndProc = UpDownWindowProc;
1181 wndClass.cbClsExtra = 0;
1182 wndClass.cbWndExtra = sizeof(UPDOWN_INFO*);
1183 wndClass.hCursor = LoadCursorW( 0, (LPWSTR)IDC_ARROW );
1184 wndClass.hbrBackground = (HBRUSH)(COLOR_BTNFACE + 1);
1185 wndClass.lpszClassName = UPDOWN_CLASSW;
1187 RegisterClassW( &wndClass );
1191 /***********************************************************************
1192 * UPDOWN_Unregister [Internal]
1194 * Unregisters the updown window class.
1196 void UPDOWN_Unregister (void)
1198 UnregisterClassW (UPDOWN_CLASSW, NULL);