Completed
Push — master ( 40c094...b6a1c4 )
by Wojtek
02:30
created

Optimizer.optimize_process()   B

Complexity

Conditions 5

Size

Total Lines 21

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 12
CRAP Score 5

Importance

Changes 2
Bugs 0 Features 0
Metric Value
cc 5
c 2
b 0
f 0
dl 0
loc 21
rs 8.439
ccs 12
cts 12
cp 1
crap 5
1
"""Basic optimizer."""
2
3 1
from grortir.externals.pyswarm.pso import pso
4 1
from grortir.main.model.core.optimization_status import OptimizationStatus
5 1
from grortir.main.optimizers.base_optimizer import BaseOptimizer
6
7
8 1
class Optimizer(BaseOptimizer):
9
    """Optimizer is object which optimize process."""
10
11 1
    def optimize_process(self):
12
        """Optimize process.
13
        Raises:
14
            AttributeError: When project has incorrect structure.
15
16
        Returns:
17
            bool: True if success, False otherwise."""
18 1
        for stage in self.ordered_stages:
19 1
            if not self.process.predecessors(stage):
20 1
                self.run_pso(stage)
21 1
            elif len(self.process.predecessors(stage)) == 1:
22 1
                predecessor = self.process.predecessors(stage)[0]
23 1
                stage.input_vector = predecessor.get_output_of_stage(
24
                    predecessor.input_vector, predecessor.control_params)
25 1
                self.run_pso(stage)
26
            else:
27 1
                raise AttributeError('Incorrect process structure.')
28
29 1
            if stage.optimization_status != OptimizationStatus.success:
30 1
                return False
31 1
        return True
32
33 1
    def run_pso(self, stage):
34
        """Run pso with predefined parameters."""
35
        pso(stage, swarmsize=self.swarm_size)
36