added code so that the Translate() function actually Translates. It now translates to/from plain and styled text. It passes most of the Translate tests, but Be's STXTTranslator includes font information with some of the styled text files it generates, and I haven't decided how to deal with that yet.

git-svn-id: file:///srv/svn/repos/haiku/trunk/current@2294 a95241bf-73f2-0310-859d-f6bbb57e9c96
This commit is contained in:
Matthew Wilber 2002-12-25 03:18:52 +00:00
parent 9c721dde42
commit 6eed52aa01

View File

@ -37,6 +37,8 @@
#define min(x,y) ((x < y) ? x : y)
#define max(x,y) ((x > y) ? x : y)
#define READ_BUFFER_SIZE 512
// The input formats that this translator supports.
translation_format gInputFormats[] = {
{
@ -257,7 +259,8 @@ STXTTranslator::OutputFormats(int32 *out_count) const
status_t
identify_stxt_header(const TranslatorStyledTextStreamHeader &header,
BPositionIO *inSource, translator_info *outInfo, uint32 outType)
BPositionIO *inSource, translator_info *outInfo, uint32 outType,
TranslatorStyledTextTextHeader *ptxtheader = NULL)
{
char buffer[20];
@ -286,6 +289,8 @@ identify_stxt_header(const TranslatorStyledTextStreamHeader &header,
SEEK_CUR);
if (seekresult < pos)
return B_NO_TRANSLATOR;
if (seekresult > pos)
return B_ERROR;
// check the STYL header (not all STXT files have this)
ssize_t read = 0;
@ -307,6 +312,14 @@ identify_stxt_header(const TranslatorStyledTextStreamHeader &header,
return B_NO_TRANSLATOR;
}
// if output TEXT header is supplied, fill it with data
if (ptxtheader) {
ptxtheader->header.magic = txtheader.header.magic;
ptxtheader->header.header_size = txtheader.header.header_size;
ptxtheader->header.data_size = txtheader.header.data_size;
ptxtheader->charset = txtheader.charset;
}
// return information about the data in the stream
outInfo->type = B_STYLED_TEXT_FORMAT;
outInfo->group = B_TRANSLATOR_TEXT;
@ -431,6 +444,155 @@ STXTTranslator::Identify(BPositionIO *inSource,
return identify_txt_header(buffer, nread, inSource, outInfo, outType);
}
status_t
translate_from_stxt(BPositionIO *inSource, BPositionIO *outDestination,
uint32 outType, const TranslatorStyledTextTextHeader &txtheader)
{
if (inSource->Seek(0, SEEK_SET) != 0)
return B_ERROR;
bool btoplain;
if (outType == B_TRANSLATOR_TEXT)
btoplain = true;
else if (outType == B_STYLED_TEXT_FORMAT)
btoplain = false;
else
return B_BAD_VALUE;
uint8 buffer[READ_BUFFER_SIZE];
ssize_t nread = 0, nwritten = 0, nreed = 0, ntotalread = 0;
// skip to the actual text data when outputting a
// plain text file
if (btoplain) {
if (inSource->Seek(32, SEEK_CUR) != 32)
return B_ERROR;
}
// Read data from inSource
// When outputing B_TRANSLATOR_TEXT, the loop stops when all of
// the text data has been read and written.
// When outputting B_STYLED_TEXT_FORMAT, the loop stops when all
// of the data from inSource has been read and written.
if (btoplain)
nreed = min(READ_BUFFER_SIZE,
txtheader.header.data_size - ntotalread);
else
nreed = READ_BUFFER_SIZE;
nread = inSource->Read(buffer, nreed);
while (nread > 0) {
nwritten = outDestination->Write(buffer, nread);
if (nwritten != nread)
return B_ERROR;
if (btoplain) {
ntotalread += nread;
nreed = min(READ_BUFFER_SIZE,
txtheader.header.data_size - ntotalread);
} else
nreed = READ_BUFFER_SIZE;
nread = inSource->Read(buffer, nreed);
}
if (btoplain && static_cast<ssize_t>(txtheader.header.data_size) !=
ntotalread)
// If not all of the text data was able to be read...
return B_NO_TRANSLATOR;
else
return B_OK;
}
// outputs the Stream and Text headers from the B_STYLED_TEXT_FORMAT
// to outDestination, setting the data_size member of the text header
// to text_data_size
status_t
output_headers(BPositionIO *outDestination, uint32 text_data_size)
{
const int32 kHeadersSize = sizeof(TranslatorStyledTextStreamHeader) +
sizeof(TranslatorStyledTextTextHeader);
status_t result;
TranslatorStyledTextStreamHeader stxtheader;
TranslatorStyledTextTextHeader txtheader;
uint8 buffer[kHeadersSize];
stxtheader.header.magic = 'STXT';
stxtheader.header.header_size = sizeof(TranslatorStyledTextStreamHeader);
stxtheader.header.data_size = 0;
stxtheader.version = 100;
memcpy(buffer, &stxtheader, stxtheader.header.header_size);
txtheader.header.magic = 'TEXT';
txtheader.header.header_size = sizeof(TranslatorStyledTextTextHeader);
txtheader.header.data_size = text_data_size;
txtheader.charset = B_UNICODE_UTF8;
memcpy(buffer + stxtheader.header.header_size, &txtheader,
txtheader.header.header_size);
// write out headers in Big Endian byte order
result = swap_data(B_UINT32_TYPE, buffer, kHeadersSize,
B_SWAP_HOST_TO_BENDIAN);
if (result == B_OK) {
ssize_t nwritten = 0;
nwritten = outDestination->Write(buffer, kHeadersSize);
if (nwritten != kHeadersSize)
return B_ERROR;
else
return B_OK;
}
return result;
}
// convert the plain text from inSource to plain or styled text
// in outDestination
status_t
translate_from_text(BPositionIO *inSource, BPositionIO *outDestination,
uint32 outType)
{
// find the length of the text
off_t size = 0;
size = inSource->Seek(0, SEEK_END);
if (size < 0)
return B_ERROR;
if (inSource->Seek(0, SEEK_SET) != 0)
return B_ERROR;
bool btoplain;
if (outType == B_TRANSLATOR_TEXT)
btoplain = true;
else if (outType == B_STYLED_TEXT_FORMAT)
btoplain = false;
else
return B_BAD_VALUE;
// output styled text headers if outputting
// in the B_STYLED_TEXT_FORMAT
if (!btoplain) {
status_t headresult;
headresult = output_headers(outDestination,
static_cast<uint32>(size));
if (headresult != B_OK)
return headresult;
}
uint8 buffer[READ_BUFFER_SIZE];
ssize_t nread = 0, nwritten = 0;
// output the actual text part of the data
nread = inSource->Read(buffer, READ_BUFFER_SIZE);
while (nread > 0) {
nwritten = outDestination->Write(buffer, nread);
if (nwritten != nread)
return B_ERROR;
nread = inSource->Read(buffer, READ_BUFFER_SIZE);
}
return B_OK;
}
// ---------------------------------------------------------------
// Translate
//
@ -472,7 +634,42 @@ STXTTranslator::Translate(BPositionIO *inSource,
if (outType != B_TRANSLATOR_TEXT && outType != B_STYLED_TEXT_FORMAT)
return B_NO_TRANSLATOR;
return B_NO_TRANSLATOR;
uint8 buffer[64];
status_t nread = 0, result;
translator_info outInfo;
// Read in the header to determine
// if the data is supported
nread = inSource->Read(buffer, 16);
if (nread < 0)
return nread;
// read in enough data to fill the stream header
if (nread == 16) {
TranslatorStyledTextStreamHeader header;
memcpy(&header, buffer, 16);
if (swap_data(B_UINT32_TYPE, &header, 16, B_SWAP_BENDIAN_TO_HOST)
!= B_OK)
return B_ERROR;
if (header.header.magic == B_STYLED_TEXT_FORMAT &&
header.header.header_size ==
sizeof(TranslatorStyledTextStreamHeader) &&
header.header.data_size == 0 &&
header.version == 100) {
TranslatorStyledTextTextHeader txtheader;
result = identify_stxt_header(header, inSource, &outInfo, outType,
&txtheader);
return translate_from_stxt(inSource, outDestination, outType,
txtheader);
}
}
// if the data is not styled text, check if it is ASCII text
result = identify_txt_header(buffer, nread, inSource, &outInfo, outType);
if (result == B_OK)
return translate_from_text(inSource, outDestination, outType);
else
return result;
}
// ---------------------------------------------------------------