import numpy as np

from ..base.experiment_abc import ExperimentABC


class ExperimentNumerical(ExperimentABC):
    def __init__(self, hof_size, popsize, save_only_best) -> None:
        ExperimentABC.__init__(self,popsize=popsize,
                               hof_size=hof_size,
                               save_only_best=save_only_best)

    def mutate(self, gen):
        return gen + np.random.normal(0, 15, len(gen))

    def cross_over(self, gen1, gen2):
        a = np.random.uniform()
        return a * gen1 + (1.0-a) * gen2

    def evaluate(self, gen):
        return -sum([x*x for x in gen])