D:/Home/Programs/iSpike/src/Writer/WriterFactory.cpp

Go to the documentation of this file.

//iSpike includes
#include "iSpike/ISpikeException.hpp"
#include <iSpike/Writer/WriterFactory.hpp>
#include <iSpike/Writer/AngleWriter.hpp>
#include <iSpike/Writer/YarpAngleWriter.hpp>
#include <iSpike/Writer/FileAngleWriter.hpp>
#include "iSpike/Log/Log.hpp"
using namespace ispike;

WriterFactory::WriterFactory(){
        //Store list of available writers
        writerList.push_back(FileAngleWriter().getWriterDescription());
        printWriters();
}


WriterFactory::WriterFactory(string ip, unsigned port){
        //Store list of available writers
        writerList.push_back(FileAngleWriter().getWriterDescription());
        writerList.push_back(YarpAngleWriter(ip, port).getWriterDescription());
        printWriters();

        //Store variables
        this->ip = ip;
        this->port = port;
}


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

vector<Description> WriterFactory::getWritersOfType(string writerType){
        vector<Description> result;
        for(unsigned i = 0; i < writerList.size(); i++){
                if(writerList[i].getType() == writerType)
                        result.push_back(writerList[i]);
        }
        return result;
}


map<string, Property> WriterFactory::getDefaultProperties(Description& desc){
        if(desc.getName() == "File Angle Writer") {
                return FileAngleWriter().getProperties();
        }
        else if(desc.getName() == "Yarp Angle Writer") {
                return YarpAngleWriter(this->ip, this->port).getProperties();
        }
        throw ISpikeException("Invalid writer");
}


Writer* WriterFactory::create(Description& desc, map<string, Property>& writerProperties){
        Writer* result;
        if(desc.getName() == "File Angle Writer") {
                result = new FileAngleWriter();
        }
        else if(desc.getName() == "Yarp Angle Writer") {
                result = new YarpAngleWriter(this->ip, this->port);
        }
        else {
                throw ISpikeException("Invalid writer");
        }
        result->initialize(writerProperties);
        return result;
}


/*--------------------------------------------------------------------*/
/*---------                 PRIVATE METHODS                    -------*/
/*--------------------------------------------------------------------*/

void WriterFactory::printWriters(){
        for(size_t i=0; i<writerList.size(); ++i)
                LOG(LOG_DEBUG)<<"Writer: "<<writerList[i].getName()<<", "<<writerList[i].getDescription();
}