Source code for evolutionary_optimization.phenotype.phenotype_model.phenotype_interface

from enum import Enum

from evolutionary_optimization.phenotype.implemented_phenotypes.booth_phenotype import BoothPhenotype
from evolutionary_optimization.phenotype.phenotype_model.abstract_phenotype import AbstractPhenotype
from evolutionary_optimization.phenotype.implemented_phenotypes.inverted_parabola_phenotype import InvertedParabolaPhenotype
from evolutionary_optimization.phenotype.implemented_phenotypes.parabola_phenotype import ParabolaPhenotype
from evolutionary_optimization.phenotype.implemented_phenotypes.saddle_point_phenotype import SaddlePointPhenotype


class Phenotypes(str, Enum):
    """Enum containing implemented phenotypes."""
    PARABOLA = "parabola"
    INVERTED_PARABOLA = "inverted_parabola"
    SADDLE_POINT = "saddle_point"
    BOOTH = "booth"

class Phenotype:
    """Map Phenotypes to their associated concrete class based on AbstractPhenotype."""
    phenotypes_dictionary = {
        Phenotypes.PARABOLA: ParabolaPhenotype,
        Phenotypes.INVERTED_PARABOLA: InvertedParabolaPhenotype,
        Phenotypes.SADDLE_POINT: SaddlePointPhenotype,
        Phenotypes.BOOTH: BoothPhenotype,
    }

    @classmethod
    def get_phenotype(cls, phenotypes_enum: Phenotypes) -> type(AbstractPhenotype):
        """Return class of desired AbstractPhenotype."""
        return cls.phenotypes_dictionary[phenotypes_enum]