ole32/tests: Additional test for RevokeDragDrop with destroyed window.
[wine] / dlls / comctl32 / theming.c
1 /*
2  * Theming - Initialization
3  *
4  * Copyright (c) 2005 by Frank Richter
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  *
20  */
21
22 #include <stdarg.h>
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "wingdi.h"
27 #include "winuser.h"
28 #include "comctl32.h"
29 #include "uxtheme.h"
30 #include "wine/debug.h"
31
32 WINE_DEFAULT_DEBUG_CHANNEL(theming);
33
34 typedef LRESULT (CALLBACK* THEMING_SUBCLASSPROC)(HWND, UINT, WPARAM, LPARAM,
35     ULONG_PTR);
36
37 extern LRESULT CALLBACK THEMING_ButtonSubclassProc (HWND, UINT, WPARAM, LPARAM,
38                                                     ULONG_PTR);
39 extern LRESULT CALLBACK THEMING_ComboSubclassProc (HWND, UINT, WPARAM, LPARAM,
40                                                    ULONG_PTR);
41 extern LRESULT CALLBACK THEMING_DialogSubclassProc (HWND, UINT, WPARAM, LPARAM,
42                                                     ULONG_PTR);
43 extern LRESULT CALLBACK THEMING_EditSubclassProc (HWND, UINT, WPARAM, LPARAM,
44                                                   ULONG_PTR);
45 extern LRESULT CALLBACK THEMING_ListBoxSubclassProc (HWND, UINT, WPARAM, LPARAM,
46                                                      ULONG_PTR);
47
48 static const WCHAR dialogClass[] = {'#','3','2','7','7','0',0};
49 static const WCHAR comboLboxClass[] = {'C','o','m','b','o','L','b','o','x',0};
50
51 static const struct ThemingSubclass
52 {
53     const WCHAR* className;
54     THEMING_SUBCLASSPROC subclassProc;
55 } subclasses[] = {
56     /* Note: list must be sorted by class name */
57     {dialogClass,          THEMING_DialogSubclassProc},
58     {WC_BUTTONW,           THEMING_ButtonSubclassProc},
59     {WC_COMBOBOXW,         THEMING_ComboSubclassProc},
60     {comboLboxClass,       THEMING_ListBoxSubclassProc},
61     {WC_EDITW,             THEMING_EditSubclassProc},
62     {WC_LISTBOXW,          THEMING_ListBoxSubclassProc}
63 };
64
65 #define NUM_SUBCLASSES        (sizeof(subclasses)/sizeof(subclasses[0]))
66
67 static WNDPROC originalProcs[NUM_SUBCLASSES];
68 static ATOM atRefDataProp;
69 static ATOM atSubclassProp;
70
71 /* Generate a number of subclass window procs.
72  * With a single proc alone, we can't really reliably find out the superclass,
73  * so have one for each subclass. The subclass number is also stored in a prop
74  * since it's needed by THEMING_CallOriginalClass(). Then, the subclass
75  * proc and ref data are fetched and the proc called.
76  */
77 #define MAKE_SUBCLASS_PROC(N)                                               \
78 static LRESULT CALLBACK subclass_proc ## N (HWND wnd, UINT msg,             \
79                                             WPARAM wParam, LPARAM lParam)   \
80 {                                                                           \
81     LRESULT result;                                                         \
82     ULONG_PTR refData;                                                      \
83     SetPropW (wnd, (LPCWSTR)MAKEINTATOM(atSubclassProp), (HANDLE)N);        \
84     refData = (ULONG_PTR)GetPropW (wnd, (LPCWSTR)MAKEINTATOM(atRefDataProp)); \
85     TRACE ("%d; (%p, %x, %lx, %lx, %lx)\n", N, wnd, msg, wParam, lParam,     \
86         refData);                                                           \
87     result = subclasses[N].subclassProc (wnd, msg, wParam, lParam, refData);\
88     TRACE ("result = %lx\n", result);                                       \
89     return result;                                                          \
90 }
91
92 MAKE_SUBCLASS_PROC(0)
93 MAKE_SUBCLASS_PROC(1)
94 MAKE_SUBCLASS_PROC(2)
95 MAKE_SUBCLASS_PROC(3)
96 MAKE_SUBCLASS_PROC(4)
97 MAKE_SUBCLASS_PROC(5)
98
99 static const WNDPROC subclassProcs[NUM_SUBCLASSES] = {
100     subclass_proc0,
101     subclass_proc1,
102     subclass_proc2,
103     subclass_proc3,
104     subclass_proc4,
105     subclass_proc5
106 };
107
108 /***********************************************************************
109  * THEMING_Initialize
110  *
111  * Register classes for standard controls that will shadow the system
112  * classes.
113  */
114 void THEMING_Initialize (void)
115 {
116     unsigned int i;
117     static const WCHAR subclassPropName[] = 
118         { 'C','C','3','2','T','h','e','m','i','n','g','S','u','b','C','l',0 };
119     static const WCHAR refDataPropName[] = 
120         { 'C','C','3','2','T','h','e','m','i','n','g','D','a','t','a',0 };
121
122     if (!IsThemeActive()) return;
123
124     atSubclassProp = GlobalAddAtomW (subclassPropName);
125     atRefDataProp = GlobalAddAtomW (refDataPropName);
126
127     for (i = 0; i < NUM_SUBCLASSES; i++)
128     {
129         WNDCLASSEXW class;
130
131         class.cbSize = sizeof(class);
132         class.style |= CS_GLOBALCLASS;
133         GetClassInfoExW (NULL, subclasses[i].className, &class);
134         originalProcs[i] = class.lpfnWndProc;
135         class.lpfnWndProc = subclassProcs[i];
136         
137         if (!class.lpfnWndProc)
138         {
139             ERR("Missing proc for class %s\n", 
140                 debugstr_w (subclasses[i].className));
141             continue;
142         }
143
144         if (!RegisterClassExW (&class))
145         {
146             ERR("Could not re-register class %s: %x\n",
147                 debugstr_w (subclasses[i].className), GetLastError ());
148         }
149         else
150         {
151             TRACE("Re-registered class %s\n", 
152                 debugstr_w (subclasses[i].className));
153         }
154     }
155 }
156
157 /***********************************************************************
158  * THEMING_Uninitialize
159  *
160  * Unregister shadow classes for standard controls.
161  */
162 void THEMING_Uninitialize (void)
163 {
164     unsigned int i;
165
166     if (!atSubclassProp) return;  /* not initialized */
167
168     for (i = 0; i < NUM_SUBCLASSES; i++)
169     {
170         UnregisterClassW (subclasses[i].className, NULL);
171     }
172 }
173
174 /***********************************************************************
175  * THEMING_CallOriginalClass
176  *
177  * Determines the original window proc and calls it.
178  */
179 LRESULT THEMING_CallOriginalClass (HWND wnd, UINT msg, WPARAM wParam, LPARAM lParam)
180 {
181     INT_PTR subclass = (INT_PTR)GetPropW (wnd, (LPCWSTR)MAKEINTATOM(atSubclassProp));
182     WNDPROC oldProc = originalProcs[subclass];
183     return CallWindowProcW (oldProc, wnd, msg, wParam, lParam);
184 }
185
186 /***********************************************************************
187  * THEMING_SetSubclassData
188  *
189  * Update the "refData" value of the subclassed window.
190  */
191 void THEMING_SetSubclassData (HWND wnd, ULONG_PTR refData)
192 {
193     SetPropW (wnd, (LPCWSTR)MAKEINTATOM(atRefDataProp), (HANDLE)refData);
194 }