PsoAlgorithm

Complexity

Total Complexity

3

Size/Duplication

Total Lines	18
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	run()	0	9	2
A	__init__()	0	5	1

"""Contain PsoAlgorithm."""
from grortir.main.model.core.optimization_status import OptimizationStatus
from grortir.main.pso.process_validator import ProcessValidator
from grortir.main.pso.whole_group_pso import WholeGroupPso


class PsoAlgorithm:
    """Optimize process with different strategies."""

    def __init__(self, process, strategy, number_of_particle=40):
        self.process = process
        self.strategy = strategy
        self.process_validator = ProcessValidator()
        self.whole_group_pso = WholeGroupPso(self.process, number_of_particle)

    def run(self):
        """Run algorithm."""
        self.process_validator.validate(self.process)
        self.process.optimizationStatus = OptimizationStatus.in_progress
        number_of_groups = self.strategy.get_actual_numbers_of_groups()
        for current_group_number in range(number_of_groups):
            current_stages = self.strategy.get_items_from_group(
                current_group_number)
            self.whole_group_pso.optimize(current_stages)