trunk/echoes/EchoesStyleModel.cpp

#include "EchoesStyleModel.h"

#include <QtAlgorithms>
#include <QHash>
#include <QFileInfo>
#include <QDir>

#include "TreeItem.h"
#include "EchoesPhrase.h"
#include "envelope/Envelope.h"

using namespace stand::echoes;

EchoesStyleModel::EchoesStyleModel(QObject *parent) :
    TreeItemModel(parent)
{
    onDestroy();
}

EchoesStyleModel::~EchoesStyleModel()
{
    QList<stand::utility::envelope::Envelope *> values = _envelopes.values();
    qDeleteAll(values);
}

void EchoesStyleModel::onDestroy()
{
    QList<stand::utility::envelope::Envelope *> es = _envelopes.values();
    qDeleteAll(es);
    _envelopes.clear();
    _data.clear();

    TreeItemModel::onDestroy();
    _root = new stand::model::TreeItem("Style root");
    _root->appendChild(new stand::model::TreeItem("Envelopes", _root));
    _root->appendChild(new stand::model::TreeItem("Phrases", _root));
}

void EchoesStyleModel::onRead()
{
    if(!_root)
    {
        onDestroy();
    }
    stand::model::TreeItem *phrases = _root->findChildItem("Phrases");
    stand::model::TreeItem *envelopes = _root->findChildItem("Envelopes");
    if(!phrases)
    {
        phrases = new stand::model::TreeItem("Phrases", _root);
        _root->appendChild(phrases);
    }
    if(!envelopes)
    {
        envelopes = new stand::model::TreeItem("Envelopes", _root);
        _root->appendChild(envelopes);
    }

    // ���������������������������������������������
    for(int i = 0; i < envelopes->childCount(); i++)
    {
        stand::model::TreeItem *e = envelopes->child(i);
        _registerEnvelope(e);
    }

    for(int i = 0; i < phrases->childCount(); i++)
    {
        stand::model::TreeItem *p = phrases->child(i);
        _registerPhrase(p);
    }
}

void EchoesStyleModel::_registerEnvelope(stand::model::TreeItem *e)
{
    QFileInfo info(this->path());
    QString path = info.absoluteDir().absoluteFilePath(e->data(_Path, 0).toString());
    QString name = e->name();
    stand::utility::envelope::Envelope *env = new stand::utility::envelope::Envelope();
    if(!env->read(path.toLocal8Bit().data()))
    {
        qDebug("EchoesStyleModel::registerEnvelop(%s); // envelope not found", path.toUtf8().data());
        delete env;
        return;
    }
    appendEnvelope(name, path, env);
}

void EchoesStyleModel::_registerPhrase(stand::model::TreeItem *p)
{
    EchoesNoteList notes;
    for(int i = 0; i < p->childCount(); i++)
    {
        stand::model::TreeItem *item = p->child(i);
        notes.append(EchoesNote(item->data(_MsPosition, 0).toDouble(), item->data(_MsLength, 0).toDouble(), item->data(_NoteNumber, 0).toInt()));
    }
    if(!notes.empty())
    {
        EchoesPhrase phrase(notes);
        _Data d = {envelope(p->data(_PitchEnvelopeName, 0).toString()), envelope(p->data(_VolumeEnvelopeName, 0).toString())};
        _data.append(QPair<EchoesPhrase, _Data>(phrase, d));
    }
}

EchoesSegment EchoesStyleModel::findNearest(const EchoesPhrase &key) const
{
    if(_data.empty())
    {
        return EchoesSegment();
    }
    int index = 0;
    double distance = _data[0].first.distance(key);
    for(int i = 0; i < _data.size(); i++)
    {
        double currentDistance = _data[i].first.distance(key);
        if(distance < currentDistance)
        {
            index = i;
            distance = currentDistance;
        }
    }
    return EchoesSegment(&(_data[index].first), _data[index].second.pitch, _data[index].second.volume);
}


void EchoesStyleModel::appendEnvelope(const QString &name, const QString &path, stand::utility::envelope::Envelope *env)
{
    // ���������������������
    if(_envelopes.contains(name))
    {
        qDebug("EchoesStyleModel::appendEvelope(\"%s\"); // name duplication", name.toUtf8().data());
        return;
    }
    // ������������������������������������������������
    stand::model::TreeItem *envelopes = _root->findChildItem("Envelopes");
    if(!envelopes)
    {
        envelopes = new stand::model::TreeItem("Envelopes", _root);
        _root->appendChild(envelopes);
    }

    // ���������������������������������������
    if(!env)
    {
        env = new stand::utility::envelope::Envelope();
        QFileInfo info(this->path());
        QDir current = info.dir();
        if(!env->read((current.absoluteFilePath(path)).toLocal8Bit().data()))
        {
            delete env;
            return;
        }
    }
    _envelopes[name] = env;

    // ���������������������������������������������
    if(!envelopes->findChildItem(name))
    {
        stand::model::TreeItem *item = new stand::model::TreeItem(name, envelopes);
        item->setData(_Path, path, 0);
        envelopes->appendChild(item);
    }
}

void EchoesStyleModel::appendPhrase(const QString &pitchName, const QString &volumeName, const EchoesPhrase &phrase)
{
    if(phrase.notes().empty())
    {
        qDebug("EchoesStyleModel::appendPhrase(); // Phrsae has no note.");
        return;
    }
    if(!_envelopes.contains(pitchName) || !_envelopes.contains(volumeName))
    {
        qDebug("EchoesStyleModel::appendPhrase(\"%s\", \"%s\"); // envelope not found", pitchName.toUtf8().data(), volumeName.toUtf8().data());
        return;
    }
    /* ��������������������� */
    stand::model::TreeItem *phrases = _root->findChildItem("Phrases");
    if(!phrases)
    {
        phrases = new stand::model::TreeItem("Phrases", _root);
        _root->appendChild(phrases);
    }

    stand::model::TreeItem *notes = new stand::model::TreeItem("Notes", phrases);
    phrases->appendChild(notes);

    notes->setData(_PitchEnvelopeName, pitchName, 0);
    notes->setData(_VolumeEnvelopeName, volumeName, 0);

    for(int i = 0; i < phrase.notes().size(); i++)
    {
        stand::model::TreeItem *note = new stand::model::TreeItem("Note" + QString::number(i), notes);
        note->setData(_NoteNumber, phrase.notes().at(i).noteNumber(), 0);
        note->setData(_MsPosition, phrase.notes().at(i).msPosition(), 0);
        note->setData(_MsLength, phrase.notes().at(i).msLength(), 0);
        notes->appendChild(note);
    }
    /* ��������������������� */
    _Data d = {envelope(pitchName), envelope(volumeName)};
    _data.append(QPair<EchoesPhrase, _Data>(phrase, d));
}

stand::utility::envelope::Envelope *EchoesStyleModel::envelope(const QString &key)
{
    return _envelopes.value(key, NULL);
}

QString EchoesStyleModel::metaData(MetaTypes type)
{
    if(!_root)
    {
        return QString();
    }
    return _root->data(type, 0).toString();
}

bool EchoesStyleModel::write(const QString &path) const
{
    QFileInfo info(path);
    QDir current = info.absoluteDir();

    stand::model::TreeItem *envelopes = _root->findChildItem("Envelopes");
    if(envelopes)
    {
        for(int i = 0; i < envelopes->childCount(); i++)
        {
            stand::model::TreeItem *item = envelopes->child(i);
            stand::utility::envelope::Envelope *e = _envelopes[item->name()];
            if(!e)
            {
                continue;
            }
            e->write(current.absoluteFilePath(item->data(0, 0).toString()).toLocal8Bit().data());
        }
    }
    return TreeItemModel::write(path);
}