PSO_Optimizer.h

Go to the documentation of this file.

//*************************************************************************//
//                                                                         //
//  LibCudaOptimize                                                        //
//  Copyright (C) 2012 Ibislab, University of Parma                        //
//  Authors: Youssef S.G. Nashed, Roberto Ugolotti                         //
//                                                                         //
//  You should have received a copy of the GNU General Public License      //
//  along with this program.  If not, see <http://www.gnu.org/licenses/>   //
//                                                                         //
//*************************************************************************//

#ifndef PSO_Optimizer_H
#define PSO_Optimizer_H

#include "IOptimizer.h"
#include "IOptimizer.cuh"

/*open the namespace
  this code should be after all the #include lines and before the first line of the code in each file.
  by Hamid. 23/07/2012*/
#ifdef LIBCUDAOPTIMIZE_NAMESPACE
OPEN_NAMESPACE
#endif

typedef enum 
{
  LBEST,                
  GBEST,                
  STAR,                 
  RANDOM,               
  VON_NEUMANN   
} TOPOLOGY;

class PSO_Optimizer :
        public virtual IOptimizer
{
protected:

#ifndef PRIVATE_DOXYGEN

        SolutionSet m_bestSolutionSet;
        float* m_dVelocities;

        uint32_t* m_dLocalBestIDs;
        uint32_t* m_dUpdate;

        //PSO parameters
        float m_W;
        float m_C1;
        float m_C2;
        
        TOPOLOGY m_Topology;
#endif //private doxygen
        virtual void initSolutions(dim3,dim3);
        virtual void fitnessEvaluation(dim3,dim3,bool first=false);
        virtual void step(dim3,dim3);
        virtual void update(dim3,dim3,bool first=false);
        virtual void findBest();

public:
        PSO_Optimizer(EvalFuncPtr fPtr=NULL, unsigned int dn=PROBLEM_DIMENSIONS, unsigned int sn=NUM_SETS, unsigned int pn=NUM_SOLUTIONS);
        virtual ~PSO_Optimizer();
        
        virtual bool init();
        virtual void free();

        void setInertia(float w)                                {m_W = w;}
        void setC1(float c1)                                    {m_C1 = c1;}
        void setC2(float c2)                                    {m_C2 = c2;}
        void setTopology (TOPOLOGY topology)    {m_Topology=topology;}

        float* getBestSolution()                                                        {return m_bestSolutionSet.getHostBestPositions();}
        float* getBestSolution(unsigned int sid)                        {checkSetNumber(sid); return m_bestSolutionSet.getHostBestPositions(sid);}
        float* getBestFitnesses()                                                       {return m_bestSolutionSet.getHostBestFitnesses();}
        float  getBestFitness(unsigned int sid)                         {checkSetNumber(sid); return m_bestSolutionSet.getHostBestFitnesses(sid);}

};

/*close the namespace
  this code should be at the end of the file
  by Hamid. 23/07/2012*/
#ifdef LIBCUDAOPTIMIZE_NAMESPACE
CLOSE_NAMESPACE
#endif

#endif