/**
* Copyright (C) 2010-2017 Gordon Fraser, Andrea Arcuri and EvoSuite
* contributors
*
* This file is part of EvoSuite.
*
* EvoSuite is free software: you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation, either version 3.0 of the License, or
* (at your option) any later version.
*
* EvoSuite is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with EvoSuite. If not, see <http://www.gnu.org/licenses/>.
*/
package org.evosuite.ga.comparators;
import java.util.Comparator;
import org.evosuite.ga.Chromosome;
import org.evosuite.ga.FitnessFunction;
/**
* Sort a Collection of Chromosomes by their fitness value
*
* @author José Campos
*/
public class SortByFitness
implements Comparator<Chromosome>
{
private FitnessFunction<?> ff;
private boolean order;
/**
*
* @param ff
* @param des descending order
*/
public SortByFitness(FitnessFunction<?> ff, boolean desc) {
this.ff = ff;
this.order = desc;
}
@Override
public int compare(Chromosome c1, Chromosome c2)
{
if (c1 == null)
return 1;
else if (c2 == null)
return -1;
double objetive1 = c1.getFitness(this.ff);
double objetive2 = c2.getFitness(this.ff);
if (this.order) {
if (objetive1 < objetive2)
return 1;
else if (objetive1 > objetive2)
return -1;
else
return 0;
}
else {
if (objetive1 < objetive2)
return -1;
else if (objetive1 > objetive2)
return 1;
else
return 0;
}
}
}