desdeo 2.0.0__py3-none-any.whl → 2.1.0__py3-none-any.whl

desdeo/emo/operators/termination.py

@@ -11,6 +11,8 @@ Warning:
     super().do method to increment the generation counter _before_ conducting the termination check.
 """
 
+import time
+from abc import abstractmethod
 from collections.abc import Sequence
 
 from desdeo.tools.message import (
@@ -20,7 +22,7 @@ from desdeo.tools.message import (
     Message,
     TerminatorMessageTopics,
 )
-from desdeo.tools.patterns import Subscriber
+from desdeo.tools.patterns import Publisher, Subscriber
 
 
 class BaseTerminator(Subscriber):
@@ -52,21 +54,22 @@ class BaseTerminator(Subscriber):
         """Return the message topics that the terminator is interested in."""
         return [EvaluatorMessageTopics.NEW_EVALUATIONS, GeneratorMessageTopics.NEW_EVALUATIONS]
 
-    def __init__(self, **kwargs):
+    def __init__(self, publisher: Publisher):
         """Initialize a termination criterion."""
-        super().__init__(**kwargs)
+        super().__init__(publisher=publisher, verbosity=1)
         self.current_generation: int = 1
         self.current_evaluations: int = 0
         self.max_generations: int = 0
         self.max_evaluations: int = 0
 
-    def check(self) -> bool | None:
+    def check(self) -> bool:
         """Check if the termination criterion is reached.
 
         Returns:
             bool: True if the termination criterion is reached, False otherwise.
         """
         self.current_generation += 1
+        self.notify()
 
     def state(self) -> Sequence[Message]:
         """Return the state of the termination criterion."""
@@ -109,7 +112,7 @@ class BaseTerminator(Subscriber):
         """
         if not isinstance(message, IntMessage):
             return
-        if not isinstance(message.topic, EvaluatorMessageTopics) or isinstance(message.topic, GeneratorMessageTopics):
+        if not (isinstance(message.topic, EvaluatorMessageTopics) or isinstance(message.topic, GeneratorMessageTopics)):
             return
         if (
             message.topic == EvaluatorMessageTopics.NEW_EVALUATIONS  # NOQA: PLR1714
@@ -121,23 +124,19 @@ class BaseTerminator(Subscriber):
 class MaxGenerationsTerminator(BaseTerminator):
     """A class for a termination criterion based on the number of generations."""
 
-    def __init__(self, max_generations: int, **kwargs):
+    def __init__(self, max_generations: int, publisher: Publisher):
         """Initialize a termination criterion based on the number of generations.
 
         Args:
             max_generations (int): the maximum number of generations.
-            kwargs: Additional keyword arguments. These are passed to the Subscriber class. At the very least, the
-                publisher must be passed. See the Subscriber class for more information.
+            publisher (Publisher): The publisher to which the terminator will publish its state.
         """
-        super().__init__(**kwargs)
+        super().__init__(publisher=publisher)
         self.max_generations = max_generations
 
     def check(self) -> bool:
         """Check if the termination criterion based on the number of generations is reached.
 
-        Args:
-            new_generation (bool, optional): Increment the generation counter. Defaults to True.
-
         Returns:
             bool: True if the termination criterion is reached, False otherwise.
         """
@@ -151,24 +150,24 @@ class MaxGenerationsTerminator(BaseTerminator):
 class MaxEvaluationsTerminator(BaseTerminator):
     """A class for a termination criterion based on the number of evaluations."""
 
-    def __init__(self, max_evaluations: int, **kwargs):
+    def __init__(self, max_evaluations: int, publisher: Publisher):
         """Initialize a termination criterion based on the number of evaluations.
 
         Looks for messages with key "num_evaluations" to update the number of evaluations.
 
         Args:
             max_evaluations (int): the maximum number of evaluations.
-            kwargs: Additional keyword arguments. These are passed to the Subscriber class. At the very least, the
+            publisher (Publisher): The publisher to which the terminator will publish its state.
                 publisher must be passed. See the Subscriber class for more information.
         """
-        super().__init__(**kwargs)
+        super().__init__(publisher=publisher)
         if not isinstance(max_evaluations, int) or max_evaluations < 0:
             raise ValueError("max_evaluations must be a non-negative integer")
         self.max_evaluations = max_evaluations
         self.current_evaluations = 0
 
     def check(self) -> bool:
-        """Check if the termination criterion based on the number of evaluations is reached.
+        """Check if the termination criterion based on the number of generations is reached.
 
         Returns:
             bool: True if the termination criterion is reached, False otherwise.
@@ -176,3 +175,112 @@ class MaxEvaluationsTerminator(BaseTerminator):
         super().check()
         self.notify()
         return self.current_evaluations >= self.max_evaluations
+
+
+class CompositeTerminator(BaseTerminator):
+    """Combines multiple terminators using logical AND or OR."""
+
+    def __init__(self, terminators: list[BaseTerminator], publisher: Publisher, mode: str = "any"):
+        """Initialize a composite termination criterion.
+
+        Args:
+            terminators (list[BaseTerminator]): List of BaseTerminator instances.
+            publisher (Publisher): Publisher for passing messages.
+            mode (str): "any" (terminate if any) or "all" (terminate if all). By default, "any".
+        """
+        super().__init__(publisher=publisher)
+        self.terminators = terminators
+        for t in self.terminators:
+            t.notify = lambda: None  # Reset the notify method so that individual terminators do not send notifications
+        types = [type(t) for t in self.terminators]
+        # Assert that all terminators are unique
+        if len(types) != len(set(types)):
+            raise ValueError("All terminators must be unique.")
+        max_generations = [t.max_generations for t in self.terminators if isinstance(t, MaxGenerationsTerminator)]
+        if max_generations:
+            self.max_generations = max(max_generations)
+        max_evaluations = [t.max_evaluations for t in self.terminators if isinstance(t, MaxEvaluationsTerminator)]
+        if max_evaluations:
+            self.max_evaluations = max(max_evaluations)
+        self.mode = mode
+
+    def check(self) -> bool:
+        """Check if the termination criterion is reached.
+
+        Returns:
+            bool: True if the termination criterion is reached, False otherwise.
+        """
+        super().check()
+        results = [t.check() for t in self.terminators]
+        if self.mode == "all":
+            return all(results)
+        return any(results)
+
+
+class ExternalCheckTerminator(BaseTerminator):
+    """A termination criterion that checks an external condition."""
+
+    def __init__(self, check_function, publisher: Publisher):
+        """Initialize the external check terminator.
+
+        Args:
+            check_function (callable): A function that returns True if the termination condition is met.
+            publisher (Publisher): The publisher to send messages to.
+        """
+        super().__init__(publisher=publisher)
+        self.check_function = check_function
+
+    def check(self) -> bool:
+        """Check if the termination condition is met.
+
+        Returns:
+            bool: True if the termination condition is met, False otherwise.
+        """
+        super().check()
+        self.notify()
+        return self.check_function()
+
+
+class MaxTimeTerminator(BaseTerminator):
+    """A termination criterion based on the maximum elapsed time."""
+
+    @property
+    def provided_topics(self) -> dict[int, Sequence[TerminatorMessageTopics]]:
+        """Return the topics provided by the terminator.
+
+        Returns:
+            dict[int, Sequence[TerminatorMessageTopics]]: The topics provided by the terminator.
+        """
+        return {
+            0: [],
+            1: [
+                TerminatorMessageTopics.GENERATION,
+                TerminatorMessageTopics.EVALUATION,
+            ],
+        }
+
+    def __init__(self, max_time_in_seconds: float, publisher: Publisher):
+        """Initialize the maximum time terminator.
+
+        Args:
+            max_time_in_seconds (float): The maximum elapsed time in seconds.
+            publisher (Publisher): The publisher to which the terminator will publish its state.
+        """
+        super().__init__(publisher=publisher)
+        if not isinstance(max_time_in_seconds, float) or max_time_in_seconds < 0:
+            raise ValueError("max_time must be a non-negative float")
+        self.max_time = max_time_in_seconds
+        self.start_time = None
+
+    def check(self) -> bool:
+        """Check if the termination criterion based on the maximum elapsed time is reached.
+
+        Returns:
+            bool: True if the termination criterion is reached, False otherwise.
+        """
+        super().check()
+        self.notify()
+        if self.start_time is None:
+            self.start_time = time.perf_counter()
+        elapsed_time = time.perf_counter() - self.start_time
+        return elapsed_time >= self.max_time

desdeo/emo/options/__init__.py

@@ -0,0 +1,108 @@
+from .crossover import (
+    BlendAlphaCrossoverOptions,
+    BoundedExponentialCrossoverOptions,
+    CrossoverOptions,
+    LocalCrossoverOptions,
+    SimulatedBinaryCrossoverOptions,
+    SingleArithmeticCrossoverOptions,
+    SinglePointBinaryCrossoverOptions,
+    UniformIntegerCrossoverOptions,
+    UniformMixedIntegerCrossoverOptions,
+    crossover_constructor,
+)
+from .generator import (
+    GeneratorOptions,
+    LHSGeneratorOptions,
+    RandomBinaryGeneratorOptions,
+    RandomGeneratorOptions,
+    RandomIntegerGeneratorOptions,
+    RandomMixedIntegerGeneratorOptions,
+    generator_constructor,
+)
+from .mutation import (
+    BinaryFlipMutationOptions,
+    BoundedPolynomialMutationOptions,
+    IntegerRandomMutationOptions,
+    MixedIntegerRandomMutationOptions,
+    MPTMutationOptions,
+    MutationOptions,
+    NonUniformMutationOptions,
+    PowerMutationOptions,
+    SelfAdaptiveGaussianMutationOptions,
+    mutation_constructor,
+)
+from .repair import RepairOptions, repair_constructor
+from .scalar_selection import (
+    RouletteWheelSelectionOptions,
+    ScalarSelectionOptions,
+    TournamentSelectionOptions,
+    scalar_selector_constructor,
+)
+from .selection import (
+    IBEASelectorOptions,
+    NSGA2SelectorOptions,
+    NSGA3SelectorOptions,
+    ReferenceVectorOptions,
+    RVEASelectorOptions,
+    SelectorOptions,
+    selection_constructor,
+)
+from .termination import (
+    CompositeTerminatorOptions,
+    ExternalCheckTerminatorOptions,
+    MaxEvaluationsTerminatorOptions,
+    MaxGenerationsTerminatorOptions,
+    MaxTimeTerminatorOptions,
+    TerminatorOptions,
+    terminator_constructor,
+)
+
+__all__ = [  # noqa: RUF022
+    "BinaryFlipMutationOptions",
+    "BlendAlphaCrossoverOptions",
+    "BoundedExponentialCrossoverOptions",
+    "BoundedPolynomialMutationOptions",
+    "CompositeTerminatorOptions",
+    "CrossoverOptions",
+    "ExternalCheckTerminatorOptions",
+    "GeneratorOptions",
+    "IBEASelectorOptions",
+    "IntegerRandomMutationOptions",
+    "LHSGeneratorOptions",
+    "LocalCrossoverOptions",
+    "MaxEvaluationsTerminatorOptions",
+    "MaxGenerationsTerminatorOptions",
+    "MaxTimeTerminatorOptions",
+    "MixedIntegerRandomMutationOptions",
+    "MPTMutationOptions",
+    "MutationOptions",
+    "NSGA2SelectorOptions",
+    "NSGA3SelectorOptions",
+    "NonUniformMutationOptions",
+    "PowerMutationOptions",
+    "RandomBinaryGeneratorOptions",
+    "RandomGeneratorOptions",
+    "RandomIntegerGeneratorOptions",
+    "RandomMixedIntegerGeneratorOptions",
+    "ReferenceVectorOptions",
+    "RouletteWheelSelectionOptions",
+    "RVEASelectorOptions",
+    "ScalarSelectionOptions",
+    "scalar_selector_constructor",
+    "SelfAdaptiveGaussianMutationOptions",
+    "SelectorOptions",
+    "SimulatedBinaryCrossoverOptions",
+    "SingleArithmeticCrossoverOptions",
+    "SinglePointBinaryCrossoverOptions",
+    "TerminatorOptions",
+    "terminator_constructor",
+    "TournamentSelectionOptions",
+    "UniformIntegerCrossoverOptions",
+    "UniformMixedIntegerCrossoverOptions",
+    "crossover_constructor",
+    "generator_constructor",
+    "mutation_constructor",
+    "selection_constructor",
+    "RepairOptions",
+    "repair_constructor",
+]