#include "pch.h"
#include "TraceService.h"
#include <locale.h>
#include <cstdio>
#pragma warning(disable:4996)
///////////////////////////////////////////////////////////////////////////
//颜色定义
#define COLOR_TIME RGB(0,0,0)
#define COLOR_NORMAL RGB(125,125,125)
#define COLOR_WARN RGB(255,128,0)
#define COLOR_EXCEPTION RGB(200,0,0)
#define COLOR_DEBUG RGB(0,128,128)
#define EVENT_LEVEL_COUNT 5
///////////////////////////////////////////////////////////////////////////
//追踪服务
///////////////////////////////////////////////////////////////////////////
static ITraceService* g_pITraceService = NULL;
//函数定义
///////////////////////////////////////////////////////////////////////////
//构造函数
CTraceService::CTraceService()
{
}
//服务配置
///////////////////////////////////////////////////////////////////////////
//设置服务
bool CTraceService::SetTraceService(ITraceService * pIUnknownEx)
{
g_pITraceService = pIUnknownEx;
return NULL != g_pITraceService;
}
//获取服务
VOID* CTraceService::GetTraceServiceManager()
{
ASSERT(g_pITraceService);
if (g_pITraceService) return NULL;
return (VOID*)g_pITraceService;
}
//追踪控制
bool CTraceService::EnableTrace(enTraceLevel TraceLevel, bool bEnableTrace)
{
return false;
}
//功能函数
///////////////////////////////////////////////////////////////////////////
//追踪信息
bool CTraceService::TraceString(LPCTSTR pszString, enTraceLevel TraceLevel)
{
if ((g_pITraceService))
return g_pITraceService->TraceString(pszString, TraceLevel);
}
bool CTraceService::TraceStringEx(enTraceLevel TraceLevel, LPCTSTR pszFormat, ...)
{
TCHAR szBuffer[1024] = { 0 };
va_list arglist;
va_start(arglist, pszFormat);
_vstprintf_s(szBuffer, CountArray(szBuffer), pszFormat, arglist);
va_end(arglist);
if (g_pITraceService)
return g_pITraceService->TraceString(szBuffer, TraceLevel);
return false;
}
///////////////////////////////////////////////////////////////////////////
//追踪服务
///////////////////////////////////////////////////////////////////////////
BEGIN_MESSAGE_MAP(CTraceServiceControl, CRichEditCtrl)
ON_WM_RBUTTONDOWN()
ON_COMMAND(IDM_MENU0, &CTraceServiceControl::OnCopyString)
ON_COMMAND(IDM_MENU1, &CTraceServiceControl::OnSelectAll)
ON_COMMAND(IDM_MENU2, &CTraceServiceControl::OnDeleteString)
ON_COMMAND(IDM_MENU3, &CTraceServiceControl::OnClearAll)
ON_COMMAND(IDM_MENU4, &CTraceServiceControl::OnSaveString)
END_MESSAGE_MAP()
//函数定义
///////////////////////////////////////////////////////////////////////////
//构造函数
CTraceServiceControl::CTraceServiceControl()
{
//CTraceService::SetTraceService(this);
}
//析构函数
CTraceServiceControl::~CTraceServiceControl()
{
CTraceService::SetTraceService(NULL);
}
//基础接口
///////////////////////////////////////////////////////////////////////////
//接口查询
VOID* CTraceServiceControl::QueryInterface()
{
//QueryInterface(ITraceService, Guid, dwQueryVer);
//QueryInterface_IUNKNOWNEX(ITraceService, Guid, dwQueryVer);
return this;
}
//信息接口
///////////////////////////////////////////////////////////////////////////
//追踪信息
bool CTraceServiceControl::TraceString(LPCTSTR pszString, enTraceLevel TraceLevel)
{
//变量定义
CHARFORMAT2 CharFormat;
ZeroMemory(&CharFormat, sizeof(CharFormat));
//构造数据
CharFormat.cbSize = sizeof(CharFormat);
CharFormat.dwMask = CFM_COLOR | CFM_BACKCOLOR;
CharFormat.crTextColor = RGB(0, 0, 0);
CharFormat.crBackColor = RGB(255, 255, 255);
lstrcpyn(CharFormat.szFaceName, TEXT("宋体"), sizeof(CharFormat.szFaceName));
//获取时间
SYSTEMTIME SystemTime;
TCHAR szTimeBuffer[1024 * 2] = { 0 };
GetLocalTime(&SystemTime);
_sntprintf(szTimeBuffer,sizeof(szTimeBuffer),TEXT("【 %04d-%02d-%02d %02d:%02d:%02d 】"), SystemTime.wYear, SystemTime.wMonth, SystemTime.wDay, SystemTime.wHour, SystemTime.wMinute, SystemTime.wSecond);
_sntprintf(szTimeBuffer, sizeof(szTimeBuffer), TEXT("%s%s\n"), szTimeBuffer, pszString);
CharFormat.crTextColor = GetTraceColor(TraceLevel);
return InsertString(szTimeBuffer, CharFormat);
}
//重载函数
///////////////////////////////////////////////////////////////////////////
//绑定函数
VOID CTraceServiceControl::PreSubclassWindow()
{
__super::PreSubclassWindow();
// 在这里设置服务，此时控件已完全创建
CTraceService::SetTraceService(this);
// 测试输出
//CTraceService::TraceString(_T("CTraceServiceControl 已初始化\n"), TraceLevel_Normal);
}
//功能函数
///////////////////////////////////////////////////////////////////////////
//加载消息
bool CTraceServiceControl::LoadMessage(LPCTSTR pszFileName)
{
if (static_cast<CTraceServiceControl*>(g_pITraceService) == NULL) return false;
return static_cast<CTraceServiceControl*>(g_pITraceService)->LoadMessage(pszFileName);
}
//保存信息
bool CTraceServiceControl::SaveMessage(LPCTSTR pszFileName)
{
if (static_cast<CTraceServiceControl*>(g_pITraceService) == NULL) return false;
return static_cast<CTraceServiceControl*>(g_pITraceService)->SaveMessage(pszFileName);
}
//设置参数
bool CTraceServiceControl::SetParameter(LONG lMaxLineCount, LONG lReserveLineCount)
{
m_lMaxLineCount = lMaxLineCount; //最大行数
m_lReserveLineCount = lReserveLineCount; //保留函数
return true;
}
//辅助函数
///////////////////////////////////////////////////////////////////////////
//配置服务
VOID CTraceServiceControl::InitializeService()
{
}
//获取颜色
COLORREF CTraceServiceControl::GetTraceColor(enTraceLevel TraceLevel)
{
switch (TraceLevel)
{
case TraceLevel_Info:
return RGB(133, 124, 129);
break;
case TraceLevel_Normal:
return RGB(133, 124, 129);
break;
case TraceLevel_Warning:
return RGB(255, 0, 0);
break;
case TraceLevel_Exception:
return RGB(255, 0, 0);
break;
case TraceLevel_Debug:
return RGB(19, 127, 140);
break;
}
return RGB(0, 0, 0);
}
//字符判断
bool EFFicacyUrlChar(TCHAR chChar)
{
//特殊字符
if (chChar == TEXT('.')) return true;
if (chChar == TEXT('=')) return true;
if (chChar == TEXT('+')) return true;
if (chChar == TEXT('?')) return true;
if (chChar == TEXT('#')) return true;
if (chChar == TEXT('%')) return true;
if (chChar == TEXT('/')) return true;
if (chChar == TEXT(':')) return true;
if (chChar == TEXT('&')) return true;
//字符范围
if ((chChar >= TEXT('a')) && (chChar <= TEXT('z'))) return true;
if ((chChar >= TEXT('A')) && (chChar <= TEXT('Z'))) return true;
if ((chChar >= TEXT('0')) && (chChar <= TEXT('9'))) return true;
return false;
}
//地址判断
bool EFFicacyUrlString(LPCTSTR pszUrl)
{
for (WORD i = 0; i < lstrlen(pszUrl); i++)
{
if(!EFFicacyUrlChar(pszUrl[i])) return false;
}
return true;
}
//插入字串
bool CTraceServiceControl::InsertString(LPCTSTR pszString, CHARFORMAT2& CharFormat)
{
//保存状态
if (GetTextLength() >= 1024 * 512)
OnClearAll();
//插入消息
//变量定义
bool bResumeSelect;
CHARRANGE CharRange;
//保存状态
SetSel(-1L, -1L);
GetSel(CharRange.cpMin, CharRange.cpMax);
bResumeSelect = (CharRange.cpMax != CharRange.cpMin);
//搜索变量
LPCTSTR pszHttp = TEXT("http://");
const INT nHttpLength = lstrlen(pszHttp);
const INT nStringLength = lstrlen(pszString);
//索引定义
INT nStringStart = 0;
INT nStringPause = 0;
//字符解释
for (INT i = 0; i < nStringLength; i++)
{
//变量定义
INT nUrlPause = i;
//地址判断
if (((i + nHttpLength) < nStringLength) && (memcmp(&pszString[i], pszHttp, nHttpLength * sizeof(TCHAR)) == 0))
{
//设置索引
nUrlPause = i + nHttpLength;
//地址搜索
while (nUrlPause < nStringLength)
{
//字符判断
if (EFFicacyUrlChar(pszString[nUrlPause]) == true)
{
nUrlPause++;
continue;
}
break;
}
}
//终止字符
if (nUrlPause <= (i + nHttpLength)) nStringPause = (i + 1);
//插入字符
if ((i == (nStringLength - 1)) || (nUrlPause > (i + nHttpLength)))
{
//普通字符
if (nStringPause > nStringStart)
{
//获取缓冲
CString strNormalString;
LPTSTR pszNormalString = strNormalString.GetBuffer(nStringPause - nStringStart + 1);
//拷贝字符
pszNormalString[nStringPause - nStringStart] = 0;
CopyMemory(pszNormalString, &pszString[nStringStart], (nStringPause - nStringStart) * sizeof(TCHAR));
//释放缓冲
strNormalString.ReleaseBuffer();
//插入消息
SetSel(-1L, -1L);
SetWordCharFormat(CharFormat);
ReplaceSel((LPCTSTR)strNormalString);
}
//设置索引
nStringStart = __max(i, nUrlPause);
}
//设置索引
i += (nUrlPause - i);
}
//状态设置
if (bResumeSelect == true) SetSel(CharRange);
else PostMessage(WM_VSCROLL, SB_BOTTOM, 0);
return true;
}
//回调函数
///////////////////////////////////////////////////////////////////////////
//加载回调
DWORD CALLBACK CTraceServiceControl::LoadCallBack(DWORD dwCookie, LPBYTE pbBuff, LONG cb, LONG* pcb)
{
return 0;
}
//保存回调
DWORD CALLBACK CTraceServiceControl::SaveCallBack(DWORD dwCookie, LPBYTE pbBuff, LONG cb, LONG* pcb)
{
return 0;
}
//菜单命令
///////////////////////////////////////////////////////////////////////////
//删除消息
VOID CTraceServiceControl::OnClearAll()
{
SetWindowText(NULL);
}
//全部选择
VOID CTraceServiceControl::OnSelectAll()
{
__super::SetSel(0, __super::GetTextLength());
}
//拷贝字符
VOID CTraceServiceControl::OnCopyString()
{
__super::Copy();
}
//保存信息
VOID CTraceServiceControl::OnSaveString()
{
char* old_local = _strdup(setlocale(LC_CTYPE, NULL));
setlocale(LC_CTYPE, "chs");
CFileDialog dlg(FALSE, TEXT("*.txt"), NULL, 4 | 2, TEXT("信息|*.txt||"));
if (dlg.DoModal() == IDOK)
{
CString m_filename = dlg.GetPathName();
CString strInput;
GetWindowText(strInput);
CFile file(m_filename, CFile::modeCreate | CFile::modeWrite);
#ifdef UNICODE
WCHAR wFlag = 0xFEFF;
file.Write(&wFlag, sizeof(WCHAR));
#endif //UNICODE
file.Write(strInput, strInput.GetLength());
file.Close();
}
setlocale(LC_CTYPE, old_local);
free(old_local);
}
//删除字符
VOID CTraceServiceControl::OnDeleteString()
{
ReplaceSel(NULL, true);
}
//消息映射
///////////////////////////////////////////////////////////////////////////
//建立消息
INT CTraceServiceControl::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
/*return __super::OnCreate(lpCreateStruct);*/
if (__super::OnCreate(lpCreateStruct) == -1)
return -1;
CTraceService::SetTraceService(this);
return 0;
}
//右键消息
void CTraceServiceControl::OnRButtonDown(UINT nFlags, CPoint point)
{
CMenu menu;
// 创建弹出菜单
if (menu.CreatePopupMenu())
{
ClientToScreen(&point);
CHARRANGE sel;
GetSel(sel);
int nTextLength = GetTextLength();
// 启用删除菜单 - 当有选中文本时可以删除
BOOL bCanDelete = (sel.cpMax != sel.cpMin); // 有选中文本才能删除
menu.AppendMenu(MF_STRING | ((sel.cpMax != sel.cpMin) ? 0 : (MF_DISABLED | MF_GRAYED)),
IDM_MENU0, TEXT("复制(&C)\tCtrl+C"));
menu.AppendMenu(MF_STRING | ((nTextLength > 0 && (sel.cpMax - sel.cpMin) < nTextLength) ? 0 : (MF_DISABLED | MF_GRAYED)),
IDM_MENU1, TEXT("全选(&A)\tCtrl+A"));
// 启用删除菜单（当有选中文本时）
menu.AppendMenu(MF_STRING | (bCanDelete ? 0 : (MF_DISABLED | MF_GRAYED)),
IDM_MENU2, TEXT("删除(&D)\tDel"));
menu.AppendMenu(MF_STRING | ((nTextLength > 0) ? 0 : (MF_DISABLED | MF_GRAYED)),
IDM_MENU3, TEXT("清除信息"));
menu.AppendMenu(MF_SEPARATOR);
menu.AppendMenu(MF_STRING | ((nTextLength > 0) ? 0 : (MF_DISABLED | MF_GRAYED)),
IDM_MENU4, TEXT("保存信息"));
// 显示菜单
menu.TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, this);
}
// 调用基类处理
__super::OnRButtonDown(nFlags, point);
}
//追踪消息
LRESULT CTraceServiceControl::OnTraceServiceMessage(WPARAM wParam, LPARAM lParam)
{
return 0;
}
最新发布
本文详细介绍了如何通过子类化窗口来实现对已有窗口的定制功能，如输入检测等。文章解释了子类化的基本原理，并提供了具体实现步骤。
扫一扫
838
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
