/*************************************************************************** * Copyright (C) 2008-2012 by Andrzej Rybczak * * electricityispower@gmail.com * * * * This program is free software; you can redistribute it and/or modify * * it under the terms of the GNU General Public License as published by * * the Free Software Foundation; either version 2 of the License, or * * (at your option) any later version. * * * * This program 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 General Public License for more details. * * * * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the * * Free Software Foundation, Inc., * * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA. * ***************************************************************************/ #include #include #ifdef WIN32 # include #else # include # include #endif #include "error.h" #include "utility/string.h" #include "utility/wide_string.h" #include "window.h" namespace NC {// void initScreen(GNUC_UNUSED const char *window_title, bool enable_colors) { const int ColorsTable[] = { COLOR_BLACK, COLOR_RED, COLOR_GREEN, COLOR_YELLOW, COLOR_BLUE, COLOR_MAGENTA, COLOR_CYAN, COLOR_WHITE }; # ifdef XCURSES Xinitscr(1, const_cast(&window_title)); # else initscr(); # endif // XCURSES if (has_colors() && enable_colors) { start_color(); use_default_colors(); int num = 1; # ifdef USE_PDCURSES int i = 0; # else int i = -1; # endif // USE_PDCURSES for (; i < 8; ++i) for (int j = 0; j < 8; ++j) init_pair(num++, ColorsTable[j], i < 0 ? i : ColorsTable[i]); } noecho(); cbreak(); curs_set(0); } void destroyScreen() { curs_set(1); endwin(); } Window::Window(size_t startx, size_t starty, size_t width, size_t height, const std::string &title, Color color, Border border) : m_window(0), m_border_window(0), m_start_x(startx), m_start_y(starty), m_width(width), m_height(height), m_window_timeout(-1), m_color(color), m_bg_color(clDefault), m_base_color(color), m_base_bg_color(clDefault), m_border(border), m_get_string_helper(0), m_title(title), m_history(0), m_bold_counter(0), m_underline_counter(0), m_reverse_counter(0), m_alt_charset_counter(0) { if (m_start_x > size_t(COLS) || m_start_y > size_t(LINES) || m_width+m_start_x > size_t(COLS) || m_height+m_start_y > size_t(LINES)) FatalError("Constructed window is bigger than terminal size"); if (m_border != brNone) { m_border_window = newpad(m_height, m_width); wattron(m_border_window, COLOR_PAIR(m_border)); box(m_border_window, 0, 0); m_start_x++; m_start_y++; m_width -= 2; m_height -= 2; } if (!m_title.empty()) { m_start_y += 2; m_height -= 2; } m_window = newpad(m_height, m_width); setColor(m_color); keypad(m_window, 1); } Window::Window(const Window &rhs) : m_window(dupwin(rhs.m_window)) , m_border_window(dupwin(rhs.m_border_window)) , m_start_x(rhs.m_start_x) , m_start_y(rhs.m_start_y) , m_width(rhs.m_width) , m_height(rhs.m_height) , m_window_timeout(rhs.m_window_timeout) , m_color(rhs.m_color) , m_bg_color(rhs.m_bg_color) , m_base_color(rhs.m_base_color) , m_base_bg_color(rhs.m_base_bg_color) , m_border(rhs.m_border) , m_get_string_helper(rhs.m_get_string_helper) , m_title(rhs.m_title) , m_color_stack(rhs.m_color_stack) , m_input_queue(rhs.m_input_queue) , m_fds(rhs.m_fds) , m_history(rhs.m_history ? new std::list(*rhs.m_history) : 0) , m_bold_counter(rhs.m_bold_counter) , m_underline_counter(rhs.m_underline_counter) , m_reverse_counter(rhs.m_reverse_counter) , m_alt_charset_counter(rhs.m_alt_charset_counter) { } Window::Window(Window &&rhs) : m_window(rhs.m_window) , m_border_window(rhs.m_border_window) , m_start_x(rhs.m_start_x) , m_start_y(rhs.m_start_y) , m_width(rhs.m_width) , m_height(rhs.m_height) , m_window_timeout(rhs.m_window_timeout) , m_color(rhs.m_color) , m_bg_color(rhs.m_bg_color) , m_base_color(rhs.m_base_color) , m_base_bg_color(rhs.m_base_bg_color) , m_border(rhs.m_border) , m_get_string_helper(rhs.m_get_string_helper) , m_title(std::move(rhs.m_title)) , m_color_stack(std::move(rhs.m_color_stack)) , m_input_queue(std::move(rhs.m_input_queue)) , m_fds(std::move(rhs.m_fds)) , m_history(rhs.m_history) , m_bold_counter(rhs.m_bold_counter) , m_underline_counter(rhs.m_underline_counter) , m_reverse_counter(rhs.m_reverse_counter) , m_alt_charset_counter(rhs.m_alt_charset_counter) { rhs.m_window = 0; rhs.m_border_window = 0; rhs.m_history = 0; } Window &Window::operator=(Window rhs) { std::swap(m_window, rhs.m_window); std::swap(m_border_window, rhs.m_border_window); std::swap(m_start_x, rhs.m_start_x); std::swap(m_start_y, rhs.m_start_y); std::swap(m_width, rhs.m_width); std::swap(m_height, rhs.m_height); std::swap(m_window_timeout, rhs.m_window_timeout); std::swap(m_color, rhs.m_color); std::swap(m_bg_color, rhs.m_bg_color); std::swap(m_base_color, rhs.m_base_color); std::swap(m_base_bg_color, rhs.m_base_bg_color); std::swap(m_border, rhs.m_border); std::swap(m_get_string_helper, rhs.m_get_string_helper); std::swap(m_title, rhs.m_title); std::swap(m_color_stack, rhs.m_color_stack); std::swap(m_input_queue, rhs.m_input_queue); std::swap(m_fds, rhs.m_fds); std::swap(m_history, rhs.m_history); std::swap(m_bold_counter, rhs.m_bold_counter); std::swap(m_underline_counter, rhs.m_underline_counter); std::swap(m_reverse_counter, rhs.m_reverse_counter); std::swap(m_alt_charset_counter, rhs.m_alt_charset_counter); return *this; } Window::~Window() { delwin(m_window); delwin(m_border_window); delete m_history; } void Window::setColor(Color fg, Color bg) { if (fg == clDefault) fg = m_base_color; if (fg != clDefault) wattron(m_window, COLOR_PAIR(bg*8+fg)); else wattroff(m_window, COLOR_PAIR(m_color)); m_color = fg; m_bg_color = bg; } void Window::setBaseColor(Color fg, Color bg) { m_base_color = fg; m_base_bg_color = bg; } void Window::setBorder(Border border) { if (border == brNone && m_border != brNone) { delwin(m_border_window); m_start_x--; m_start_y--; m_height += 2; m_width += 2; recreate(m_width, m_height); } else if (border != brNone && m_border == brNone) { m_border_window = newpad(m_height, m_width); wattron(m_border_window, COLOR_PAIR(border)); box(m_border_window,0,0); m_start_x++; m_start_y++; m_height -= 2; m_width -= 2; recreate(m_width, m_height); } else { wattron(m_border_window,COLOR_PAIR(border)); box(m_border_window, 0, 0); } m_border = border; } void Window::setTitle(const std::string &new_title) { if (m_title == new_title) { return; } else if (!new_title.empty() && m_title.empty()) { m_start_y += 2; m_height -= 2; recreate(m_width, m_height); } else if (new_title.empty() && !m_title.empty()) { m_start_y -= 2; m_height += 2; recreate(m_width, m_height); } m_title = new_title; } void Window::createHistory() { if (!m_history) m_history = new std::list; } void Window::deleteHistory() { delete m_history; m_history = 0; } void Window::recreate(size_t width, size_t height) { delwin(m_window); m_window = newpad(height, width); setTimeout(m_window_timeout); setColor(m_color, m_bg_color); keypad(m_window, 1); } void Window::moveTo(size_t new_x, size_t new_y) { m_start_x = new_x; m_start_y = new_y; if (m_border != brNone) { m_start_x++; m_start_y++; } if (!m_title.empty()) m_start_y += 2; } void Window::adjustDimensions(size_t width, size_t height) { if (m_border != brNone) { delwin(m_border_window); m_border_window = newpad(height, width); wattron(m_border_window, COLOR_PAIR(m_border)); box(m_border_window, 0, 0); width -= 2; height -= 2; } if (!m_title.empty()) height -= 2; m_height = height; m_width = width; } void Window::resize(size_t new_width, size_t new_height) { adjustDimensions(new_width, new_height); recreate(m_width, m_height); } void Window::showBorder() const { if (m_border != brNone) { ::refresh(); prefresh(m_border_window, 0, 0, getStarty(), getStartX(), m_start_y+m_height, m_start_x+m_width); } if (!m_title.empty()) { if (m_border != brNone) attron(COLOR_PAIR(m_border)); else attron(COLOR_PAIR(m_base_color)); mvhline(m_start_y-1, m_start_x, 0, m_width); attron(A_BOLD); mvhline(m_start_y-2, m_start_x, 32, m_width); // clear title line mvaddstr(m_start_y-2, m_start_x, m_title.c_str()); attroff(COLOR_PAIR(m_border) | A_BOLD); } ::refresh(); } void Window::display() { showBorder(); refresh(); } void Window::refresh() { prefresh(m_window, 0, 0, m_start_y, m_start_x, m_start_y+m_height-1, m_start_x+m_width-1); } void Window::clear() { werase(m_window); } void Window::bold(bool bold_state) const { (bold_state ? wattron : wattroff)(m_window, A_BOLD); } void Window::underline(bool underline_state) const { (underline_state ? wattron : wattroff)(m_window, A_UNDERLINE); } void Window::reverse(bool reverse_state) const { (reverse_state ? wattron : wattroff)(m_window, A_REVERSE); } void Window::altCharset(bool altcharset_state) const { (altcharset_state ? wattron : wattroff)(m_window, A_ALTCHARSET); } void Window::setTimeout(int timeout) { m_window_timeout = timeout; wtimeout(m_window, timeout); } void Window::addFDCallback(int fd, void (*callback)()) { m_fds.push_back(std::make_pair(fd, callback)); } void Window::clearFDCallbacksList() { m_fds.clear(); } bool Window::FDCallbacksListEmpty() const { return m_fds.empty(); } int Window::readKey() { int result; // if there are characters in input queue, get them and // return immediately. if (!m_input_queue.empty()) { result = m_input_queue.front(); m_input_queue.pop(); return result; } // in pdcurses polling stdin doesn't work, so we can't poll // both stdin and other file descriptors in one select. the // workaround is to set the timeout of select to 0, poll // other file descriptors and then wait for stdin input with // the given timeout. unfortunately, this results in delays // since ncmpcpp doesn't see that data arrived while waiting // for input from stdin, but it seems there is no better option. fd_set fdset; FD_ZERO(&fdset); # if !defined(USE_PDCURSES) FD_SET(STDIN_FILENO, &fdset); timeval timeout = { m_window_timeout/1000, (m_window_timeout%1000)*1000 }; # else timeval timeout = { 0, 0 }; # endif int fd_max = STDIN_FILENO; for (FDCallbacks::const_iterator it = m_fds.begin(); it != m_fds.end(); ++it) { if (it->first > fd_max) fd_max = it->first; FD_SET(it->first, &fdset); } if (select(fd_max+1, &fdset, 0, 0, m_window_timeout < 0 ? 0 : &timeout) > 0) { # if !defined(USE_PDCURSES) result = FD_ISSET(STDIN_FILENO, &fdset) ? wgetch(m_window) : ERR; # endif // !USE_PDCURSES for (FDCallbacks::const_iterator it = m_fds.begin(); it != m_fds.end(); ++it) if (FD_ISSET(it->first, &fdset)) it->second(); } # if !defined(USE_PDCURSES) else result = ERR; # else result = wgetch(m_window); # endif return result; } void Window::pushChar(int ch) { m_input_queue.push(ch); } std::string Window::getString(const std::string &base, size_t length_, size_t width, bool encrypted) { int input; size_t beginning, maxbeginning, minx, x, real_x, y, maxx, real_maxx; getyx(m_window, y, x); minx = real_maxx = maxx = real_x = x; width--; if (width == size_t(-1)) width = m_width-x-1; curs_set(1); std::wstring wbase = ToWString(base); std::wstring *tmp = &wbase; std::list::iterator history_it = m_history->end(); std::string tmp_in; wchar_t wc_in; bool gotoend = 1; bool block_scrolling = 0; // disable scrolling if wide chars are used for (std::wstring::const_iterator it = tmp->begin(); it != tmp->end(); ++it) if (wcwidth(*it) > 1) block_scrolling = 1; beginning = -1; do { if (tmp->empty()) block_scrolling = 0; maxbeginning = block_scrolling ? 0 : (tmp->length() < width ? 0 : tmp->length()-width); maxx = minx + (wideLength(*tmp) < width ? wideLength(*tmp) : width); real_maxx = minx + (tmp->length() < width ? tmp->length() : width); if (beginning > maxbeginning) beginning = maxbeginning; if (gotoend) { size_t real_real_maxx = minx; size_t biggest_x = minx+width; if (block_scrolling && maxx >= biggest_x) { size_t i = 0; for (std::wstring::const_iterator it = tmp->begin(); i < width; ++it, ++real_real_maxx) i += wcwidth(*it); } else real_real_maxx = real_maxx; real_x = real_real_maxx; x = block_scrolling ? (maxx > biggest_x ? biggest_x : maxx) : maxx; beginning = maxbeginning; gotoend = 0; } mvwhline(m_window, y, minx, ' ', width+1); if (!encrypted) mvwprintw(m_window, y, minx, "%ls", tmp->substr(beginning, width+1).c_str()); else mvwhline(m_window, y, minx, '*', maxx-minx); if (m_get_string_helper) { if (!m_get_string_helper(*tmp)) break; } wmove(m_window, y, x); prefresh(m_window, 0, 0, m_start_y, m_start_x, m_start_y+m_height-1, m_start_x+m_width-1); input = readKey(); switch (input) { case ERR: case KEY_MOUSE: break; case KEY_UP: if (m_history && !encrypted && history_it != m_history->begin()) { while (--history_it != m_history->begin()) if (!history_it->empty()) break; tmp = &*history_it; gotoend = 1; } break; case KEY_DOWN: if (m_history && !encrypted && history_it != m_history->end()) { while (++history_it != m_history->end()) if (!history_it->empty()) break; tmp = &(history_it == m_history->end() ? wbase : *history_it); gotoend = 1; } break; case KEY_RIGHT: { if (x < maxx) { real_x++; x += std::max(wcwidth((*tmp)[beginning+real_x-minx-1]), 1); } else if (beginning < maxbeginning) beginning++; break; } case KEY_CTRL_H: case KEY_BACKSPACE: case KEY_BACKSPACE_2: { if (x <= minx && !beginning) break; } case KEY_LEFT: { if (x > minx) { real_x--; x -= std::max(wcwidth((*tmp)[beginning+real_x-minx]), 1); } else if (beginning > 0) beginning--; if (input != KEY_CTRL_H && input != KEY_BACKSPACE && input != KEY_BACKSPACE_2) break; // backspace = left & delete. } case KEY_DC: { if ((real_x-minx)+beginning == tmp->length()) break; tmp->erase(tmp->begin()+(real_x-minx)+beginning); if (beginning && beginning == maxbeginning && real_x < maxx) { real_x++; x++; } break; } case KEY_HOME: { real_x = x = minx; beginning = 0; break; } case KEY_END: { gotoend = 1; break; } case KEY_ENTER: break; case KEY_CTRL_U: tmp->clear(); real_maxx = maxx = real_x = x = minx; maxbeginning = beginning = 0; break; default: { if (tmp->length() >= length_) break; tmp_in += input; if (int(mbrtowc(&wc_in, tmp_in.c_str(), MB_CUR_MAX, 0)) < 0) break; int wcwidth_res = wcwidth(wc_in); if (wcwidth_res > 1) block_scrolling = 1; if (wcwidth_res > 0) // is char printable? we want to ignore things like Ctrl-?, Fx etc. { if ((real_x-minx)+beginning >= tmp->length()) { tmp->push_back(wc_in); if (!beginning) { real_x++; x += wcwidth(wc_in); } beginning++; gotoend = 1; } else { tmp->insert(tmp->begin()+(real_x-minx)+beginning, wc_in); if (x < maxx) { real_x++; x += wcwidth(wc_in); } else if (beginning < maxbeginning) beginning++; } } tmp_in.clear(); } } } while (input != KEY_ENTER); curs_set(0); if (m_history && !encrypted) { if (history_it != m_history->end()) { m_history->push_back(*history_it); tmp = &m_history->back(); m_history->erase(history_it); } else m_history->push_back(*tmp); } return ToString(*tmp); } void Window::goToXY(int x, int y) { wmove(m_window, y, x); } int Window::getX() { return getcurx(m_window); } int Window::getY() { return getcury(m_window); } bool Window::hasCoords(int &x, int &y) { # ifndef USE_PDCURSES return wmouse_trafo(m_window, &y, &x, 0); # else // wmouse_trafo is broken in pdcurses, use our own implementation size_t u_x = x, u_y = y; if (u_x >= m_start_x && u_x < m_start_x+m_width && u_y >= m_start_y && u_y < m_start_y+m_height) { x -= m_start_x; y -= m_start_y; return true; } return false; # endif } size_t Window::getWidth() const { if (m_border != brNone) return m_width+2; else return m_width; } size_t Window::getHeight() const { size_t height = m_height; if (m_border != brNone) height += 2; if (!m_title.empty()) height += 2; return height; } size_t Window::getStartX() const { if (m_border != brNone) return m_start_x-1; else return m_start_x; } size_t Window::getStarty() const { size_t starty = m_start_y; if (m_border != brNone) starty--; if (!m_title.empty()) starty -= 2; return starty; } const std::string &Window::getTitle() const { return m_title; } Color Window::getColor() const { return m_color; } Border Window::getBorder() const { return m_border; } int Window::getTimeout() const { return m_window_timeout; } void Window::scroll(Scroll where) { idlok(m_window, 1); scrollok(m_window, 1); switch (where) { case Scroll::Up: wscrl(m_window, 1); break; case Scroll::Down: wscrl(m_window, -1); break; case Scroll::PageUp: wscrl(m_window, m_width); break; case Scroll::PageDown: wscrl(m_window, -m_width); break; default: break; } idlok(m_window, 0); scrollok(m_window, 0); } Window &Window::operator<<(Colors colors) { if (colors.fg == clEnd || colors.bg == clEnd) { *this << clEnd; return *this; } m_color_stack.push(colors); setColor(colors.fg, colors.bg); return *this; } Window &Window::operator<<(Color color) { switch (color) { case clDefault: while (!m_color_stack.empty()) m_color_stack.pop(); setColor(m_base_color, m_base_bg_color); break; case clEnd: if (!m_color_stack.empty()) m_color_stack.pop(); if (!m_color_stack.empty()) setColor(m_color_stack.top().fg, m_color_stack.top().bg); else setColor(m_base_color, m_base_bg_color); break; default: m_color_stack.push(Colors(color, clDefault)); setColor(m_color_stack.top().fg, m_color_stack.top().bg); } return *this; } Window &Window::operator<<(Format format) { switch (format) { case fmtNone: bold((m_bold_counter = 0)); reverse((m_reverse_counter = 0)); altCharset((m_alt_charset_counter = 0)); break; case fmtBold: bold(++m_bold_counter); break; case fmtBoldEnd: if (--m_bold_counter <= 0) bold((m_bold_counter = 0)); break; case fmtUnderline: underline(++m_underline_counter); break; case fmtUnderlineEnd: if (--m_underline_counter <= 0) underline((m_underline_counter = 0)); break; case fmtReverse: reverse(++m_reverse_counter); break; case fmtReverseEnd: if (--m_reverse_counter <= 0) reverse((m_reverse_counter = 0)); break; case fmtAltCharset: altCharset(++m_alt_charset_counter); break; case fmtAltCharsetEnd: if (--m_alt_charset_counter <= 0) altCharset((m_alt_charset_counter = 0)); break; } return *this; } Window &Window::operator<<(int (*f)(WINDOW *)) { f(m_window); return *this; } Window &Window::operator<<(XY coords) { goToXY(coords.x, coords.y); return *this; } Window &Window::operator<<(const char *s) { for (const char *c = s; *c != '\0'; ++c) wprintw(m_window, "%c", *c); return *this; } Window &Window::operator<<(char c) { wprintw(m_window, "%c", c); return *this; } Window &Window::operator<<(const wchar_t *ws) { for (const wchar_t *wc = ws; *wc != L'\0'; ++wc) wprintw(m_window, "%lc", *wc); return *this; } Window &Window::operator<<(wchar_t wc) { wprintw(m_window, "%lc", wc); return *this; } Window &Window::operator<<(int i) { wprintw(m_window, "%d", i); return *this; } Window &Window::operator<<(double d) { wprintw(m_window, "%f", d); return *this; } Window &Window::operator<<(const std::string &s) { // for some reason passing whole string at once with "%s" doesn't work // (string is cut in the middle, probably due to limitation of ncurses' // internal buffer?), so we need to pass characters in a loop. for (auto it = s.begin(); it != s.end(); ++it) wprintw(m_window, "%c", *it); return *this; } Window &Window::operator<<(const std::wstring &ws) { for (auto it = ws.begin(); it != ws.end(); ++it) wprintw(m_window, "%lc", *it); return *this; } Window &Window::operator<<(size_t s) { wprintw(m_window, "%zu", s); return *this; } }