selection.cpp

/*-----------------------------------------------------------------------------
 * File: selection.cpp
 *-----------------------------------------------------------------------------
 * Copyright (c) 2004-2005  Kazuo Ishii <k-ishii@wb4.so-net.ne.jp>
 *				- original version
 *
 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *---------------------------------------------------------------------------*/
#include "ckw.h"
#include "encoding.h"

static int	gSelectMode = 0;
static COORD	gSelectPos = { -1, -1 }; // pick point
static SMALL_RECT gSelectRect = { -1, -1, -1, -1 }; // expanded selection area

static const wchar_t WORD_BREAK_CHARS[] = {
	L' ',  L'\t', L'\"', L'&', L'\'', L'(', L')', L'*',
	L',',  L';',  L'<',  L'=', L'>',  L'?', L'@', L'[',
	L'\\', L']',  L'^',  L'`', L'{',  L'}', L'~',
	0x3000,	// 全角スペース
	0x3001,	// 読点
	0x3002,	// 句点
	/**/
	0,
};

/*****************************************************************************/

#define SCRN_InvalidArea(x,y) \
	(y < gCSI->srWindow.Top    ||	\
	 y > gCSI->srWindow.Bottom ||	\
	 x < gCSI->srWindow.Left   ||	\
	 x > gCSI->srWindow.Right)

/**
 * 指定表示座標のCHAR_INFOを取得
 * @param x [in] 表示x座標
 * @param y [in] 表示y座標
 * @param is_half [out] 全角の途中か否か
 * @return CHAR_INF
 *
 */
static inline CHAR_INFO* SELECT_GetScrn(int x, int y, int &is_half) {

	// Y座標
	CHAR_INFO* ptr = gScreen + CSI_WndCols(gCSI) * (y - gCSI->srWindow.Top);

	// Xは実際チェックしないと確定しない
	is_half = 0;
	for(int pos=gCSI->srWindow.Left ; pos < x ; ptr++) {
		if(ptr->Attributes & COMMON_LVB_TRAILING_BYTE ||
		   ptr->Attributes & COMMON_LVB_LEADING_BYTE) {
			pos++;
			continue;
		}
		if(is_dblchar_cjk(ptr->Char.UnicodeChar) == true) {
			pos+=2;
			if(pos > x) {
				is_half = 1;
			}
		} else {
			pos++;
		}
	}

	return ptr;
}

/**
 * 表示座標から論理座標へ
 * @param x [in] 表示x座標
 * @param y [in] 表示y座標
 * @return 論理x座標(0 origin)
 */
static inline int charpos_to_strpos(int x, int y) {
	// Y座標
	CHAR_INFO* ptr = gScreen + CSI_WndCols(gCSI) * (y - gCSI->srWindow.Top);

	// Xは実際チェックしないと確定しない
	int pos_x = 0;
	for(int pos=gCSI->srWindow.Left; pos < x ; ptr++, pos_x++) {
		if(ptr->Attributes & COMMON_LVB_TRAILING_BYTE ||
			ptr->Attributes & COMMON_LVB_LEADING_BYTE) {
			pos++;
			continue;
		}
		if(is_dblchar_cjk(ptr->Char.UnicodeChar) == true) {
			pos+=2;
		} else {
			pos++;
		}
	}
	return pos_x;
}

static void __select_word_expand_left()
{
	if(SCRN_InvalidArea(gSelectRect.Left, gSelectRect.Top))
		return;
	int is_half;
	CHAR_INFO* base = SELECT_GetScrn(gSelectRect.Left, gSelectRect.Top, is_half);
	CHAR_INFO* ptr = base;
	int c = gSelectRect.Left;
	if (is_half) {
		ptr-=1;
	}
	else if (!(ptr->Attributes & COMMON_LVB_TRAILING_BYTE) &&
		!(ptr->Attributes & COMMON_LVB_LEADING_BYTE) &&
		is_dblchar_cjk(ptr->Char.UnicodeChar)) {
		c++;
	}

	for( ; c >= gCSI->srWindow.Left ; c--, ptr--) {
		if(wcschr(WORD_BREAK_CHARS, ptr->Char.UnicodeChar)) {
			c++;
			break;
		}
		if (ptr->Attributes & COMMON_LVB_TRAILING_BYTE ||
			ptr->Attributes & COMMON_LVB_LEADING_BYTE) {
			continue;
		}
		if (is_dblchar_cjk(ptr->Char.UnicodeChar) == true) {
			c--;
		}
	}
	if(c < 0)
		c = 0;

	if(gSelectRect.Left > c)
		gSelectRect.Left = c;
}

static void __select_word_expand_right()
{
	if(SCRN_InvalidArea(gSelectRect.Right, gSelectRect.Bottom))
		return;
	int is_half;
	CHAR_INFO* base = SELECT_GetScrn(gSelectRect.Right, gSelectRect.Bottom, is_half);
	CHAR_INFO* ptr = base;
	int c = gSelectRect.Right;
	if (is_half) {
		c++;
	}

	for( ; c <= gCSI->srWindow.Right ; c++, ptr++) {
		if(wcschr(WORD_BREAK_CHARS, ptr->Char.UnicodeChar)) {
			break;
		}
		if (ptr->Attributes & COMMON_LVB_TRAILING_BYTE ||
			ptr->Attributes & COMMON_LVB_LEADING_BYTE) {
			continue;
		}
		if (is_dblchar_cjk(ptr->Char.UnicodeChar) == true) {
			c++;
		}
	}

	if(gSelectRect.Right < c)
		gSelectRect.Right = c;
}

static void __select_char_expand()
{
	int is_half;
	CHAR_INFO* base;

	if(SCRN_InvalidArea(gSelectRect.Left, gSelectRect.Top)) {
	}
	else if (gSelectRect.Left - 1 >= gCSI->srWindow.Left) {
		base = SELECT_GetScrn(gSelectRect.Left, gSelectRect.Top, is_half);
		// マルチバイトの前半
		if(base->Attributes & COMMON_LVB_TRAILING_BYTE) {
			gSelectRect.Left--;
		}
		else if(!(base->Attributes & COMMON_LVB_LEADING_BYTE)) {
			// サロゲートの前半
			if (IS_LOW_SURROGATE(base->Char.UnicodeChar) == true) {
				gSelectRect.Left--;
			}
			// 全角の後半
			if (is_half) {
				gSelectRect.Left--;
			}
		}
	}

	if(SCRN_InvalidArea(gSelectRect.Right, gSelectRect.Bottom)) {
	}
	else {
		base  = SELECT_GetScrn(gSelectRect.Right, gSelectRect.Bottom, is_half);
		// マルチバイトの前半
		if(base->Attributes & COMMON_LVB_TRAILING_BYTE) {
			gSelectRect.Right++;
		} else if (!(base->Attributes & COMMON_LVB_LEADING_BYTE)) {
			// サロゲートの前半
			if(IS_LOW_SURROGATE(base->Char.UnicodeChar) == true) {
				gSelectRect.Right++;
			}
			// 全角の後半
			if (is_half) {
				gSelectRect.Right++;
			}
		}
	}
}

inline void __select_expand()
{
	if(gSelectMode == 0) {
		// 選択範囲
		__select_char_expand();
	}
	else if(gSelectMode == 1) {
		// ワード抽出(dboule click)
		__select_word_expand_left();
		__select_word_expand_right();
	}
	else if(gSelectMode == 2) {
		// 1行選択(triple click)
		gSelectRect.Left = gCSI->srWindow.Left;
		gSelectRect.Right = gCSI->srWindow.Right+1;
	}
}

// 実dot座標 to コンソール仮想座標
static void window_to_charpos(int& x, int& y)
{
	x -= gBorderSize;
	y -= gBorderSize;
	if(x < 0) x = 0;
	if(y < 0) y = 0;
	x /= gFontW;
	y /= gFontH;
	x += gCSI->srWindow.Left;
	y += gCSI->srWindow.Top;
	if(x > gCSI->srWindow.Right)  x = gCSI->srWindow.Right+1;
	if(y > gCSI->srWindow.Bottom) y = gCSI->srWindow.Bottom;
}

/*****************************************************************************/

static inline bool __select_invalid()
{
	return ( gSelectRect.Top > gSelectRect.Bottom ||
	         (gSelectRect.Top == gSelectRect.Bottom &&
	         gSelectRect.Left >= gSelectRect.Right) );
}

/*----------*/
bool	selectionGetArea(SMALL_RECT& sr)
{
	if( __select_invalid() )
		return(false);
	sr = gSelectRect;
	return(true);
}

/*----------*/
void	selectionClear(HWND hWnd)
{
	if( __select_invalid() )
		return;
	gSelectRect.Left = gSelectRect.Right = \
	gSelectRect.Top = gSelectRect.Bottom = 0;
	InvalidateRect(hWnd, nullptr, FALSE);
}

/*----------*/
wchar_t * selectionGetString()
{
	if( __select_invalid() )
		return(nullptr);

	int nb, y;

	if(gSelectRect.Top == gSelectRect.Bottom) {
		nb = gSelectRect.Right - gSelectRect.Left;
	}
	else {
		nb = gCSI->srWindow.Right - gSelectRect.Left+1;
		for(y = gSelectRect.Top+1 ; y <= gSelectRect.Bottom-1 ; y++)
			nb += CSI_WndCols(gCSI);
		nb += gSelectRect.Right - gCSI->srWindow.Left;
	}

	COORD      size = { CSI_WndCols(gCSI), 1 };
	CHAR_INFO* work = new CHAR_INFO[ size.X ];
	wchar_t*   buffer = new wchar_t[ nb +32 ];
	wchar_t*   wp = buffer;
	COORD      pos = { 0,0 };
	SMALL_RECT sr = { gCSI->srWindow.Left, 0, gCSI->srWindow.Right, 0 };

	*wp = 0;

	if(gSelectRect.Top == gSelectRect.Bottom) {
		// 1行
		sr.Top = sr.Bottom = gSelectRect.Top;
		ReadConsoleOutput(gStdOut, work, size, pos, &sr);
		SHORT Left  = charpos_to_strpos(gSelectRect.Left, gSelectRect.Top);
		SHORT Right = charpos_to_strpos(gSelectRect.Right, gSelectRect.Top);
		copyChar(wp, work, Left, Right-1, false);
	}
	else {
		// 複数行
		sr.Top = sr.Bottom = gSelectRect.Top;
		ReadConsoleOutput(gStdOut, work, size, pos, &sr);
		SHORT Left  = charpos_to_strpos(gSelectRect.Left, gSelectRect.Top);
		SHORT Right = charpos_to_strpos(gCSI->srWindow.Right, gSelectRect.Top);
		copyChar(wp, work, Left, Right);
		for(y = gSelectRect.Top+1 ; y <= gSelectRect.Bottom-1 ; y++) {
			sr.Top = sr.Bottom = y;
			ReadConsoleOutput(gStdOut, work, size, pos, &sr);
			copyChar(wp, work, gCSI->srWindow.Left, gCSI->srWindow.Right);
		}
		sr.Top = sr.Bottom = gSelectRect.Bottom;
		ReadConsoleOutput(gStdOut, work, size, pos, &sr);
		Left  = charpos_to_strpos(gCSI->srWindow.Left, gSelectRect.Top);
		Right = charpos_to_strpos(gSelectRect.Right, gSelectRect.Top);
		copyChar(wp, work, Left, Right-1, false);
	}

	delete [] work;
	return(buffer);
}

/*----------*/
void	onLBtnDown(HWND hWnd, int x, int y)
{
	static DWORD	prev_time = 0;
	static int	prevX = -100;
	static int	prevY = -100;

	{
		/* calc click count */
		DWORD now_time = GetTickCount();
		DWORD stime;
		if(prev_time > now_time)
			stime = now_time + ~prev_time+1;
		else
			stime = now_time - prev_time;
		if(stime <= GetDoubleClickTime()) {
			int sx = (prevX > x) ? prevX-x : x-prevX;
			int sy = (prevY > y) ? prevY-y : y-prevY;
			if(sx <= GetSystemMetrics(SM_CXDOUBLECLK) &&
			   sy <= GetSystemMetrics(SM_CYDOUBLECLK)) {
				if(++gSelectMode > 2)
					gSelectMode = 0;
			}
			else {
				gSelectMode = 0;
			}
		}
		else {
			gSelectMode = 0;
		}
		prev_time = now_time;
		prevX = x;
		prevY = y;
	}

	if(!gScreen || !gCSI)
		return;
	window_to_charpos(x, y);
	SetCapture(hWnd);

	gSelectPos.X = x;
	gSelectPos.Y = y;
	gSelectRect.Left = gSelectRect.Right = x;
	gSelectRect.Top  = gSelectRect.Bottom = y;

	__select_expand();
	InvalidateRect(hWnd, nullptr, FALSE);
}

/*----------*/
void	onLBtnUp(HWND hWnd, int x, int y)
{
	if(hWnd != GetCapture())
		return;
	ReleaseCapture();
	if(!gScreen || !gCSI)
		return;
	//window_to_charpos(x, y);

	wchar_t* str = selectionGetString();
	if(!str) return;

	copyStringToClipboard( hWnd, str );

	delete [] str;
}

/*----------*/
void	onMouseMove(HWND hWnd, int x, int y)
{
	if(hWnd != GetCapture())
		return;
	if(!gScreen || !gCSI)
		return;
	window_to_charpos(x, y);

	SMALL_RECT bak = gSelectRect;

	if(y < gSelectPos.Y || (y == gSelectPos.Y && x < gSelectPos.X)) {
		gSelectRect.Left   = x;
		gSelectRect.Top    = y;
		gSelectRect.Right  = gSelectPos.X;
		gSelectRect.Bottom = gSelectPos.Y;
	}
	else {
		gSelectRect.Left   = gSelectPos.X;
		gSelectRect.Top    = gSelectPos.Y;
		gSelectRect.Right  = x;
		gSelectRect.Bottom = y;
	}
	__select_expand();

	if(memcmp(&bak, &gSelectRect, sizeof(bak))) {
		InvalidateRect(hWnd, nullptr, FALSE);
	}
}

/* EOF */