aboutsummaryrefslogtreecommitdiff
path: root/scripts/loop.py
blob: 712b0549c47278a32007c541c1c11c84efd6f13e (plain)
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#!/usr/bin/env python3
import os
import sys
import subprocess
import signal
from threading import Thread

from conf import conf
from conf import sf
import initialize
import solution
import kernel
import boot
import exceptions

def step():
	phs = phase_get()
	if phs == 0 or phs == 1:
		phase_message(1)
		initialize.all()
		phase_set(2)
	elif phs == 2:
		phase_message(2)
		phase_set(3)
	elif phs == 3:
		phase_message(3)
		try:
			solution.apply()
		except exceptions.NoApplicableSolution:
			try:
				os.mkdir(sf(conf.result_folder))
			except FileExistsError:
				pass
			print('\nAll done.')
			exit(0)
		phase_set(4)
	elif phs == 4:
		phase_message(4)
		phase_set(5)
	elif phs == 5:
		phase_message(5)
		try:
			kernel.config()
		except exceptions.ConfigurationError:
			if not conf.ignore_misconfig:
				print("Configuration mismatch. Exiting.")
				sys.exit(-2)
		phase_set(6)
	elif phs == 6:
		phase_message(6)
		if conf.only_config:
			phase_set(3)
		else:
			phase_set(7)
	elif phs == 7:
		phase_message(7)
		kernel.make()
		phase_set(8)
	elif phs == 8:
		phase_message(8)
		phase_set(9)
	elif phs == 9:
		phase_message(9)
		boot.boot()
		phase_set(10)
	elif phs == 10:
		phase_message(10)
		phase_set(3)

# Phase #
phases = ("Not Initialized",		#0
		  "Initializing",			#1
		  "Initialized",			#2
		  "Solution applying",		#3
		  "Solution applied",		#4
		  "Kernel configuration",	#5
		  "Kernel configured",		#6
		  "Kernel build",			#7
		  "Kernel built",			#8
		  "System boot",			#9
		  "Benchmark successful"	#10
		  )

def phase_get():
	try:
		with open(sf(conf.phase_file)) as f:
			txtPhase = f.readline().rstrip()
		if not txtPhase in phases:
			raise PhaseMismatch()
		return phases.index(txtPhase)
	except FileNotFoundError:
		return 0

def phase_set(phs):
	# TODO
	try:
		global thr
		if thr.term:
			return
	except NameError:
		pass
	with open(sf(conf.phase_file), 'w') as f:
		f.write(phases[phs])

def phase_message(phs):
	"Prints message signaling running phase_"
	print("-- " + phases[phs])

# Iteration #
def iteration_reset():
	with open(sf(conf.iteration_file), 'w') as f:
		f.write('0')

def iteration_inc():
	with open(sf(conf.iteration_file), 'r') as f:
		it = int(f.readline())
	it += 1
	with open(sf(conf.iteration_file), 'w') as f:
		f.write(str(it))

# Thread #
class mainThread(Thread):
	def __init__(self, name):
		Thread.__init__(self, name=name)
		self.term = False
	def run(self):
		if conf.step_by_step:
			step()
		elif conf.single_loop:
			while not phase_get() == 2:
				step()
			step()
			while not phase_get() == 2:
				step()
		else:
			while not self.term:
				step()

def loop_term():
	global thr
	thr.term = True

def sigterm_handler(_signo, _stack_frame):
	loop_term()

def loop():
	global thr
	thr = mainThread("thred")
	thr.start()
	try:
		thr.join()
	except KeyboardInterrupt:
		loop_term()

#################################################################################

if __name__ == '__main__':
	signal.signal(signal.SIGTERM, sigterm_handler)
	loop()