View GitHub Repository grortir

Schedule Inspection
Subscribe
Completed
Pull Request — master (#43)
by Wojtek
06:31
created

PsoAlgorithm.run()   A

↳ Parent: PsoAlgorithm

Complexity

Conditions 2

Size

Total Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 2
Bugs 0 Features 0
Metric Value
cc 2
c 2
b 0
f 0
dl 0
loc 9
rs 9.6666
1 
"""Contain PsoAlgorithm."""
2 
from grortir.main.model.core.optimization_status import OptimizationStatus
3 
from grortir.main.pso.process_validator import ProcessValidator
4 
from grortir.main.pso.whole_group_pso import WholeGroupPso
5 

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    6
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    7
                

                                    
                                                    
                
                
                
class PsoAlgorithm:

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    8
                

                                    
                                                    
                
                
                
    """Optimize process with different strategies."""

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    9
                

                                    
                                                    
                
                
                

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    10
                

                                    
                                                    
                
                
                
    def __init__(self, process, strategy, number_of_particle=40):

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    11
                

                                    
                                                    
                
                
                
        self.process = process

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    12
                

                                    
                                                    
                
                
                
        self.strategy = strategy

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    13
                

                                    
                                                    
                
                
                
        self.process_validator = ProcessValidator()

            


            
                                                                                                            
                            
            
                                    
            
            

                
                    14
                

                                    
                                                    
                
                
                
        self.whole_group_pso = WholeGroupPso(self.process, number_of_particle)

            


            
                                                                                                            
                                                                
            
                                    
            
            

                
                    15
                

                                    
                                                    
                
                
                

            


            
                                                        
            
                                    
            
            

                
                    16
                

                                    
                                                    
                
                
                
    def run(self):

            


            
                                                        
            
                                    
            
            

                
                    17
                

                                    
                                                    
                
                
                
        """Run algorithm."""

            


            
                                                        
            
                                    
            
            

                
                    18
                

                                    
                                                    
                
                
                
        self.process_validator.validate(self.process)

            


            
                                                        
            
                                    
            
            

                
                    19
                

                                    
                                                    
                
                
                
        self.process.optimizationStatus = OptimizationStatus.in_progress

            


            
                                                        
            
                                    
            
            

                
                    20
                

                                    
                                                    
                
                
                
        number_of_groups = self.strategy.get_actual_numbers_of_groups()

            


            
                                                        
            
                                    
            
            

                
                    21
                

                                    
                                                    
                
                
                
        for current_group_number in range(number_of_groups):

            


            
                                                        
            
                                    
            
            

                
                    22
                

                                    
                                                    
                
                
                
            current_stages = self.strategy.get_items_from_group(

            


            
                                                        
            
                                    
            
            

                
                    23
                

                                    
                                                    
                
                
                
                current_group_number)

            


            
                                                        
            
                                    
            
            

                
                    24
                

                                    
                                                    
                
                
                
            self.whole_group_pso.optimize(current_stages)

            


            
                                                        
            
                                    
            
            

                
                    25