epucksensors.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>                         *
 *  Onofrio Gigliotta <onofrio.gigliotta@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 "epucksensors.h"
#include "configurationhelper.h"
#include "logger.h"
#include "graphicalwobject.h"

namespace farsa {

EpuckSensor::EpuckSensor(ConfigurationParameters& params, QString prefix) :
    Sensor(params, prefix),
    m_epuckResource("robot"),
    m_neuronsIteratorResource("neuronsIterator")
{
    // Reading parameters
    m_epuckResource = ConfigurationHelper::getString(params, prefix + "epuck", m_epuckResource);
    m_neuronsIteratorResource = ConfigurationHelper::getString(params, prefix + "neuronsIterator", m_neuronsIteratorResource);

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

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

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

    // Saving parameters
    params.startObjectParameters(prefix, "EpuckSensor", this);
    params.createParameter(prefix, "epuck", m_epuckResource);
    params.createParameter(prefix, "neuronsIterator", m_neuronsIteratorResource);
}

void EpuckSensor::describe(QString type)
{
    // Calling parent function
    Sensor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The base class for e-puck sensors");
    d.describeString("epuck").def("robot").help("the name of the resource associated with the e-puck 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 EpuckSensor::resourceChanged(QString resourceName, ResourceChangeType changeType)
{
    // Calling parent function
    Sensor::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;
    }
}

EpuckProximityIRSensor::EpuckProximityIRSensor(ConfigurationParameters& params, QString prefix) :
    EpuckSensor(params, prefix),
    m_robot(NULL),
    m_neuronsIterator(NULL)
{
}

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

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

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

void EpuckProximityIRSensor::describe(QString type)
{
    // Calling parent function
    EpuckSensor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The infrared proximity sensors of the e-puck robot", "The infrared proximity sensors of the e-puck robot. These are the very short range IR sensors all around the base");
}

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

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

    m_neuronsIterator->setCurrentBlock(name());
    for (int i = 0; i < size(); i++, m_neuronsIterator->nextNeuron()) {
        m_neuronsIterator->setInput(m_robot->proximityIRSensorController()->activation(i));
    }
}

int EpuckProximityIRSensor::size()
{
    return 8;
}

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

    if (changeType == Deleted) {
        return;
    }

    if (resourceName == m_epuckResource) {
        m_robot = getResource<PhyEpuck>();

        // Eanbling sensors
        m_robot->proximityIRSensorController()->setEnabled(true);
    } 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("ir" + QString::number(i), 0.0, 1.0, Qt::red);
        }
    } else {
        Logger::info("Unknown resource " + resourceName + " for " + name());
    }
}

EpuckGroundIRSensor::EpuckGroundIRSensor(ConfigurationParameters& params, QString prefix) :
    EpuckSensor(params, prefix),
    m_robot(NULL),
    m_neuronsIterator(NULL)
{
}

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

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

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

void EpuckGroundIRSensor::describe(QString type)
{
    // Calling parent function
    EpuckSensor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The infrared ground sensors of the e-puck robot", "The infrared ground sensors of the e-puck robot. These are the three ground sensors in the frontal part of the robot.");
}

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

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

    m_neuronsIterator->setCurrentBlock(name());
    for (int i = 0; i < size(); i++, m_neuronsIterator->nextNeuron()) {
        m_neuronsIterator->setInput(m_robot->groundIRSensorController()->activation(i));
    }
}

int EpuckGroundIRSensor::size()
{
    return 3;
}

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

    if (changeType == Deleted) {
        return;
    }

    if (resourceName == m_epuckResource) {
        m_robot = getResource<PhyEpuck>();

        // Eanbling sensors
        m_robot->groundIRSensorController()->setEnabled(true);
    } 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("gb" + QString::number(i), 0.0, 1.0, Qt::red);
        }
    } else {
        Logger::info("Unknown resource " + resourceName + " for " + name());
    }
}

EpuckLinearCameraSensor::EpuckLinearCameraSensor(ConfigurationParameters& params, QString prefix) :
    EpuckSensor(params, prefix),
    m_robot(NULL),
    m_neuronsIterator(NULL),
    m_numReceptors(ConfigurationHelper::getInt(params, prefix + "numReceptors", 8)),
    m_aperture(ConfigurationHelper::getDouble(params, prefix + "aperture", 36.0f)),
    m_camera(NULL),
    m_drawCamera(ConfigurationHelper::getBool(params, prefix + "drawCamera", true))
{
}

EpuckLinearCameraSensor::~EpuckLinearCameraSensor()
{
    // Deleting the camera
    delete m_camera;
}

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

    // Saving parameters
    params.startObjectParameters(prefix, "EpuckLinearCameraSensor", this);
    params.createParameter(prefix, "numReceptors", QString::number(m_numReceptors));
    params.createParameter(prefix, "aperture", QString::number(m_aperture));
    params.createParameter(prefix, "drawCamera", (m_drawCamera ? "true" : "false"));
}

void EpuckLinearCameraSensor::describe(QString type)
{
    // Calling parent function
    EpuckSensor::describe(type);

    // Describing our parameters
    Descriptor d = addTypeDescription(type, "The linear camera sensor of the e-puck robot", "This is a linear camera with configurable aperture");
    d.describeInt("numReceptors").def(8).limits(1, MaxInteger).help("The number of receptors of the sensor", "Each receptor returns three values, one for each of the three colors (red, green, blue). This means that the size returned by this sensor is 3 * numReceptors (default is 8)");
    d.describeReal("aperture").def(360.0f).limits(0.0f, 360.0f).help("The aperture of the camera in degrees", "The real e-puck has a camera with an aperture of 36 degrees, but here you can use any value up to 360° (default is 36)");
    d.describeBool("drawCamera").def(true).help("Whether to draw the camera or not");
}

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

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

    // Updating the camera
    m_camera->update();

    // Reading activations: first the red one, then the green one and finally the blue one
    m_neuronsIterator->setCurrentBlock(name());
    for (int i = 0; i < m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
        m_neuronsIterator->setInput(m_camera->colorForReceptor(i).redF());
    }
    for (int i = 0; i < m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
        m_neuronsIterator->setInput(m_camera->colorForReceptor(i).greenF());
    }
    for (int i = 0; i < m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
        m_neuronsIterator->setInput(m_camera->colorForReceptor(i).blueF());
    }
}

int EpuckLinearCameraSensor::size()
{
    return m_numReceptors * 3;
}

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

    if (changeType == Deleted) {
        // Deleting the camera if the robot was deleted
        if (resourceName == m_epuckResource) {
            delete m_camera;
            m_camera = NULL;
        }

        return;
    }

    if (resourceName == m_epuckResource) {
        m_robot = getResource<PhyEpuck>();

        // Now we can also create the camera
        wMatrix mtr = wMatrix::roll(-PI_GRECO / 2.0);
#ifdef __GNUC__
    #warning QUI INVECE DI UNA COSTANTE, CALCOLARSI UNA POSIZIONE DALLE DIMENSIONI DEL ROBOT
#endif
        mtr.w_pos.z = 0.06;
        m_camera = new LinearCamera(m_robot, mtr, toRad(m_aperture), m_numReceptors, Qt::black);

        // Sharing resources with the camera
        m_camera->shareResourcesWith(this);
        m_camera->drawCamera(m_drawCamera);
    } else if (resourceName == m_neuronsIteratorResource) {
        m_neuronsIterator = getResource<NeuronsIterator>();
        m_neuronsIterator->setCurrentBlock(name());
        int i = 0;
        for (; i < m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
            m_neuronsIterator->setGraphicProperties("lr" + QString::number(i % m_numReceptors), 0.0, 1.0, Qt::red);
        }
        for (; i < 2 * m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
            m_neuronsIterator->setGraphicProperties("lg" + QString::number(i % m_numReceptors), 0.0, 1.0, Qt::red);
        }
        for (; i < 3 * m_numReceptors; i++, m_neuronsIterator->nextNeuron()) {
            m_neuronsIterator->setGraphicProperties("lb" + QString::number(i % m_numReceptors), 0.0, 1.0, Qt::red);
        }
    } else {
        Logger::info("Unknown resource " + resourceName + " for " + name());
    }
}

}