testpso.cpp

Go to the documentation of this file.

#include <stdio.h>
#include <time.h>  
#include <string>
#include <fstream>  
#include <iostream>
#include "pso/pso.h"

using namespace std;
using namespace pso;

int main()
{
    Particle*p[40];  
    float w;
  
    Particle*temp;  
    float randX,randY;  
    srand((int)time(NULL));  
    for(int i=0;i<40;++i)  
    {
        randX=(rand()/(float)RAND_MAX)*30.0f;  
        randY=(rand()/(float)RAND_MAX)*30.0f;  
        cout<<"randX="<<randX<<endl;  
        cout<<"randY="<<randY<<endl;  
  
        p[i]=new Particle(randX,randY);  
        cout<<"The temp info is X:"<<p[i]->getX()<<" Y:"<<p[i]->getY()<<endl;  
      
    }  

    Coordinate gBest;
    int bestIndex=0;  
    float bestP;
    bestP=p[0]->getP();  
    gBest=p[0]->getPBest();  
    for(int i=1;i<40;++i)  
    {  
        if(p[i]->getP()<bestP)  
        {  
            bestP=p[i]->getP();  
            gBest=p[i]->getPBest();  
            bestIndex=i;  
        }  
    }  

    cout<<"Now the initial gBest is X:"<<gBest.x<<" Y:"<<gBest.y<<endl;  
    cout<<"And the p[0] is X:"<<p[0]->getX()<<" Y:"<<p[0]->getY()<<endl;  
    cout<<"And the p[39] is X:"<<p[39]->getX()<<" Y:"<<p[39]->getY()<<endl;  
    cout<<"Now p[0].p="<<p[0]->getP()<<endl;

    for(int k=0;k<100;++k)
    {
        w=0.9f-(0.9f-0.4f)*k/99.0f;

        for(int i=0;i<40;++i)  
        {  
            temp=p[i];  
            temp->setV(gBest,w);  
            temp->setCoordinate();  
            temp->setP();
        }  
        bestP=p[0]->getP();  
        gBest=p[0]->getPBest();  
        for(int i=1;i<40;++i)  
        {  
            temp=p[i];  
            if(temp->getP()<bestP)  
            {  
                bestP=temp->getP();  
                gBest=temp->getPBest();  
                bestIndex=i;  
            }
        }  
        cout<<"Now gBest is X:"<<gBest.x<<" Y:"<<gBest.y<<" and the minP="<<p[bestIndex]->getP()<<endl;  
        cout<<"bestIndex="<<bestIndex<<endl;  
    }
  
    getchar();
    return 0;
}