2a816665d7
git-svn-id: svn://scm.gforge.inria.fr/svnroot/paradiseo@901 331e1502-861f-0410-8da2-ba01fb791d7f
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
#include <peo>
|
|
#include <es.h>
|
|
|
|
typedef eoRealParticle < double >Indi;
|
|
|
|
double f (const Indi & _indi)
|
|
{
|
|
double sum;
|
|
sum=_indi[1]-pow(_indi[0],2);
|
|
sum=100*pow(sum,2);
|
|
sum+=pow((1-_indi[0]),2);
|
|
return (-sum);
|
|
}
|
|
|
|
template <class EOT>
|
|
class eoResizerInit: public eoInit<EOT>
|
|
{
|
|
public:
|
|
|
|
typedef typename EOT::AtomType AtomType;
|
|
|
|
eoResizerInit(unsigned _size)
|
|
: size(_size){}
|
|
|
|
virtual void operator()(EOT& chrom)
|
|
{
|
|
chrom.resize(size);
|
|
chrom.invalidate();
|
|
}
|
|
private :
|
|
unsigned size;
|
|
};
|
|
|
|
|
|
int main( int __argc, char** __argv )
|
|
{
|
|
|
|
peo :: init( __argc, __argv );
|
|
const unsigned int VEC_SIZE = 2;
|
|
const unsigned int POP_SIZE = 20;
|
|
const unsigned int NEIGHBORHOOD_SIZE= 6;
|
|
const unsigned int MAX_GEN = 300;
|
|
const double INIT_POSITION_MIN = -2.0;
|
|
const double INIT_POSITION_MAX = 2.0;
|
|
const double INIT_VELOCITY_MIN = -1.;
|
|
const double INIT_VELOCITY_MAX = 1.;
|
|
const double omega = 1;
|
|
const double C1 = 0.5;
|
|
const double C2 = 2.;
|
|
rng.reseed (time(0));
|
|
eoGenContinue < Indi > genContPara (MAX_GEN);
|
|
eoCombinedContinue <Indi> continuatorPara (genContPara);
|
|
eoCheckPoint<Indi> checkpoint(continuatorPara);
|
|
eoUniformGenerator < double >uGen (INIT_POSITION_MIN, INIT_POSITION_MAX);
|
|
eoUniformGenerator < double >sGen (INIT_VELOCITY_MIN, INIT_VELOCITY_MAX);
|
|
eoVelocityInitFixedLength < Indi > veloRandom (VEC_SIZE, sGen);
|
|
eoFirstIsBestInit < Indi > localInit;
|
|
eoRealVectorBounds bndsFlight(VEC_SIZE,INIT_POSITION_MIN,INIT_POSITION_MAX);
|
|
eoStandardFlight < Indi > flight(bndsFlight);
|
|
eoLinearTopology<Indi> topology(NEIGHBORHOOD_SIZE);
|
|
eoRealVectorBounds bnds(VEC_SIZE,INIT_VELOCITY_MIN,INIT_VELOCITY_MAX);
|
|
eoStandardVelocity < Indi > velocity (topology,omega,C1,C2,bnds);
|
|
|
|
eoResizerInit<Indi> sizeInit(VEC_SIZE);
|
|
// need at least a size initialization
|
|
eoPop < Indi > pop(POP_SIZE,sizeInit);
|
|
eoInitFixedLength < Indi > randomSeq (VEC_SIZE, uGen);
|
|
peoMultiStart <Indi> initRandom (randomSeq);
|
|
peoWrapper random (initRandom,pop);
|
|
initRandom.setOwner(random);
|
|
peo :: run( );
|
|
peo :: finalize( );
|
|
|
|
// Parallel algorithm
|
|
|
|
peo :: init (__argc, __argv);
|
|
peoEvalFunc<Indi, double, const Indi& > plainEval(f);
|
|
peoPopEval< Indi > eval(plainEval);
|
|
eoInitializer <Indi> init(eval,veloRandom,localInit,topology,pop);
|
|
eoSyncEasyPSO <Indi> psa(init,checkpoint,eval, velocity, flight);
|
|
peoWrapper parallelPSO( psa, pop);
|
|
eval.setOwner(parallelPSO);
|
|
peo :: run();
|
|
peo :: finalize();
|
|
if (getNodeRank()==1)
|
|
{
|
|
pop.sort();
|
|
std::cout << "Final population :\n" << pop << std::endl;
|
|
}
|
|
}
|