diff options
Diffstat (limited to 'ui/tray_icon.cpp')
| -rw-r--r-- | ui/tray_icon.cpp | 213 |
1 files changed, 213 insertions, 0 deletions
diff --git a/ui/tray_icon.cpp b/ui/tray_icon.cpp new file mode 100644 index 00000000..464bf7ef --- /dev/null +++ b/ui/tray_icon.cpp @@ -0,0 +1,213 @@ +// ************************************************************************** +// * This file is part of the FreeFileSync project. It is distributed under * +// * GNU General Public License: http://www.gnu.org/licenses/gpl.html * +// * Copyright (C) 2008-2010 ZenJu (zhnmju123 AT gmx.de) * +// ************************************************************************** +// +#include "tray_icon.h" +#include "../library/resources.h" +#include "small_dlgs.h" +#include <wx/taskbar.h> +#include <cmath> +#include <wx/image.h> +#include <wx/menu.h> +#include <wx/icon.h> //req. by Linux + +namespace +{ +inline +int roundNum(double d) //little rounding function +{ + return static_cast<int>(d < 0 ? d - .5 : d + .5); +} + + +wxIcon generateIcon(size_t percent) //generate icon with progress indicator +{ + percent = std::min(percent, static_cast<size_t>(100u)); //handle invalid input + +#ifdef FFS_WIN + static const wxBitmap trayIcon = GlobalResources::getInstance().getImageByName(wxT("FFS_tray_win.png")); +#elif defined FFS_LINUX + static const wxBitmap trayIcon = GlobalResources::getInstance().getImageByName(wxT("FFS_tray_linux.png")); +#endif + + const int indicatorHeight = roundNum((trayIcon.GetHeight() * percent) / 100.0); + + //minor optimization + static std::pair<int, wxIcon> buffer = std::make_pair(-1, wxNullIcon); + if (buffer.first == indicatorHeight) + return buffer.second; + + if ( trayIcon.GetWidth() > 0 && + trayIcon.GetHeight() > 0) + { + static const int indicatorWidth = trayIcon.GetWidth() * .25; + const int indicatorXBegin = ceil((trayIcon.GetWidth() - indicatorWidth) / 2.0); + const int indicatorYBegin = trayIcon.GetHeight() - indicatorHeight; + + wxImage genImage(trayIcon.ConvertToImage()); + + //draw progress indicator: do NOT use wxDC::DrawRectangle! Doesn't respect alpha in Windows, but does in Linux! + //We need a simple, working solution: + unsigned char* const data = genImage.GetData(); + for (int row = indicatorYBegin; row < genImage.GetHeight(); ++row) + { + for (int col = indicatorXBegin; col < indicatorXBegin + indicatorWidth; ++col) + { + unsigned char* const pixelBegin = data + (row * genImage.GetWidth() + col) * 3; + pixelBegin[0] = 255; //red + pixelBegin[1] = 255; //green + pixelBegin[2] = 0; //blue + } + } + + if (genImage.HasAlpha()) + { + unsigned char* const alpha = genImage.GetAlpha(); + //make progress indicator fully opaque: + for (int row = indicatorYBegin; row < genImage.GetHeight(); ++row) + ::memset(alpha + row * genImage.GetWidth() + indicatorXBegin, wxIMAGE_ALPHA_OPAQUE, indicatorWidth); + } + + wxIcon genIcon; + genIcon.CopyFromBitmap(wxBitmap(genImage)); + + //fill buffer + buffer.first = indicatorHeight; + buffer.second = genIcon; + + return genIcon; + } + + //fallback + wxIcon defaultIcon; + defaultIcon.CopyFromBitmap(trayIcon); + + //fill buffer + buffer.first = indicatorHeight; + buffer.second = defaultIcon; + + return defaultIcon; +} +} + + +//------------------------------------------------------------------------------------------------ +enum Selection +{ + CONTEXT_RESTORE, + CONTEXT_ABOUT +}; + + +class MinimizeToTray::TaskBarImpl : public wxTaskBarIcon +{ +public: + TaskBarImpl(MinimizeToTray* parent) : parent_(parent) {} + + void parentHasDied() + { + parent_ = NULL; + } +private: + virtual wxMenu* CreatePopupMenu() + { + if (!parent_) + return NULL; + + wxMenu* contextMenu = new wxMenu; + contextMenu->Append(CONTEXT_ABOUT, _("&About...")); + contextMenu->AppendSeparator(); + contextMenu->Append(CONTEXT_RESTORE, _("&Restore")); + //event handling + contextMenu->Connect(wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(MinimizeToTray::OnContextMenuSelection), NULL, parent_); + + return contextMenu; //ownership transferred to caller + } + + MinimizeToTray* parent_; +}; + + +MinimizeToTray::MinimizeToTray(wxTopLevelWindow* callerWnd, wxWindow* secondWnd) : + callerWnd_(callerWnd), + secondWnd_(secondWnd), + trayIcon(new TaskBarImpl(this)) +{ + trayIcon->SetIcon(generateIcon(0), wxT("FreeFileSync")); + trayIcon->Connect(wxEVT_TASKBAR_LEFT_DCLICK, wxCommandEventHandler(MinimizeToTray::OnDoubleClick), NULL, this); //register double-click + + if (callerWnd_) + callerWnd_->Hide(); + if (secondWnd_) + secondWnd_->Hide(); +} + + +MinimizeToTray::~MinimizeToTray() +{ + resumeFromTray(); +} + + +void MinimizeToTray::resumeFromTray() //remove trayIcon and restore windows: MinimizeToTray is now a zombie object... +{ + if (trayIcon) + { + if (secondWnd_) + secondWnd_->Show(); + + if (callerWnd_) //usecase: avoid dialog flashing in batch silent mode + { + callerWnd_->Iconize(false); + callerWnd_->Show(); + callerWnd_->Raise(); + callerWnd_->SetFocus(); + } + trayIcon->RemoveIcon(); //hide icon until final deletion takes place + trayIcon->Disconnect(wxEVT_TASKBAR_LEFT_DCLICK, wxCommandEventHandler(MinimizeToTray::OnDoubleClick), NULL, this); + trayIcon->parentHasDied(); //TaskBarImpl (potentially) has longer lifetime than MinimizeToTray: avoid callback! + + //use wxWidgets delayed destruction: delete during next idle loop iteration (handle late window messages, e.g. when double-clicking) + if (!wxPendingDelete.Member(trayIcon)) + wxPendingDelete.Append(trayIcon); + + trayIcon = NULL; //avoid reentrance + } +} + + +void MinimizeToTray::setToolTip(const wxString& toolTipText, size_t percent) +{ + if (trayIcon) + trayIcon->SetIcon(generateIcon(percent), toolTipText); +} + + +void MinimizeToTray::keepHidden() +{ + callerWnd_ = NULL; + secondWnd_ = NULL; +} + + +void MinimizeToTray::OnContextMenuSelection(wxCommandEvent& event) +{ + const Selection eventId = static_cast<Selection>(event.GetId()); + switch (eventId) + { + case CONTEXT_ABOUT: + ffs3::showAboutDialog(); + break; + case CONTEXT_RESTORE: + resumeFromTray(); + } +} + + +void MinimizeToTray::OnDoubleClick(wxCommandEvent& event) +{ + resumeFromTray(); +} + |