|
1
|
|
|
"""Contain PsoAlgorithm.""" |
|
2
|
1 |
|
from grortir.main.model.core.optimization_status import OptimizationStatus |
|
3
|
1 |
|
from grortir.main.pso.process_validator import ProcessValidator |
|
4
|
1 |
|
from grortir.main.pso.whole_group_pso import WholeGroupPso |
|
5
|
|
|
|
|
6
|
|
|
|
|
7
|
1 |
|
class PsoAlgorithm: |
|
8
|
|
|
"""Optimize process with different strategies.""" |
|
9
|
|
|
|
|
10
|
1 |
|
def __init__(self, process, grouping_strategy, optimization_startegy, |
|
11
|
|
|
number_of_particle=40): |
|
12
|
1 |
|
self.process = process |
|
13
|
1 |
|
self.grouping_strategy = grouping_strategy |
|
14
|
1 |
|
self.optimization_strategy = optimization_startegy |
|
15
|
1 |
|
self.process_validator = ProcessValidator() |
|
16
|
1 |
|
self.whole_group_pso = WholeGroupPso(self.process, number_of_particle) |
|
17
|
|
|
|
|
18
|
1 |
|
def run(self): |
|
19
|
|
|
"""Run algorithm.""" |
|
20
|
1 |
|
self.process_validator.validate(self.process) |
|
21
|
1 |
|
self.process.optimization_status = OptimizationStatus.in_progress |
|
22
|
1 |
|
number_of_groups = self.grouping_strategy.get_actual_numbers_of_groups() |
|
23
|
1 |
|
for current_group_number in range(number_of_groups): |
|
24
|
1 |
|
current_stages = self.grouping_strategy.get_items_from_group( |
|
25
|
|
|
current_group_number) |
|
26
|
1 |
|
group_optimization_strategy = self.optimization_strategy. \ |
|
27
|
|
|
get_group_optimization_strategy(current_stages) |
|
28
|
1 |
|
self.whole_group_pso.optimize(current_stages, |
|
29
|
|
|
group_optimization_strategy) |
|
30
|
|
|
self._post_processing() |
|
31
|
|
|
|
|
32
|
|
|
def _post_processing(self): |
|
|
|
|
|
|
33
|
|
|
final_status = OptimizationStatus.success |
|
34
|
|
|
for stage in self.grouping_strategy.ordered_stages: |
|
35
|
|
|
if stage.optimization_status != OptimizationStatus.success: |
|
36
|
|
|
final_status = OptimizationStatus.failed |
|
37
|
|
|
self.process.optimization_status = final_status |
|
38
|
|
|
|
qbahn /
grortir
The coding style of this project requires that you add a docstring to this code element. Below, you find an example for methods:
If you would like to know more about docstrings, we recommend to read PEP-257: Docstring Conventions.