Click here to Skip to main content
15,895,746 members
Search within:


Articles / Desktop Programming / MFC
 

Enhancements of Matthew R. Millers COptionTree

Rate me:
Please Sign up or sign in to vote.
4.76/5 (15 votes)
26 Apr 2006CPOL3 min read 120.5K   2.6K   58  
This article demonstrates an enhanced version of Matthew R. Millers Option Tree
 
// COptionTree
//
// License
// -------
// This code is provided "as is" with no expressed or implied warranty.
// 
// You may use this code in a commercial product with or without acknowledgement.
// However you may not sell this code or any modification of this code, this includes 
// commercial libraries and anything else for profit.
//
// I would appreciate a notification of any bugs or bug fixes to help the control grow.
//
// History:
// --------
//	See License.txt for full history information.
//
//
// Copyright (c) 1999-2002 
// ComputerSmarts.net 
// mattrmiller@computersmarts.net

#include "stdafx.h"
#include "OptionTreeSpinnerButton.h"

// Added Headers
#include "OptionTree.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

/////////////////////////////////////////////////////////////////////////////
// COptionTreeSpinnerButton

COptionTreeSpinnerButton::COptionTreeSpinnerButton()
{
	// Intialize variables
	m_otSpinnerOption = NULL;
	m_dRangeTop = 100;
	m_dRangeBottom = 0;
	m_rcButtonTop = CRect(0, 0, 0, 0);
	m_rcButtonBottom = CRect(0, 0, 0, 0);
	m_bBottomPressed = FALSE;
	m_bTopPressed = FALSE;
	m_bFirstRepeat = FALSE;
	m_nRepeatDelay = 0;
	m_nRepeatRate = 0;
	m_dwOptions = NULL;
}

COptionTreeSpinnerButton::~COptionTreeSpinnerButton()
{
}


BEGIN_MESSAGE_MAP(COptionTreeSpinnerButton, CWnd)
	//{{AFX_MSG_MAP(COptionTreeSpinnerButton)
	ON_WM_ERASEBKGND()
	ON_WM_PAINT()
	ON_WM_LBUTTONUP()
	ON_WM_LBUTTONDOWN()
	ON_WM_CREATE()
	ON_WM_DESTROY()
	ON_WM_MOVE()
	ON_WM_SIZE()
	ON_MESSAGE(OT_NOTIFY_FORCEREDRAW, WM_ForceRedraw)
	ON_MESSAGE(OT_NOTIFY_UP, WM_EditUp)
	ON_MESSAGE(OT_NOTIFY_DOWN, WM_EditDown)	
	ON_WM_TIMER()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// COptionTreeSpinnerButton message handlers

void COptionTreeSpinnerButton::SetSpinnerOptionsOwner(COptionTree *otOption)
{
	// Save pointer
	m_otSpinnerOption = otOption;
}

BOOL COptionTreeSpinnerButton::OnEraseBkgnd(CDC* pDC) 
{
	// Naa, we like flicker free better
	return FALSE;
}

void COptionTreeSpinnerButton::SetRange(double dBottom, double dTop)
{
	// Save variables
	m_dRangeBottom = dBottom;
	m_dRangeTop = dTop;
}

void COptionTreeSpinnerButton::OnPaint() 
{
	// Make sure options aren't NULL
	if (m_otSpinnerOption == NULL)
	{
		return;
	}

	// Declare variables
	CPaintDC dc(this);
	CDC* pDCMem = new CDC;
	CBitmap bpMem;
	CBitmap *bmOld;
	COLORREF crOld;
	HGDIOBJ hOldBrush;
	int nOldBack;
	CRect rcButtonTop, rcButtonBottom, rcClient;
	CString strText;
	HGDIOBJ hOld;

	// Get client rectangle
	GetClientRect(rcClient);

	// Calculate rectangle
	// -- Top
	rcButtonTop.top = rcClient.top;
	rcButtonTop.left = (rcClient.right - 2) - OT_SPINNER_WIDTH;
	rcButtonTop.right = rcClient.right - 2;
	rcButtonTop.bottom = rcClient.Height() / 2;
	m_rcButtonTop = rcButtonTop;
	// -- Bottom
	rcButtonBottom.top = rcButtonTop.bottom;
	rcButtonBottom.left = (rcClient.right - 2) - OT_SPINNER_WIDTH;
	rcButtonBottom.right = rcClient.right - 2;
	rcButtonBottom.bottom = rcClient.bottom;
	m_rcButtonBottom = rcButtonBottom;

	// Create DC
	pDCMem->CreateCompatibleDC(&dc);
	
	// Create bitmap
	bpMem.CreateCompatibleBitmap(&dc, rcClient.Width(), rcClient.Height());

	// Select bitmap
	bmOld = pDCMem->SelectObject(&bpMem);

	// Set background mode
	nOldBack = pDCMem->SetBkMode(TRANSPARENT);

	// Set text color
	crOld = pDCMem->SetTextColor(GetSysColor(COLOR_WINDOWTEXT));

	// Select font
	hOld = pDCMem->SelectObject(m_otSpinnerOption->GetNormalFont());

	// Draw control background
	hOldBrush = pDCMem->SelectObject(GetSysColorBrush(COLOR_BTNFACE));
	pDCMem->PatBlt(rcClient.left, rcClient.top, rcClient.Width(), rcClient.Height(), PATCOPY);

	// Wrap around
	if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
	{
		// -- Draw top button
		// -- -- Pressed
		if (m_bTopPressed == TRUE)
		{
			pDCMem->DrawFrameControl(&rcButtonTop, DFC_SCROLL, DFCS_PUSHED | DFCS_SCROLLUP); 
		}
		// -- -- UnPressed
		else
		{
			pDCMem->DrawFrameControl(&rcButtonTop, DFC_SCROLL, DFCS_SCROLLUP); 
		}

		// -- Draw bottom button
		// -- -- Pressed
		if (m_bBottomPressed == TRUE)
		{
			pDCMem->DrawFrameControl(&rcButtonBottom, DFC_SCROLL, DFCS_PUSHED | DFCS_SCROLLDOWN); 
		}
		// -- -- UnPressed
		else
		{
			pDCMem->DrawFrameControl(&rcButtonBottom, DFC_SCROLL, DFCS_SCROLLDOWN); 
		}
	}
	// No Wrap around
	else
	{
		// -- Draw top button
		if (_GetValue() >= m_dRangeTop)
		{
			pDCMem->DrawFrameControl(&rcButtonTop, DFC_SCROLL, DFCS_INACTIVE | DFCS_SCROLLUP); 
		}
		else
		{
			// -- -- Pressed
			if (m_bTopPressed == TRUE)
			{
				pDCMem->DrawFrameControl(&rcButtonTop, DFC_SCROLL, DFCS_PUSHED | DFCS_SCROLLUP); 
			}
			// -- -- UnPressed
			else
			{
				pDCMem->DrawFrameControl(&rcButtonTop, DFC_SCROLL, DFCS_SCROLLUP); 
			}
		}

		// -- Draw bottom button
		if (_GetValue() <= m_dRangeBottom)
		{
			pDCMem->DrawFrameControl(&rcButtonBottom, DFC_SCROLL, DFCS_INACTIVE | DFCS_SCROLLDOWN); 
		}
		else
		{
			// -- -- Pressed
			if (m_bBottomPressed == TRUE)
			{
				pDCMem->DrawFrameControl(&rcButtonBottom, DFC_SCROLL, DFCS_PUSHED | DFCS_SCROLLDOWN); 
			}
			// -- -- UnPressed
			else
			{
				pDCMem->DrawFrameControl(&rcButtonBottom, DFC_SCROLL, DFCS_SCROLLDOWN); 
			}
		}
	}

	// Copy to screen
	dc.BitBlt(0, 0, rcClient.Width(), rcClient.Height(), pDCMem, 0, 0, SRCCOPY);

	// Restore GDI ojects
	pDCMem->SelectObject(bmOld);
	pDCMem->SelectObject(hOldBrush);
	pDCMem->SetBkMode(nOldBack);
	pDCMem->SelectObject(hOld);
	pDCMem->SetTextColor(crOld);	

	// Delete objects
	if (pDCMem->GetSafeHdc() != NULL)
	{
		pDCMem->DeleteDC();
	}
	delete pDCMem;
	if (bpMem.GetSafeHandle() != NULL)
	{
		bpMem.DeleteObject();
	}
}

void COptionTreeSpinnerButton::GetRange(double &dBottom, double &dTop)
{
	// Save variables
	dBottom = m_dRangeBottom;
	dTop = m_dRangeTop;
}

void COptionTreeSpinnerButton::OnLButtonUp(UINT nFlags, CPoint point) 
{
	// Kill timer
	KillTimer(OT_TIMER);

	// Clear pressed
	m_bBottomPressed = FALSE;
	m_bTopPressed = FALSE;

	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();
	
	CWnd::OnLButtonUp(nFlags, point);
}

void COptionTreeSpinnerButton::OnLButtonDown(UINT nFlags, CPoint point) 
{
	// Clear pressed
	m_bBottomPressed = FALSE;
	m_bTopPressed = FALSE;

	// Save point
	m_ptSavePoint = point;

	// See if we have pressed a button
	if (m_rcButtonTop.PtInRect(point) == TRUE)
	{
		// -- Mark bool
		m_bTopPressed = TRUE;

		// -- Update value
		// -- -- Wrap around
		if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
		{
			if ((_GetValue() + 1) > m_dRangeTop)
			{
				SetEditDouble(m_dRangeBottom);
			}
			else
			{
				SetEditDouble(_GetValue() + 1);
			}

		}
		// -- -- No wrap around
		else
		{
			if ((_GetValue() + 1) <= m_dRangeTop)
			{
				SetEditDouble(_GetValue() + 1);
			}
		}
	}
	else if (m_rcButtonBottom.PtInRect(point) == TRUE)
	{
		// -- Mark bool
		m_bBottomPressed = TRUE;

		// -- Update value
		// -- -- Wrap around
		if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
		{
			if ((_GetValue() - 1) < m_dRangeBottom)
			{
				SetEditDouble(m_dRangeTop);
			}
			else
			{
				SetEditDouble(_GetValue() - 1);
			}
		}
		// -- -- No wrap around
		else
		{
			if ((_GetValue() - 1) >= m_dRangeBottom)
			{
				SetEditDouble(_GetValue() - 1);
			}
		}
	}

	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();

	// Set repeat timer
	KillTimer(OT_TIMER);
	m_bFirstRepeat = TRUE;
	SetTimer(OT_TIMER, m_nRepeatDelay, NULL);
	
	CWnd::OnLButtonDown(nFlags, point);
}

int COptionTreeSpinnerButton::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	// Declare variables
	CRect rcEdit;
	DWORD dwStyle = WS_CHILD | WS_VISIBLE | ES_AUTOHSCROLL | ES_NUMBER;
	int nSetting;
	DWORD dwSetting;
	
	// Create window
	if (CWnd::OnCreate(lpCreateStruct) == -1)
	{
		return -1;
	}

	// Calculate edit rectangle
	GetClientRect(rcEdit);
	rcEdit.right = (rcEdit.right - 2) - OT_SPINNER_WIDTH;
	
	// Create edit
	if(m_ctlEdit.Create(dwStyle, rcEdit, this, 1000) == FALSE)
	{
		return -1;
	}

	// Set selection
	m_ctlEdit.SetSel(0, 0);

	// Set font
	m_ctlEdit.SetFont(m_otSpinnerOption->GetNormalFont(), TRUE);

	//  Modify style
	m_ctlEdit.ModifyStyleEx(0, WS_EX_CLIENTEDGE, SWP_FRAMECHANGED);

	// Set Owner spinner
	m_ctlEdit.SetOwnerSpinner(this);

	// Get keyboard repeat info
	if (SystemParametersInfo(SPI_GETKEYBOARDDELAY, 0, &nSetting, 0) == FALSE)
	{
		m_nRepeatDelay = 500;
	}
	m_nRepeatDelay = (nSetting + 1) * 250;
	if (SystemParametersInfo(SPI_GETKEYBOARDSPEED, 0, &dwSetting, 0) == FALSE)
	{
		m_nRepeatRate = 200;
	}
	m_nRepeatRate = 400 - (dwSetting * 12);
	
	return 0;
}

void COptionTreeSpinnerButton::SetEditDouble(double dValue)
{
	// Declare variables
	CString strText;

	// Convert string
	strText.Format("%.0f", dValue);

	// Modify style
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowText(strText);
	}
}

void COptionTreeSpinnerButton::SetEditInt(int nValue)
{
	// Declare variables
	CString strText;

	// Convert string
	strText.Format("%d", nValue);

	// Modify style
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowText(strText);
	}
}

void COptionTreeSpinnerButton::SetEditFloat(float fValue)
{
	// Declare variables
	CString strText;

	// Convert string
	strText.Format("%.0f", fValue);

	// Modify style
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowText(strText);
	}
}

void COptionTreeSpinnerButton::SetEditDword(DWORD dwValue)
{
	// Declare variables
	CString strText;

	// Convert string
	strText.Format("%d", dwValue);

	// Modify style
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowText(strText);
	}
}

void COptionTreeSpinnerButton::SetEditLong(long lValue)
{
	// Declare variables
	CString strText;

	// Convert string
	strText.Format("%.0f", lValue);

	// Modify style
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowText(strText);
	}
}

BOOL COptionTreeSpinnerButton::GetEditFloat(float &fReturn)
{
	// Declare variables
	CString strWindowText;

	// Set blank
	fReturn = 0;

	// Get window text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()) == TRUE)
	{
		m_ctlEdit.GetWindowText(strWindowText);
	}
	else
	{
		return FALSE;
	}

	// See if string is numeric
	if (!IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		return FALSE;
	}

	// Convert string
	fReturn = (float) atof(strWindowText);

	return TRUE;
}

BOOL COptionTreeSpinnerButton::GetEditDouble(double &dReturn)
{
	// Declare variables
	CString strWindowText;

	// Set blank
	dReturn = 0;

	// Get window text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.GetWindowText(strWindowText);
	}
	else
	{
		return FALSE;
	}

	// See if string is numeric
	if (!IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		return FALSE;
	}

	// Convert string
	dReturn = atof(strWindowText);

	return TRUE;
}

BOOL COptionTreeSpinnerButton::GetEditLong(long &lReturn)
{
	// Declare variables
	CString strWindowText;

	// Set blank
	lReturn = 0;

	// Get window text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.GetWindowText(strWindowText);
	}
	else
	{
		return FALSE;
	}

	// See if string is numeric
	if (!IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		return FALSE;
	}

	// Convert string
	lReturn = atol(strWindowText);

	return TRUE;
}

BOOL COptionTreeSpinnerButton::GetEditInt(int &nReturn)
{
	// Declare variables
	CString strWindowText;

	// Set blank
	nReturn = 0;

	// Get window text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.GetWindowText(strWindowText);
	}
	else
	{
		return FALSE;
	}

	// See if string is numeric
	if (!IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		return FALSE;
	}

	// Convert string
	nReturn = atoi(strWindowText);

	return TRUE;
}

BOOL COptionTreeSpinnerButton::GetEditDword(DWORD &dwReturn)
{
	// Declare variables
	CString strWindowText;

	// Set blank
	dwReturn = 0;

	// Get window text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.GetWindowText(strWindowText);
	}
	else
	{
		return FALSE;
	}

	// See if string is numeric
	if (!IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		return FALSE;
	}

	// Convert string
	dwReturn = (DWORD) atoi(strWindowText);

	return TRUE;
}

void COptionTreeSpinnerButton::OnDestroy() 
{
	// Destroy edit window
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.DestroyWindow();
	}	
	
	CWnd::OnDestroy();	
}

void COptionTreeSpinnerButton::ResizeEdit()
{
	// Declare variables
	CRect rcEdit;

	// Calculate edit rectangle
	GetClientRect(rcEdit);
	rcEdit.right = (rcEdit.right - 2) - OT_SPINNER_WIDTH;

	// Set window position
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.SetWindowPos(NULL, rcEdit.left, rcEdit.top, rcEdit.Width(), rcEdit.Height(), SWP_NOZORDER | SWP_SHOWWINDOW);
	}
}

void COptionTreeSpinnerButton::OnMove(int x, int y) 
{
	// Resize edit
	ResizeEdit();

	CWnd::OnMove(x, y);	
  // move buttons
}

void COptionTreeSpinnerButton::OnSize(UINT nType, int cx, int cy) 
{	
	// Resize edit
	ResizeEdit();

	CWnd::OnSize(nType, cx, cy);
	
}

CEdit* COptionTreeSpinnerButton::GetEdit()
{
	// Return pointer
	return &m_ctlEdit;
}

BOOL COptionTreeSpinnerButton::IsStringNumeric(CString strString)
{
	// See if string is numeric or not
	if (strString.FindOneOf("1234567890") == -1 || strString.FindOneOf("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ`~[]{}-_=+\\|'/?>,<") != -1)
	{
		return FALSE;
	}

	return TRUE;
}

double COptionTreeSpinnerButton::_GetValue()
{
	// Declare variables
	double dValue;

	// Get double value
	GetEditDouble(dValue);

	return dValue;
}

LRESULT COptionTreeSpinnerButton::WM_ForceRedraw(WPARAM wParam, LPARAM lParam)
{
	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();

	return 0;
}

LRESULT COptionTreeSpinnerButton::WM_EditUp(WPARAM wParam, LPARAM lParam)
{
	// Update value
	// -- Wrap around
	if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
	{
		if ((_GetValue() + 1) > m_dRangeTop)
		{
			SetEditDouble(m_dRangeBottom);
		}
		else
		{
			SetEditDouble(_GetValue() + 1);
		}

	}
	// -- No wrap around
	else
	{
		if ((_GetValue() + 1) <= m_dRangeTop)
		{
			SetEditDouble(_GetValue() + 1);
		}
	}

	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();

	return 0;
}

LRESULT COptionTreeSpinnerButton::WM_EditDown(WPARAM wParam, LPARAM lParam)
{
	// Update value
	// -- Wrap around
	if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
	{
		if ((_GetValue() - 1) < m_dRangeBottom)
		{
			SetEditDouble(m_dRangeTop);
		}
		else
		{
			SetEditDouble(_GetValue() - 1);
		}
	}
	// -- No wrap around
	else
	{
		if ((_GetValue() - 1) >= m_dRangeBottom)
		{
			SetEditDouble(_GetValue() - 1);
		}
	}

	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();

	return 0;
}

CString COptionTreeSpinnerButton::GetEditText()
{
	// Declare variables
	CString strText = _T("");

	// Get edit text
	if (IsWindow(m_ctlEdit.GetSafeHwnd()))
	{
		m_ctlEdit.GetWindowText(strText);
	}

	return strText;
}

void COptionTreeSpinnerButton::OnTimer(UINT nIDEvent) 
{
	// See what timer event
	if (nIDEvent == OT_TIMER)
	{
		// -- See if first time
		if (m_bFirstRepeat == TRUE)
		{
			KillTimer(OT_TIMER);
			m_bFirstRepeat = FALSE;
			SetTimer(OT_TIMER, m_nRepeatRate, NULL);
		}

		RepeatButton();
	}
	
	CWnd::OnTimer(nIDEvent);
}

void COptionTreeSpinnerButton::RepeatButton()
{
	// Clear pressed
	m_bBottomPressed = FALSE;
	m_bTopPressed = FALSE;

	// See if we have pressed a button
	if (m_rcButtonTop.PtInRect(m_ptSavePoint) == TRUE)
	{
		// -- Mark bool
		m_bTopPressed = TRUE;

		// -- Update value
		// -- -- Wrap around
		if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
		{
			if ((_GetValue() + 1) > m_dRangeTop)
			{
				SetEditDouble(m_dRangeBottom);
			}
			else
			{
				SetEditDouble(_GetValue() + 1);
			}

		}
		// -- -- No wrap around
		else
		{
			if ((_GetValue() + 1) <= m_dRangeTop)
			{
				SetEditDouble(_GetValue() + 1);
			}
		}
	}
	else if (m_rcButtonBottom.PtInRect(m_ptSavePoint) == TRUE)
	{
		// -- Mark bool
		m_bBottomPressed = TRUE;

		// -- Update value
		// -- -- Wrap around
		if (GetOption(OT_EDIT_WRAPAROUND) == TRUE)
		{
			if ((_GetValue() - 1) < m_dRangeBottom)
			{
				SetEditDouble(m_dRangeTop);
			}
			else
			{
				SetEditDouble(_GetValue() - 1);
			}
		}
		// -- -- No wrap around
		else
		{
			if ((_GetValue() - 1) >= m_dRangeBottom)
			{
				SetEditDouble(_GetValue() - 1);
			}
		}
	}

	// Force redraw
	Invalidate();

	// Update window
	UpdateWindow();
}

BOOL COptionTreeSpinnerButton::GetOption(DWORD dwOption)
{
	// Return option
	return (m_dwOptions & dwOption) ? TRUE : FALSE;
}

void COptionTreeSpinnerButton::SetOption(DWORD dwOption, BOOL bSet)
{
	// Set option
	if (bSet == TRUE)
	{
		m_dwOptions |= dwOption;
	}
	else
	{
		m_dwOptions &= ~dwOption;
	}
}

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The Code Project Open License (CPOL)


Written By
CppChinaski
Web Developer
Germany Germany
This member has not yet provided a Biography. Assume it's interesting and varied, and probably something to do with programming.

Comments and Discussions

Permalink
Advertise
Privacy
Cookies
Terms of Use
Layout: fixed | fluid

Posted 27 Mar 2006

Article Copyright 2006 by CppChinaski
Everything else Copyright © CodeProject, 1999-2024

Web02 2.8:2024-04-02:1