kheperamotors.cpp

/********************************************************************************
 *  FARSA Experiments Library                                                   *
 *  Copyright (C) 2007-2012                                                     *
 *  Gianluca Massera <emmegian@yahoo.it>                                        *
 *  Stefano Nolfi <stefano.nolfi@istc.cnr.it>                                   *
 *  Tomassino Ferrauto <tomassino.ferrauto@istc.cnr.it>                         *
 *                                                                              *
 *  This program is free software; you can redistribute it and/or modify        *
 *  it under the terms of the GNU General Public License as published by        *
 *  the Free Software Foundation; either version 2 of the License, or           *
 *  (at your option) any later version.                                         *
 *                                                                              *
 *  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.                                *
 *                                                                              *
 *  You should have received a copy of the GNU General Public License           *
 *  along with this program; if not, write to the Free Software                 *
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA  *
 ********************************************************************************/

#include "kheperamotors.h"
#include "configurationhelper.h"
#include "randomgenerator.h"
#include "logger.h"

namespace farsa {

KheperaMotor::KheperaMotor(ConfigurationParameters& params, QString prefix) :
    Motor(params, prefix),
    m_kheperaResource("robot"),
    m_neuronsIteratorResource("neuronsIterator")
{
    // Reading parameters
    m_kheperaResource = ConfigurationHelper::getString(params, prefix + "khepera", m_kheperaResource);
    m_neuronsIteratorResource = ConfigurationHelper::getString(params, prefix + "neuronsIterator", m_neuronsIteratorResource);

    // Declaring the resources that are needed here
    usableResources(QStringList() << m_kheperaResource << m_neuronsIteratorResource);
}

KheperaMotor::~KheperaMotor()
{
    // Nothing to do here
}

void KheperaMotor::save(ConfigurationParameters& params, QString prefix)
{
    // Calling parent function
    Motor::save(params, prefix);

    // Saving parameters
    params.startObjectParameters(prefix, "KheperaMotor", this);
    params.createParameter(prefix, "khepera", m_kheperaResource);
    params.createParameter(prefix, "neuronsIterator", m_neuronsIteratorResource);
}

void KheperaMotor::describe(QString type)
{
    // Calling parent function
    Motor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The base class for khepera motors");
    d.describeString("khepera").def("robot").help("The name of the resource associated with the Khepera robot to use (default is \"robot\")");
    d.describeString("neuronsIterator").def("neuronsIterator").help("The name of the resource associated with the neural network iterator (default is \"neuronsIterator\")");
}

void KheperaMotor::resourceChanged(QString resourceName, ResourceChangeType changeType)
{
    // Calling parent function
    Motor::resourceChanged(resourceName, changeType);

    // Here we only check whether the resource has been deleted and reset the check flag, the
    // actual work is done in subclasses
    if (changeType == Deleted) {
        resetNeededResourcesCheck();
        return;
    }
}

KheperaWheelVelocityMotor::KheperaWheelVelocityMotor(ConfigurationParameters& params, QString prefix) :
    KheperaMotor(params, prefix),
    m_robot(NULL),
    m_neuronsIterator(NULL)
{
}

KheperaWheelVelocityMotor::~KheperaWheelVelocityMotor()
{
    // Nothing to do here
}

void KheperaWheelVelocityMotor::save(ConfigurationParameters& params, QString prefix)
{
    // Calling parent function
    KheperaMotor::save(params, prefix);

    // Saving parameters
    params.startObjectParameters(prefix, "KheperaWheelVelocityMotor", this);
}

void KheperaWheelVelocityMotor::describe(QString type)
{
    // Calling parent function
    KheperaMotor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The motor controlling the velocity of the wheels of the khepera robot");
}

void KheperaWheelVelocityMotor::update()
{
    // Checking all resources we need exist
    checkAllNeededResourcesExist();

    // Acquiring the lock to get resources
    ResourcesLocker locker( this );

    // Getting limit velocities for wheels
    double minSpeed1;
    double minSpeed2;
    double maxSpeed1;
    double maxSpeed2;
    m_robot->wheelsController()->getSpeedLimits(minSpeed1, minSpeed2, maxSpeed1, maxSpeed2);

    // Computing desired wheel velocities
    m_neuronsIterator->setCurrentBlock(name());
    const double v1 = (maxSpeed1 - minSpeed1) * m_neuronsIterator->getOutput() + minSpeed1;
    m_neuronsIterator->nextNeuron();
    const double v2 = (maxSpeed2 - minSpeed2) * m_neuronsIterator->getOutput() + minSpeed2;

    m_robot->wheelsController()->setSpeeds(v1, v2);
}

int KheperaWheelVelocityMotor::size()
{
    return 2;
}

void KheperaWheelVelocityMotor::resourceChanged(QString resourceName, ResourceChangeType changeType)
{
    // Calling parent function
    KheperaMotor::resourceChanged(resourceName, changeType);

    if (changeType == Deleted) {
        return;
    }

    if (resourceName == m_kheperaResource) {
        m_robot = getResource<PhyKhepera>();
    } else if (resourceName == m_neuronsIteratorResource) {
        m_neuronsIterator = getResource<NeuronsIterator>();
        m_neuronsIterator->setCurrentBlock(name());
        for (int i = 0; i < size(); i++, m_neuronsIterator->nextNeuron()) {
            m_neuronsIterator->setGraphicProperties("W" + QString::number(i), 0.0, 1.0, Qt::red);
        }
    } else {
        Logger::info("Unknown resource " + resourceName + " for " + name());
    }
}

}