reactos/modules/rosapps/applications/explorer-old/taskbar/startmenu.cpp

2299 lines
56 KiB
C++

/*
* Copyright 2003, 2004 Martin Fuchs
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
//
// Explorer clone
//
// startmenu.cpp
//
// Explorer start menu
//
// Martin Fuchs, 19.08.2003
//
// Credits: Thanks to Everaldo (http://www.everaldo.com) for his nice looking icons.
//
#include <precomp.h>
#include "desktopbar.h"
#include "startmenu.h"
#include "../dialogs/searchprogram.h"
#include "../dialogs/settings.h"
#define SHELLPATH_CONTROL_PANEL TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}")
#define SHELLPATH_PRINTERS TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}")
#define SHELLPATH_NET_CONNECTIONS TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{7007ACC7-3202-11D1-AAD2-00805FC1270E}")
StartMenu::StartMenu(HWND hwnd, int icon_size)
: super(hwnd),
_icon_size(icon_size)
{
_next_id = IDC_FIRST_MENU;
_submenu_id = 0;
_border_left = 0;
_border_top = 0;
_bottom_max = INT_MAX;
_floating_btn = false;
_arrow_btns = false;
_scroll_mode = SCROLL_NOT;
_scroll_pos = 0;
_invisible_lines = 0;
_last_pos = WindowRect(hwnd).pos();
#ifdef _LIGHT_STARTMENU
_selected_id = -1;
_last_mouse_pos = 0;
#endif
}
StartMenu::StartMenu(HWND hwnd, const StartMenuCreateInfo& create_info, int icon_size)
: super(hwnd),
_create_info(create_info),
_icon_size(icon_size)
{
for(StartMenuFolders::const_iterator it=create_info._folders.begin(); it!=create_info._folders.end(); ++it)
if (*it)
_dirs.push_back(ShellDirectory(GetDesktopFolder(), *it, _hwnd));
_next_id = IDC_FIRST_MENU;
_submenu_id = 0;
_border_left = 0;
_border_top = create_info._border_top;
_bottom_max = INT_MAX;
_floating_btn = create_info._border_top? true: false;
_arrow_btns = false;
_scroll_mode = SCROLL_NOT;
_scroll_pos = 0;
_invisible_lines = 0;
_last_pos = WindowRect(hwnd).pos();
#ifdef _LIGHT_STARTMENU
_selected_id = -1;
_last_mouse_pos = 0;
#endif
}
StartMenu::~StartMenu()
{
SendParent(PM_STARTMENU_CLOSED);
}
// We need this wrapper function for s_wcStartMenu, it calls the WIN32 API,
// though static C++ initializers are not allowed for Winelib applications.
BtnWindowClass& StartMenu::GetWndClasss()
{
static BtnWindowClass s_wcStartMenu(CLASSNAME_STARTMENU);
return s_wcStartMenu;
}
Window::CREATORFUNC_INFO StartMenu::s_def_creator = STARTMENU_CREATOR(StartMenu);
HWND StartMenu::Create(int x, int y, const StartMenuFolders& folders, HWND hwndParent, LPCTSTR title,
CREATORFUNC_INFO creator, void* info, const String& filter)
{
UINT style, ex_style;
int top_height;
if (hwndParent) {
style = WS_POPUP|WS_THICKFRAME|WS_CLIPCHILDREN|WS_VISIBLE;
ex_style = 0;
top_height = STARTMENU_TOP_BTN_SPACE;
} else {
style = WS_POPUP|WS_CAPTION|WS_SYSMENU|WS_CLIPCHILDREN|WS_VISIBLE;
ex_style = WS_EX_TOOLWINDOW;
top_height = 0;
}
int icon_size = ICON_SIZE_SMALL;
RECT rect = {x, y-STARTMENU_LINE_HEIGHT(icon_size)-top_height, x+STARTMENU_WIDTH_MIN, y};
#ifndef _LIGHT_STARTMENU
rect.top += STARTMENU_LINE_HEIGHT(icon_size);
#endif
AdjustWindowRectEx(&rect, style, FALSE, ex_style);
StartMenuCreateInfo create_info;
create_info._folders = folders;
create_info._border_top = top_height;
create_info._creator = creator;
create_info._info = info;
create_info._filter = filter;
if (title)
create_info._title = title;
HWND hwnd = Window::Create(creator, &create_info, ex_style, GetWndClasss(), title,
style, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, hwndParent);
// make sure the window is not off the screen
MoveVisible(hwnd);
return hwnd;
}
LRESULT StartMenu::Init(LPCREATESTRUCT pcs)
{
try {
AddEntries();
if (super::Init(pcs))
return 1;
// create buttons for registered entries in _entries
for(ShellEntryMap::const_iterator it=_entries.begin(); it!=_entries.end(); ++it) {
const StartMenuEntry& sme = it->second;
bool hasSubmenu = false;
for(ShellEntrySet::const_iterator it=sme._entries.begin(); it!=sme._entries.end(); ++it)
if ((*it)->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
hasSubmenu = true;
#ifdef _LIGHT_STARTMENU
_buttons.push_back(SMBtnInfo(sme, it->first, hasSubmenu));
#else
AddButton(sme._title, sme._hIcon, hasSubmenu, it->first);
#endif
}
#ifdef _LIGHT_STARTMENU
if (_buttons.empty())
#else
if (!GetWindow(_hwnd, GW_CHILD))
#endif
AddButton(ResString(IDS_EMPTY), ICID_NONE, false, 0, false);
#ifdef _LIGHT_STARTMENU
ResizeToButtons();
#endif
#ifdef _LAZY_ICONEXTRACT
PostMessage(_hwnd, PM_UPDATE_ICONS, 0, 0);
#endif
} catch(COMException& e) {
HandleException(e, pcs->hwndParent); // destroys the start menu window while switching focus
}
return 0;
}
void StartMenu::AddEntries()
{
for(StartMenuShellDirs::iterator it=_dirs.begin(); it!=_dirs.end(); ++it) {
StartMenuDirectory& smd = *it;
ShellDirectory& dir = smd._dir;
if (!dir._scanned) {
WaitCursor wait;
#ifdef _LAZY_ICONEXTRACT
dir.smart_scan(SORT_NAME, SCAN_DONT_EXTRACT_ICONS); // lazy icon extraction, try to read directly from filesystem
#else
dir.smart_scan(SORT_NAME);
#endif
}
AddShellEntries(dir, -1, smd._ignore);
}
}
static LPTSTR trim_path_slash(LPTSTR path)
{
LPTSTR p = path;
while(*p)
++p;
if (p>path && (p[-1]=='\\' || p[-1]=='/'))
*--p = '\0';
return path;
}
void StartMenu::AddShellEntries(const ShellDirectory& dir, int max, const String& ignore)
{
TCHAR ignore_path[MAX_PATH], ignore_dir[MAX_PATH], ignore_name[_MAX_FNAME], ignore_ext[_MAX_EXT];
TCHAR dir_path[MAX_PATH];
if (!ignore.empty()) {
_tsplitpath_s(ignore, ignore_path, COUNTOF(ignore_path), ignore_dir, COUNTOF(ignore_dir), ignore_name, COUNTOF(ignore_name), ignore_ext, COUNTOF(ignore_ext));
_tcscat(ignore_path, ignore_dir);
_tcscat(ignore_name, ignore_ext);
dir.get_path(dir_path, COUNTOF(dir_path));
if (_tcsicmp(trim_path_slash(dir_path), trim_path_slash(ignore_path)))
*ignore_name = '\0';
} else
*ignore_name = '\0';
String lwr_filter = _create_info._filter;
lwr_filter.toLower();
int cnt = 0;
for(Entry*entry=dir._down; entry; entry=entry->_next) {
// hide files like "desktop.ini"
if (entry->_shell_attribs & SFGAO_HIDDEN)
//not appropriate for drive roots: if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN)
continue;
// hide "Programs" subfolders if requested
if (*ignore_name && !_tcsicmp(entry->_data.cFileName, ignore_name))
continue;
// only 'max' entries shall be added.
if (++cnt == max)
break;
// filter only non-directory entries
if (!(entry->_data.dwFileAttributes&FILE_ATTRIBUTE_DIRECTORY) && !lwr_filter.empty()) {
String lwr_name = entry->_data.cFileName;
String lwr_disp = entry->_display_name;
lwr_name.toLower();
lwr_disp.toLower();
if (!_tcsstr(lwr_name,lwr_filter) && !_tcsstr(lwr_disp,lwr_filter))
continue;
}
if (entry->_etype == ET_SHELL)
AddEntry(dir._folder, static_cast<ShellEntry*>(entry));
else
AddEntry(dir._folder, entry);
}
}
LRESULT StartMenu::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
switch(nmsg) {
case WM_PAINT: {
PaintCanvas canvas(_hwnd);
Paint(canvas);
break;}
case WM_SIZE:
ResizeButtons(LOWORD(lparam)-_border_left);
break;
case WM_MOVE: {
POINTS pos;
pos.x = LOWORD(lparam);
pos.y = HIWORD(lparam);
// move open submenus of floating menus
if (_submenu) {
int dx = pos.x - _last_pos.x;
int dy = pos.y - _last_pos.y;
if (dx || dy) {
WindowRect rt(_submenu);
SetWindowPos(_submenu, 0, rt.left+dx, rt.top+dy, 0, 0, SWP_NOSIZE|SWP_NOACTIVATE);
//MoveVisible(_submenu);
}
}
_last_pos.x = pos.x;
_last_pos.y = pos.y;
goto def;}
case WM_NCHITTEST: {
LRESULT res = super::WndProc(nmsg, wparam, lparam);
if (res>=HTSIZEFIRST && res<=HTSIZELAST)
return HTCLIENT; // disable window resizing
return res;}
case WM_LBUTTONDOWN: {
RECT rect;
// check mouse cursor for coordinates of floating button
GetFloatingButtonRect(&rect);
if (PtInRect(&rect, Point(lparam))) {
// create a floating copy of the current start menu
WindowRect pos(_hwnd);
///@todo do something similar to StartMenuRoot::TrackStartmenu() in order to automatically close submenus when clicking on the desktop background
StartMenu::Create(pos.left+3, pos.bottom-3, _create_info._folders, 0, _create_info._title, _create_info._creator, _create_info._info);
CloseStartMenu();
}
#ifdef _LIGHT_STARTMENU
int id = ButtonHitTest(Point(lparam));
if (id)
Command(id, BN_CLICKED);
#endif
break;}
case WM_SYSCOMMAND:
if ((wparam&0xFFF0) == SC_SIZE)
return 0; // disable window resizing
goto def;
case WM_ACTIVATEAPP:
// close start menu when activating another application
if (!wparam)
CloseStartMenu();
break; // don't call super::WndProc in case "this" has been deleted
case WM_CANCELMODE:
CloseStartMenu();
#ifdef _LIGHT_STARTMENU
if (_scroll_mode != SCROLL_NOT) {
ReleaseCapture();
KillTimer(_hwnd, 0);
}
#endif
break;
#ifdef _LIGHT_STARTMENU
case WM_MOUSEMOVE: {
// automatically set the focus to startmenu entries when moving the mouse over them
if (lparam != _last_mouse_pos) { // don't process WM_MOUSEMOVE when opening submenus using keyboard navigation
Point pt(lparam);
if (_arrow_btns) {
RECT rect_up, rect_down;
GetArrowButtonRects(&rect_up, &rect_down, _icon_size);
SCROLL_MODE scroll_mode = SCROLL_NOT;
if (PtInRect(&rect_up, pt))
scroll_mode = SCROLL_UP;
else if (PtInRect(&rect_down, pt))
scroll_mode = SCROLL_DOWN;
if (scroll_mode != _scroll_mode) {
if (scroll_mode == SCROLL_NOT) {
ReleaseCapture();
KillTimer(_hwnd, 0);
} else {
CloseSubmenus();
SetTimer(_hwnd, 0, 150, NULL); // 150 ms scroll interval
SetCapture(_hwnd);
}
_scroll_mode = scroll_mode;
}
}
int new_id = ButtonHitTest(pt);
if (new_id>0 && new_id!=_selected_id)
SelectButton(new_id);
_last_mouse_pos = lparam;
}
break;}
case WM_TIMER:
if (_scroll_mode == SCROLL_UP) {
if (_scroll_pos > 0) {
--_scroll_pos;
InvalidateRect(_hwnd, NULL, TRUE);
}
} else {
if (_scroll_pos <= _invisible_lines) {
++_scroll_pos;
InvalidateRect(_hwnd, NULL, TRUE);
}
}
break;
case WM_KEYDOWN:
ProcessKey(wparam);
break;
#else
case PM_STARTENTRY_FOCUSED: { ///@todo use TrackMouseEvent() and WM_MOUSEHOVER to wait a bit before opening submenus
BOOL hasSubmenu = wparam;
HWND hctrl = (HWND)lparam;
// automatically open submenus
if (hasSubmenu) {
UpdateWindow(_hwnd); // draw focused button before waiting on submenu creation
//SendMessage(_hwnd, WM_COMMAND, MAKEWPARAM(GetDlgCtrlID(hctrl),BN_CLICKED), (LPARAM)hctrl);
Command(GetDlgCtrlID(hctrl), BN_CLICKED);
} else {
// close any open submenu
CloseOtherSubmenus();
}
break;}
#endif
#ifdef _LAZY_ICONEXTRACT
case PM_UPDATE_ICONS:
UpdateIcons(/*wparam*/);
break;
#endif
case PM_STARTENTRY_LAUNCHED:
if (GetWindowStyle(_hwnd) & WS_CAPTION) // don't automatically close floating menus
return 0;
// route message to the parent menu and close menus after launching an entry
if (!SendParent(nmsg, wparam, lparam))
CloseStartMenu();
return 1; // signal that we have received and processed the message
case PM_STARTMENU_CLOSED:
_submenu = 0;
break;
case PM_SELECT_ENTRY:
SelectButtonIndex(0, wparam!=0);
break;
#ifdef _LIGHT_STARTMENU
case WM_CONTEXTMENU: {
Point screen_pt(lparam), clnt_pt=screen_pt;
ScreenToClient(_hwnd, &clnt_pt);
int id = ButtonHitTest(clnt_pt);
if (id) {
StartMenuEntry& sme = _entries[id];
for(ShellEntrySet::iterator it=sme._entries.begin(); it!=sme._entries.end(); ++it) {
Entry* entry = *it;
if (entry) {
CHECKERROR(entry->do_context_menu(_hwnd, screen_pt, _cm_ifs)); // may close start menu because of focus loss
///@todo refresh on successfull context menu execution?
break; ///@todo handle context menu for more than one entry
}
}
}
break;}
#endif
default: def:
return super::WndProc(nmsg, wparam, lparam);
}
return 0;
}
#ifdef _LIGHT_STARTMENU
int StartMenu::ButtonHitTest(POINT pt)
{
ClientRect clnt(_hwnd);
const int icon_size = _icon_size;
RECT rect = {_border_left, _border_top, clnt.right, STARTMENU_LINE_HEIGHT(icon_size)};
if (pt.x<rect.left || pt.x>rect.right)
return 0;
for(SMBtnVector::const_iterator it=_buttons.begin()+_scroll_pos; it!=_buttons.end(); ++it) {
const SMBtnInfo& info = *it;
if (rect.top > pt.y)
break;
rect.bottom = rect.top + (info._id==-1? STARTMENU_SEP_HEIGHT(icon_size): STARTMENU_LINE_HEIGHT(icon_size));
if (rect.bottom > _bottom_max)
break;
if (pt.y < rect.bottom) // PtInRect(&rect, pt)
return info._id;
rect.top = rect.bottom;
}
return 0;
}
void StartMenu::InvalidateSelection()
{
if (_selected_id <= 0)
return;
ClientRect clnt(_hwnd);
const int icon_size = _icon_size;
RECT rect = {_border_left, _border_top, clnt.right, STARTMENU_LINE_HEIGHT(icon_size)};
for(SMBtnVector::const_iterator it=_buttons.begin()+_scroll_pos; it!=_buttons.end(); ++it) {
const SMBtnInfo& info = *it;
rect.bottom = rect.top + (info._id==-1? STARTMENU_SEP_HEIGHT(icon_size): STARTMENU_LINE_HEIGHT(icon_size));
if (info._id == _selected_id) {
InvalidateRect(_hwnd, &rect, TRUE);
break;
}
rect.top = rect.bottom;
}
}
const SMBtnInfo* StartMenu::GetButtonInfo(int id) const
{
for(SMBtnVector::const_iterator it=_buttons.begin(); it!=_buttons.end(); ++it)
if (it->_id == id)
return &*it;
return NULL;
}
bool StartMenu::SelectButton(int id, bool open_sub)
{
if (id == -1)
return false;
if (id == _selected_id)
return true;
InvalidateSelection();
const SMBtnInfo* btn = GetButtonInfo(id);
if (btn && btn->_enabled) {
_selected_id = id;
InvalidateSelection();
// automatically open submenus
if (btn->_hasSubmenu) {
if (open_sub)
OpenSubmenu();
} else
CloseOtherSubmenus(); // close any open submenu
return true;
} else {
_selected_id = -1;
return false;
}
}
bool StartMenu::OpenSubmenu(bool select_first)
{
if (_selected_id == -1)
return false;
InvalidateSelection();
const SMBtnInfo* btn = GetButtonInfo(_selected_id);
// automatically open submenus
if (btn->_hasSubmenu) {
//@@ allows destroying of startmenu when processing PM_UPDATE_ICONS -> GPF
UpdateWindow(_hwnd); // draw focused button before waiting on submenu creation
Command(_selected_id, BN_CLICKED);
if (select_first && _submenu)
SendMessage(_submenu, PM_SELECT_ENTRY, (WPARAM)false, 0);
return true;
} else
return false;
}
int StartMenu::GetSelectionIndex()
{
if (_selected_id == -1)
return -1;
for(int i=0; i<(int)_buttons.size(); ++i)
if (_buttons[i]._id == _selected_id)
return i;
return -1;
}
bool StartMenu::SelectButtonIndex(int idx, bool open_sub)
{
if (idx>=0 && idx<(int)_buttons.size())
return SelectButton(_buttons[idx]._id, open_sub);
else
return false;
}
void StartMenu::ProcessKey(int vk)
{
switch(vk) {
case VK_RETURN:
if (_selected_id)
Command(_selected_id, BN_CLICKED);
break;
case VK_UP:
Navigate(-1);
break;
case VK_DOWN:
Navigate(+1);
break;
case VK_HOME:
SelectButtonIndex(0, false);
break;
case VK_END:
SelectButtonIndex(_buttons.size()-1, false);
break;
case VK_LEFT:
if (_submenu)
CloseOtherSubmenus();
else if (!(GetWindowStyle(_hwnd) & WS_CAPTION)) // don't automatically close floating menus
DestroyWindow(_hwnd);
break;
case VK_RIGHT:
OpenSubmenu(true);
break;
case VK_ESCAPE:
CloseStartMenu();
break;
default:
if (vk>='0' && vk<='Z')
JumpToNextShortcut(vk);
}
}
bool StartMenu::Navigate(int step)
{
int idx = GetSelectionIndex();
if (idx == -1)
{
if (step > 0)
idx = 0 - step;
else
idx = _buttons.size() - step;
}
for(;;) {
idx += step;
if (_buttons.size()<=1 && (idx<0 || idx>(int)_buttons.size()))
break;
if (idx < 0)
idx += _buttons.size();
if (idx > (int)_buttons.size())
idx -= _buttons.size()+1;
if (SelectButtonIndex(idx, false))
return true;
}
return false;
}
bool StartMenu::JumpToNextShortcut(char c)
{
int cur_idx = GetSelectionIndex();
if (cur_idx == -1)
cur_idx = 0;
int first_found = 0;
int found_more = 0;
SMBtnVector::const_iterator cur_it = _buttons.begin();
cur_it += cur_idx + 1;
// first search down from current item...
SMBtnVector::const_iterator it = cur_it;
for(; it!=_buttons.end(); ++it) {
const SMBtnInfo& btn = *it;
if (!btn._title.empty() && toupper((TBYTE)btn._title.at(0)) == c) {
if (!first_found)
first_found = btn._id;
else
++found_more;
}
}
// ...now search from top to the current item
it = _buttons.begin();
for(; it!=_buttons.end() && it!=cur_it; ++it) {
const SMBtnInfo& btn = *it;
if (!btn._title.empty() && toupper((TBYTE)btn._title.at(0)) == c) {
if (!first_found)
first_found = btn._id;
else
++found_more;
}
}
if (first_found) {
SelectButton(first_found);
if (!found_more)
Command(first_found, BN_CLICKED);
return true;
} else
return false;
}
#endif // _LIGHT_STARTMENU
bool StartMenu::GetButtonRect(int id, PRECT prect) const
{
#ifdef _LIGHT_STARTMENU
ClientRect clnt(_hwnd);
const int icon_size = _icon_size;
RECT rect = {_border_left, _border_top, clnt.right, STARTMENU_LINE_HEIGHT(icon_size)};
for(SMBtnVector::const_iterator it=_buttons.begin()+_scroll_pos; it!=_buttons.end(); ++it) {
const SMBtnInfo& info = *it;
rect.bottom = rect.top + (info._id==-1? STARTMENU_SEP_HEIGHT(icon_size): STARTMENU_LINE_HEIGHT(icon_size));
if (info._id == id) {
*prect = rect;
return true;
}
rect.top = rect.bottom;
}
return false;
#else
HWND btn = GetDlgItem(_hwnd, id);
if (btn) {
GetWindowRect(btn, prect);
ScreenToClient(_hwnd, prect);
return true;
} else
return false;
#endif
}
void StartMenu::DrawFloatingButton(HDC hdc)
{
static ResIconEx floatingIcon(IDI_FLOATING, 8, 4);
ClientRect clnt(_hwnd);
DrawIconEx(hdc, clnt.right-12, 0, floatingIcon, 8, 4, 0, 0, DI_NORMAL);
}
void StartMenu::GetFloatingButtonRect(LPRECT prect)
{
GetClientRect(_hwnd, prect);
prect->right -= 4;
prect->left = prect->right - 8;
prect->bottom = 4;
}
void StartMenu::DrawArrows(HDC hdc, int icon_size)
{
int cx = icon_size / 2;
int cy = icon_size / 4;
ResIconEx arrowUpIcon(IDI_ARROW_UP, cx, cy);
ResIconEx arrowDownIcon(IDI_ARROW_DOWN, cx, cy);
ClientRect clnt(_hwnd);
DrawIconEx(hdc, clnt.right/2-cx/2, _floating_btn?3:1, arrowUpIcon, cx, cy, 0, 0, DI_NORMAL);
DrawIconEx(hdc, clnt.right/2-cx/2, clnt.bottom-cy-1, arrowDownIcon, cx, cy, 0, 0, DI_NORMAL);
}
void StartMenu::GetArrowButtonRects(LPRECT prect_up, LPRECT prect_down, int icon_size)
{
int cx = icon_size / 2;
int cy = icon_size / 4;
GetClientRect(_hwnd, prect_up);
*prect_down = *prect_up;
// prect_up->left = prect_up->right/2 - cx/2;
// prect_up->right = prect_up->left + cy;
prect_up->right -= cx;
prect_up->top = _floating_btn? cy-1: 1;
prect_up->bottom = prect_up->top + cy;
// prect_down->left = prect_down->right/2 - cx/2;
// prect_down->right = prect_down->left + cy;
prect_down->right -= cx;
prect_down->top = prect_down->bottom - cy - 1;
}
void StartMenu::Paint(PaintCanvas& canvas)
{
if (_floating_btn)
DrawFloatingButton(canvas);
#ifdef _LIGHT_STARTMENU
if (_arrow_btns)
DrawArrows(canvas, _icon_size);
ClientRect clnt(_hwnd);
const int icon_size = _icon_size;
RECT rect = {_border_left, _border_top, clnt.right, STARTMENU_LINE_HEIGHT(icon_size)};
int sep_width = rect.right-rect.left - 4;
FontSelection font(canvas, GetStockFont(DEFAULT_GUI_FONT));
BkMode bk_mode(canvas, TRANSPARENT);
for(SMBtnVector::const_iterator it=_buttons.begin()+_scroll_pos; it!=_buttons.end(); ++it) {
const SMBtnInfo& btn = *it;
if (rect.top > canvas.rcPaint.bottom)
break;
if (btn._id == -1) { // a separator?
rect.bottom = rect.top + STARTMENU_SEP_HEIGHT(icon_size);
if (rect.bottom > _bottom_max)
break;
BrushSelection brush_sel(canvas, GetSysColorBrush(COLOR_BTNSHADOW));
PatBlt(canvas, rect.left+2, rect.top+STARTMENU_SEP_HEIGHT(icon_size)/2-1, sep_width, 1, PATCOPY);
SelectBrush(canvas, GetSysColorBrush(COLOR_BTNHIGHLIGHT));
PatBlt(canvas, rect.left+2, rect.top+STARTMENU_SEP_HEIGHT(icon_size)/2, sep_width, 1, PATCOPY);
} else {
rect.bottom = rect.top + STARTMENU_LINE_HEIGHT(icon_size);
if (rect.bottom > _bottom_max)
break;
if (rect.top >= canvas.rcPaint.top)
DrawStartMenuButton(canvas, rect, btn._title, btn, btn._id==_selected_id, false, _icon_size);
}
rect.top = rect.bottom;
}
#endif
}
#ifdef _LAZY_ICONEXTRACT
void StartMenu::UpdateIcons(/*int idx*/)
{
UpdateWindow(_hwnd);
#ifdef _SINGLE_ICONEXTRACT
//if (idx >= 0)
int idx = _scroll_pos;
for(; idx<(int)_buttons.size(); ++idx) {
SMBtnInfo& btn = _buttons[idx];
if (btn._icon_id==ICID_UNKNOWN && btn._id>0) {
StartMenuEntry& sme = _entries[btn._id];
btn._icon_id = ICID_NONE;
for(ShellEntrySet::iterator it=sme._entries.begin(); it!=sme._entries.end(); ++it) {
Entry* entry = *it;
if (entry->_icon_id == ICID_UNKNOWN)
entry->_icon_id = entry->safe_extract_icon(ICF_FROM_ICON_SIZE(_icon_size));
if (entry->_icon_id > ICID_NONE) {
btn._icon_id = (ICON_ID)/*@@*/ entry->_icon_id;
RECT rect;
GetButtonRect(btn._id, &rect);
if (rect.bottom > _bottom_max)
break;
WindowCanvas canvas(_hwnd);
DrawStartMenuButton(canvas, rect, NULL, btn, btn._id==_selected_id, false, _icon_size);
//InvalidateRect(_hwnd, &rect, FALSE);
//UpdateWindow(_hwnd);
//break;
break;
}
}
}
}
// if (++idx < (int)_buttons.size())
// PostMessage(_hwnd, PM_UPDATE_ICONS, idx, 0);
#else
int icons_extracted = 0;
int icons_updated = 0;
for(StartMenuShellDirs::iterator it=_dirs.begin(); it!=_dirs.end(); ++it) {
ShellDirectory& dir = it->_dir;
icons_extracted += dir.extract_icons(icon_size);
}
if (icons_extracted) {
for(ShellEntryMap::iterator it1=_entries.begin(); it1!=_entries.end(); ++it1) {
StartMenuEntry& sme = it1->second;
if (!sme._hIcon) {
sme._hIcon = (HICON)-1;
for(ShellEntrySet::const_iterator it2=sme._entries.begin(); it2!=sme._entries.end(); ++it2) {
const Entry* sm_entry = *it2;
if (sm_entry->_hIcon) {
sme._hIcon = sm_entry->_hIcon;
break;
}
}
}
}
for(SMBtnVector::iterator it=_buttons.begin(); it!=_buttons.end(); ++it) {
SMBtnInfo& info = *it;
if (info._id>0 && !info._hIcon) {
info._hIcon = _entries[info._id]._hIcon;
++icons_updated;
}
}
}
if (icons_updated) {
InvalidateRect(_hwnd, NULL, FALSE);
UpdateWindow(_hwnd);
}
#endif
}
#endif
// resize child button controls to accomodate for new window size
void StartMenu::ResizeButtons(int cx)
{
HDWP hdwp = BeginDeferWindowPos(10);
for(HWND ctrl=GetWindow(_hwnd,GW_CHILD); ctrl; ctrl=GetNextWindow(ctrl,GW_HWNDNEXT)) {
ClientRect rt(ctrl);
if (rt.right != cx) {
int height = rt.bottom - rt.top;
// special handling for separator controls
if (!height && (GetWindowStyle(ctrl)&SS_TYPEMASK)==SS_ETCHEDHORZ)
height = 2;
hdwp = DeferWindowPos(hdwp, ctrl, 0, 0, 0, cx, height, SWP_NOMOVE|SWP_NOZORDER|SWP_NOACTIVATE);
}
}
EndDeferWindowPos(hdwp);
}
int StartMenu::Command(int id, int code)
{
#ifndef _LIGHT_STARTMENU
switch(id) {
case IDCANCEL:
CloseStartMenu(id);
break;
default: {
#endif
ShellEntryMap::iterator found = _entries.find(id);
if (found != _entries.end()) {
ActivateEntry(id, found->second._entries);
return 0;
}
return super::Command(id, code);
#ifndef _LIGHT_STARTMENU
}
}
return 0;
#endif
}
ShellEntryMap::iterator StartMenu::AddEntry(const String& title, ICON_ID icon_id, Entry* entry)
{
// search for an already existing subdirectory entry with the same name
if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
for(ShellEntryMap::iterator it=_entries.begin(); it!=_entries.end(); ++it) {
StartMenuEntry& sme = it->second;
if (!_tcsicmp(sme._title, title)) ///@todo speed up by using a map indexed by name
{
for(ShellEntrySet::iterator it2=sme._entries.begin(); it2!=sme._entries.end(); ++it2) {
if ((*it2)->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
// merge the new shell entry with the existing of the same name
sme._entries.insert(entry);
return it;
}
}
}
}
}
ShellEntryMap::iterator sme = AddEntry(title, icon_id);
sme->second._entries.insert(entry);
return sme;
}
ShellEntryMap::iterator StartMenu::AddEntry(const String& title, ICON_ID icon_id, int id)
{
if (id == -1)
id = ++_next_id;
StartMenuEntry sme;
sme._title = title;
sme._icon_id = icon_id;
ShellEntryMap::iterator it = _entries.insert(make_pair(id, sme)).first;
return it;
}
ShellEntryMap::iterator StartMenu::AddEntry(const ShellFolder folder, ShellEntry* entry)
{
ICON_ID icon_id;
if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
icon_id = ICID_APPS;
else
icon_id = (ICON_ID)/*@@*/ entry->_icon_id;
return AddEntry(folder.get_name(entry->_pidl), icon_id, entry);
}
ShellEntryMap::iterator StartMenu::AddEntry(const ShellFolder folder, Entry* entry)
{
ICON_ID icon_id;
if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
icon_id = ICID_APPS;
else
icon_id = (ICON_ID)/*@@*/ entry->_icon_id;
return AddEntry(entry->_display_name, icon_id, entry);
}
void StartMenu::AddButton(LPCTSTR title, ICON_ID icon_id, bool hasSubmenu, int id, bool enabled)
{
#ifdef _LIGHT_STARTMENU
_buttons.push_back(SMBtnInfo(title, icon_id, id, hasSubmenu, enabled));
#else
DWORD style = enabled? WS_VISIBLE|WS_CHILD|BS_OWNERDRAW: WS_VISIBLE|WS_CHILD|BS_OWNERDRAW|WS_DISABLED;
WindowRect rect(_hwnd);
ClientRect clnt(_hwnd);
// increase window height to make room for the new button
rect.top -= STARTMENU_LINE_HEIGHT(icon_size);
// move down if we are too high now
if (rect.top < 0) {
rect.top += STARTMENU_LINE_HEIGHT(icon_size);
rect.bottom += STARTMENU_LINE_HEIGHT(icon_size);
}
WindowCanvas canvas(_hwnd);
FontSelection font(canvas, GetStockFont(DEFAULT_GUI_FONT));
// widen window, if it is too small
int text_width = GetStartMenuBtnTextWidth(canvas, title, _hwnd) + icon_size + 10/*placeholder*/ + 16/*arrow*/;
int cx = clnt.right - _border_left;
if (text_width > cx)
rect.right += text_width-cx;
MoveWindow(_hwnd, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, TRUE);
StartMenuCtrl(_hwnd, _border_left, clnt.bottom, rect.right-rect.left-_border_left,
title, id, g_Globals._icon_cache.get_icon(icon_id)._hIcon, hasSubmenu, style);
#endif
}
void StartMenu::AddSeparator()
{
#ifdef _LIGHT_STARTMENU
_buttons.push_back(SMBtnInfo(NULL, ICID_NONE, -1, false));
#else
WindowRect rect(_hwnd);
ClientRect clnt(_hwnd);
// increase window height to make room for the new separator
rect.top -= STARTMENU_SEP_HEIGHT(icon_size);
// move down if we are too high now
if (rect.top < 0) {
rect.top += STARTMENU_LINE_HEIGHT(icon_size);
rect.bottom += STARTMENU_LINE_HEIGHT(icon_size);
}
MoveWindow(_hwnd, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, TRUE);
StartMenuSeparator(_hwnd, _border_left, clnt.bottom, rect.right-rect.left-_border_left);
#endif
}
bool StartMenu::CloseOtherSubmenus(int id)
{
if (_submenu) {
if (IsWindow(_submenu)) {
if (_submenu_id == id)
return false;
else {
_submenu_id = 0;
DestroyWindow(_submenu);
// _submenu should be reset automatically by PM_STARTMENU_CLOSED, but safety first...
}
}
_submenu = 0;
}
return true;
}
void StartMenu::CreateSubmenu(int id, LPCTSTR title, CREATORFUNC_INFO creator, void* info)
{
CreateSubmenu(id, StartMenuFolders(), title, creator, info);
}
bool StartMenu::CreateSubmenu(int id, int folder_id, LPCTSTR title, CREATORFUNC_INFO creator, void* info)
{
try {
SpecialFolderPath folder(folder_id, _hwnd);
StartMenuFolders new_folders;
new_folders.push_back(folder);
CreateSubmenu(id, new_folders, title, creator, info);
return true;
} catch(COMException&) {
// ignore Exception and don't display anything
CloseOtherSubmenus(id);
_buttons[GetSelectionIndex()]._enabled = false; // disable entries for non-existing folders
return false;
}
}
bool StartMenu::CreateSubmenu(int id, int folder_id1, int folder_id2, LPCTSTR title, CREATORFUNC_INFO creator, void* info)
{
StartMenuFolders new_folders;
try {
new_folders.push_back(SpecialFolderPath(folder_id1, _hwnd));
} catch(COMException&) {
}
try {
new_folders.push_back(SpecialFolderPath(folder_id2, _hwnd));
} catch(COMException&) {
}
if (!new_folders.empty()) {
CreateSubmenu(id, new_folders, title, creator, info);
return true;
} else {
CloseOtherSubmenus(id);
_buttons[GetSelectionIndex()]._enabled = false; // disable entries for non-existing folders
return false;
}
}
void StartMenu::CreateSubmenu(int id, const StartMenuFolders& new_folders, LPCTSTR title, CREATORFUNC_INFO creator, void* info)
{
// Only open one submenu at a time.
if (!CloseOtherSubmenus(id))
return;
RECT rect;
int x, y;
if (GetButtonRect(id, &rect)) {
ClientToScreen(_hwnd, &rect);
x = rect.right; // Submenus should overlap their parent a bit.
const int icon_size = _icon_size;
y = rect.top+STARTMENU_LINE_HEIGHT(icon_size) +_border_top/*own border*/ -STARTMENU_TOP_BTN_SPACE/*border of new submenu*/;
} else {
WindowRect pos(_hwnd);
x = pos.right;
y = pos.top;
}
_submenu_id = id;
_submenu = StartMenu::Create(x, y, new_folders, _hwnd, title, creator, info, _create_info._filter);
}
void StartMenu::ActivateEntry(int id, const ShellEntrySet& entries)
{
StartMenuFolders new_folders;
String title;
for(ShellEntrySet::const_iterator it=entries.begin(); it!=entries.end(); ++it) {
Entry* entry = const_cast<Entry*>(*it);
if (entry->_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
///@todo If the user explicitly clicked on a submenu, display this folder as floating start menu.
if (entry->_etype == ET_SHELL)
new_folders.push_back(entry->create_absolute_pidl());
else {
TCHAR path[MAX_PATH];
if (entry->get_path(path, COUNTOF(path)))
new_folders.push_back(path);
}
if (title.empty())
title = entry->_display_name;
} else {
// The entry is no subdirectory, so there can only be one shell entry.
assert(entries.size()==1);
HWND hparent = GetParent(_hwnd);
ShellPath shell_path = entry->create_absolute_pidl();
// close start menus when launching the selected entry
CloseStartMenu(id);
///@todo launch in the background; specify correct HWND for error message box titles
SHELLEXECUTEINFO shexinfo;
shexinfo.cbSize = sizeof(SHELLEXECUTEINFO);
shexinfo.fMask = SEE_MASK_IDLIST; // SEE_MASK_INVOKEIDLIST is also possible.
shexinfo.hwnd = hparent;
shexinfo.lpVerb = NULL;
shexinfo.lpFile = NULL;
shexinfo.lpParameters = NULL;
shexinfo.lpDirectory = NULL;
shexinfo.nShow = SW_SHOWNORMAL;
shexinfo.lpIDList = &*shell_path;
// add PIDL to the recent file list
SHAddToRecentDocs(SHARD_PIDL, shexinfo.lpIDList);
if (!ShellExecuteEx(&shexinfo))
display_error(hparent, GetLastError());
// we may have deleted 'this' - ensure we leave the loop and function
return;
}
}
if (!new_folders.empty()) {
// Only open one submenu at a time.
if (!CloseOtherSubmenus(id))
return;
CreateSubmenu(id, new_folders, title);
}
}
/// close all windows of the start menu popup
void StartMenu::CloseStartMenu(int id)
{
if (!(GetWindowStyle(_hwnd) & WS_CAPTION)) { // don't automatically close floating menus
if (!SendParent(PM_STARTENTRY_LAUNCHED, id, (LPARAM)_hwnd))
DestroyWindow(_hwnd);
} else if (_submenu) // instead close submenus of floating parent menus
CloseSubmenus();
}
int GetStartMenuBtnTextWidth(HDC hdc, LPCTSTR title, HWND hwnd)
{
RECT rect = {0, 0, 0, 0};
DrawText(hdc, title, -1, &rect, DT_SINGLELINE|DT_NOPREFIX|DT_CALCRECT);
return rect.right-rect.left;
}
#ifdef _LIGHT_STARTMENU
void DrawStartMenuButton(HDC hdc, const RECT& rect, LPCTSTR title, const SMBtnInfo& btn, bool has_focus, bool pushed, int icon_size)
#else
void DrawStartMenuButton(HDC hdc, const RECT& rect, LPCTSTR title, HICON hIcon,
bool hasSubmenu, bool enabled, bool has_focus, bool pushed, int icon_size);
#endif
{
UINT style = DFCS_BUTTONPUSH;
if (!btn._enabled)
style |= DFCS_INACTIVE;
POINT iconPos = {rect.left+2, (rect.top+rect.bottom-icon_size)/2};
RECT textRect = {rect.left+icon_size+4, rect.top+2, rect.right-4, rect.bottom-4};
if (pushed) {
style |= DFCS_PUSHED;
++iconPos.x; ++iconPos.y;
++textRect.left; ++textRect.top;
++textRect.right; ++textRect.bottom;
}
int bk_color_idx = COLOR_BTNFACE;
int text_color_idx = COLOR_BTNTEXT;
if (has_focus) {
bk_color_idx = COLOR_HIGHLIGHT;
text_color_idx = COLOR_HIGHLIGHTTEXT;
}
COLORREF bk_color = GetSysColor(bk_color_idx);
HBRUSH bk_brush = GetSysColorBrush(bk_color_idx);
if (title)
FillRect(hdc, &rect, bk_brush);
if (btn._icon_id > ICID_NONE)
g_Globals._icon_cache.get_icon(btn._icon_id).draw(hdc, iconPos.x, iconPos.y, icon_size, icon_size, bk_color, bk_brush/*, icon_size*/);
// draw submenu arrow at the right
if (btn._hasSubmenu) {
ResIconEx arrowIcon(IDI_ARROW, icon_size, icon_size);
ResIconEx selArrowIcon(IDI_ARROW_SELECTED, icon_size, icon_size);
DrawIconEx(hdc, rect.right-icon_size, iconPos.y,
has_focus? selArrowIcon: arrowIcon,
icon_size, icon_size, 0, bk_brush, DI_NORMAL);
}
if (title) {
BkMode bk_mode(hdc, TRANSPARENT);
if (!btn._enabled) // dis->itemState & (ODS_DISABLED|ODS_GRAYED)
DrawGrayText(hdc, &textRect, title, DT_SINGLELINE|DT_NOPREFIX|DT_VCENTER);
else {
TextColor lcColor(hdc, GetSysColor(text_color_idx));
DrawText(hdc, title, -1, &textRect, DT_SINGLELINE|DT_NOPREFIX|DT_VCENTER);
}
}
}
#ifdef _LIGHT_STARTMENU
void StartMenu::ResizeToButtons()
{
WindowRect rect(_hwnd);
WindowCanvas canvas(_hwnd);
FontSelection font(canvas, GetStockFont(DEFAULT_GUI_FONT));
const int icon_size = _icon_size;
int max_width = STARTMENU_WIDTH_MIN;
int height = 0;
for(SMBtnVector::const_iterator it=_buttons.begin(); it!=_buttons.end(); ++it) {
int w = GetStartMenuBtnTextWidth(canvas, it->_title, _hwnd);
if (w > max_width)
max_width = w;
if (it->_id == -1)
height += STARTMENU_SEP_HEIGHT(icon_size);
else
height += STARTMENU_LINE_HEIGHT(icon_size);
}
// calculate new window size
int text_width = max_width + icon_size + 10/*placeholder*/ + 16/*arrow*/;
RECT rt_hgt = {rect.left, rect.bottom-_border_top-height, rect.left+_border_left+text_width, rect.bottom};
AdjustWindowRectEx(&rt_hgt, GetWindowStyle(_hwnd), FALSE, GetWindowExStyle(_hwnd));
// ignore movement, only look at the size change
rect.right = rect.left + (rt_hgt.right-rt_hgt.left);
rect.top = rect.bottom - (rt_hgt.bottom-rt_hgt.top);
// move down if we are too high
if (rect.top < 0) {
int dy = -rect.top;
rect.top += dy;
rect.bottom += dy;
}
// enable scroll mode for long start menus, which span more than the whole screen height
int cyscreen = GetSystemMetrics(SM_CYSCREEN);
int bottom_max = 0;
if (rect.bottom > cyscreen) {
_arrow_btns = true;
_invisible_lines = (rect.bottom-cyscreen+(STARTMENU_LINE_HEIGHT(icon_size)+1))/STARTMENU_LINE_HEIGHT(icon_size)+1;
rect.bottom -= _invisible_lines * STARTMENU_LINE_HEIGHT(icon_size);
bottom_max = rect.bottom;
if (_floating_btn)
rect.bottom += 6; // lower scroll arrow
else {
_border_top += 6; // upper scroll arrow
rect.bottom += 2*6; // upper+lower scroll arrow
}
}
MoveWindow(_hwnd, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, TRUE);
if (bottom_max) {
POINT pt = {0, bottom_max};
ScreenToClient(_hwnd, &pt);
_bottom_max = pt.y;
}
}
#else // _LIGHT_STARTMENU
LRESULT StartMenuButton::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
switch(nmsg) {
case WM_MOUSEMOVE:
// automatically set the focus to startmenu entries when moving the mouse over them
if (GetFocus()!=_hwnd && !(GetWindowStyle(_hwnd)&WS_DISABLED))
SetFocus(_hwnd);
break;
case WM_SETFOCUS:
PostParent(PM_STARTENTRY_FOCUSED, _hasSubmenu, (LPARAM)_hwnd);
goto def;
case WM_CANCELMODE:
// route WM_CANCELMODE to the startmenu window
return SendParent(nmsg, wparam, lparam);
default: def:
return super::WndProc(nmsg, wparam, lparam);
}
return 0;
}
void StartMenuButton::DrawItem(LPDRAWITEMSTRUCT dis)
{
TCHAR title[BUFFER_LEN];
GetWindowText(_hwnd, title, BUFFER_LEN);
DrawStartMenuButton(dis->hDC, dis->rcItem, title, _hIcon,
_hasSubmenu,
!(dis->itemState & ODS_DISABLED),
dis->itemState&ODS_FOCUS? true: false,
dis->itemState&ODS_SELECTED? true: false);
}
#endif
StartMenuRoot::StartMenuRoot(HWND hwnd, const StartMenuRootCreateInfo& info)
: super(hwnd, info._icon_size),
_hwndStartButton(0)
{
if (!g_Globals._SHRestricted || !SHRestricted(REST_NOCOMMONGROUPS))
try {
// insert directory "All Users\Start Menu"
ShellDirectory cmn_startmenu(GetDesktopFolder(), SpecialFolderPath(CSIDL_COMMON_STARTMENU, _hwnd), _hwnd);
_dirs.push_back(StartMenuDirectory(cmn_startmenu, (LPCTSTR)SpecialFolderFSPath(CSIDL_COMMON_PROGRAMS, _hwnd)));
} catch(COMException&) {
// ignore exception and don't show additional shortcuts
}
try {
// insert directory "<user name>\Start Menu"
ShellDirectory usr_startmenu(GetDesktopFolder(), SpecialFolderPath(CSIDL_STARTMENU, _hwnd), _hwnd);
_dirs.push_back(StartMenuDirectory(usr_startmenu, (LPCTSTR)SpecialFolderFSPath(CSIDL_PROGRAMS, _hwnd)));
} catch(COMException&) {
// ignore exception and don't show additional shortcuts
}
ReadLogoSize();
}
void StartMenuRoot::ReadLogoSize()
{
// read size of logo bitmap
BITMAP bmp_hdr;
GetObject(ResBitmap(GetLogoResId()), sizeof(BITMAP), &bmp_hdr);
_logo_size.cx = bmp_hdr.bmWidth;
_logo_size.cy = bmp_hdr.bmHeight;
// cache logo width
_border_left = _logo_size.cx + 1;
}
static void CalculateStartPos(HWND hwndOwner, RECT& rect, int icon_size)
{
WindowRect pos(hwndOwner);
rect.left = pos.left;
rect.top = pos.top-STARTMENU_LINE_HEIGHT(icon_size)-4;
rect.right = pos.left+STARTMENU_WIDTH_MIN;
rect.bottom = pos.top;
#ifndef _LIGHT_STARTMENU
rect.top += STARTMENU_LINE_HEIGHT(icon_size);
#endif
}
HWND StartMenuRoot::Create(HWND hwndOwner, int icon_size)
{
RECT rect;
CalculateStartPos(hwndOwner, rect, icon_size);
StartMenuRootCreateInfo create_info;
create_info._icon_size = icon_size;
return Window::Create(WINDOW_CREATOR_INFO(StartMenuRoot,StartMenuRootCreateInfo), &create_info, 0, GetWndClasss(), TITLE_STARTMENU,
WS_POPUP|WS_THICKFRAME|WS_CLIPCHILDREN,
rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, hwndOwner);
}
void StartMenuRoot::TrackStartmenu()
{
MSG msg;
HWND hwnd = _hwnd;
#ifdef _LIGHT_STARTMENU
_selected_id = -1;
#endif
#ifdef _LIGHT_STARTMENU
// recalculate start menu root position
RECT rect;
CalculateStartPos(_hwndStartButton, rect, _icon_size);
SetWindowPos(hwnd, 0, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, 0);
ResizeToButtons();
#endif
// show previously hidden start menu
ShowWindow(hwnd, SW_SHOW);
SetForegroundWindow(hwnd);
while(IsWindow(hwnd) && IsWindowVisible(hwnd)) {
if (!GetMessage(&msg, 0, 0, 0)) {
PostQuitMessage(msg.wParam);
break;
}
// Check for a mouse click on any window, that is not part of the start menu
if (msg.message==WM_LBUTTONDOWN || msg.message==WM_MBUTTONDOWN || msg.message==WM_RBUTTONDOWN) {
StartMenu* menu_wnd = NULL;
for(HWND hwnd=msg.hwnd; hwnd; hwnd=GetParent(hwnd)) {
menu_wnd = WINDOW_DYNAMIC_CAST(StartMenu, hwnd);
if (menu_wnd)
break;
}
if (!menu_wnd) {
CloseStartMenu();
break;
}
}
try {
if (pretranslate_msg(&msg))
continue;
if (dispatch_dialog_msg(&msg))
continue;
TranslateMessage(&msg);
try {
DispatchMessage(&msg);
} catch(COMException& e) {
HandleException(e, _hwnd);
}
} catch(COMException& e) {
HandleException(e, _hwnd);
}
}
}
int StartMenuRoot::Command(int id, int code)
{
return super::Command(id, code);
}
LRESULT StartMenuRoot::Init(LPCREATESTRUCT pcs)
{
// add buttons for entries in _entries
if (super::Init(pcs))
return 1;
AddSeparator();
// insert hard coded start entries
AddButton(ResString(IDS_PROGRAMS), ICID_APPS, true, IDC_PROGRAMS);
AddButton(ResString(IDS_DOCUMENTS), ICID_DOCUMENTS, true, IDC_DOCUMENTS);
if (!g_Globals._SHRestricted || !SHRestricted(REST_NORECENTDOCSMENU))
AddButton(ResString(IDS_RECENT), ICID_RECENT, true, IDC_RECENT);
AddButton(ResString(IDS_FAVORITES), ICID_FAVORITES, true, IDC_FAVORITES);
AddButton(ResString(IDS_SETTINGS), ICID_CONFIG, true, IDC_SETTINGS);
AddButton(ResString(IDS_BROWSE), ICID_FOLDER, true, IDC_BROWSE);
if (!g_Globals._SHRestricted || !SHRestricted(REST_NOFIND))
AddButton(ResString(IDS_SEARCH), ICID_SEARCH, true, IDC_SEARCH);
AddButton(ResString(IDS_START_HELP), ICID_INFO, false, IDC_START_HELP);
if (!g_Globals._SHRestricted || !SHRestricted(REST_NORUN))
AddButton(ResString(IDS_LAUNCH), ICID_ACTION, false, IDC_LAUNCH);
AddSeparator();
if (!g_Globals._SHRestricted || SHRestricted(REST_STARTMENULOGOFF) != 1)
AddButton(ResString(IDS_LOGOFF), ICID_LOGOFF, false, IDC_LOGOFF);
#ifdef __REACTOS__
AddButton(ResString(IDS_RESTART), ICID_RESTART, false, IDC_RESTART);
#endif
if (!g_Globals._SHRestricted || !SHRestricted(REST_NOCLOSE))
AddButton(ResString(IDS_SHUTDOWN), ICID_SHUTDOWN, false, IDC_SHUTDOWN);
#ifndef __REACTOS__
AddButton(ResString(IDS_TERMINATE), ICID_LOGOFF, false, IDC_TERMINATE);
#endif
#ifdef _LIGHT_STARTMENU
// set the window size to fit all buttons
ResizeToButtons();
#endif
return 0;
}
void StartMenuRoot::AddEntries()
{
super::AddEntries();
AddButton(ResString(IDS_EXPLORE), ICID_EXPLORER, false, IDC_EXPLORE);
}
LRESULT StartMenuRoot::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
switch(nmsg) {
case WM_PAINT: {
PaintCanvas canvas(_hwnd);
Paint(canvas);
break;}
case WM_DISPLAYCHANGE:
// re-evaluate logo size using the correct color depth
ReadLogoSize();
break;
default:
return super::WndProc(nmsg, wparam, lparam);
}
return 0;
}
void StartMenuRoot::Paint(PaintCanvas& canvas)
{
MemCanvas mem_dc;
ResBitmap bmp(GetLogoResId());
BitmapSelection sel(mem_dc, bmp);
ClientRect clnt(_hwnd);
int h = min(_logo_size.cy, clnt.bottom);
RECT rect = {0, 0, _logo_size.cx, clnt.bottom-h};
HBRUSH hbr = CreateSolidBrush(GetPixel(mem_dc, 0, 0));
FillRect(canvas, &rect, hbr);
DeleteObject(hbr);
PatBlt(canvas, _logo_size.cx, 0, 1, clnt.bottom, WHITENESS);
BitBlt(canvas, 0, clnt.bottom-h, _logo_size.cx, h, mem_dc, 0, ( h<_logo_size.cy ? _logo_size.cy-h : 0) , SRCCOPY);
super::Paint(canvas);
}
UINT StartMenuRoot::GetLogoResId()
{
WindowCanvas dc(_hwnd);
int clr_bits = GetDeviceCaps(dc, BITSPIXEL);
if (clr_bits > 8)
return IDB_LOGOV;
else if (clr_bits > 4)
return IDB_LOGOV256;
else
return IDB_LOGOV16;
}
void StartMenuRoot::CloseStartMenu(int id)
{
if (_submenu)
CloseSubmenus();
ShowWindow(_hwnd, SW_HIDE);
}
bool StartMenuRoot::IsStartMenuVisible() const
{
return IsWindowVisible(_hwnd) != FALSE;
}
void StartMenuRoot::ProcessKey(int vk)
{
switch(vk) {
case VK_LEFT:
if (_submenu)
CloseOtherSubmenus();
// don't close start menu root
break;
default:
super::ProcessKey(vk);
}
}
int StartMenuHandler::Command(int id, int code)
{
switch(id) {
// start menu root
case IDC_PROGRAMS:
CreateSubmenu(id, CSIDL_COMMON_PROGRAMS, CSIDL_PROGRAMS, ResString(IDS_PROGRAMS));
break;
case IDC_EXPLORE:
CloseStartMenu(id);
explorer_show_frame(SW_SHOWNORMAL);
break;
case IDC_LAUNCH:
CloseStartMenu(id);
ShowLaunchDialog(g_Globals._hwndDesktopBar);
break;
case IDC_DOCUMENTS:
CreateSubmenu(id, CSIDL_PERSONAL, ResString(IDS_DOCUMENTS));
break;
case IDC_RECENT:
CreateSubmenu(id, CSIDL_RECENT, ResString(IDS_RECENT), STARTMENU_CREATOR(RecentStartMenu));
break;
case IDC_FAVORITES:
#ifndef _SHELL32_FAVORITES
CreateSubmenu(id, ResString(IDS_FAVORITES), STARTMENU_CREATOR(FavoritesMenu), &static_cast<BookmarkList&>(g_Globals._favorites));
#else
CreateSubmenu(id, CSIDL_COMMON_FAVORITES, CSIDL_FAVORITES, ResString(IDS_FAVORITES));
#endif
break;
case IDC_BROWSE:
CreateSubmenu(id, ResString(IDS_BROWSE), STARTMENU_CREATOR(BrowseMenu));
break;
case IDC_SETTINGS:
CreateSubmenu(id, ResString(IDS_SETTINGS), STARTMENU_CREATOR(SettingsMenu));
break;
case IDC_SEARCH:
CreateSubmenu(id, ResString(IDS_SEARCH), STARTMENU_CREATOR(SearchMenu));
break;
case IDC_START_HELP:
CloseStartMenu(id);
MessageBox(g_Globals._hwndDesktopBar, TEXT("Help not yet implemented"), ResString(IDS_TITLE), MB_OK);
break;
case IDC_LOGOFF:
CloseStartMenu(id);
ShowLogoffDialog(g_Globals._hwndDesktopBar);
break;
#ifndef __REACTOS__
case IDC_TERMINATE:
DestroyWindow(g_Globals._hwndDesktopBar);
DestroyWindow(g_Globals._hwndDesktop);
break;
#endif
case IDC_SHUTDOWN:
CloseStartMenu(id);
ShowExitWindowsDialog(g_Globals._hwndDesktop);
break;
case IDC_RESTART:
CloseStartMenu(id);
ShowRestartDialog(g_Globals._hwndDesktop, EWX_REBOOT);
/* An alternative way to do restart without shell32 help */
//launch_file(_hwnd, TEXT("shutdown.exe"), SW_HIDE, TEXT("-r"));
break;
// settings menu
case ID_DESKTOPBAR_SETTINGS:
CloseStartMenu(id);
ExplorerPropertySheet(g_Globals._hwndDesktopBar);
break;
case IDC_CONTROL_PANEL: {
CloseStartMenu(id);
#ifndef ROSSHELL
#ifndef _NO_MDI
XMLPos explorer_options = g_Globals.get_cfg("general/explorer");
bool mdi = XMLBool(explorer_options, "mdi", true);
if (mdi)
MDIMainFrame::Create(TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}"), 0);
else
#endif
SDIMainFrame::Create(TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}"), 0);
#else
launch_file(_hwnd, SHELLPATH_CONTROL_PANEL);
#endif
break;}
case IDC_SETTINGS_MENU:
CreateSubmenu(id, CSIDL_CONTROLS, ResString(IDS_SETTINGS_MENU));
break;
case IDC_PRINTERS: {
CloseStartMenu(id);
#ifndef ROSSHELL
#ifndef _NO_MDI
XMLPos explorer_options = g_Globals.get_cfg("general/explorer");
bool mdi = XMLBool(explorer_options, "mdi", true);
if (mdi)
MDIMainFrame::Create(TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}"), 0);
else
#endif
SDIMainFrame::Create(TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}"), 0);
#else
launch_file(_hwnd, SHELLPATH_PRINTERS);
#endif
break;}
#if 0 ///@todo use printer start menu folder per default and allow opening "printers" cabinet window using the context menu
case IDC_PRINTERS_MENU:
CreateSubmenu(id, CSIDL_PRINTERS, CSIDL_PRINTHOOD, ResString(IDS_PRINTERS));
/* StartMenuFolders new_folders;
try {
new_folders.push_back(ShellPath(TEXT("::{20D04FE0-3AEA-1069-A2D8-08002B30309D}\\::{21EC2020-3AEA-1069-A2DD-08002B30309D}\\::{2227A280-3AEA-1069-A2DE-08002B30309D}")));
} catch(COMException&) {
}
CreateSubmenu(id, new_folders, ResString(IDS_PRINTERS));*/
break;
#endif
case IDC_ADMIN:
#ifndef ROSSHELL
CreateSubmenu(id, CSIDL_COMMON_ADMINTOOLS, CSIDL_ADMINTOOLS, ResString(IDS_ADMIN));
//CloseStartMenu(id);
//MainFrame::Create(SpecialFolderPath(CSIDL_COMMON_ADMINTOOLS, _hwnd), OWM_PIDL);
#else
launch_file(_hwnd, SpecialFolderFSPath(CSIDL_COMMON_ADMINTOOLS, _hwnd));
#endif
break;
case IDC_CONNECTIONS:{
CloseStartMenu(id);
#ifndef ROSSHELL
#ifndef _NO_MDI
XMLPos explorer_options = g_Globals.get_cfg("general/explorer");
bool mdi = XMLBool(explorer_options, "mdi", true);
if (mdi)
MDIMainFrame::Create(SHELLPATH_NET_CONNECTIONS, 0);
else
#endif
SDIMainFrame::Create(SHELLPATH_NET_CONNECTIONS, 0);
#else
launch_file(_hwnd, SHELLPATH_NET_CONNECTIONS);
#endif
break;}
// browse menu
case IDC_NETWORK:
#ifdef __REACTOS__ ///@todo to be removed when network browsing will be implemented in shell namespace
MessageBox(0, TEXT("not yet implemented"), ResString(IDS_TITLE), MB_OK);
#else
CreateSubmenu(id, CSIDL_NETWORK, ResString(IDS_NETWORK));
#endif
break;
case IDC_DRIVES:
///@todo exclude removable drives
CreateSubmenu(id, CSIDL_DRIVES, ResString(IDS_DRIVES));
break;
// search menu
case IDC_SEARCH_PROGRAM:
CloseStartMenu(id);
Dialog::DoModal(IDD_SEARCH_PROGRAM, WINDOW_CREATOR(FindProgramDlg));
break;
case IDC_SEARCH_FILES:
CloseStartMenu(id);
ShowSearchDialog();
break;
case IDC_SEARCH_COMPUTER:
CloseStartMenu(id);
ShowSearchComputer();
break;
default:
return super::Command(id, code);
}
return 0;
}
void StartMenuHandler::ShowSearchDialog()
{
#ifndef __REACTOS__ ///@todo to be removed when SHFindFiles() will be implemented in shell32.dll
static DynamicFct<SHFINDFILES> SHFindFiles(TEXT("SHELL32"), 90);
if (SHFindFiles)
(*SHFindFiles)(NULL, NULL);
else
#endif
MessageBox(0, TEXT("SHFindFiles() not yet implemented in SHELL32"), ResString(IDS_TITLE), MB_OK);
}
void StartMenuHandler::ShowSearchComputer()
{
#ifndef __REACTOS__ ///@todo to be removed when SHFindComputer() will be implemented in shell32.dll
static DynamicFct<SHFINDCOMPUTER> SHFindComputer(TEXT("SHELL32"), 91);
if (SHFindComputer)
(*SHFindComputer)(NULL, NULL);
else
#endif
MessageBox(0, TEXT("SHFindComputer() not yet implemented in SHELL32"), ResString(IDS_TITLE), MB_OK);
}
struct RunDialogThread : public Thread
{
int Run();
};
int RunDialogThread::Run()
{
static DynamicFct<RUNFILEDLG> RunFileDlg(TEXT("SHELL32"), 61);
// RunFileDlg needs owner window to properly position dialog
// that window will be disabled so we can't use DesktopBar
RECT rect = {0};
#ifndef TASKBAR_AT_TOP
rect.top = GetSystemMetrics(SM_CYSCREEN) - DESKTOPBARBAR_HEIGHT;
#endif
rect.right = GetSystemMetrics(SM_CXSCREEN);
rect.bottom = rect.top + DESKTOPBARBAR_HEIGHT;
Static dlgOwner(0, 0, rect.left, rect.top, rect.right-rect.left, rect.bottom-rect.top, 0, 0);
// Show "Run..." dialog
if (RunFileDlg) {
(*RunFileDlg)(dlgOwner, 0, NULL, NULL, NULL, RFF_CALCDIRECTORY);
}
DestroyWindow(dlgOwner);
return 0;
}
void StartMenuHandler::ShowLaunchDialog(HWND hwndOwner)
{
RunDialogThread * rdt = new RunDialogThread();
rdt->Start();
}
void StartMenuHandler::ShowLogoffDialog(HWND hwndOwner)
{
static DynamicFct<LOGOFFWINDOWSDIALOG> LogoffWindowsDialog(TEXT("SHELL32"), 54);
// static DynamicFct<RESTARTWINDOWSDLG> RestartDialog(TEXT("SHELL32"), 59);
if (LogoffWindowsDialog)
(*LogoffWindowsDialog)(0);
/* The RestartDialog function prompts about some system setting change. This is not what we want to display here.
else if (RestartDialog)
return (*RestartDialog)(hwndOwner, (LPWSTR)L"You selected <Log Off>.\n\n", EWX_LOGOFF) == 1; ///@todo ANSI string conversion if needed
*/
else
MessageBox(hwndOwner, TEXT("LogoffWindowsDialog() not yet implemented in SHELL32"), ResString(IDS_TITLE), MB_OK);
}
void ShowExitWindowsDialog(HWND hwndOwner)
{
static DynamicFct<EXITWINDOWSDLG> ExitWindowsDialog(TEXT("SHELL32"), 60);
if (ExitWindowsDialog)
(*ExitWindowsDialog)(hwndOwner);
else
MessageBox(hwndOwner, TEXT("ExitWindowsDialog() not yet implemented in SHELL32"), ResString(IDS_TITLE), MB_OK);
}
void StartMenuHandler::ShowRestartDialog(HWND hwndOwner, UINT flags)
{
static DynamicFct<RESTARTWINDOWSDLG> RestartDlg(TEXT("SHELL32"), 59);
if (RestartDlg)
(*RestartDlg)(hwndOwner, (LPWSTR)L"You selected restart.\n\n", flags);
else
MessageBox(hwndOwner, TEXT("RestartDlg() not yet implemented in SHELL32"), ResString(IDS_TITLE), MB_OK);
}
void SettingsMenu::AddEntries()
{
super::AddEntries();
#if defined(ROSSHELL) || defined(__REACTOS__) // __REACTOS__ to be removed when printers will be implemented
//TODO AddButton(ResString(IDS_PRINTERS), ICID_PRINTER, false, IDC_PRINTERS_MENU);
#else
//TODO AddButton(ResString(IDS_PRINTERS), ICID_PRINTER, true, IDC_PRINTERS_MENU);
#endif
AddButton(ResString(IDS_CONNECTIONS), ICID_NETCONNS, false, IDC_CONNECTIONS);
AddButton(ResString(IDS_ADMIN), ICID_ADMIN, true, IDC_ADMIN);
if (!g_Globals._SHRestricted || !SHRestricted(REST_NOCONTROLPANEL))
AddButton(ResString(IDS_SETTINGS_MENU), ICID_CONFIG, true, IDC_SETTINGS_MENU);
AddButton(ResString(IDS_DESKTOPBAR_SETTINGS), ICID_DESKSETTING, false, ID_DESKTOPBAR_SETTINGS);
AddButton(ResString(IDS_PRINTERS), ICID_PRINTER, false, IDC_PRINTERS);
if (!g_Globals._SHRestricted || !SHRestricted(REST_NOCONTROLPANEL))
AddButton(ResString(IDS_CONTROL_PANEL), ICID_CONTROLPAN, false, IDC_CONTROL_PANEL);
}
void BrowseMenu::AddEntries()
{
super::AddEntries();
if (!g_Globals._SHRestricted || !SHRestricted(REST_NONETHOOD)) // or REST_NOENTIRENETWORK ?
#if defined(ROSSHELL) || defined(__REACTOS__) // __REACTOS__ to be removed when printer/network will be implemented
AddButton(ResString(IDS_NETWORK), ICID_NETWORK, false, IDC_NETWORK);
#else
AddButton(ResString(IDS_NETWORK), ICID_NETWORK, true, IDC_NETWORK);
#endif
AddButton(ResString(IDS_DRIVES), ICID_FOLDER, true, IDC_DRIVES);
}
void SearchMenu::AddEntries()
{
super::AddEntries();
AddButton(ResString(IDS_SEARCH_FILES), ICID_SEARCH_DOC, false, IDC_SEARCH_FILES);
if (!g_Globals._SHRestricted || !SHRestricted(REST_HASFINDCOMPUTERS))
AddButton(ResString(IDS_SEARCH_COMPUTER),ICID_COMPUTER, false, IDC_SEARCH_COMPUTER);
AddButton(ResString(IDS_SEARCH_PRG), ICID_APPS, false, IDC_SEARCH_PROGRAM);
}
void RecentStartMenu::AddEntries()
{
for(StartMenuShellDirs::iterator it=_dirs.begin(); it!=_dirs.end(); ++it) {
StartMenuDirectory& smd = *it;
ShellDirectory& dir = smd._dir;
if (!dir._scanned) {
WaitCursor wait;
#ifdef _LAZY_ICONEXTRACT
dir.smart_scan(SORT_NAME, SCAN_DONT_EXTRACT_ICONS);
#else
dir.smart_scan(SORT_NAME);
#endif
}
dir.sort_directory(SORT_DATE);
AddShellEntries(dir, RECENT_DOCS_COUNT, smd._ignore); ///@todo read max. count of entries from registry
}
}
#ifndef _SHELL32_FAVORITES
void FavoritesMenu::AddEntries()
{
super::AddEntries();
String lwr_filter = _create_info._filter;
lwr_filter.toLower();
for(BookmarkList::iterator it=_bookmarks.begin(); it!=_bookmarks.end(); ++it) {
BookmarkNode& node = *it;
int id = ++_next_id;
_entries[id] = node;
if (node._type == BookmarkNode::BMNT_FOLDER) {
BookmarkFolder& folder = *node._pfolder;
AddButton(folder._name, ICID_FOLDER, true, id);
} else if (node._type == BookmarkNode::BMNT_BOOKMARK) {
Bookmark& bookmark = *node._pbookmark;
ICON_ID icon = ICID_NONE;
if (!bookmark._icon_path.empty())
icon = g_Globals._icon_cache.extract(bookmark._icon_path, bookmark._icon_idx);
// filter non-directory entries
if (!lwr_filter.empty()) {
String lwr_name = bookmark._name;
String lwr_desc = bookmark._description;
String lwr_url = bookmark._url;
lwr_name.toLower();
lwr_desc.toLower();
lwr_url.toLower();
if (!_tcsstr(lwr_name,lwr_filter) && !_tcsstr(lwr_desc,lwr_filter) && !_tcsstr(lwr_url,lwr_filter))
continue;
}
AddButton(bookmark._name, icon!=ICID_NONE?icon:ICID_BOOKMARK, false, id);
}
}
}
int FavoritesMenu::Command(int id, int code)
{
BookmarkMap::iterator found = _entries.find(id);
if (found != _entries.end()) {
BookmarkNode& node = found->second;
if (node._type == BookmarkNode::BMNT_FOLDER) {
BookmarkFolder& folder = *node._pfolder;
if (CloseOtherSubmenus(id))
CreateSubmenu(id, folder._name, STARTMENU_CREATOR(FavoritesMenu), &static_cast<BookmarkList&>(folder._bookmarks));
} else if (node._type == BookmarkNode::BMNT_BOOKMARK) {
Bookmark& bookmark = *node._pbookmark;
String url = bookmark._url;
HWND hparent = GetParent(_hwnd);
CloseStartMenu(id);
launch_file(hparent, url, SW_SHOWNORMAL);
}
return 0;
}
return super::Command(id, code);
}
#endif