DatasetRecorder.h

Go to the documentation of this file.

/*
 * Copyright (C) 2007-2011 RobotCub Consortium, European Commission FP6 Project IST-004370
 * author:  Arjan Gijsberts
 * email:   arjan.gijsberts@iit.it
 * website: www.robotcub.org
 * Permission is granted to copy, distribute, and/or modify this program
 * under the terms of the GNU General Public License, version 2 or any
 * later version published by the Free Software Foundation.
 *
 * A copy of the license can be found at
 * http://www.robotcub.org/icub/license/gpl.txt
 *
 * This program 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 General
 * Public License for more details
 */
 
#ifndef LM_DATASETRECORDER__
#define LM_DATASETRECORDER__
 
#include <fstream>
 
#include "iCub/learningMachine/IMachineLearner.h"
 
 
namespace iCub {
namespace learningmachine {
 
class DatasetRecorder : public IMachineLearner {
private:
    std::string filename;
 
    std::ofstream stream;
 
    int precision;
 
    int sampleCount;
 
public:
    DatasetRecorder() : filename("dataset.dat"), precision(8), sampleCount(0) {
        this->setName("Recorder");
    }
 
    DatasetRecorder(const DatasetRecorder& other)
      : IMachineLearner(other), filename(other.filename),
        precision(other.precision), sampleCount(other.sampleCount) {
    }
 
    virtual ~DatasetRecorder() {
        if(!this->stream.is_open()) {
            this->stream.close();
        }
    }
 
    DatasetRecorder& operator=(const DatasetRecorder& other);
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual void feedSample(const yarp::sig::Vector& input, const yarp::sig::Vector& output);
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual void train() { }
 
    /*
     * Inherited from IMachineLearner.
     */
    Prediction predict(const yarp::sig::Vector& input) {
        return Prediction();
    }
 
    /*
     * Inherited from IMachineLearner.
     */
    void reset() {
        this->stream.close();
        this->sampleCount = 0;
    }
 
    /*
     * Inherited from IMachineLearner.
     */
    DatasetRecorder* clone() {
        return new DatasetRecorder(*this);
    }
 
    /*
     * Inherited from IMachineLearner.
     */
    std::string getInfo();
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual std::string getConfigHelp();
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual void writeBottle(yarp::os::Bottle& bot) const;
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual void readBottle(yarp::os::Bottle& bot);
 
    /*
     * Inherited from IMachineLearner.
     */
    virtual bool configure(yarp::os::Searchable& config);
};
 
} // learningmachine
} // iCub
#endif