#include "config.h"
#include "TextCodecQt.h"
#include "PlatformString.h"
#include <wtf/text/CString.h>
#include <qset.h>
namespace WebCore {
static QSet<QByteArray> *unique_names = 0;
static const char *getAtomicName(const QByteArray &name)
{
if (!unique_names)
unique_names = new QSet<QByteArray>;
unique_names->insert(name);
return unique_names->find(name)->constData();
}
void TextCodecQt::registerEncodingNames(EncodingNameRegistrar registrar)
{
QList<int> mibs = QTextCodec::availableMibs();
for (int i = 0; i < mibs.size(); ++i) {
QTextCodec *c = QTextCodec::codecForMib(mibs.at(i));
const char *name = getAtomicName(c->name());
registrar(name, name);
QList<QByteArray> aliases = c->aliases();
for (int i = 0; i < aliases.size(); ++i) {
const char *a = getAtomicName(aliases.at(i));
registrar(a, name);
}
}
}
static PassOwnPtr<TextCodec> newTextCodecQt(const TextEncoding& encoding, const void*)
{
return adoptPtr(new TextCodecQt(encoding));
}
void TextCodecQt::registerCodecs(TextCodecRegistrar registrar)
{
QList<int> mibs = QTextCodec::availableMibs();
for (int i = 0; i < mibs.size(); ++i) {
QTextCodec *c = QTextCodec::codecForMib(mibs.at(i));
const char *name = getAtomicName(c->name());
registrar(name, newTextCodecQt, 0);
}
}
TextCodecQt::TextCodecQt(const TextEncoding& encoding)
: m_encoding(encoding)
{
m_codec = QTextCodec::codecForName(m_encoding.name());
}
TextCodecQt::~TextCodecQt()
{
}
String TextCodecQt::decode(const char* bytes, size_t length, bool flush, bool , bool& sawError)
{
#if OS(SYMBIAN)
static const int MaxInputChunkSize = 32 * 1024;
#else
static const int MaxInputChunkSize = 1024 * 1024;
#endif
const char* buf = bytes;
const char* end = buf + length;
String unicode("");
while (buf < end) {
int size = end - buf;
size = qMin(size, MaxInputChunkSize);
QString decoded = m_codec->toUnicode(buf, size, &m_state);
unicode.append(reinterpret_cast_ptr<const UChar*>(decoded.unicode()), decoded.length());
buf += size;
}
sawError = m_state.invalidChars != 0;
if (flush) {
m_state.flags = QTextCodec::DefaultConversion;
m_state.remainingChars = 0;
m_state.invalidChars = 0;
}
return unicode;
}
CString TextCodecQt::encode(const UChar* characters, size_t length, UnencodableHandling handling)
{
QTextCodec::ConverterState state;
state.flags = QTextCodec::ConversionFlags(QTextCodec::ConvertInvalidToNull | QTextCodec::IgnoreHeader);
if (!length)
return "";
QByteArray ba = m_codec->fromUnicode(reinterpret_cast<const QChar*>(characters), length, &state);
if (state.invalidChars) {
state.invalidChars = 0;
state.remainingChars = 0;
int len = 0;
ba.clear();
for (size_t pos = 0; pos < length; ++pos) {
QByteArray tba = m_codec->fromUnicode(reinterpret_cast<const QChar*>(characters), ++len, &state);
if (state.remainingChars)
continue;
if (state.invalidChars) {
UnencodableReplacementArray replacement;
getUnencodableReplacement(characters[0], handling, replacement);
tba.replace('\0', replacement);
state.invalidChars = 0;
}
ba.append(tba);
characters += len;
len = 0;
state.remainingChars = 0;
}
}
return CString(ba.constData(), ba.length());
}
}