SolutionSet.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 __SOLUTION_SET_H
#define __SOLUTION_SET_H

#include <vector>
#include <stdint.h>
#include "FitnessSet.h"


#ifdef LIBCUDAOPTIMIZE_NAMESPACE
OPEN_NAMESPACE
#endif


class SolutionSet
{
private:
        unsigned int m_setNumber;
        unsigned int m_solutionNumber;
        unsigned int m_problemDimension;
        unsigned int m_actualSolutionSize;

    FitnessSet m_fitnessSet;

        float* m_dPositions;

        std::vector<float> m_hPositions;
        std::vector<float> m_hFitnesses;
        std::vector<float> m_hBestFitnesses;
        std::vector<float> m_hBestPositions;

        float* m_dGlobalBestFitness;
        uint32_t* m_dGlobalBestID;

    bool m_hostPositionsUpdate;
    bool m_hostFitnessesUpdate;
    bool m_hostBestPositionsUpdate;
    bool m_hostBestFitnessesUpdate;

    std::string m_loggerFile;
    LOG_LEVEL m_verbosity;

    void invalidateAllUpdateFlags();
    void invalidatePositionUpdateFlags();
    void invalidateFitnessUpdateFlags();

        void allocateMemory();

        void cpyDeviceToHost(float* dst, const float* src, unsigned int count);
        void cpyHostToDevice(float* dst, const float* src, unsigned int count);
        void checkSetNumber(unsigned int sId) const;
        void checkSolutionNumber(unsigned int sId) const;

public:
        SolutionSet(unsigned int dn, unsigned int sn, unsigned int pn);
        virtual ~SolutionSet();
    
    FitnessSet* getFitnessSet() { return &m_fitnessSet; }

    void init(LOG_LEVEL verbosity, std::string loggerFile);

        void free();
        void findBests(bool maximize=false);

        void setSetNumber(unsigned int sn){m_setNumber = sn;}
        void setSolutionNumber(unsigned int pn){m_solutionNumber = pn;}
        void setProblemDimension(unsigned int dn){m_problemDimension = dn; m_actualSolutionSize = iAlignUp(dn, 32);}
        void setParams(unsigned int sn, unsigned int pn, unsigned int dn)
        {setSetNumber(sn);setSolutionNumber(pn);setProblemDimension(dn);}
        void setParams(const SolutionSet& src)
        {setParams(src.getSetNumber(), src.getSolutionNumber(), src.getProblemDimension());}
        void setDevicePositions(float* ptr, POINTER_LOCATION location=HOST_POINTER, unsigned int sid = ALL_SETS);
        void setDevicePositionsPtr(float* ptr){invalidatePositionUpdateFlags(); m_dPositions=ptr;}
        void setDeviceFitnesses(float* ptr, POINTER_LOCATION location=HOST_POINTER, unsigned int sid = ALL_SETS);
        void setDeviceFitnessesPtr(float* ptr){invalidateFitnessUpdateFlags(); m_fitnessSet.setPtr(ptr);}
    void setDeviceFitnesses(FitnessSet fitn);
    
        unsigned int getSetNumber() const {return m_setNumber;}
        unsigned int getSolutionNumber() const {return m_solutionNumber;}
        unsigned int getProblemDimension() const {return m_problemDimension;}
        unsigned int getActualSolutionSize() const {return m_actualSolutionSize;}
    float* getDevicePositions() {invalidatePositionUpdateFlags(); return m_dPositions;}
        const float* getDevicePositionsConst() const {return m_dPositions;}
        float* getDeviceFitnesses() {invalidateFitnessUpdateFlags(); return m_fitnessSet.get();}
    const float* getDeviceFitnessesConst() const {return m_fitnessSet.get();}
    float* getDeviceBestFitnesses() {m_hostBestFitnessesUpdate = false; return m_dGlobalBestFitness;}
        const float* getDeviceBestFitnessesConst() const {return m_dGlobalBestFitness;}
    uint32_t* getDeviceBestIDs() {m_hostBestFitnessesUpdate = m_hostBestPositionsUpdate = false; return m_dGlobalBestID;}
        const uint32_t* getDeviceBestIDsConst() const {return m_dGlobalBestID;}
    float* getDevicePositions(unsigned int sid) {checkSetNumber(sid); invalidatePositionUpdateFlags(); return m_dPositions+(sid*m_solutionNumber*m_actualSolutionSize);}
        const float* getDevicePositionsConst(unsigned int sid) const {checkSetNumber(sid); return m_dPositions+(sid*m_solutionNumber*m_actualSolutionSize);}
    float* getDeviceFitnesses(unsigned int sid) {checkSetNumber(sid); invalidateFitnessUpdateFlags(); return m_fitnessSet.get(sid);}
        const float* getDeviceFitnessesConst(unsigned int sid) const {checkSetNumber(sid); return m_fitnessSet.get(sid);}
    float* getDeviceBestFitnesses(unsigned int sid) {checkSetNumber(sid); m_hostBestFitnessesUpdate = false; return m_dGlobalBestFitness+sid;}
        const float* getDeviceBestFitnessesConst(unsigned int sid) const {checkSetNumber(sid); return m_dGlobalBestFitness+sid;}
    uint32_t* getDeviceBestIDs(unsigned int sid) {checkSetNumber(sid); m_hostBestFitnessesUpdate = m_hostBestPositionsUpdate = false; return m_dGlobalBestID+sid;}
        const uint32_t* getDeviceBestIDsConst(unsigned int sid)const {checkSetNumber(sid); return m_dGlobalBestID+sid;}
    float* getDevicePositions(unsigned int sid, unsigned int pid) {checkSetNumber(sid); checkSolutionNumber(pid); invalidatePositionUpdateFlags(); return m_dPositions+(((sid*m_solutionNumber)+pid)*m_actualSolutionSize);}
    const float* getDevicePositionsConst(unsigned int sid, unsigned int pid) const {checkSetNumber(sid); checkSolutionNumber(pid); return m_dPositions+(((sid*m_solutionNumber)+pid)*m_actualSolutionSize);}
    float* getDeviceFitnesses(unsigned int sid, unsigned int pid) {checkSetNumber(sid); checkSolutionNumber(pid); invalidateFitnessUpdateFlags(); return m_fitnessSet.get(sid, pid);}
    const float* getDeviceFitnessesConst(unsigned int sid, unsigned int pid) const {checkSetNumber(sid); checkSolutionNumber(pid); return m_fitnessSet.get(sid,pid);}

        float* getHostFitnesses();
        float* getHostBestFitnesses();
        float* getHostBestPositions();
        
        float* getHostFitnesses(unsigned int sid);
        float  getHostBestFitnesses(unsigned int sid);
        float* getHostBestPositions(unsigned int sid);

        void copyPositionsToHost();
        const float* getHostPositions(unsigned int sid=ALL_SETS, unsigned int pid=ALL_SOLUTIONS);
        const float* getHostPositionsConst(unsigned int sid=ALL_SETS, unsigned int pid=ALL_SOLUTIONS) const;
        float getHostFitnesses(unsigned int sid, unsigned int pid);
};

/*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