| Index: icu46/source/test/intltest/textfile.cpp
|
| ===================================================================
|
| --- icu46/source/test/intltest/textfile.cpp (revision 0)
|
| +++ icu46/source/test/intltest/textfile.cpp (revision 0)
|
| @@ -0,0 +1,179 @@
|
| +/*
|
| +**********************************************************************
|
| +* Copyright (c) 2004, International Business Machines
|
| +* Corporation and others. All Rights Reserved.
|
| +**********************************************************************
|
| +* Author: Alan Liu
|
| +* Created: March 19 2004
|
| +* Since: ICU 3.0
|
| +**********************************************************************
|
| +*/
|
| +#include "textfile.h"
|
| +#include "cmemory.h"
|
| +#include "cstring.h"
|
| +#include "intltest.h"
|
| +#include "util.h"
|
| +
|
| +// If the symbol CCP is defined, then the 'name' and 'encoding'
|
| +// constructor parameters are copied. Otherwise they are aliased.
|
| +// #define CCP
|
| +
|
| +TextFile::TextFile(const char* _name, const char* _encoding, UErrorCode& ec) :
|
| + file(0),
|
| + name(0), encoding(0),
|
| + buffer(0),
|
| + capacity(0),
|
| + lineNo(0)
|
| +{
|
| + if (U_FAILURE(ec) || _name == 0 || _encoding == 0) {
|
| + if (U_SUCCESS(ec)) {
|
| + ec = U_ILLEGAL_ARGUMENT_ERROR;
|
| + }
|
| + return;
|
| + }
|
| +
|
| +#ifdef CCP
|
| + name = uprv_malloc(uprv_strlen(_name) + 1);
|
| + encoding = uprv_malloc(uprv_strlen(_encoding) + 1);
|
| + if (name == 0 || encoding == 0) {
|
| + ec = U_MEMORY_ALLOCATION_ERROR;
|
| + return;
|
| + }
|
| + uprv_strcpy(name, _name);
|
| + uprv_strcpy(encoding, _encoding);
|
| +#else
|
| + name = (char*) _name;
|
| + encoding = (char*) _encoding;
|
| +#endif
|
| +
|
| + const char* testDir = IntlTest::getSourceTestData(ec);
|
| + if (U_FAILURE(ec)) {
|
| + return;
|
| + }
|
| + if (!ensureCapacity((int32_t)(uprv_strlen(testDir) + uprv_strlen(name) + 1))) {
|
| + ec = U_MEMORY_ALLOCATION_ERROR;
|
| + return;
|
| + }
|
| + uprv_strcpy(buffer, testDir);
|
| + uprv_strcat(buffer, name);
|
| +
|
| + file = T_FileStream_open(buffer, "rb");
|
| + if (file == 0) {
|
| + ec = U_ILLEGAL_ARGUMENT_ERROR;
|
| + return;
|
| + }
|
| +}
|
| +
|
| +TextFile::~TextFile() {
|
| + if (file != 0) T_FileStream_close(file);
|
| + if (buffer != 0) uprv_free(buffer);
|
| +#ifdef CCP
|
| + uprv_free(name);
|
| + uprv_free(encoding);
|
| +#endif
|
| +}
|
| +
|
| +UBool TextFile::readLine(UnicodeString& line, UErrorCode& ec) {
|
| + if (T_FileStream_eof(file)) {
|
| + return FALSE;
|
| + }
|
| + // Note: 'buffer' may change after ensureCapacity() is called,
|
| + // so don't use
|
| + // p=buffer; *p++=c;
|
| + // but rather
|
| + // i=; buffer[i++]=c;
|
| + int32_t n = 0;
|
| + for (;;) {
|
| + int c = T_FileStream_getc(file); // sic: int, not int32_t
|
| + if (c < 0 || c == 0xD || c == 0xA) {
|
| + // consume 0xA following 0xD
|
| + if (c == 0xD) {
|
| + c = T_FileStream_getc(file);
|
| + if (c != 0xA && c >= 0) {
|
| + T_FileStream_ungetc(c, file);
|
| + }
|
| + }
|
| + break;
|
| + }
|
| + if (!setBuffer(n++, c, ec)) return FALSE;
|
| + }
|
| + if (!setBuffer(n++, 0, ec)) return FALSE;
|
| + UnicodeString str(buffer, encoding);
|
| + // Remove BOM in first line, if present
|
| + if (lineNo == 0 && str[0] == 0xFEFF) {
|
| + str.remove(0, 1);
|
| + }
|
| + ++lineNo;
|
| + line = str.unescape();
|
| + return TRUE;
|
| +}
|
| +
|
| +UBool TextFile::readLineSkippingComments(UnicodeString& line, UErrorCode& ec,
|
| + UBool trim) {
|
| + for (;;) {
|
| + if (!readLine(line, ec)) return FALSE;
|
| + // Skip over white space
|
| + int32_t pos = 0;
|
| + ICU_Utility::skipWhitespace(line, pos, TRUE);
|
| + // Ignore blank lines and comment lines
|
| + if (pos == line.length() || line.charAt(pos) == 0x23/*'#'*/) {
|
| + continue;
|
| + }
|
| + // Process line
|
| + if (trim) line.remove(0, pos);
|
| + return TRUE;
|
| + }
|
| +}
|
| +
|
| +/**
|
| + * Set buffer[index] to c, growing buffer if necessary. Return TRUE if
|
| + * successful.
|
| + */
|
| +UBool TextFile::setBuffer(int32_t index, char c, UErrorCode& ec) {
|
| + if (capacity <= index) {
|
| + if (!ensureCapacity(index+1)) {
|
| + ec = U_MEMORY_ALLOCATION_ERROR;
|
| + return FALSE;
|
| + }
|
| + }
|
| + buffer[index] = c;
|
| + return TRUE;
|
| +}
|
| +
|
| +/**
|
| + * Make sure that 'buffer' has at least 'mincapacity' bytes.
|
| + * Return TRUE upon success. Upon return, 'buffer' may change
|
| + * value. In any case, previous contents are preserved.
|
| + */
|
| +UBool TextFile::ensureCapacity(int32_t mincapacity) {
|
| + if (capacity >= mincapacity) {
|
| + return TRUE;
|
| + }
|
| +
|
| + // Grow by factor of 2 to prevent frequent allocation
|
| + // Note: 'capacity' may be 0
|
| + int32_t i;
|
| + for (i = capacity || 1024; i < mincapacity; ) {
|
| + i <<= 1;
|
| + if (i < 0) {
|
| + i = 0x7FFFFFFF;
|
| + break;
|
| + }
|
| + }
|
| + mincapacity = i;
|
| +
|
| + // Simple realloc() no good; contents not preserved
|
| + // Note: 'buffer' may be 0
|
| + char* newbuffer = (char*) uprv_malloc(mincapacity);
|
| + if (newbuffer == 0) {
|
| + return FALSE;
|
| + }
|
| + if (buffer != 0) {
|
| + uprv_strncpy(newbuffer, buffer, capacity);
|
| + uprv_free(buffer);
|
| + }
|
| + buffer = newbuffer;
|
| + capacity = mincapacity;
|
| + return TRUE;
|
| +}
|
| +
|
|
|
| Property changes on: icu46/source/test/intltest/textfile.cpp
|
| ___________________________________________________________________
|
| Added: svn:eol-style
|
| + LF
|
|
|
|
|