survivor_selection.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
#include <stdexcept>
#include <vector>
 
#include "pulsatrix/individual.hpp"
 
namespace pulsatrix {
 
template <typename Genotype, typename FitnessT>
std::vector<Individual<Genotype, FitnessT>> GenerationalReplacement(
    const std::vector<Individual<Genotype, FitnessT>>& /*population*/,
    std::vector<Individual<Genotype, FitnessT>> offspring, size_t mu) {
    if (offspring.size() != mu) {
        throw std::invalid_argument("GenerationalReplacement: offspring.size() must equal mu");
    }
    return offspring;
}
 
template <typename Genotype, typename FitnessT>
std::vector<Individual<Genotype, FitnessT>> MuPlusLambdaReplacement(
    const std::vector<Individual<Genotype, FitnessT>>& population,
    std::vector<Individual<Genotype, FitnessT>> offspring, size_t mu) {
    if (population.size() + offspring.size() < mu) {
        throw std::invalid_argument(
            "MuPlusLambdaReplacement: population.size() + offspring.size() must be >= mu");
    }
    std::vector<Individual<Genotype, FitnessT>> combined = population;
    combined.insert(combined.end(), std::make_move_iterator(offspring.begin()),
                     std::make_move_iterator(offspring.end()));
    std::partial_sort(combined.begin(), combined.begin() + static_cast<std::ptrdiff_t>(mu),
                       combined.end(),
                       [](const auto& a, const auto& b) { return a.fitness > b.fitness; });
    combined.resize(mu);
    return combined;
}
 
template <typename Genotype, typename FitnessT>
std::vector<Individual<Genotype, FitnessT>> MuCommaLambdaReplacement(
    const std::vector<Individual<Genotype, FitnessT>>& /*population*/,
    std::vector<Individual<Genotype, FitnessT>> offspring, size_t mu) {
    if (offspring.size() < mu) {
        throw std::invalid_argument("MuCommaLambdaReplacement: offspring.size() must be >= mu");
    }
    std::partial_sort(offspring.begin(), offspring.begin() + static_cast<std::ptrdiff_t>(mu),
                       offspring.end(),
                       [](const auto& a, const auto& b) { return a.fitness > b.fitness; });
    offspring.resize(mu);
    return offspring;
}
 
}  // namespace pulsatrix