nana/source/gui/notifier.cpp

369 lines
8.6 KiB
C++

/*
* Implementation of Notifier
* Nana C++ Library(http://www.nanapro.org)
* Copyright(C) 2003-2015 Jinhao(cnjinhao@hotmail.com)
*
* Distributed under the Boost Software License, Version 1.0.
* (See accompanying file LICENSE_1_0.txt or copy at
* http://www.boost.org/LICENSE_1_0.txt)
*
* @file: nana/gui/notifier.cpp
* @contributors:
* Jan
* Benjamin Navarro(pr#81)
*/
#include <nana/deploy.hpp>
#include <nana/gui/programming_interface.hpp>
#include <nana/gui/notifier.hpp>
#include <nana/gui/timer.hpp>
#include <unordered_map>
#include <unordered_set>
#if defined(NANA_MINGW) && defined(STD_THREAD_NOT_SUPPORTED)
#include <nana/std_mutex.hpp>
#else
#include <mutex>
#endif
#include <nana/detail/platform_spec_selector.hpp>
#if defined(NANA_LINUX) || defined(NANA_MACOS)
#include <nana/system/platform.hpp>
#include <iostream>
#endif
namespace nana
{
typedef std::lock_guard<std::recursive_mutex> lock_guard;
struct notifier::implement
{
nana::timer ani_timer;
native_window_type native_handle;
window handle;
event_handle evt_destroy;
unsigned short id;
detail::notifier_events events;
bool icon_added = false;
std::size_t play_index;
#if defined(NANA_WINDOWS)
HICON icon_handle = nullptr;
std::vector<HICON> icons;
void set_icon(HICON icon)
{
if (icon_handle)
{
NOTIFYICONDATA icon_data;
memset(&icon_data, 0, sizeof icon_data);
icon_data.cbSize = sizeof icon_data;
icon_data.hWnd = reinterpret_cast<HWND>(native_handle);
icon_data.uID = id;
icon_data.uFlags = NIF_MESSAGE | NIF_ICON;
icon_data.uCallbackMessage = nana::detail::messages::tray;
icon_data.hIcon = icon;
::Shell_NotifyIcon(icon_added ? NIM_MODIFY : NIM_ADD, &icon_data);
icon_added = true;
}
}
#endif
};
arg_notifier::operator nana::arg_mouse() const
{
arg_mouse arg;
arg.evt_code = evt_code;
arg.ctrl = false;
arg.shift = false;
arg.left_button = left_button;
arg.mid_button = mid_button;
arg.right_button = right_button;
arg.pos = pos;
arg.window_handle = (notifier_ptr ? notifier_ptr->handle() : nullptr);
return arg;
}
class notifications
{
struct notifier_data
{
::nana::notifier * notifier_ptr;
detail::notifier_events* evt_ptr;
};
struct window_notifiers
{
std::unordered_map<unsigned short, notifier_data> idtable;
};
public:
static notifications& instance()
{
static notifications obj;
return obj;
}
unsigned short register_wd(::nana::notifier* ntf_ptr, native_window_type native_handle, detail::notifier_events* evt_ptr)
{
lock_guard lock(mutex_);
auto i = wd_table_.find(native_handle);
if (i == wd_table_.end())
{
auto & data = wd_table_[native_handle].idtable[0];
data.notifier_ptr = ntf_ptr;
data.evt_ptr = evt_ptr;
return 0;
}
auto & idtable = i->second.idtable;
auto id = static_cast<unsigned short>(idtable.size());
const auto orig_id = id;
while (idtable.count(id) != 0)
{
if (orig_id == ++id)
throw std::runtime_error("Full");
}
auto & data = idtable[id];
data.notifier_ptr = ntf_ptr;
data.evt_ptr = evt_ptr;
return id;
}
void cancel(native_window_type wd, unsigned short id)
{
lock_guard lock(mutex_);
auto i_wd = wd_table_.find(wd);
if (i_wd == wd_table_.end())
return;
auto & idtable = i_wd->second.idtable;
auto i_id = idtable.find(id);
if (i_id == idtable.end())
return;
idtable.erase(i_id);
if (idtable.empty())
wd_table_.erase(i_wd);
}
void emit(native_window_type wd, unsigned short id, const arg_notifier& arg_basic)
{
lock_guard lock(mutex_);
auto i_wd = wd_table_.find(wd);
if (i_wd == wd_table_.end())
return;
auto & idtable = i_wd->second.idtable;
auto i_id = idtable.find(id);
if (i_id == idtable.end())
return;
auto arg = arg_basic;
arg.notifier_ptr = i_id->second.notifier_ptr;
auto evt_ptr = i_id->second.evt_ptr;
switch (arg.evt_code)
{
case event_code::mouse_down:
evt_ptr->mouse_down.emit(arg);
break;
case event_code::mouse_up:
evt_ptr->mouse_up.emit(arg);
break;
case event_code::mouse_leave:
evt_ptr->mouse_leave.emit(arg);
break;
case event_code::mouse_move:
evt_ptr->mouse_move.emit(arg);
break;
case event_code::dbl_click:
evt_ptr->dbl_click.emit(arg);
break;
default:
break;
}
}
private:
mutable std::recursive_mutex mutex_;
std::unordered_map<native_window_type, window_notifiers> wd_table_;
};
#if defined(NANA_WINDOWS)
void notifications_window_proc(HWND wd, WPARAM wparam, LPARAM lparam)
{
arg_notifier arg;
switch (lparam)
{
case WM_LBUTTONDBLCLK:
case WM_MBUTTONDBLCLK:
case WM_RBUTTONDBLCLK:
arg.evt_code = event_code::dbl_click;
arg.left_button = (lparam == WM_LBUTTONDBLCLK);
arg.mid_button = (lparam == WM_MBUTTONDBLCLK);
arg.right_button = (lparam == WM_RBUTTONDBLCLK);
break;
case WM_LBUTTONDOWN:
case WM_MBUTTONDOWN:
case WM_RBUTTONDOWN:
arg.evt_code = event_code::mouse_down;
arg.left_button = (lparam == WM_LBUTTONDOWN);
arg.mid_button = (lparam == WM_MBUTTONDOWN);
arg.right_button = (lparam == WM_RBUTTONDOWN);
break;
case WM_LBUTTONUP:
case WM_MBUTTONUP:
case WM_RBUTTONUP:
arg.evt_code = event_code::mouse_up;
arg.left_button = (lparam == WM_LBUTTONUP);
arg.mid_button = (lparam == WM_MBUTTONUP);
arg.right_button = (lparam == WM_RBUTTONUP);
break;
case WM_MOUSEMOVE:
case WM_MOUSELEAVE:
arg.evt_code = (WM_MOUSEMOVE == lparam ? event_code::mouse_move : event_code::mouse_leave);
arg.left_button = false;
arg.mid_button = false;
arg.right_button = false;
break;
default:
return;
}
::POINT pos;
::GetCursorPos(&pos);
arg.pos.x = pos.x;
arg.pos.y = pos.y;
notifications::instance().emit(reinterpret_cast<native_window_type>(wd), static_cast<unsigned short>(wparam), arg);
}
#endif
//class notifier
notifier::notifier(window wd)
: impl_(new implement)
{
impl_->handle = wd;
impl_->native_handle = API::root(wd);
if (!impl_->native_handle)
throw std::invalid_argument("Invalid window handle");
impl_->ani_timer.elapse([this]
{
#if defined(NANA_WINDOWS)
if (impl_->play_index >= impl_->icons.size())
impl_->play_index = 0;
auto ico = impl_->icons[impl_->play_index++];
impl_->set_icon(ico);
#endif
});
impl_->evt_destroy = API::events(wd).destroy([this]
{
close();
});
impl_->id = notifications::instance().register_wd(this, impl_->native_handle, &impl_->events);
}
notifier::~notifier()
{
close();
delete impl_;
}
void notifier::close()
{
if (nullptr == impl_->native_handle)
return;
#if defined(NANA_WINDOWS)
NOTIFYICONDATA icon_data;
memset(&icon_data, 0, sizeof icon_data);
icon_data.cbSize = sizeof icon_data;
icon_data.hWnd = reinterpret_cast<HWND>(impl_->native_handle);
icon_data.uID = impl_->id;
::Shell_NotifyIcon(NIM_DELETE, &icon_data);
if (impl_->icon_handle)
::DestroyIcon(impl_->icon_handle);
for (auto handle : impl_->icons)
::DestroyIcon(handle);
#endif
API::umake_event(impl_->evt_destroy);
notifications::instance().cancel(impl_->native_handle, impl_->id);
impl_->native_handle = nullptr;
}
void notifier::text(const nana::string& str)
{
#if defined(NANA_WINDOWS)
NOTIFYICONDATA icon_data;
memset(&icon_data, 0, sizeof icon_data);
icon_data.cbSize = sizeof icon_data;
icon_data.hWnd = reinterpret_cast<HWND>(impl_->native_handle);
icon_data.uID = impl_->id;
icon_data.uFlags = NIF_MESSAGE | NIF_TIP;
icon_data.uCallbackMessage = nana::detail::messages::tray;
strcpy(icon_data.szTip, str.data());
::Shell_NotifyIcon(impl_->icon_added ? NIM_MODIFY : NIM_ADD, &icon_data);
impl_->icon_added = true;
#endif
}
void notifier::icon(const nana::string& icon_file)
{
#if defined(NANA_WINDOWS)
auto pre_icon = impl_->icon_handle;
auto ico = (HICON)::LoadImage(0, icon_file.data(), IMAGE_ICON, 0, 0, LR_LOADFROMFILE);
if (ico)
{
impl_->icon_handle = ico;
impl_->ani_timer.stop();
impl_->play_index = 0;
impl_->set_icon(impl_->icon_handle);
::DestroyIcon(pre_icon);
}
#endif
}
void notifier::insert_icon(const nana::string& icon_file)
{
#if defined(NANA_WINDOWS)
auto icon = (HICON)::LoadImage(0, icon_file.data(), IMAGE_ICON, 0, 0, LR_LOADFROMFILE);
impl_->icons.push_back(icon);
#endif
}
void notifier::period(unsigned ms)
{
#if defined(NANA_WINDOWS)
if (ms && impl_->icons.size())
{
ms /= static_cast<unsigned>(impl_->icons.size());
impl_->ani_timer.interval(ms < 16 ? 16 : ms);
impl_->ani_timer.start();
}
else
impl_->ani_timer.stop();
#endif
}
detail::notifier_events& notifier::events()
{
return impl_->events;
}
window notifier::handle() const
{
return impl_->handle;
}
//end notifier
}