net.sourceforge.cilib.measurement.single.diversity.centerinitialisationstrategies.SpatialCenterInitialisationStrategy 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

There is a newer version: 0.8

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

import java.util.Iterator;
import net.sourceforge.cilib.entity.Entity;
import net.sourceforge.cilib.entity.Topology;
import net.sourceforge.cilib.type.types.container.Vector;

/**
 * Returns the center of a given topology where the center is the average position
 * of all entities in the topology.
 */
public class SpatialCenterInitialisationStrategy implements CenterInitialisationStrategy {

    /**
     * {@inheritDoc}
     */
    @Override
    public Vector getCenter(Topology topology) {
        int numberOfEntities = topology.size();
        Iterator averageIterator = topology.iterator();
        Entity entity = averageIterator.next();
        Vector averageEntityPosition = (Vector) entity.getCandidateSolution().getClone();

        while (averageIterator.hasNext()) {
            entity = averageIterator.next();
            Vector entityContents = (Vector) entity.getCandidateSolution();
            averageEntityPosition = averageEntityPosition.plus(entityContents);
        }

        averageEntityPosition.divide(numberOfEntities);

        return averageEntityPosition;
    }
}