potential_calculator.h

Go to the documentation of this file.

#ifndef HIGH_PRECISION_PSO_STATISTICS_POTENTIAL_CALCULATOR_H_
#define HIGH_PRECISION_PSO_STATISTICS_POTENTIAL_CALCULATOR_H_

#include <gmp.h>
#include <string>
#include <vector>

#include "general/particle.h"
#include "statistics/specific_statistical_evaluations.h"

namespace highprecisionpso {

class PotentialCalculator : public SpecificStatisticalEvaluation {
public:
    virtual mpf_t* Evaluate(Particle* p, int dim) = 0;
    std::vector<std::vector<mpf_t*> > Evaluate();
};

class PotentialFunctionDifference : public PotentialCalculator {
public:
    mpf_t* Evaluate(Particle* p, int dim);
    std::string GetName();
};

class PotentialAbsVelocityPlusDistToGlobalAttractor : public PotentialCalculator {
public:
    PotentialAbsVelocityPlusDistToGlobalAttractor(double scale);
    mpf_t* Evaluate(Particle* p, int dim);
    std::string GetName();

private:
    const double scale_;
};

class PotentialSqrtAbsVelocityPlusSqrtDistToGlobalAttractor : public PotentialCalculator {
public:
    PotentialSqrtAbsVelocityPlusSqrtDistToGlobalAttractor(double scale);
    mpf_t* Evaluate(Particle* p, int dim);
    std::string GetName();

private:
    const double scale_;
};

} // namespace highprecisionpso

#endif /* HIGH_PRECISION_PSO_STATISTICS_POTENTIAL_CALCULATOR_H_ */