Grey-Wolf-Optimization-algorithm/wolf.py at master · setareA/Grey-Wolf-Optimization-algorithm · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/python

import copy
import database
import sqlite3


class Wolf:

    def __init__(self, num, filename, programs, num_of_nodes,UCE, mode):

        self.num = num
        self.file = filename
        self.program_list = copy.deepcopy(programs)
        self.genes = []
        self.num_of_nodes = num_of_nodes
        self.UCE = UCE  # unit cost of energy consumption by the time required
        # for the server to complete the different tasks
        self.alfa = 100
        self.beta = 10

        if mode == 1:
            for i in range(self.num):
                self.genes.append(database.select_random_node())  # select one of the nodes
            with open('first_random_genes.txt', 'a') as out:
                out.write(str(self.genes) + '\n\n' + "#################" + '\n\n')
            self.calc_fitness()

    def calc_fitness(self):

        fit = 0
        healthy_genes = []
        ccc = sqlite3.connect(self.file)
        for i in range(self.num):
            prog = self.program_list[i]
            for row in ccc.execute("SELECT * FROM NODES WHERE ID = " + str(self.genes[i])):

                if row[2] >= prog.get_ram() and row[3] >= prog.get_cpu() and row[4] == prog.get_cpu_type():
                    self.program_list[i].set_node()
                    fit += 1
                    healthy_genes.append(i)
                    ccc.execute("UPDATE NODES SET RAM = " + str(row[2] - prog.get_ram()) + ", CPU = " + str(row[3] - prog.get_cpu()) + " WHERE ID = " + str(self.genes[i]))

        ccc.close()

        all_nodes = list(range(1, self.num_of_nodes + 1))
        on_servers = len(healthy_genes)

        self.fitness = (self.alfa * on_servers * self.UCE + self.beta * (self.num - fit))
        with open('fitness.txt', 'a') as out:
            out.write(str(self.fitness)+"  fit : +"+str(fit)+" on_servers : "+str(on_servers)+'\n\n' + "#################" + '\n\n')

    def get_fitness(self):
        return self.fitness

    def update(self, x):
        f1 = self.get_fitness()
        if f1 > x:

            damaged_genes = []
            healthy_genes = []
            ccc = sqlite3.connect(self.file)
            for i in range(self.num):
                prog = self.program_list[i]
                for row in ccc.execute("SELECT * FROM NODES WHERE ID = " + str(self.genes[i])):

                    if row[2] >= prog.get_ram() and row[3] >= prog.get_cpu() and row[4] == prog.get_cpu_type():
                        self.program_list[i].set_node()
                        healthy_genes.append(i)
                        ccc.execute("UPDATE NODES SET RAM = " + str(row[2] - prog.get_ram()) + ", CPU = " + str(row[3] - prog.get_cpu()) + " WHERE ID = " + str(self.genes[i]))
                    else:
                        damaged_genes.append(i)
            count = 0
            for k in damaged_genes:
                if count == 1:
                    break
                else:
                    prog = self.program_list[k]

                    for row in ccc.execute("SELECT * FROM NODES WHERE RAM > " + str(prog.get_ram()) + " AND cpu > " + str(prog.get_cpu()) + " AND CPU_TYPE = " + str(prog.get_cpu_type())):

                        if k in healthy_genes:
                            f1 -= self.beta

                        else:
                            f1 -= self.beta
                            f1 += self.alfa
                        if abs(f1 - x) < 15 or f1 < x:
                            self.program_list[k].set_node()
                            self.genes[k] = row[0]
                            ccc.execute("UPDATE NODES SET RAM = " + str(row[2] - prog.get_ram()) + ", CPU = " + str(row[3] - prog.get_cpu()) + " WHERE ID = " + str(row[0]))
                            count = 1
                            break

            ccc.close()
            self.calc_fitness()

    def get_genes(self):
        return self.genes

    def get_empty_genes(self):
        l = []
        for i in range(self.num):
            if self.program_list[i].get_node() == False:
                l.append(i)
        return l

    def calc_empty_nodes(self):

        healthy_genes = []
        ccc = sqlite3.connect(self.file)
        for i in range(self.num):
            prog = self.program_list[i]
            for row in ccc.execute("SELECT * FROM NODES WHERE ID = " + str(self.genes[i])):

                if row[2] >= prog.get_ram() and row[3] >= prog.get_cpu() and row[4] == prog.get_cpu_type():
                    self.program_list[i].set_node()
                    healthy_genes.append(i)
                    ccc.execute("UPDATE NODES SET RAM = " + str(row[2] - prog.get_ram()) + ", CPU = " + str(row[3] - prog.get_cpu()) + " WHERE ID = " + str(self.genes[i]))

        ccc.close()

        all_nodes = list(range(1, self.num_of_nodes + 1))
        empty_nodes_count = len(set(all_nodes) - set(healthy_genes))

        return empty_nodes_count