net.sourceforge.cilib.measurement.single.diversity.centerinitialisationstrategies.SpatialCenterInitialisationStrategy Maven / Gradle / Ivy

Go to download

/**           __  __
 *    _____ _/ /_/ /_    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.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(fj.data.List topology) {
        int numberOfEntities = topology.length();
        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);
        }

        return averageEntityPosition.divide(numberOfEntities);
    }
}