基于毫秒的时间控件（C++/MFC）

#include "StdAfx.h" #include "TimeWnd.h" CTimeWnd::CTimeWnd(void) { for (int i=0; i<valCount; i++) { m_nValue[i] = 0; m_Rect[i].SetRectEmpty(); } m_nValueMax[valHours] = 24; m_nValueMax[valMinutes] = 60; m_nValueMax[valSeconds] = 60; m_nValueMax[valMilliseconds] = 1000; m_dPower[valHours] = 3600; m_dPower[valMinutes] = 60; m_dPower[valSeconds] = 1; m_dPower[valMilliseconds] = 0.001; m_nCurSel = -1; m_nOldSel = 0; m_pFont = NULL; m_bFocus = false; } CTimeWnd::~CTimeWnd(void) { } BEGIN_MESSAGE_MAP(CTimeWnd, CWnd) ON_WM_PAINT() ON_WM_CREATE() ON_WM_LBUTTONDOWN() ON_WM_KEYDOWN() ON_NOTIFY(UDN_DELTAPOS, 1051, &CTimeWnd::OnDeltaposSpin1) ON_WM_KILLFOCUS() ON_WM_SIZE() ON_WM_CAPTURECHANGED() ON_WM_ENABLE() END_MESSAGE_MAP() void CTimeWnd::OnDeltaposSpin1(NMHDR *pNMHDR, LRESULT *pResult) { LPNMUPDOWN pNMUpDown = reinterpret_cast<LPNMUPDOWN>(pNMHDR); *pResult = 0; SetFocus(); ChangeSpin(pNMUpDown->iDelta*(-1)); } void CTimeWnd::ChangeSpin(int nDelta) { if(m_nCurSel == -1) { m_nCurSel = m_nOldSel; } if (nDelta > 0) { m_nValue[m_nCurSel] = (m_nValue[m_nCurSel]+nDelta)%m_nValueMax[m_nCurSel]; } else { m_nValue[m_nCurSel] = (m_nValue[m_nCurSel]+nDelta+m_nValueMax[m_nCurSel])%m_nValueMax[m_nCurSel]; } ReDraw(true); } void CTimeWnd::OnPaint() { CPaintDC dc(this); // device context for painting CRect rcBounds; GetClientRect(&rcBounds); CPen penWhite; penWhite.CreatePen(PS_SOLID, 1, RGB(127,157,185)); CPen *pOldPen = dc.SelectObject(&penWhite); dc.FillSolidRect(rcBounds, RGB(255,255,255)); dc.MoveTo(rcBounds.left, rcBounds.top); dc.LineTo(rcBounds.right-1, rcBounds.top); dc.LineTo(rcBounds.right-1, rcBounds.bottom-1); dc.LineTo(rcBounds.left, rcBounds.bottom-1); dc.LineTo(rcBounds.left, rcBounds.top); CString str = NULL; int xoffset = 5; CRect rc; rc.SetRectEmpty(); for(int i=0; i<valCount; i++) { str = GetString(i); rc = rcBounds; DrawSingleString(&dc,str,rc,&xoffset,(i == m_nCurSel)); m_Rect[i] = rc; if (i != valCount-1) { rc = rcBounds; str = _T(":"); DrawSingleString(&dc, str, rc, &xoffset); } } dc.SelectObject(pOldPen); } void CTimeWnd::PreSubclassWindow() { CWnd::PreSubclassWindow(); } int CTimeWnd::OnCreate(LPCREATESTRUCT lpCreateStruct) { if (CWnd::OnCreate(lpCreateStruct) == -1) return -1; m_Spin.Create(UDS_ALIGNRIGHT|UDS_AUTOBUDDY|WS_VISIBLE, CRect(0,0,20,19), this, 1051); m_Spin.SetBuddy(this); return 0; } void CTimeWnd::OnLButtonDown(UINT nFlags, CPoint point) { for (int i=0; i<valCount; i++) { if (m_Rect[i].PtInRect(point)) { if (m_nCurSel == i)break; ReDraw(false); m_nCurSel = i; ReDraw(true); break; } } SetFocus(); m_bFocus = true; CWnd::OnLButtonDown(nFlags, point); } void CTimeWnd::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags) { if (m_nCurSel == -1) { m_nCurSel = m_nOldSel; } if ((nChar>='0') && (nChar<='9')) { ChangeChar(nChar, '0'); } else if ((nChar>=VK_NUMPAD0) && (nChar<=VK_NUMPAD9)) { ChangeChar(nChar, VK_NUMPAD0); } ReDraw(true); CWnd::OnKeyDown(nChar, nRepCnt, nFlags); } CString CTimeWnd::GetString(int nType) { CString str = NULL; int nCount = 0; int nZero = 0; nCount = GetNumberCount(m_nValueMax[nType]-1); nZero = nCount - GetNumberCount(m_nValue[nType]) ; m_nValue[nType] %= m_nValueMax[nType]; str.Format(_T("%d"), m_nValue[nType]); while (nZero--) { str = _T("0") + str; } return str; } void CTimeWnd::DrawSingleString(CDC *pDC, CString &str, CRect &rc, int *pOffset/*= -1*/, bool bSelect/* = false*/) { int nHeight = rc.Height(); int offset = 0; if (pOffset == NULL) { offset = rc.left; } else { offset = *pOffset; } CFont *pOldFont = NULL; if (m_pFont) { pOldFont = pDC->SelectObject(m_pFont); } pDC->DrawText(str, rc, DT_CALCRECT); rc.MoveToXY(offset, (nHeight - rc.Height())/2); COLORREF clr; int nMode = 0; BOOL bEnable = IsWindowEnabled(); if (bSelect && bEnable) { pDC->FillSolidRect(rc, RGB(49,106,197)); clr = pDC->SetTextColor(RGB(255,255,255)); nMode = pDC->SetBkMode(TRANSPARENT); } else { pDC->FillSolidRect(rc, RGB(255,255,255)); } if (bEnable) pDC->DrawState(CPoint(rc.left, rc.top), CSize(rc.Width(), rc.Height()), str, DSS_NORMAL, TRUE, 0, (HBRUSH)NULL); else pDC->DrawState(CPoint(rc.left, rc.top), CSize(rc.Width(), rc.Height()), str, DSS_DISABLED, TRUE, 0, (HBRUSH)NULL); offset += rc.Width(); if (bSelect) { pDC->SetTextColor(clr); pDC->SetBkMode(nMode); } if (pOffset) *pOffset = offset; if (pOldFont) pDC->SelectObject(pOldFont); } void CTimeWnd::OnKillFocus(CWnd *pNewWnd) { CWnd::OnKillFocus(pNewWnd); ReDraw(false); m_nOldSel = m_nCurSel; m_nCurSel = -1; m_bFocus = false; } void CTimeWnd::ReDraw(bool bNeedBk/*= true*/) { if (m_nCurSel == -1) { return; } CDC *pDC = GetDC(); CString str = GetString(m_nCurSel); CRect rcBounds; GetClientRect(&rcBounds); rcBounds.left = m_Rect[m_nCurSel].left; rcBounds.right = m_Rect[m_nCurSel].right; DrawSingleString(pDC, str, rcBounds, NULL, bNeedBk); ReleaseDC(pDC); } void CTimeWnd::SetFont(CFont *pFont, BOOL bRedraw/*=false*/) { m_pFont = pFont; if (bRedraw) { Invalidate(); } } CFont *CTimeWnd::GetFont() { return m_pFont; } void CTimeWnd::ChangeChar(UINT nChar, int refer) { if (!IsCountValid(m_nCurSel)) return; int iCharValue = nChar - refer; int iTemVal = m_nValue[m_nCurSel] * 10 + iCharValue; m_nValue[m_nCurSel] = (iTemVal >= m_nValueMax[m_nCurSel]) ? iCharValue : iTemVal; } BOOL CTimeWnd::PreTranslateMessage(MSG *pMsg) { if (pMsg->message == WM_KEYDOWN) { if (pMsg->wParam == VK_UP) { ChangeSpin(1); return true; } if (pMsg->wParam == VK_DOWN) { ChangeSpin(-1); return true; } if (pMsg->wParam == VK_LEFT) { ReDraw(false); m_nCurSel = (m_nCurSel + valCount - 1) % valCount; ReDraw(true); return true; } if (pMsg->wParam == VK_RIGHT) { ReDraw(false); m_nCurSel = (m_nCurSel + 1) % valCount; ReDraw(true); return true; } if (pMsg->wParam == VK_END) { m_nValue[m_nCurSel] = m_nValueMax[m_nCurSel]; } if (pMsg->wParam == VK_HOME) { m_nValue[m_nCurSel] = 0; } } return CWnd::PreTranslateMessage(pMsg); } double CTimeWnd::GetTime() { double dTime = 0; for (int i=0; i<valCount; i++) { dTime += m_nValue[i] * m_dPower[i]; } return dTime; } int CTimeWnd::GetValue(int nType) { if (!IsCountValid(nType)) return 0; else return m_nValue[nType]; } void CTimeWnd::SetValue(int nType, int nVal) { if ((!IsCountValid(nType)) || nVal<0) return; else m_nValue[nType] = nVal % m_nValueMax[nType]; } void CTimeWnd::SetTime(double dTime) { int nTemVal = 0; for (int i=0; i<valCount; i++) { m_nValue[i] = (int)((dTime - nTemVal) / m_dPower[i]); nTemVal += m_nValue[i]; m_nValue[i] %= m_nValueMax[i]; } } bool CTimeWnd::IsFocus() { return m_bFocus; } void CTimeWnd::OnSize(UINT nType, int cx, int cy) { CWnd::OnSize(nType, cx, cy); m_Spin.MoveWindow(cx-19, 1, 18, cy-2); } void CTimeWnd::OnCaptureChanged(CWnd *pWnd) { CWnd::OnCaptureChanged(pWnd); } void CTimeWnd::OnEnable(BOOL bEnable) { CWnd::OnEnable(bEnable); Invalidate(); m_Spin.EnableWindow(bEnable); } int CTimeWnd::GetNumberCount(int num) { int i = 0; if (num == 0) { return 1; } while (num) { num /= 10; i++; }; return i; } bool CTimeWnd::IsCountValid(int nCount) { if ((nCount>=0) && (nCount<valCount)) { return true; } return false; }