desktopswitcher.cpp

   1 // $Id: desktopswitcher.cpp,v 1.2 2003/08/03 16:56:29 nedko Exp $
   2 //
   3 // Desktop Switcher
   4 // Copyright (C) 2000,2001,2002 Nedko Arnaudov <nedko@users.sourceforge.net>
   5 //
   6 // This program is free software; you can redistribute it and/or modify
   7 // it under the terms of the GNU General Public License as published by
   8 // the Free Software Foundation; either version 2 of the License, or
   9 // (at your option) any later version.
  10 //
  11 // This program 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
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License
  17 // along with this program; if not, write to the Free Software
  18 // Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19
  20 #include "ph.h"
  21 #include "DesktopSwitcher.h"
  22 #include "Desktop.h"
  23
  24 CDesktopSwitcher::CDesktopSwitcher()
  25 {
  26   m_ppDesktops = NULL;
  27   m_hExitEvent = NULL;
  28 }
  29
  30 CDesktopSwitcher::~CDesktopSwitcher()
  31 {
  32   ASSERT(!m_ppDesktops);
  33   if (m_ppDesktops)
  34     delete m_ppDesktops;
  35
  36   ASSERT(!m_hExitEvent);
  37   if (m_hExitEvent)
  38     VERIFY(CloseHandle(m_hExitEvent));
  39 }
  40
  41 HRESULT CDesktopSwitcher::Go(unsigned int nDesktops)
  42 {
  43   DWORD dwError;
  44
  45   if (m_hExitEvent)
  46     return E_UNEXPECTED;        // call this method only once
  47
  48   if (nDesktops <= 1 || nDesktops > 12)
  49     return E_INVALIDARG;
  50
  51   m_nDesktops = nDesktops;
  52
  53   m_hExitEvent = CreateEvent(NULL,TRUE,FALSE,NULL);
  54   if (!m_hExitEvent)
  55   {
  56     dwError = GetLastError();
  57     return HRESULT_FROM_WIN32(dwError);
  58   }
  59
  60   m_ppDesktops = new CDesktop * [nDesktops];
  61   if (!m_ppDesktops)
  62     return E_OUTOFMEMORY;
  63
  64   m_nCurrentDesktopIndex = 0;
  65
  66   HRESULT hr;
  67   char pszDesktopName[256] = "";
  68
  69   HKEY hKey;
  70   // open/create the key
  71   RegCreateKeyEx(HKEY_CURRENT_USER, DESKTOP_NAMES_KEY, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_QUERY_VALUE|KEY_SET_VALUE,NULL,&hKey,NULL);
  72   char NameBuffer[256];
  73
  74   for (unsigned int i = 0 ; i < nDesktops ; i++)
  75   {
  76     // try to get desktop name from registry
  77     _ultot(i+1,NameBuffer,10);
  78     DWORD dwSize = 256;
  79     DWORD dwType;
  80     if (hKey == NULL || RegQueryValueEx(hKey,NameBuffer,NULL,&dwType,(BYTE *)pszDesktopName,&dwSize) != ERROR_SUCCESS && dwType != REG_SZ)
  81       sprintf(pszDesktopName,"Desktop %u",i+1);
  82
  83     m_ppDesktops[i] = new CDesktop(this,i?pszDesktopName:"Default",pszDesktopName,hr);
  84     if (!m_ppDesktops[i])
  85     {
  86       Exit();
  87       return E_OUTOFMEMORY;
  88     }
  89
  90     if (FAILED(hr))
  91     {
  92       Exit();
  93       return E_OUTOFMEMORY;
  94     }
  95   }
  96
  97   switch(WaitForSingleObject(m_hExitEvent,INFINITE))
  98   {
  99   case WAIT_OBJECT_0:
 100     break;
 101   case WAIT_FAILED:
 102     {
 103       dwError = GetLastError();
 104       return HRESULT_FROM_WIN32(dwError);
 105     }
 106   default:
 107     ASSERT(FALSE);
 108     return E_UNEXPECTED;
 109   }
 110
 111   m_ppDesktops[0]->SwitchTo();
 112
 113   for (i = 0 ; i < nDesktops ; i ++)
 114     m_ppDesktops[i]->Exit();
 115
 116   delete m_ppDesktops;
 117   m_ppDesktops = NULL;
 118
 119   VERIFY(CloseHandle(m_hExitEvent));
 120   m_hExitEvent= NULL;
 121
 122   return S_OK;
 123 }
 124
 125 CDesktop * CDesktopSwitcher::GetDefaultDesktop()
 126 {
 127   if (!m_ppDesktops)
 128     return NULL;
 129
 130   return m_ppDesktops[0];
 131 }
 132
 133 void CDesktopSwitcher::Exit()
 134 {
 135   ASSERT(m_hExitEvent);
 136   VERIFY(SetEvent(m_hExitEvent));
 137 }
 138
 139 HRESULT CDesktopSwitcher::HotKeyPressed(WORD wVKey, WORD wModifiers)
 140 {
 141   HRESULT hr;
 142   if (wVKey >= VK_F1 && wVKey < VK_F1 + m_nDesktops && wModifiers == (MOD_ALT|MOD_CONTROL))
 143   {
 144     m_nCurrentDesktopIndex = wVKey - VK_F1;
 145     hr = m_ppDesktops[m_nCurrentDesktopIndex]->SwitchTo();
 146     if (FAILED(hr))
 147       return hr;
 148   }
 149   else if (wVKey == VK_RETURN && wModifiers == MOD_WIN)
 150   {
 151     for (unsigned int i = 0 ; i < m_nDesktops ; i ++)
 152       m_ppDesktops[i]->WindowToggleVisible();
 153
 154     return S_OK;
 155   }
 156   else if (wVKey == VK_ADD && wModifiers == MOD_WIN)
 157   {
 158     m_nCurrentDesktopIndex = (m_nCurrentDesktopIndex+1) % m_nDesktops;
 159     hr = m_ppDesktops[m_nCurrentDesktopIndex]->SwitchTo();
 160     if (FAILED(hr))
 161       return hr;
 162
 163     return S_OK;
 164   }
 165   else if (wVKey == VK_SUBTRACT && wModifiers == MOD_WIN)
 166   {
 167     if (m_nCurrentDesktopIndex)
 168       m_nCurrentDesktopIndex--;
 169     else
 170       m_nCurrentDesktopIndex = m_nDesktops-1;
 171
 172     hr = m_ppDesktops[m_nCurrentDesktopIndex]->SwitchTo();
 173     if (FAILED(hr))
 174       return hr;
 175
 176     return S_OK;
 177   }
 178   else if (wVKey == 'R' && wModifiers == (MOD_WIN|MOD_SHIFT))
 179   {
 180     m_ppDesktops[m_nCurrentDesktopIndex]->RunPrompt();
 181     return S_OK;
 182   }
 183   else if (wVKey == VK_ESCAPE && wModifiers == (MOD_WIN))
 184   {
 185     Exit();
 186     return S_OK;
 187   }
 188
 189   return S_FALSE;
 190 }
 191
 192 void CDesktopSwitcher::RegisterHotKeys(HWND hWnd)
 193 {
 194   for (unsigned int i=0 ; i < m_nDesktops ; i++)
 195     RegisterHotKey (hWnd, VK_F1+i, MOD_ALT|MOD_CONTROL, VK_F1+i);
 196
 197   RegisterHotKey(hWnd, VK_RETURN, MOD_WIN, VK_RETURN);
 198   RegisterHotKey(hWnd, VK_ADD, MOD_WIN, VK_ADD);
 199   RegisterHotKey(hWnd, VK_SUBTRACT, MOD_WIN, VK_SUBTRACT);
 200   RegisterHotKey(hWnd, 'R', MOD_WIN|MOD_SHIFT, 'R');
 201   RegisterHotKey(hWnd, VK_ESCAPE, MOD_WIN, VK_ESCAPE);
 202 }