qskinny/inputcontext/QskHunspellCompositionModel.cpp

#include "QskHunspellCompositionModel.h"
#include <QVector>

#include "hunspell.h"

class QskHunspellCompositionModel::PrivateData
{
public:
    Hunhandle* hunspellHandle;
    QVector< QString > candidates;
};

QskHunspellCompositionModel::QskHunspellCompositionModel( QskInputContext* context ):
    Inherited( Words, context ),
    m_data( new PrivateData() )
{
#if 1
    // TODO: loading the language specific one depending on the locale

    m_data->hunspellHandle = Hunspell_create(
        "/usr/share/hunspell/en_US.aff",
        "/usr/share/hunspell/en_US.dic" );
#endif
}

QskHunspellCompositionModel::~QskHunspellCompositionModel()
{
    Hunspell_destroy( m_data->hunspellHandle );
}

int QskHunspellCompositionModel::candidateCount() const
{
    return m_data->candidates.count();
}

QString QskHunspellCompositionModel::candidate( int pos ) const
{
    return m_data->candidates[ pos ];
}

void QskHunspellCompositionModel::resetCandidates()
{
    if ( !m_data->candidates.isEmpty() )
    {
        m_data->candidates.clear();
        Q_EMIT candidatesChanged();
    }
}

void QskHunspellCompositionModel::requestCandidates( const QString& text )
{
    char** suggestions;
    const QByteArray word = text.toUtf8(); // ### do we need to check the encoding

    const int count = Hunspell_suggest(
        m_data->hunspellHandle, &suggestions, word.constData() );

    QVector< QString > candidates;
    candidates.reserve( count );

    for( int i = 0; i < count; i++ )
    {
        const QString suggestion = QString::fromUtf8( suggestions[i] ); // ### encoding?

        if( suggestion.startsWith( text ) )
            candidates.prepend( suggestion );
        else
            candidates.append( suggestion );
    }

    Hunspell_free_list( m_data->hunspellHandle, &suggestions, count );

    m_data->candidates = candidates;
    Q_EMIT candidatesChanged();
}