FitnessFunctions.java example

Explorer
evosuite-master
/**
 * 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.coverage;

import org.evosuite.Properties;
import org.evosuite.Properties.Criterion;
import org.evosuite.coverage.ambiguity.AmbiguityCoverageFactory;
import org.evosuite.coverage.ambiguity.AmbiguityCoverageSuiteFitness;
import org.evosuite.coverage.branch.BranchCoverageFactory;
import org.evosuite.coverage.branch.BranchCoverageSuiteFitness;
import org.evosuite.coverage.branch.OnlyBranchCoverageFactory;
import org.evosuite.coverage.branch.OnlyBranchCoverageSuiteFitness;
import org.evosuite.coverage.cbranch.CBranchFitnessFactory;
import org.evosuite.coverage.cbranch.CBranchSuiteFitness;
import org.evosuite.coverage.dataflow.AllDefsCoverageFactory;
import org.evosuite.coverage.dataflow.AllDefsCoverageSuiteFitness;
import org.evosuite.coverage.dataflow.DefUseCoverageFactory;
import org.evosuite.coverage.dataflow.DefUseCoverageSuiteFitness;
import org.evosuite.coverage.exception.ExceptionCoverageFactory;
import org.evosuite.coverage.exception.ExceptionCoverageSuiteFitness;
import org.evosuite.coverage.exception.TryCatchCoverageFactory;
import org.evosuite.coverage.exception.TryCatchCoverageSuiteFitness;
import org.evosuite.coverage.ibranch.IBranchFitnessFactory;
import org.evosuite.coverage.ibranch.IBranchSuiteFitness;
import org.evosuite.coverage.io.input.InputCoverageFactory;
import org.evosuite.coverage.io.input.InputCoverageSuiteFitness;
import org.evosuite.coverage.io.output.OutputCoverageFactory;
import org.evosuite.coverage.io.output.OutputCoverageSuiteFitness;
import org.evosuite.coverage.line.LineCoverageFactory;
import org.evosuite.coverage.line.LineCoverageSuiteFitness;
import org.evosuite.coverage.line.OnlyLineCoverageSuiteFitness;
import org.evosuite.coverage.method.*;
import org.evosuite.coverage.mutation.*;
import org.evosuite.coverage.readability.ReadabilitySuiteFitness;
import org.evosuite.coverage.rho.RhoCoverageFactory;
import org.evosuite.coverage.rho.RhoCoverageSuiteFitness;
import org.evosuite.coverage.statement.StatementCoverageFactory;
import org.evosuite.coverage.statement.StatementCoverageSuiteFitness;
import org.evosuite.regression.RegressionSuiteFitness;
import org.evosuite.testcase.TestFitnessFunction;
import org.evosuite.testsuite.TestSuiteFitnessFunction;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * factory class for fitness functions
 * @author mattia
 *
 */
public class FitnessFunctions {
	
	private static Logger logger = LoggerFactory.getLogger(FitnessFunctions.class);

	/**
	 * <p>
	 * getFitnessFunction
	 * </p>
	 * 
	 * @param criterion
	 *            a {@link org.evosuite.Properties.Criterion} object.
	 * @return a {@link org.evosuite.testsuite.TestSuiteFitnessFunction} object.
	 */
	public static TestSuiteFitnessFunction getFitnessFunction(Criterion criterion) {
		switch (criterion) {
		case STRONGMUTATION:
			return new StrongMutationSuiteFitness();
		case WEAKMUTATION:
			return new WeakMutationSuiteFitness();
		case MUTATION:
			return new StrongMutationSuiteFitness();
		case ONLYMUTATION:
			return new OnlyMutationSuiteFitness();
		case DEFUSE:
			return new DefUseCoverageSuiteFitness();
		case BRANCH:
			return new BranchCoverageSuiteFitness();
		case CBRANCH:
			return new CBranchSuiteFitness();
		case IBRANCH:
			return new IBranchSuiteFitness();
		case STATEMENT:
			return new StatementCoverageSuiteFitness();
		case RHO:
			return new RhoCoverageSuiteFitness();
		case AMBIGUITY:
			return new AmbiguityCoverageSuiteFitness();
		case ALLDEFS:
			return new AllDefsCoverageSuiteFitness();
		case EXCEPTION:
			return new ExceptionCoverageSuiteFitness();
		case REGRESSION:
			return new RegressionSuiteFitness();
		case READABILITY:
			return new ReadabilitySuiteFitness();
		case ONLYBRANCH:
			return new OnlyBranchCoverageSuiteFitness();
		case METHODTRACE:
			return new MethodTraceCoverageSuiteFitness();
		case METHOD:
			return new MethodCoverageSuiteFitness();
		case METHODNOEXCEPTION:
			return new MethodNoExceptionCoverageSuiteFitness();
		case ONLYLINE:
			return new OnlyLineCoverageSuiteFitness();
		case LINE:
			return new LineCoverageSuiteFitness();
		case OUTPUT:
			return new OutputCoverageSuiteFitness();
		case INPUT:
			return new InputCoverageSuiteFitness();
		case TRYCATCH:
			return new TryCatchCoverageSuiteFitness();
		default:
			logger.warn("No TestSuiteFitnessFunction defined for " + Properties.CRITERION
			        + " using default one (BranchCoverageSuiteFitness)");
			return new BranchCoverageSuiteFitness();
		}
	}
	
	/**
	 * <p>
	 * getFitnessFactory
	 * </p>
	 * 
	 * @param crit
	 *            a {@link org.evosuite.Properties.Criterion} object.
	 * @return a {@link org.evosuite.coverage.TestFitnessFactory} object.
	 */
	public static TestFitnessFactory<? extends TestFitnessFunction> getFitnessFactory(
	        Criterion crit) {
		switch (crit) {
		case STRONGMUTATION:
		case MUTATION:
			return new MutationFactory();
		case WEAKMUTATION:
			return new MutationFactory(false);
		case ONLYMUTATION:
			return new OnlyMutationFactory();
		case DEFUSE:
			return new DefUseCoverageFactory();
		case BRANCH:
			return new BranchCoverageFactory();
		case CBRANCH:
			return new CBranchFitnessFactory();
		case IBRANCH:
			return new IBranchFitnessFactory();
		case STATEMENT:
			return new StatementCoverageFactory();
		case RHO:
			return new RhoCoverageFactory();
		case AMBIGUITY:
			return new AmbiguityCoverageFactory();
		case ALLDEFS:
			return new AllDefsCoverageFactory();
		case EXCEPTION:
			return new ExceptionCoverageFactory();
		case ONLYBRANCH:
			return new OnlyBranchCoverageFactory();
		case METHODTRACE:
			return new MethodTraceCoverageFactory();
		case METHOD:
			return new MethodCoverageFactory();
		case METHODNOEXCEPTION:
			return new MethodNoExceptionCoverageFactory();
		case LINE:
			return new LineCoverageFactory();
		case ONLYLINE:
			return new LineCoverageFactory();
		case OUTPUT:
			return new OutputCoverageFactory();
		case INPUT:
			return new InputCoverageFactory();
		case TRYCATCH:
			return new TryCatchCoverageFactory();
		default:
			logger.warn("No TestFitnessFactory defined for " + crit
			        + " using default one (BranchCoverageFactory)");
			return new BranchCoverageFactory();
		}
	}
}