xref: /reactos/dll/shellext/stobject/volume.cpp (revision 14d3b53c) 
1 /*
2  * PROJECT:     ReactOS system libraries
3  * LICENSE:     GPL - See COPYING in the top level directory
4  * FILE:        dll/shellext/stobject/volume.cpp
5  * PURPOSE:     Volume notification icon handler
6  * PROGRAMMERS: David Quintana <gigaherz@gmail.com>
7  */
8 
9 #include "precomp.h"
10 
11 #include <mmddk.h>
12 
13 HICON g_hIconVolume;
14 HICON g_hIconMute;
15 
16 HMIXER g_hMixer;
17 UINT   g_mixerId;
18 DWORD  g_mixerLineID;
19 DWORD  g_muteControlID;
20 
21 UINT g_mmDeviceChange;
22 
23 static BOOL g_IsMute = FALSE;
24 
25 static HRESULT __stdcall Volume_FindMixerControl(CSysTray * pSysTray)
26 {
27     MMRESULT result;
28     UINT mixerId = 0;
29     DWORD waveOutId = 0;
30     DWORD param2 = 0;
31 
32     TRACE("Volume_FindDefaultMixerID\n");
33 
34     result = waveOutMessage((HWAVEOUT)UlongToHandle(WAVE_MAPPER), DRVM_MAPPER_PREFERRED_GET, (DWORD_PTR)&waveOutId, (DWORD_PTR)&param2);
35     if (result)
36         return E_FAIL;
37 
38     if (waveOutId == (DWORD)-1)
39     {
40         TRACE("WARNING: waveOut has no default device, trying with first available device...\n", waveOutId);
41 
42         mixerId = 0;
43     }
44     else
45     {
46         TRACE("waveOut default device is %d\n", waveOutId);
47 
48         result = mixerGetID((HMIXEROBJ)UlongToHandle(waveOutId), &mixerId, MIXER_OBJECTF_WAVEOUT);
49         if (result)
50             return E_FAIL;
51 
52         TRACE("mixerId for waveOut default device is %d\n", mixerId);
53     }
54 
55     g_mixerId = mixerId;
56     return S_OK;
57 
58     MIXERCAPS mixerCaps;
59     MIXERLINE mixerLine;
60     MIXERCONTROL mixerControl;
61     MIXERLINECONTROLS mixerLineControls;
62 
63     g_mixerLineID = -1;
64     g_muteControlID = -1;
65 
66     if (mixerGetDevCapsW(g_mixerId, &mixerCaps, sizeof(mixerCaps)))
67         return E_FAIL;
68 
69     if (mixerCaps.cDestinations == 0)
70         return S_FALSE;
71 
72     TRACE("mixerCaps.cDestinations %d\n", mixerCaps.cDestinations);
73 
74     DWORD idx;
75     for (idx = 0; idx < mixerCaps.cDestinations; idx++)
76     {
77         mixerLine.cbStruct = sizeof(mixerLine);
78         mixerLine.dwDestination = idx;
79         if (!mixerGetLineInfoW((HMIXEROBJ)UlongToHandle(g_mixerId), &mixerLine, 0))
80         {
81             if (mixerLine.dwComponentType >= MIXERLINE_COMPONENTTYPE_DST_SPEAKERS &&
82                 mixerLine.dwComponentType <= MIXERLINE_COMPONENTTYPE_DST_HEADPHONES)
83                 break;
84             TRACE("Destination %d was not speakers or headphones.\n");
85         }
86     }
87 
88     if (idx >= mixerCaps.cDestinations)
89         return E_FAIL;
90 
91     TRACE("Valid destination %d found.\n");
92 
93     g_mixerLineID = mixerLine.dwLineID;
94 
95     mixerLineControls.cbStruct = sizeof(mixerLineControls);
96     mixerLineControls.dwLineID = mixerLine.dwLineID;
97     mixerLineControls.cControls = 1;
98     mixerLineControls.dwControlType = MIXERCONTROL_CONTROLTYPE_MUTE;
99     mixerLineControls.pamxctrl = &mixerControl;
100     mixerLineControls.cbmxctrl = sizeof(mixerControl);
101 
102     if (mixerGetLineControlsW((HMIXEROBJ)UlongToHandle(g_mixerId), &mixerLineControls, MIXER_GETLINECONTROLSF_ONEBYTYPE))
103         return E_FAIL;
104 
105     TRACE("Found control id %d for mute: %d\n", mixerControl.dwControlID);
106 
107     g_muteControlID = mixerControl.dwControlID;
108 
109     return S_OK;
110 }
111 
112 HRESULT Volume_IsMute()
113 {
114     MIXERCONTROLDETAILS mixerControlDetails;
115 
116     if (g_mixerId != (UINT)-1 && g_muteControlID != (DWORD)-1)
117     {
118         BOOL detailsResult = 0;
119         mixerControlDetails.cbStruct = sizeof(mixerControlDetails);
120         mixerControlDetails.hwndOwner = 0;
121         mixerControlDetails.dwControlID = g_muteControlID;
122         mixerControlDetails.cChannels = 1;
123         mixerControlDetails.paDetails = &detailsResult;
124         mixerControlDetails.cbDetails = sizeof(detailsResult);
125         if (mixerGetControlDetailsW((HMIXEROBJ)UlongToHandle(g_mixerId), &mixerControlDetails, 0))
126             return E_FAIL;
127 
128         TRACE("Obtained mute status %d\n", detailsResult);
129 
130         g_IsMute = detailsResult != 0;
131     }
132 
133     return S_OK;
134 }
135 
136 HRESULT STDMETHODCALLTYPE Volume_Init(_In_ CSysTray * pSysTray)
137 {
138     HRESULT hr;
139     WCHAR strTooltip[128];
140 
141     TRACE("Volume_Init\n");
142 
143     if (!g_hMixer)
144     {
145         hr = Volume_FindMixerControl(pSysTray);
146         if (FAILED(hr))
147             return hr;
148 
149         g_mmDeviceChange = RegisterWindowMessageW(L"winmm_devicechange");
150     }
151 
152     g_hIconVolume = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_VOLUME));
153     g_hIconMute = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_VOLMUTE));
154 
155     Volume_IsMute();
156 
157     HICON icon;
158     if (g_IsMute)
159         icon = g_hIconMute;
160     else
161         icon = g_hIconVolume;
162 
163     LoadStringW(g_hInstance, IDS_VOL_VOLUME, strTooltip, _countof(strTooltip));
164     return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_VOLUME, icon, strTooltip);
165 }
166 
167 HRESULT STDMETHODCALLTYPE Volume_Update(_In_ CSysTray * pSysTray)
168 {
169     BOOL PrevState;
170 
171     TRACE("Volume_Update\n");
172 
173     PrevState = g_IsMute;
174     Volume_IsMute();
175 
176     if (PrevState != g_IsMute)
177     {
178         WCHAR strTooltip[128];
179         HICON icon;
180         if (g_IsMute)
181         {
182             icon = g_hIconMute;
183             LoadStringW(g_hInstance, IDS_VOL_MUTED, strTooltip, _countof(strTooltip));
184         }
185         else
186         {
187             icon = g_hIconVolume;
188             LoadStringW(g_hInstance, IDS_VOL_VOLUME, strTooltip, _countof(strTooltip));
189         }
190 
191         return pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_VOLUME, icon, strTooltip);
192     }
193     else
194     {
195         return S_OK;
196     }
197 }
198 
199 HRESULT STDMETHODCALLTYPE Volume_Shutdown(_In_ CSysTray * pSysTray)
200 {
201     TRACE("Volume_Shutdown\n");
202 
203     return pSysTray->NotifyIcon(NIM_DELETE, ID_ICON_VOLUME, NULL, NULL);
204 }
205 
206 HRESULT Volume_OnDeviceChange(_In_ CSysTray * pSysTray, WPARAM wParam, LPARAM lParam)
207 {
208     return Volume_FindMixerControl(pSysTray);
209 }
210 
211 static void _RunVolume(BOOL bTray)
212 {
213     ShellExecuteW(NULL,
214                   NULL,
215                   L"sndvol32.exe",
216                   bTray ? L"/t" : NULL,
217                   NULL,
218                   SW_SHOWNORMAL);
219 }
220 
221 static void _RunMMCpl()
222 {
223     ShellExecuteW(NULL, NULL, L"mmsys.cpl", NULL, NULL, SW_NORMAL);
224 }
225 
226 static void _ShowContextMenu(CSysTray * pSysTray)
227 {
228     WCHAR strAdjust[128];
229     WCHAR strOpen[128];
230     LoadStringW(g_hInstance, IDS_VOL_OPEN, strOpen, _countof(strOpen));
231     LoadStringW(g_hInstance, IDS_VOL_ADJUST, strAdjust, _countof(strAdjust));
232 
233     HMENU hPopup = CreatePopupMenu();
234     AppendMenuW(hPopup, MF_STRING, IDS_VOL_OPEN, strOpen);
235     AppendMenuW(hPopup, MF_STRING, IDS_VOL_ADJUST, strAdjust);
236     SetMenuDefaultItem(hPopup, IDS_VOL_OPEN, FALSE);
237 
238     DWORD flags = TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTALIGN | TPM_BOTTOMALIGN;
239     POINT pt;
240     SetForegroundWindow(pSysTray->GetHWnd());
241     GetCursorPos(&pt);
242 
243     DWORD id = TrackPopupMenuEx(hPopup, flags,
244         pt.x, pt.y,
245         pSysTray->GetHWnd(), NULL);
246 
247     DestroyMenu(hPopup);
248 
249     switch (id)
250     {
251     case IDS_VOL_OPEN:
252         _RunVolume(FALSE);
253         break;
254     case IDS_VOL_ADJUST:
255         _RunMMCpl();
256         break;
257     }
258 }
259 
260 HRESULT STDMETHODCALLTYPE Volume_Message(_In_ CSysTray * pSysTray, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT &lResult)
261 {
262     if (uMsg == g_mmDeviceChange)
263         return Volume_OnDeviceChange(pSysTray, wParam, lParam);
264 
265     switch (uMsg)
266     {
267         case WM_USER + 220:
268             TRACE("Volume_Message: WM_USER+220\n");
269             if (wParam == VOLUME_SERVICE_FLAG)
270             {
271                 if (lParam)
272                 {
273                     pSysTray->EnableService(VOLUME_SERVICE_FLAG, TRUE);
274                     return Volume_Init(pSysTray);
275                 }
276                 else
277                 {
278                     pSysTray->EnableService(VOLUME_SERVICE_FLAG, FALSE);
279                     return Volume_Shutdown(pSysTray);
280                 }
281             }
282             return S_FALSE;
283 
284         case WM_USER + 221:
285             TRACE("Volume_Message: WM_USER+221\n");
286             if (wParam == VOLUME_SERVICE_FLAG)
287             {
288                 lResult = (LRESULT)pSysTray->IsServiceEnabled(VOLUME_SERVICE_FLAG);
289                 return S_OK;
290             }
291             return S_FALSE;
292 
293         case WM_TIMER:
294             if (wParam == VOLUME_TIMER_ID)
295             {
296                 KillTimer(pSysTray->GetHWnd(), VOLUME_TIMER_ID);
297                 _RunVolume(TRUE);
298             }
299             break;
300 
301         case ID_ICON_VOLUME:
302             TRACE("Volume_Message uMsg=%d, w=%x, l=%x\n", uMsg, wParam, lParam);
303 
304             Volume_Update(pSysTray);
305 
306             switch (lParam)
307             {
308                 case WM_LBUTTONDOWN:
309                     SetTimer(pSysTray->GetHWnd(), VOLUME_TIMER_ID, GetDoubleClickTime(), NULL);
310                     break;
311 
312                 case WM_LBUTTONUP:
313                     break;
314 
315                 case WM_LBUTTONDBLCLK:
316                     KillTimer(pSysTray->GetHWnd(), VOLUME_TIMER_ID);
317                     _RunVolume(FALSE);
318                     break;
319 
320                 case WM_RBUTTONDOWN:
321                     break;
322 
323                 case WM_RBUTTONUP:
324                     _ShowContextMenu(pSysTray);
325                     break;
326 
327                 case WM_RBUTTONDBLCLK:
328                     break;
329 
330                 case WM_MOUSEMOVE:
331                     break;
332             }
333             return S_OK;
334 
335         default:
336             TRACE("Volume_Message received for unknown ID %d, ignoring.\n");
337             return S_FALSE;
338     }
339 
340     return S_FALSE;
341 }
342