net.sourceforge.cilib.pso.guideprovider.ArithmeticGuideProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cilib-library Show documentation

A library of composable components enabling simpler Computational Intelligence

The newest version!

/**           __  __
 *    _____ _/ /_/ /_    Computational Intelligence Library (CIlib)
 *   / ___/ / / / __ \   (c) CIRG @ UP
 *  / /__/ / / / /_/ /   http://cilib.net
 *  \___/_/_/_/_.___/
 */
package net.sourceforge.cilib.pso.guideprovider;

import java.util.Iterator;

import net.sourceforge.cilib.algorithm.AbstractAlgorithm;
import net.sourceforge.cilib.entity.Topologies;
import net.sourceforge.cilib.entity.comparator.SocialBestFitnessComparator;
import net.sourceforge.cilib.math.random.UniformDistribution;
import net.sourceforge.cilib.pso.PSO;
import net.sourceforge.cilib.pso.particle.Particle;
import net.sourceforge.cilib.type.types.Bounds;
import net.sourceforge.cilib.type.types.Numeric;
import net.sourceforge.cilib.type.types.Real;
import net.sourceforge.cilib.type.types.container.StructuredType;
import net.sourceforge.cilib.type.types.container.Vector;
import fj.P;
import fj.P1;

public class ArithmeticGuideProvider implements GuideProvider {

    private boolean perDimension;

    public ArithmeticGuideProvider() {
        perDimension = false;
    }

    public GuideProvider getClone() {
        return this;
    }

    public StructuredType get(Particle particle) {
        PSO pso = (PSO) AbstractAlgorithm.get();
        fj.data.List topology = pso.getTopology();
        Particle gbestParticle = Topologies.getBestEntity(topology, new SocialBestFitnessComparator());
        Vector gbest = (Vector) gbestParticle.getBestPosition();
        Vector pbest = (Vector) particle.getBestPosition();
        P1 p1, p2;

        if (!perDimension) {
            UniformDistribution uniform = new UniformDistribution();
            double r = uniform.getRandomNumber();

            p1 = P.p(r);
            p2 = P.p(1.0 - r);
        } else {
            Vector r = Vector.newBuilder().copyOf(
                Vector.fill(Real.valueOf(0.0, new Bounds(0.0, 1.0)), particle.getDimension())
            ).buildRandom();

            Vector oneMinusR = Vector.fill(1.0, particle.getDimension()).subtract(r);

            p1 = new VectorP(r);
            p2 = new VectorP(oneMinusR);
        }

        return pbest.multiply(p1).plus(gbest.multiply(p2));
    }

    private static class VectorP extends P1 {

        private Iterator i;

        public VectorP(Vector v) {
            i = v.iterator();
        }

        @Override
        public Number _1() {
            return i.next().doubleValue();
        }

    }

    public boolean getPerDimension() {
        return perDimension;
    }

    public void setPerDimension(boolean perDimension) {
        this.perDimension = perDimension;
    }
}