使用EM_STREAMIN和EM_STREAMOUT在VC++中处理RichEdit控件的RTF内容
在VC++开发中,若需对RichEdit控件中的富文本格式(RTF)内容进行保存或加载,可通过发送EM_STREAMIN和EM_STREAMOUT消息实现。这两个消息依赖于EDITSTREAM结构体和用户定义的回调函数,实现数据流的输入与输出。
对于从RichEdit控件提取RTF格式文本的操作,推荐使用std::string作为接收容器。由于系统可能多次调用回调函数分段传输数据,使用string可方便地通过append方法动态拼接内容,避免手动管理内存分配和指针偏移带来的复杂性。
DWORD CALLBACK CXTREdit::StreamOutCallback(DWORD_PTR dwCookie, LPBYTE pbBuffer, LONG cbSize, LONG* pcbWritten)
{
std::string* pOutput = reinterpret_cast<std::string*>(dwCookie);
char* dataChunk = reinterpret_cast<char*>(pbBuffer);
// 获取实际写入长度(通常为cbSize,但以null结尾字符串为准)
LONG actualLength = 0;
for (LONG i = 0; i < cbSize && dataChunk[i] != '\0'; ++i)
++actualLength;
pOutput->append(dataChunk, actualLength);
*pcbWritten = actualLength;
return 0;
}
int CXTREdit::GetRichTextRtf(std::string& rtfData)
{
EDITSTREAM stream = { 0 };
stream.dwCookie = reinterpret_cast<DWORD_PTR>(&rtfData);
stream.pfnCallback = StreamOutCallback;
const int formatFlags = SF_RTF;
SendMessage(m_hWndREdit, EM_STREAMOUT, formatFlags, reinterpret_cast<LPARAM>(&stream));
return static_cast<int>(rtfData.length());
}
当向RichEdit控件写入RTF内容时,应传入指向原始字符串的指针值,并在回调函数中使用双重指针机制跟踪读取进度。这样既保证了原始指针变量不被修改,又能在分段处理过程中正确推进数据位置。
DWORD CALLBACK CXTREdit::StreamInCallback(DWORD_PTR dwCookie, LPBYTE pbBuffer, LONG cbBuffer, LONG* pcbRead)
{
char** ppSource = reinterpret_cast<char**>(dwCookie);
const LONG remaining = static_cast<LONG>(strlen(*ppSource));
const LONG bytesToRead = (remaining < cbBuffer) ? remaining : cbBuffer;
if (bytesToRead > 0)
{
memcpy(pbBuffer, *ppSource, bytesToRead);
*ppSource += bytesToRead; // 移动源指针
}
*pcbRead = bytesToRead;
return 0;
}
int CXTREdit::SetRichTextRtf(const char* rtfContent)
{
// 复制原始指针用于内部移动,不影响外部传入参数
char* sourcePtr = const_cast<char*>(rtfContent);
EDITSTREAM stream = { 0 };
stream.dwCookie = reinterpret_cast<DWORD_PTR>(&sourcePtr);
stream.pfnCallback = StreamInCallback;
const int formatFlags = SF_RTF | SFF_PLAINRTF; // 可选:忽略默认字符格式
SendMessage(m_hWndREdit, EM_STREAMIN, formatFlags, reinterpret_cast<LPARAM>(&stream));
return static_cast<int>(strlen(rtfContent));
}