WholeGroupPso

Complexity

Total Complexity

3

Size/Duplication

Total Lines	20
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	2
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	3	1
A	optimize()	0	13	2

"""Contains WholeGroupPso class."""
from grortir.main.pso.optimization_controller import OptimizationController
from grortir.main.pso.swarm import Swarm


class WholeGroupPso(object):
    """Optimize whole group."""

    def __init__(self, process, number_of_particles):
        self.process = process
        self.number_of_particles = number_of_particles

    def optimize(self, ordered_stages_to_optimize):
        """Optimize whole group of stages.

            Parameters:
                ordered_stages_to_optimize (list): List of stages which need
                 to be optimized
        """
        swarm = Swarm(self.process, ordered_stages_to_optimize,
                      self.number_of_particles)
        swarm.initialize()
        while OptimizationController.should_continue(
                ordered_stages_to_optimize):
            swarm.do_single_iteration()