ArchiveDao.cpp

Go to the documentation of this file.

#include "ArchiveDao.h"
#include "GlobalVariables.h"
#include "SpikeStreamDBException.h"
#include "Util.h"
using namespace spikestream;

ArchiveDao::ArchiveDao(const DBInfo& dbInfo) : AbstractDao(dbInfo){
}


ArchiveDao::~ArchiveDao(){
}


/*----------------------------------------------------------*/
/*-----                 PUBLIC METHODS                 -----*/
/*----------------------------------------------------------*/

void ArchiveDao::addArchive(ArchiveInfo& archInfo){
        //Get date at which archive is being created.
        archInfo.setDateTime(QDateTime::currentDateTime());

        //Create the archive
    QSqlQuery query = getQuery("INSERT INTO Archives (StartTime, NetworkID, Description) VALUES (" + QString::number(archInfo.getDateTime().toTime_t()) + ", " + QString::number(archInfo.getNetworkID()) + ", '" + archInfo.getDescription() + "')");
    executeQuery(query);

        //Check id is correct and add to archive info if it is
    int lastInsertID = query.lastInsertId().toInt();
    if(lastInsertID >= START_ARCHIVE_ID)
                archInfo.setID(lastInsertID);
    else
                throw SpikeStreamDBException("Insert ID for Archives is invalid: " + QString::number(lastInsertID));
}


void ArchiveDao::addArchiveData(unsigned int archiveID, unsigned int timeStep, const QString& firingNeuronString){
    executeQuery("INSERT INTO ArchiveData(ArchiveID, TimeStep, FiringNeurons) VALUES (" + QString::number(archiveID) + ", " + QString::number(timeStep) + ", '"  + firingNeuronString + "')");
}


void ArchiveDao::addArchiveData(unsigned int archiveID, unsigned int timeStep, const QList<unsigned>& firingNeuronList){

//      QByteArray ba;
//      QFile f(fileName);
//      if(f.open(QIODevice::ReadOnly))
//      {
//      ba = f.readAll();
//      f.close();
//      }
//
//      // Writing the image into table
//      QSqlDatabase::database().transaction();
//      QSqlQuery query;
//      query.prepare( "INSERT INTO picture ( IMAGE ) VALUES (:IMAGE)" );
//      query.bindValue(":IMAGE", ba);

        //FIXME: SORT THIS OUT WHEN I CONVERT TO BLOB
        //Build firing neuron string
        QString tmpStr;
        QList<unsigned>::const_iterator endList = firingNeuronList.end();
        for(QList<unsigned>::const_iterator iter = firingNeuronList.begin(); iter != endList; ++iter)
                tmpStr += QString::number(*iter) + ",";
        tmpStr.truncate(tmpStr.size() - 1);
        addArchiveData(archiveID, timeStep, tmpStr);
}


void ArchiveDao::deleteArchive(unsigned int archiveID){
    executeQuery("DELETE FROM Archives WHERE ArchiveID = " + QString::number(archiveID));
}


void ArchiveDao::deleteAllArchives(){
        executeQuery("DELETE FROM Archives");
}


QList<ArchiveInfo> ArchiveDao::getArchivesInfo(unsigned int networkID){
    QSqlQuery query = getQuery("SELECT ArchiveID, StartTime, Description FROM Archives WHERE NetworkID=" + QString::number(networkID) + " ORDER BY StartTime");
    executeQuery(query);
    QList<ArchiveInfo> tmpList;
    for(int i=0; i<query.size(); ++i){
                query.next();
                unsigned int archiveID = Util::getUInt(query.value(0).toString());
                tmpList.append(
                                ArchiveInfo(
                                                archiveID,
                                                networkID,
                                                Util::getUInt(query.value(1).toString()),//Start time as a unix timestamp
                                                query.value(2).toString()//Description
                                                )
                                );
    }
    return tmpList;
}


int ArchiveDao::getArchiveSize(unsigned int archiveID){
    QSqlQuery query = getQuery("SELECT COUNT(*) FROM ArchiveData WHERE ArchiveID=" + QString::number(archiveID));
    executeQuery(query);
    query.next();
    unsigned int archiveSize = Util::getInt(query.value(0).toString());
    return archiveSize;
}


unsigned int ArchiveDao::getMaxTimeStep(unsigned int archiveID){
    QSqlQuery query = getQuery("SELECT MAX(TimeStep) FROM ArchiveData WHERE ArchiveID=" + QString::number(archiveID));
    executeQuery(query);
    query.next();
    return Util::getUInt(query.value(0).toString());
}


unsigned int ArchiveDao::getMinTimeStep(unsigned int archiveID){
    QSqlQuery query = getQuery("SELECT MIN(TimeStep) FROM ArchiveData WHERE ArchiveID=" + QString::number(archiveID));
    executeQuery(query);
    query.next();
    return Util::getUInt(query.value(0).toString());
}


QList<unsigned> ArchiveDao::getFiringNeuronIDs(unsigned int archiveID, unsigned int timeStep){
    QSqlQuery query = getQuery("SELECT FiringNeurons FROM ArchiveData WHERE TimeStep=" + QString::number(timeStep) + " AND ArchiveID=" + QString::number(archiveID));
    executeQuery(query);
        query.next();

        //Build list of firing neuron ids
        QList<unsigned> newList;
        QStringList strList = query.value(0).toString().split(",", QString::SkipEmptyParts);
        QStringListIterator iter(strList);
        while (iter.hasNext()){
                newList.append(Util::getUInt(iter.next()));
        }

        //Return the list we have built
        return newList;
}


bool ArchiveDao::networkHasArchives(unsigned int networkID){
    QSqlQuery query = getQuery("SELECT COUNT(*) FROM Archives WHERE NetworkID=" + QString::number(networkID));
    executeQuery(query);
    query.next();
    if(query.value(0).toUInt() == 0)
                return false;
    return true;
}


void ArchiveDao::setArchiveProperties(unsigned archiveID, const QString &description){
        executeQuery("UPDATE Archives SET Description='" + description + "' WHERE ArchiveID=" + QString::number(archiveID));
}