Move the bulk of TextFileWriter's logic to libaegisub.

Originally committed to SVN as r4660.
This commit is contained in:
Thomas Goyne 2010-07-07 02:41:46 +00:00
parent 02237f2a7a
commit 929fa83dd9
14 changed files with 384 additions and 124 deletions

View file

@ -232,6 +232,10 @@
RelativePath="..\..\tests\libaegisub_iconv.cpp"
>
</File>
<File
RelativePath="..\..\tests\libaegisub_line_iterator.cpp"
>
</File>
<File
RelativePath="..\..\tests\libaegisub_mru.cpp"
>

View file

@ -18,6 +18,8 @@
/// @brief Wrapper for libiconv to present a more C++-friendly API
/// @ingroup libaegisub
#pragma once
#ifndef LAGI_PRE
#include <string.h>
#include <memory>

View file

@ -0,0 +1,218 @@
// Copyright (c) 2010, Thomas Goyne <plorkyeran@aegisub.org>
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
// $Id$
/// @file line_iterator.h
/// @brief An iterator over lines in a stream
/// @ingroup libaegisub
#pragma once
#if !defined(AGI_PRE) && !defined(LAGI_PRE)
#include <iterator>
#ifdef _WIN32
#include <memory>
#else
#include <tr1/memory>
#endif
#include <sstream>
#include <stdint.h>
#endif
#include <libaegisub/charset_conv.h>
namespace agi {
/// @class line_iterator
/// @brief An iterator over lines in a stream
template<typename OutputType = std::string>
class line_iterator : public std::iterator<std::input_iterator_tag, OutputType> {
std::istream *stream; ///< Stream to iterator over
bool valid; ///< Are there any more values to read?
OutputType value; ///< Value to return when this is dereference
std::string encoding; ///< Encoding of source stream
std::tr1::shared_ptr<agi::charset::IconvWrapper> conv;
int cr; ///< CR character in the source encoding
int lf; ///< LF character in the source encoding
int width; ///< width of LF character in the source encoding
/// @brief Convert a string to the output type
/// @param str Line read from the file
///
/// line_iterator users can either ensure that operator>> is defined for
/// their desired output type or simply provide a specialization of this
/// method which does the conversion.
inline bool convert(std::string &str);
/// Called after construction for specializations that need to do things
void init() { };
/// @brief Get the next line from the stream
/// @param[out] str String to fill with the next line
void getline(std::string &str);
/// @brief Get the next value from the stream
void next();
public:
/// @brief Constructor
/// @param stream The stream to read from. The calling code is responsible
/// for ensuring that the stream remains valid for the
/// lifetime of the iterator and that it get cleaned up.
/// @param encoding Encoding of the text read from the stream
line_iterator(std::istream &stream, std::string encoding = "utf-8")
: stream(&stream)
, valid(true)
, encoding(encoding)
, cr(0)
, lf(0)
, width(0)
{
agi::charset::IconvWrapper c("utf-8", encoding.c_str());
c.Convert("\r", 1, reinterpret_cast<char *>(&cr), sizeof(int));
c.Convert("\n", 1, reinterpret_cast<char *>(&lf), sizeof(int));
width = c.RequiredBufferSize("\n");
if (encoding != "utf-8") {
conv.reset(new agi::charset::IconvWrapper(encoding.c_str(), "utf-8"));
}
init();
++(*this);
}
/// @brief Invalid iterator constructor; use for end iterator
line_iterator()
: stream(0)
, valid(false)
{
}
/// @brief Copy constructor
/// @param that line_iterator to copy from
line_iterator(line_iterator<OutputType> const& that)
: stream(that.stream)
, valid(that.valid)
, value(that.value)
, encoding(that.encoding)
, conv(that.conv)
, cr(that.cr)
, lf(that.lf)
, width(that.width)
{
}
OutputType operator*() const {
return value;
}
line_iterator<OutputType>& operator++() {
next();
return *this;
}
line_iterator<OutputType> operator++(int) {
line_iterator<OutputType> tmp(*this);
++(*this);
return tmp;
}
bool operator==(line_iterator<OutputType> const& rgt) const {
return valid == rgt.valid;
}
bool operator!=(line_iterator<OutputType> const& rgt) const {
return !operator==(rgt);
}
// typedefs needed by some stl algorithms
typedef OutputType* pointer;
typedef OutputType& reference;
typedef const OutputType* const_pointer;
typedef const OutputType& const_reference;
line_iterator<OutputType> operator=(line_iterator<OutputType> that) {
using std::swap;
swap(*this, that);
return *this;
}
void swap(line_iterator<OutputType> &that) throw() {
using std::swap;
swap(stream, that.stream);
swap(valid, that.valid);
swap(value, that.value);
swap(encoding, that.encoding);
swap(conv, that.conv);
swap(lf, that.lf);
swap(cr, that.cr);
swap(width, that.width);
}
};
template<typename OutputType>
void line_iterator<OutputType>::getline(std::string &str) {
union {
int32_t chr;
char buf[4];
};
for (;;) {
chr = 0;
#ifdef _WIN32
int read = stream->rdbuf()->_Sgetn_s(buf, 4, width);
#else
int read = stream->rdbuf()->sgetn(buf, width);
#endif
if (read < width) {
for (int i = 0; i < read; i++) {
str += buf[i];
}
stream->setstate(std::ios::eofbit);
return;
}
if (chr == cr) continue;
if (chr == lf) return;
for (int i = 0; i < read; i++) {
str += buf[i];
}
}
}
template<typename OutputType>
void line_iterator<OutputType>::next() {
if (!valid) return;
if (!stream->good()) {
valid = false;
return;
}
std::string str;
getline(str);
if (conv.get()) {
str = conv->Convert(str);
}
if (!convert(str)) {
next();
return;
}
}
template<typename OutputType>
inline bool line_iterator<OutputType>::convert(std::string &str) {
std::istringstream ss(str);
ss >> value;
return !ss.fail();
}
template<>
inline bool line_iterator<std::string>::convert(std::string &str) {
value = str;
return true;
}
template<typename T>
void swap(agi::line_iterator<T> &lft, agi::line_iterator<T> &rgt) {
lft.swap(rgt);
}
}

View file

@ -24,6 +24,7 @@
#include <time.h>
#include <deque>
#include <memory>
#ifdef __DEPRECATED // Dodge GCC warnings
# undef __DEPRECATED
# include <strstream>

View file

@ -28,7 +28,11 @@
#include <fstream>
#include <iostream>
#include <map>
#ifdef _WIN32
#include <memory>
#else
#include <tr1/memory>
#endif
#include <sstream>
#include <string>

View file

@ -72,8 +72,13 @@
#include <iostream>
#include <list>
#include <map>
#ifdef _WIN32
#include <memory>
#else
#include <tr1/memory>
#endif
#include <set>
#include <sstream>
#include <string>
#include <utility>
#include <vector>

View file

@ -57,7 +57,6 @@
#ifdef WITH_AVISYNTH
#include "avisynth_wrap.h"
#endif
#include "charset_conv.h"
#include "compat.h"
#include "dialog_detached_video.h"
#include "dialog_search_replace.h"

View file

@ -54,7 +54,6 @@
#ifdef WITH_AUTOMATION
#include "auto4_base.h"
#endif
#include "charset_conv.h"
#include "compat.h"
#include "dialog_about.h"
#include "dialog_attachments.h"

View file

@ -1,4 +1,4 @@
// Copyright (c) 2010, Rodrigo Braz Monteiro, Thomas Goyne
// Copyright (c) 2010, Thomas Goyne
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
@ -43,120 +43,54 @@
#include <algorithm>
#include <fstream>
#include <string>
#include <wx/string.h>
#endif
#include <libaegisub/io.h>
#include <libaegisub/log.h>
#include "charset_conv.h"
#include "charset_detect.h"
#include "compat.h"
#include "text_file_reader.h"
TextFileReader::TextFileReader(wxString const& filename, wxString encoding, bool trim)
: isBinary(false)
, conv()
, trim(trim)
, readComplete(false)
, currout(0)
, outptr(0)
, currentLine(0)
: trim(trim)
, isBinary(false)
{
#ifdef __WINDOWS__
file.open(filename.wc_str(),std::ios::in | std::ios::binary);
#else
file.open(wxFNCONV(filename),std::ios::in | std::ios::binary);
#endif
if (!file.is_open()) throw L"Failed opening file for reading.";
if (encoding.IsEmpty()) encoding = CharSetDetect::GetEncoding(filename);
if (encoding.empty()) encoding = CharSetDetect::GetEncoding(filename);
if (encoding == L"binary") {
isBinary = true;
return;
}
conv.reset(new agi::charset::IconvWrapper(encoding.c_str(), "wchar_t"));
file.reset(agi::io::Open(STD_STR(filename)));
iter = agi::line_iterator<wxString>(*file, STD_STR(encoding));
}
TextFileReader::~TextFileReader() {
}
wchar_t TextFileReader::GetWChar() {
// If there's already some converted characters waiting, return the next one
if (++currout < outptr) {
return *currout;
}
if (file.eof()) return 0;
// Otherwise convert another block
char inbuf[64];
char *inptr = inbuf;
size_t inbytesleft = sizeof(inbuf) - 4;
int bytesAdded = 0;
memset(inbuf, 0, inbytesleft);
outptr = outbuf;
outbytesleft = sizeof(outbuf);
currout = outbuf;
file.read(inbuf, inbytesleft);
inbytesleft = file.gcount();
if (inbytesleft == 0)
return 0;
do {
// Without this const_cast the wrong overload is chosen
size_t ret = conv->Convert(const_cast<const char**>(&inptr), &inbytesleft, reinterpret_cast<char **>(&outptr), &outbytesleft);
if (ret != (size_t)-1) break;
int err = errno;
// If 64 chars do not fit into 256 wchar_ts the environment is so bizzare that doing
// anything is probably futile
assert(err != E2BIG);
// (Hopefully) the edge of the buffer happened to split a multibyte character, so keep
// adding one byte to the input buffer until either it succeeds or we add enough bytes to
// complete any character
if (++bytesAdded > 3)
throw wxString::Format(L"Invalid input character found near line %u", currentLine);
file.read(inptr + inbytesleft, 1);
inbytesleft++;
} while (!file.eof() && file.gcount());
if (outptr > outbuf)
return *currout;
throw wxString::Format(L"Invalid input character found near line %u", currentLine);
}
wxString TextFileReader::ReadLineFromFile() {
wxString buffer;
buffer.Alloc(1024);
currentLine++;
// Read a line
wchar_t ch;
bool first = true;
// This doesn't work for \r deliminated files, but it's very unlikely
// that we'll run into one of those
for (ch = GetWChar(); ch != L'\n' && ch != 0; ch = GetWChar()) {
if (ch == L'\r') continue;
// Skip the BOM -- we don't need it as the encoding is already known
// and it sometimes causes conversion problems
if (ch == 0xFEFF && first) continue;
buffer += ch;
first = false;
}
if (ch == 0)
readComplete = true;
if (trim) {
buffer.Trim(true);
buffer.Trim(false);
}
return buffer;
wxString str = *iter;
++iter;
if (trim) str.Trim(true).Trim(false);
if (str.StartsWith(L"\uFEFF")) str = str.Mid(1);
return str;
}
bool TextFileReader::HasMoreLines() {
return !readComplete;
namespace agi {
#ifdef _WIN32
template<> void line_iterator<wxString>::init() {
conv.reset(new agi::charset::IconvWrapper(encoding.c_str(), "utf-16le"));
}
template<> bool line_iterator<wxString>::convert(std::string &str) {
value = wxString(str.c_str(), wxMBConvUTF16LE(), str.size());
return true;
}
#else
template<> bool line_iterator<wxString>::convert(std::string &str) {
value = str;
return true;
}
#endif
}

View file

@ -1,4 +1,4 @@
// Copyright (c) 2010, Rodrigo Braz Monteiro
// Copyright (c) 2010, Thomas Goyne
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
@ -44,31 +44,15 @@
#include <wx/string.h>
#endif
namespace agi { namespace charset {
class IconvWrapper;
} }
#include <libaegisub/line_iterator.h>
/// @class TextFileReader
/// @brief A line-based text file reader
class TextFileReader {
private:
bool isBinary;
std::ifstream file;
std::auto_ptr<agi::charset::IconvWrapper> conv;
std::auto_ptr<std::ifstream> file;
bool trim;
bool readComplete;
// Iconv buffers and state
wchar_t outbuf[256];
wchar_t *currout;
wchar_t *outptr;
size_t outbytesleft;
/// Current line number
unsigned int currentLine;
/// @brief Read a single wchar_t from the file
wchar_t GetWChar();
bool isBinary;
agi::line_iterator<wxString> iter;
TextFileReader(const TextFileReader&);
TextFileReader& operator=(const TextFileReader&);
@ -86,6 +70,6 @@ public:
/// @return The line, possibly trimmed
wxString ReadLineFromFile();
/// @brief Check if there are any more lines to read
bool HasMoreLines();
bool IsBinary() { return isBinary; }
bool HasMoreLines() const { return iter != agi::line_iterator<wxString>(); }
bool IsBinary() const { return isBinary; }
};

View file

@ -13,8 +13,9 @@ run_SOURCES = \
libaegisub_access.cpp \
libaegisub_cajun.cpp \
libaegisub_iconv.cpp \
libaegisub_util.cpp \
libaegisub_mru.cpp
libaegisub_line_iterator.cpp \
libaegisub_mru.cpp \
libaegisub_util.cpp
run_SOURCES += \
*.h

View file

@ -138,10 +138,15 @@ TEST(lagi_iconv, wchar_tSupport) {
EXPECT_NO_THROW(IconvWrapper("UTF-8", "wchar_t"));
}
TEST(lagi_iconv, pretty_names) {
TEST(lagi_iconv, Roundtrip) {
std::vector<std::string> names = GetEncodingsList<std::vector<std::string> >();
for (std::vector<std::string>::iterator cur = names.begin(); cur != names.end(); ++cur) {
EXPECT_NO_THROW(IconvWrapper("utf-8", cur->c_str()));
EXPECT_NO_THROW(IconvWrapper(cur->c_str(), "utf-8"));
EXPECT_EQ(
"Jackdaws love my big sphinx of quartz",
IconvWrapper(cur->c_str(), "utf-8").Convert(
IconvWrapper("utf-8", cur->c_str()).Convert(
"Jackdaws love my big sphinx of quartz")));
}
}

View file

@ -0,0 +1,87 @@
// Copyright (c) 2010, Thomas Goyne <plorkyeran@aegisub.org>
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
//
// $Id$
/// @file libaegisub_line_iterator.cpp
/// @brief agi::line_iterator tests
/// @ingroup
#include <libaegisub/line_iterator.h>
#include <algorithm>
#include <iterator>
#include <sstream>
#include <string>
#include <vector>
#include "main.h"
#include "util.h"
using agi::line_iterator;
using namespace util;
template<typename T>
struct varg_type {
typedef T type;
};
template<> struct varg_type<std::string> {
typedef const char * type;
};
template<typename T, int N>
void expect_eq(const char (&str)[N], const char *charset, int num, ...) {
std::string string(str, N - 1);
agi::charset::IconvWrapper conv("utf-8", charset);
string = conv.Convert(string);
std::stringstream ss(string);
line_iterator<T> iter;
EXPECT_NO_THROW(iter = line_iterator<T>(ss, charset));
va_list argp;
va_start(argp, num);
for (; num > 0; --num) {
EXPECT_FALSE(iter == line_iterator<T>());
EXPECT_EQ(*iter, va_arg(argp, typename varg_type<T>::type));
EXPECT_NO_THROW(++iter);
}
va_end(argp);
EXPECT_TRUE(iter == line_iterator<T>());
}
TEST(lagi_line, int) {
std::vector<std::string> charsets = agi::charset::GetEncodingsList<std::vector<std::string> >();
for (std::vector<std::string>::iterator cur = charsets.begin(); cur != charsets.end(); ++cur) {
expect_eq<int>("1\n2\n3\n4", cur->c_str(), 4, 1, 2, 3, 4);
expect_eq<int>("1\n2\n3\n4\n", cur->c_str(), 4, 1, 2, 3, 4);
expect_eq<int>("1\n2\nb\n3\n4", cur->c_str(), 4, 1, 2, 3, 4);
expect_eq<int>("1.0\n2.0\n3.0\n4.0", cur->c_str(), 4, 1, 2, 3, 4);
expect_eq<int>(" 0x16 \n 09 \n -2", cur->c_str(), 3, 0, 9, -2);
}
}
TEST(lagi_line, double) {
std::vector<std::string> charsets = agi::charset::GetEncodingsList<std::vector<std::string> >();
for (std::vector<std::string>::iterator cur = charsets.begin(); cur != charsets.end(); ++cur) {
expect_eq<double>("1.0\n2.0", cur->c_str(), 2, 1.0, 2.0);
expect_eq<double>("#1.0\n\t2.5", cur->c_str(), 1, 2.5);
}
}
TEST(lagi_line, string) {
std::vector<std::string> charsets = agi::charset::GetEncodingsList<std::vector<std::string> >();
for (std::vector<std::string>::iterator cur = charsets.begin(); cur != charsets.end(); ++cur) {
expect_eq<std::string>("line 1\nline 2\nline 3", cur->c_str(), 3, "line 1", "line 2", "line 3");
expect_eq<std::string>(" white space ", cur->c_str(), 1, " white space ");
expect_eq<std::string>("blank\n\nlines\n", cur->c_str(), 4, "blank", "", "lines", "");
}
}

View file

@ -19,11 +19,28 @@
/// @ingroup util
#include <string>
#include <vector>
#include <stdarg.h>
namespace util {
void copy(const std::string from, const std::string to);
void remove(const std::string& file);
template<typename T>
static std::vector<T> make_vector(int len, ...) {
std::vector<T> vec(len);
va_list argp;
va_start(argp, len);
for (int i = 0; i < len; i++) {
vec[i] = va_arg(argp, T);
}
va_end(argp);
return vec;
}
} // namespace util