Files
ncmpcpp/src/window.cpp
2012-07-16 01:38:19 +02:00

953 lines
20 KiB
C++

/***************************************************************************
* 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 <cstring>
#include <cstdlib>
#ifdef WIN32
# include <winsock.h>
#else
# include <sys/select.h>
# include <unistd.h>
#endif
#include "error.h"
#include "window.h"
using namespace NCurses;
void NCurses::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<char **>(&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 NCurses::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)
: itsWindow(0),
itsWinBorder(0),
itsStartX(startx),
itsStartY(starty),
itsWidth(width),
itsHeight(height),
itsWindowTimeout(-1),
itsColor(color),
itsBgColor(clDefault),
itsBaseColor(color),
itsBaseBgColor(clDefault),
itsBorder(border),
itsGetStringHelper(0),
itsX(0),
itsY(0),
itsTitle(title),
itsHistory(0),
itsBoldCounter(0),
itsUnderlineCounter(0),
itsReverseCounter(0),
itsAltCharsetCounter(0)
{
if (itsStartX > size_t(COLS)
|| itsStartY > size_t(LINES)
|| itsWidth+itsStartX > size_t(COLS)
|| itsHeight+itsStartY > size_t(LINES))
FatalError("Constructed window is bigger than terminal size!");
if (itsBorder != brNone)
{
itsWinBorder = newpad(itsHeight, itsWidth);
wattron(itsWinBorder, COLOR_PAIR(itsBorder));
box(itsWinBorder, 0, 0);
itsStartX++;
itsStartY++;
itsWidth -= 2;
itsHeight -= 2;
}
if (!itsTitle.empty())
{
itsStartY += 2;
itsHeight -= 2;
}
itsWindow = newpad(itsHeight, itsWidth);
SetColor(itsColor);
keypad(itsWindow, 1);
}
Window::Window(const Window &w) : itsWindow(dupwin(w.itsWindow)),
itsWinBorder(dupwin(w.itsWinBorder)),
itsStartX(w.itsStartX),
itsStartY(w.itsStartY),
itsWidth(w.itsWidth),
itsHeight(w.itsHeight),
itsWindowTimeout(w.itsWindowTimeout),
itsColor(w.itsColor),
itsBgColor(w.itsBgColor),
itsBaseColor(w.itsBaseColor),
itsBaseBgColor(w.itsBaseBgColor),
itsBorder(w.itsBorder),
itsGetStringHelper(w.itsGetStringHelper),
itsX(w.itsX),
itsY(w.itsY),
itsTitle(w.itsTitle),
itsColors(w.itsColors),
itsHistory(w.itsHistory),
itsBoldCounter(w.itsBoldCounter),
itsUnderlineCounter(w.itsUnderlineCounter),
itsReverseCounter(w.itsReverseCounter),
itsAltCharsetCounter(w.itsAltCharsetCounter)
{
}
Window::~Window()
{
delwin(itsWindow);
delwin(itsWinBorder);
}
void Window::SetColor(Color fg, Color bg)
{
if (fg == clDefault)
fg = itsBaseColor;
if (fg != clDefault)
wattron(itsWindow, COLOR_PAIR(bg*8+fg));
else
wattroff(itsWindow, COLOR_PAIR(itsColor));
itsColor = fg;
itsBgColor = bg;
}
void Window::SetBaseColor(Color fg, Color bg)
{
itsBaseColor = fg;
itsBaseBgColor = bg;
}
void Window::SetBorder(Border border)
{
if (border == brNone && itsBorder != brNone)
{
delwin(itsWinBorder);
itsStartX--;
itsStartY--;
itsHeight += 2;
itsWidth += 2;
Recreate(itsWidth, itsHeight);
}
else if (border != brNone && itsBorder == brNone)
{
itsWinBorder = newpad(itsHeight, itsWidth);
wattron(itsWinBorder, COLOR_PAIR(border));
box(itsWinBorder,0,0);
itsStartX++;
itsStartY++;
itsHeight -= 2;
itsWidth -= 2;
Recreate(itsWidth, itsHeight);
}
else
{
wattron(itsWinBorder,COLOR_PAIR(border));
box(itsWinBorder, 0, 0);
}
itsBorder = border;
}
void Window::SetTitle(const std::string &new_title)
{
if (itsTitle == new_title)
{
return;
}
else if (!new_title.empty() && itsTitle.empty())
{
itsStartY += 2;
itsHeight -= 2;
Recreate(itsWidth, itsHeight);
}
else if (new_title.empty() && !itsTitle.empty())
{
itsStartY -= 2;
itsHeight += 2;
Recreate(itsWidth, itsHeight);
}
itsTitle = new_title;
}
void Window::CreateHistory()
{
if (!itsHistory)
itsHistory = new std::list<std::wstring>;
}
void Window::DeleteHistory()
{
delete itsHistory;
itsHistory = 0;
}
void Window::Recreate(size_t width, size_t height)
{
delwin(itsWindow);
itsWindow = newpad(height, width);
SetTimeout(itsWindowTimeout);
SetColor(itsColor, itsBgColor);
keypad(itsWindow, 1);
}
void Window::MoveTo(size_t new_x, size_t new_y)
{
itsStartX = new_x;
itsStartY = new_y;
if (itsBorder != brNone)
{
itsStartX++;
itsStartY++;
}
if (!itsTitle.empty())
itsStartY += 2;
}
void Window::AdjustDimensions(size_t width, size_t height)
{
if (itsBorder != brNone)
{
delwin(itsWinBorder);
itsWinBorder = newpad(height, width);
wattron(itsWinBorder, COLOR_PAIR(itsBorder));
box(itsWinBorder, 0, 0);
width -= 2;
height -= 2;
}
if (!itsTitle.empty())
height -= 2;
itsHeight = height;
itsWidth = width;
}
void Window::Resize(size_t new_width, size_t new_height)
{
AdjustDimensions(new_width, new_height);
Recreate(itsWidth, itsHeight);
}
void Window::ShowBorder() const
{
if (itsBorder != brNone)
{
refresh();
prefresh(itsWinBorder, 0, 0, GetStartY(), GetStartX(), itsStartY+itsHeight, itsStartX+itsWidth);
}
if (!itsTitle.empty())
{
if (itsBorder != brNone)
attron(COLOR_PAIR(itsBorder));
else
attron(COLOR_PAIR(itsBaseColor));
mvhline(itsStartY-1, itsStartX, 0, itsWidth);
attron(A_BOLD);
mvhline(itsStartY-2, itsStartX, 32, itsWidth); // clear title line
mvaddstr(itsStartY-2, itsStartX, itsTitle.c_str());
attroff(COLOR_PAIR(itsBorder) | A_BOLD);
}
refresh();
}
void Window::Display()
{
ShowBorder();
Refresh();
}
void Window::Refresh()
{
prefresh(itsWindow, 0, 0, itsStartY, itsStartX, itsStartY+itsHeight-1, itsStartX+itsWidth-1);
}
void Window::Clear()
{
werase(itsWindow);
}
void Window::Hide(char ch) const
{
for (size_t i = 0; i < GetHeight(); ++i)
mvhline(i+GetStartY(), GetStartX(), ch, GetWidth());
refresh();
}
void Window::Bold(bool bold_state) const
{
(bold_state ? wattron : wattroff)(itsWindow, A_BOLD);
}
void Window::Underline(bool underline_state) const
{
(underline_state ? wattron : wattroff)(itsWindow, A_UNDERLINE);
}
void Window::Reverse(bool reverse_state) const
{
(reverse_state ? wattron : wattroff)(itsWindow, A_REVERSE);
}
void Window::AltCharset(bool altcharset_state) const
{
(altcharset_state ? wattron : wattroff)(itsWindow, A_ALTCHARSET);
}
void Window::SetTimeout(int timeout)
{
itsWindowTimeout = timeout;
wtimeout(itsWindow, timeout);
}
void Window::AddFDCallback(int fd, void (*callback)())
{
itsFDs.push_back(std::make_pair(fd, callback));
}
void Window::ClearFDCallbacksList()
{
itsFDs.clear();
}
bool Window::FDCallbacksListEmpty() const
{
return itsFDs.empty();
}
void Window::ReadKey(int &read_key) const
{
// 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 = { itsWindowTimeout/1000, (itsWindowTimeout%1000)*1000 };
# else
timeval timeout = { 0, 0 };
# endif
int fd_max = STDIN_FILENO;
for (FDCallbacks::const_iterator it = itsFDs.begin(); it != itsFDs.end(); ++it)
{
if (it->first > fd_max)
fd_max = it->first;
FD_SET(it->first, &fdset);
}
if (select(fd_max+1, &fdset, 0, 0, itsWindowTimeout < 0 ? 0 : &timeout) > 0)
{
# if !defined(USE_PDCURSES)
read_key = FD_ISSET(STDIN_FILENO, &fdset) ? wgetch(itsWindow) : ERR;
# endif // !USE_PDCURSES
for (FDCallbacks::const_iterator it = itsFDs.begin(); it != itsFDs.end(); ++it)
if (FD_ISSET(it->first, &fdset))
it->second();
}
# if !defined(USE_PDCURSES)
else
read_key = ERR;
# else
read_key = wgetch(itsWindow);
# endif
}
void Window::ReadKey() const
{
wgetch(itsWindow);
}
std::string Window::GetString(const std::string &base, size_t length, size_t width, bool encrypted) const
{
int input;
size_t beginning, maxbeginning, minx, x, real_x, y, maxx, real_maxx;
getyx(itsWindow, y, x);
minx = real_maxx = maxx = real_x = x;
width--;
if (width == size_t(-1))
width = itsWidth-x-1;
curs_set(1);
std::wstring wbase = ToWString(base);
std::wstring *tmp = &wbase;
std::list<std::wstring>::iterator history_it = itsHistory->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 + (Length(*tmp) < width ? Length(*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(itsWindow, y, minx, ' ', width+1);
if (!encrypted)
mvwprintw(itsWindow, y, minx, "%ls", tmp->substr(beginning, width+1).c_str());
else
mvwhline(itsWindow, y, minx, '*', maxx-minx);
if (itsGetStringHelper)
itsGetStringHelper(*tmp);
wmove(itsWindow, y, x);
prefresh(itsWindow, 0, 0, itsStartY, itsStartX, itsStartY+itsHeight-1, itsStartX+itsWidth-1);
ReadKey(input);
// these key codes are special and should be ignored
if (input >= KEY_CTRL_A
&& input != KEY_CTRL_H
&& input != KEY_ENTER
&& input != KEY_CTRL_U
&& input <= KEY_CTRL_Z)
continue;
switch (input)
{
case ERR:
case KEY_MOUSE:
break;
case KEY_UP:
if (itsHistory && !encrypted && history_it != itsHistory->begin())
{
while (--history_it != itsHistory->begin())
if (!history_it->empty())
break;
tmp = &*history_it;
gotoend = 1;
}
break;
case KEY_DOWN:
if (itsHistory && !encrypted && history_it != itsHistory->end())
{
while (++history_it != itsHistory->end())
if (!history_it->empty())
break;
tmp = &(history_it == itsHistory->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;
if (wcwidth(wc_in) > 1)
block_scrolling = 1;
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 (itsHistory && !encrypted)
{
if (history_it != itsHistory->end())
{
itsHistory->push_back(*history_it);
tmp = &itsHistory->back();
itsHistory->erase(history_it);
}
else
itsHistory->push_back(*tmp);
}
return ToString(*tmp);
}
void Window::GetXY(int &x, int &y)
{
getyx(itsWindow, y, x);
itsX = x;
itsY = y;
}
void Window::GotoXY(int x, int y)
{
wmove(itsWindow, y, x);
itsX = x;
itsY = y;
}
int Window::X() const
{
return itsX;
}
int Window::Y() const
{
return itsY;
}
bool Window::hasCoords(int &x, int &y)
{
# ifndef USE_PDCURSES
return wmouse_trafo(itsWindow, &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 >= itsStartX && u_x < itsStartX+itsWidth
&& u_y >= itsStartY && u_y < itsStartY+itsHeight)
{
x -= itsStartX;
y -= itsStartY;
return true;
}
return false;
# endif
}
size_t Window::GetWidth() const
{
if (itsBorder != brNone)
return itsWidth+2;
else
return itsWidth;
}
size_t Window::GetHeight() const
{
size_t height = itsHeight;
if (itsBorder != brNone)
height += 2;
if (!itsTitle.empty())
height += 2;
return height;
}
size_t Window::GetStartX() const
{
if (itsBorder != brNone)
return itsStartX-1;
else
return itsStartX;
}
size_t Window::GetStartY() const
{
size_t starty = itsStartY;
if (itsBorder != brNone)
starty--;
if (!itsTitle.empty())
starty -= 2;
return starty;
}
const std::string &Window::GetTitle() const
{
return itsTitle;
}
Color Window::GetColor() const
{
return itsColor;
}
Border Window::GetBorder() const
{
return itsBorder;
}
int Window::GetTimeout() const
{
return itsWindowTimeout;
}
void Window::Scroll(Where where)
{
idlok(itsWindow, 1);
scrollok(itsWindow, 1);
switch (where)
{
case wUp:
wscrl(itsWindow, 1);
break;
case wDown:
wscrl(itsWindow, -1);
break;
case wPageUp:
wscrl(itsWindow, itsWidth);
break;
case wPageDown:
wscrl(itsWindow, -itsWidth);
break;
default:
break;
}
idlok(itsWindow, 0);
scrollok(itsWindow, 0);
}
Window &Window::operator<<(Colors colors)
{
if (colors.fg == clEnd || colors.bg == clEnd)
{
*this << clEnd;
return *this;
}
itsColors.push(colors);
SetColor(colors.fg, colors.bg);
return *this;
}
Window &Window::operator<<(Color color)
{
switch (color)
{
case clDefault:
while (!itsColors.empty())
itsColors.pop();
SetColor(itsBaseColor, itsBaseBgColor);
break;
case clEnd:
if (!itsColors.empty())
itsColors.pop();
if (!itsColors.empty())
SetColor(itsColors.top().fg, itsColors.top().bg);
else
SetColor(itsBaseColor, itsBaseBgColor);
break;
default:
itsColors.push(Colors(color, clDefault));
SetColor(itsColors.top().fg, itsColors.top().bg);
}
return *this;
}
Window &Window::operator<<(Format format)
{
switch (format)
{
case fmtNone:
Bold((itsBoldCounter = 0));
Reverse((itsReverseCounter = 0));
AltCharset((itsAltCharsetCounter = 0));
break;
case fmtBold:
Bold(++itsBoldCounter);
break;
case fmtBoldEnd:
if (--itsBoldCounter <= 0)
Bold((itsBoldCounter = 0));
break;
case fmtUnderline:
Underline(++itsUnderlineCounter);
break;
case fmtUnderlineEnd:
if (--itsUnderlineCounter <= 0)
Underline((itsUnderlineCounter = 0));
break;
case fmtReverse:
Reverse(++itsReverseCounter);
break;
case fmtReverseEnd:
if (--itsReverseCounter <= 0)
Reverse((itsReverseCounter = 0));
break;
case fmtAltCharset:
AltCharset(++itsAltCharsetCounter);
break;
case fmtAltCharsetEnd:
if (--itsAltCharsetCounter <= 0)
AltCharset((itsAltCharsetCounter = 0));
break;
}
return *this;
}
Window &Window::operator<<(int (*f)(WINDOW *))
{
f(itsWindow);
return *this;
}
Window &Window::operator<<(XY coords)
{
GotoXY(coords.x, coords.y);
return *this;
}
Window &Window::operator<<(const char *s)
{
wprintw(itsWindow, "%s", s);
return *this;
}
Window &Window::operator<<(char c)
{
wprintw(itsWindow, "%c", c);
return *this;
}
Window &Window::operator<<(const wchar_t *ws)
{
wprintw(itsWindow, "%ls", ws);
return *this;
}
Window &Window::operator<<(wchar_t wc)
{
wprintw(itsWindow, "%lc", wc);
return *this;
}
Window &Window::operator<<(int i)
{
wprintw(itsWindow, "%d", i);
return *this;
}
Window &Window::operator<<(double d)
{
wprintw(itsWindow, "%f", d);
return *this;
}
Window &Window::operator<<(const std::string &s)
{
for (std::string::const_iterator it = s.begin(); it != s.end(); ++it)
wprintw(itsWindow, "%c", *it);
return *this;
}
Window &Window::operator<<(const std::wstring &ws)
{
for (std::wstring::const_iterator it = ws.begin(); it != ws.end(); ++it)
wprintw(itsWindow, "%lc", *it);
return *this;
}
Window &Window::operator<<(size_t s)
{
wprintw(itsWindow, "%zu", s);
return *this;
}
std::string ToString(const std::wstring &ws)
{
std::string result;
char s[MB_CUR_MAX];
for (size_t i = 0; i < ws.length(); ++i)
{
int n = wcrtomb(s, ws[i], 0);
if (n > 0)
result.append(s, n);
}
return result;
}
std::wstring ToWString(const std::string &s)
{
std::wstring result;
wchar_t *ws = new wchar_t[s.length()];
const char *c_s = s.c_str();
int n = mbsrtowcs(ws, &c_s, s.length(), 0);
if (n > 0)
result.append(ws, n);
delete [] ws;
return result;
}
size_t Window::Length(const std::wstring &ws)
{
# ifdef WIN32
return ws.length();
# else
int len = wcswidth(ws.c_str(), -1);
return len < 0 ? ws.length() : len;
# endif // WIN32
}