Optimizer

Complexity

Total Complexity

9

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Test Coverage

Coverage

95%

Importance

Changes	2
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
B	optimize_process()	0	18	5
A	set_custom_optimizing_order()	0	7	3
A	__init__()	0	3	1

"""Basic optimizer."""
import networkx as nx

The import networkx could not be resolved.

from grortir.externals.pyswarm.pso import pso
from grortir.main.model.core.optimization_status import OptimizationStatus


class Optimizer(object):
    """Optimizer is object which optimize process."""

    def __init__(self, process):
        self.process = process
        self.ordered_stages = nx.topological_sort(self.process)

    def set_custom_optimizing_order(self, ordered_stages):
        """Set custom order."""
        if set(self.ordered_stages) == set(ordered_stages) and len(
                self.ordered_stages) == len(ordered_stages):
            self.ordered_stages = ordered_stages
        else:
            raise ValueError("List of stages must contain all stages.")

    def optimize_process(self):
        """Optimize process.
        Returns:
            True if success, False otherwise."""
        for stage in self.ordered_stages:
            if len(self.process.predecessors(stage)) == 0:
                pso(stage)
            elif len(self.process.predecessors(stage)) == 1:
                predecessor = self.process.predecessors(stage)[0]
                stage.input_vector = predecessor.get_output_of_stage()
                pso(stage)
            else:
                raise AttributeError('Incorrect process structure.')

            if stage.optimization_status != OptimizationStatus.success:
                return False

        return True