EncodedStreamFactory.cc

Go to the documentation of this file.

/*
 * EncodedStreamFactory.cc
 *
 * Smalltalk like class library for C++
 * EncodedStream constructor.
 *
 * Copyright (c) 2005 Milan Cermak
 */
/*
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
#include <stlib/EncodedStreamFactory.h>
#include <stlib/ByteArray.h>
#include <stlib/Dictionary.h>
#include <stlib/Locale.h>
#include <stlib/String.h>
#include <stlib/WriteStream.h>

#include <stlib/StreamEncoder.h>
#include <stlib/ByteStreamEncoder.h>
#include <stlib/UCS2StreamEncoder.h>
#include <stlib/UTF8StreamEncoder.h>
#include <stlib/WriteStream.h>

#include <stlib/ByteCharacterEncoder.h>

#include <string.h>

#ifndef DATADIR
#error DATADIR must be defined!
#endif

namespace Core {

Dictionary *EncodedStreamFactory::encoding_map = nil;

EncodedStreamFactory::EncodedStreamFactory(String *encoding)
{
    _encoding = encoding->asLowercase();
    if (_encoding->isEqual("default"))
        _encoding = Locale::current()->encoding()->asLowercase();
    normalized_encoding = nil;
    initializeAsciiEncoder();
}

/* Class-accessing protocol */
String *EncodedStreamFactory::className(void) const
{
    return new String("EncodedStreamFactory");
}

Dictionary *EncodedStreamFactory::encodingMap(void)
{
    if (encoding_map == nil)
        encoding_map = new Dictionary;
    return encoding_map;
}

/* Initialize protocol */
void EncodedStreamFactory::initializeAsciiEncoder(void)
{
    ByteArray *key = new ByteArray("usascii", 7);
    if (!encodingMap()->includesKey(key)) {
        encodingMap()->put(key, ByteCharacterEncoder::forASCII());
    }
}

/* Accessing protocol */
String *EncodedStreamFactory::encoding(void)
{
    return _encoding;
}

/* Utilities protocol */
ByteArray *EncodedStreamFactory::normalizedEncoding(void)
{
    if (normalized_encoding == nil) {
        Stream *stream = (new ByteArray(10))->writeStream();
        for (int i = 0; i < _encoding->size(); i++) {
            Character *ch = dynamic_cast<Character *>(_encoding->at(i));
            int chValue = ch->asInteger();
            if (ch->isDigit() || (chValue >= 'a' && chValue <= 'z')) {
                stream->nextPut(Integer::value(chValue));
            }
        }
        normalized_encoding = dynamic_cast<ByteArray *>(stream->contents());
    }
    return normalized_encoding;
}

/* Private protocol */
CharacterEncoder *EncodedStreamFactory::createCharacterEncoder(void)
{
    /* All charset conversions must be avoided in this method!
     * That's why this is written with C functions.
     */

    ByteArray *enc = normalizedEncoding();
    int length = 20 + enc->size() + strlen(DATADIR);
    char filename[length];
    sprintf(filename, "%s/encodings/%s.enc", DATADIR, enc->asCString());

    ByteCharacterEncoder *encoder = new ByteCharacterEncoder;
    encoder->loadFromFile(filename);
    return encoder;
}

CharacterEncoder *EncodedStreamFactory::getCharacterEncoder(void)
{
    CharacterEncoder *encoder = dynamic_cast<CharacterEncoder *>(encodingMap()->at(_encoding, nil));
    if (encoder == nil) {
        encoder = dynamic_cast<CharacterEncoder *>(encodingMap()->at(normalizedEncoding(), nil));
    }
    if (encoder == nil) {
        encoder = createCharacterEncoder();
        encodingMap()->put(normalizedEncoding(), encoder);
    }
    encodingMap()->put(_encoding, encoder);
    return encoder;
}

StreamEncoder *EncodedStreamFactory::getEncoder(void)
{
    if (strcmp(normalizedEncoding()->asCString(), "utf8") == 0)
        return new UTF8StreamEncoder;
    if (strcmp(normalizedEncoding()->asCString(), "ucs2") == 0)
        return new UCS2StreamEncoder;

    return new ByteStreamEncoder(getCharacterEncoder());
}

}; /* namespace Core */

---