RandomNumberGenerator.h

Go to the documentation of this file.

/* ============================================================
 *
 * This file is a part of CCA project
 *
 * Copyright (C) 2011 by Arne Nordmann <anordman at cor-lab dot uni-bielefeld dot de>
 *
 * This file may be licensed under the terms of the
 * GNU Lesser General Public License Version 3 (the ``LGPL''),
 * or (at your option) any later version.
 *
 * Software distributed under the License is distributed
 * on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either
 * express or implied. See the LGPL for the specific language
 * governing rights and limitations.
 *
 * You should have received a copy of the LGPL along with this
 * program. If not, go to http://www.gnu.org/licenses/lgpl.html
 * or write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * The development of this software was supported by:
 *   CoR-Lab, Research Institute for Cognition and Robotics
 *     Bielefeld University
 *
 * ============================================================ */

#pragma once

#include <string>
#include <iostream>

#include <nemo/Vector.h>
#include <nemo/Random.h>

#include "cca/Node.h"

#include "cca/dto/debug/DebugDTO.h"

namespace cca {

template<class DATATYPE>
class RandomNumberGenerator: public Node {
public:
    RandomNumberGenerator(std::string nodename) :
            Node(nodename), outputPort(), _lower(0), _upper(1), _dim(1),
                    _rng(_lower, _upper) {
        outputPort = OutputPort<DATATYPE>::create();
    }

    ~RandomNumberGenerator() {
    }

    void setRange(double lower, double upper) {
        this->_upper = upper;
        this->_lower = lower;
        this->_rng = nemo::uniform<double>(this->_lower, this->_upper);
    }

    void setDimension(unsigned int dimension) {
        this->_dim = dimension;
    }

    std::string print() const {
        std::ostringstream outstream(std::ostringstream::out);
        outstream.precision(3); // Precision when printing double values
        outstream << "<RandomNumberGenerator \"" << this->getName() << "\"";
        outstream << " (dimension: " << this->_dim << ")";
        outstream << ", range is [" << this->_lower << ", " << this->_upper
                << "]";
        outstream << ">" << std::endl;
        return outstream.str();
    }

    static NodePtr create(std::string name = "RNG") {
        NodePtr rng(new RandomNumberGenerator(name));
        return rng;
    }

protected:
    virtual void onProcess() {
        std::cout << "[RandomNumberGenerator] Processing..." << std::endl;

        nemo::RealVector values(nemo::dim(this->_dim), this->_rng);

        typename OutputPort<DATATYPE>::DataPtr out(
                new DATATYPE(values));
        outputPort->publish(out);
    }

    typename OutputPort<DATATYPE>::Ptr outputPort;

private:
    double _lower, _upper;

    unsigned int _dim;

    mutable nemo::uniform<double> _rng;
};

}