experiments/evorobot/include/evorobotexperiment.h

/********************************************************************************
 *  FARSA Experiments Library                                                   *
 *  Copyright (C) 2007-2012                                                     *
 *  Stefano Nolfi <stefano.nolfi@istc.cnr.it>                                   *
 *  Onofrio Gigliotta <onofrio.gigliotta@istc.cnr.it>                           *
 *  Gianluca Massera <emmegian@yahoo.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  *
 ********************************************************************************/

#ifndef EVOROBOTEXPERIMENT_H
#define EVOROBOTEXPERIMENT_H

#include "resourcesuser.h"
#include "evonet.h"
#include "neuroninterfaces.h"
#include "robots.h"
#include "world.h"
#include "physphere.h"
#include "phybox.h"
#include "phycylinder.h"
#include "configurationparameters.h"
#include "parametersettable.h"

#include <QCoreApplication>
#include <QVector>
#include <QMap>
#include <QString>

namespace farsa {

class Evoga;

class FARSA_EXPERIMENTS_TEMPLATE WObjectsList : public QVector<WObject*>, public Resource
{
public:
    WObjectsList() :
        QVector()
    {
    }

    WObjectsList(int size) :
        QVector(size)
    {
    }

    WObjectsList(int size, WObject* const & value) :
        QVector(size, value)
    {
    }

    WObjectsList(const QVector<WObject*>& other) :
        QVector(other)
    {
    }
};

class FARSA_EXPERIMENTS_API EvoRobotExperiment : public QObject, public ParameterSettableWithConfigureFunction, public ConcurrentResourcesUser
{
    Q_OBJECT
public:
    EvoRobotExperiment();
    virtual ~EvoRobotExperiment();
    
    void setEvoga( Evoga* ga );
    Evoga* getEvoga();
    bool inBatchRunning();
    Evonet* getNeuralNetwork();
    void setNetParameters(int *genes);
    virtual void initGeneration(int generation);
    virtual void initIndividual(int individual);
    virtual void initTrial(int trial);
    virtual void initStep( int step );
    double getFitness();
    void stopTrial();
    void skipTrial();
    void restartTrial();
    void endIndividualLife();
    virtual void afterSensorsUpdate();
    virtual void beforeMotorsUpdate();
    virtual void endStep( int step ) = 0;
    virtual void endTrial(int trial);
    virtual void endIndividual(int individual);
    virtual void endGeneration(int generation);
    void doAllTrialsForIndividual(int individual);
    
    int getGenomeLength();
    Sensor* getSensor( QString sensorName );
    Motor* getMotor( QString motorName );

    virtual void setTestingAgentAndSeed( int idindividual, int nreplica );

    enum Phases {
        INTEST,
        INEVOLUTION,
        NONE
    };
    Phases getActivityPhase() {
        return gaPhase;
    };
    void setActivityPhase( Phases newPhase ) {
        gaPhase = newPhase;
    };

    virtual void configure(ConfigurationParameters& params, QString prefix);

    virtual void save(ConfigurationParameters& params, QString prefix);

    static void describe( QString type );

    virtual void postConfigureInitialization();

    int getNTrials() const
    {
        return ntrials;
    }

    int getNSteps() const
    {
        return nsteps;
    }

    int getCurStep() const
    {
        return nstep;
    }

    int getCurTrial() const
    {
        return ntrial;
    }

protected:
    int ntrials;            
    int nsteps;         
    int nstep;          
    int ntrial;         

    void recreateWorld();
    void recreateRobot();
    void recreateNeuralNetwork();
    
    double trialFitnessValue;
    double totalFitnessValue;

    //--- it's not anymore const, because createNeuralNetwork may modify some [NET]
    //    parameters; in particual nSensors, nHiddens and nMotors
    ConfigurationParameters* savedConfigurationParameters;
    const QString* savedPrefix;

private:
    void doTrial();
    void doStep();

    // --- the attributes of the class should be private
    Evoga* ga;
    Evonet *evonet;
    EvonetIterator* neuronsIterator;
    World* world;
    Robot* robot;
    QVector<Sensor*> sensors;
    QVector<Motor*> motors;
    QMap<QString, Sensor*> sensorsMap;
    QMap<QString, Motor*> motorsMap;
    Phases gaPhase;
    bool stopCurrentTrial;
    bool skipCurrentTrial;
    bool restartCurrentTrial;
    bool endCurrentIndividualLife;
    bool batchRunning;
};

} // end namespace farsa

#endif