epicyon/fitnessFunctions.py

131 lines
4.5 KiB
Python

__filename__ = "fitnessFunctions.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.2.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Core"
import os
import time
from webapp_utils import html_header_with_external_style
from webapp_utils import html_footer
from utils import get_config_param
from utils import save_json
def fitness_performance(startTime, fitness_state: {},
fitness_id: str, watch_point: str,
debug: bool) -> None:
"""Log a performance watchpoint
"""
if 'performance' not in fitness_state:
fitness_state['performance'] = {}
if fitness_id not in fitness_state['performance']:
fitness_state['performance'][fitness_id] = {}
if watch_point not in fitness_state['performance'][fitness_id]:
fitness_state['performance'][fitness_id][watch_point] = {
"total": float(0),
"ctr": int(0)
}
time_diff = float(time.time() - startTime)
fitness_state['performance'][fitness_id][watch_point]['total'] += time_diff
fitness_state['performance'][fitness_id][watch_point]['ctr'] += 1
if fitness_state['performance'][fitness_id][watch_point]['ctr'] >= 1024:
fitness_state['performance'][fitness_id][watch_point]['total'] /= 2
fitness_state['performance'][fitness_id][watch_point]['ctr'] = \
int(fitness_state['performance'][fitness_id][watch_point]['ctr'] /
2)
if debug:
ctr = fitness_state['performance'][fitness_id][watch_point]['ctr']
total = fitness_state['performance'][fitness_id][watch_point]['total']
print('FITNESS: performance/' + fitness_id + '/' +
watch_point + '/' + str(total * 1000 / ctr))
def sorted_watch_points(fitness: {}, fitness_id: str) -> []:
"""Returns a sorted list of watchpoints
times are in mS
"""
if not fitness.get('performance'):
return []
if not fitness['performance'].get(fitness_id):
return []
result = []
for watch_point, item in fitness['performance'][fitness_id].items():
if not item.get('total'):
continue
average_time = item['total'] * 1000 / item['ctr']
average_time_str = str(average_time).zfill(8)
result.append(average_time_str + ' ' + watch_point)
result.sort(reverse=True)
return result
def html_watch_points_graph(base_dir: str, fitness: {}, fitness_id: str,
max_entries: int) -> str:
"""Returns the html for a graph of watchpoints
"""
watch_points_list = sorted_watch_points(fitness, fitness_id)
css_filename = base_dir + '/epicyon-graph.css'
if os.path.isfile(base_dir + '/graph.css'):
css_filename = base_dir + '/graph.css'
instance_title = \
get_config_param(base_dir, 'instanceTitle')
html_str = \
html_header_with_external_style(css_filename, instance_title, None)
html_str += \
'<table class="graph">\n' + \
'<caption>Watchpoints for ' + fitness_id + '</caption>\n' + \
'<thead>\n' + \
' <tr>\n' + \
' <th scope="col">Item</th>\n' + \
' <th scope="col">Percent</th>\n' + \
' </tr>\n' + \
'</thead><tbody>\n'
# get the maximum time
max_average_time = float(1)
if len(watch_points_list) > 0:
max_average_time = float(watch_points_list[0].split(' ')[0])
for watch_point in watch_points_list:
average_time = float(watch_point.split(' ')[0])
if average_time > max_average_time:
max_average_time = average_time
ctr = 0
for watch_point in watch_points_list:
name = watch_point.split(' ', 1)[1]
average_time = float(watch_point.split(' ')[0])
height_percent = int(average_time * 100 / max_average_time)
time_ms = int(average_time)
if height_percent == 0:
continue
html_str += \
'<tr style="height:' + str(height_percent) + '%">\n' + \
' <th scope="row">' + name + '</th>\n' + \
' <td><span>' + str(time_ms) + '</span></td>\n' + \
'</tr>\n'
ctr += 1
if ctr >= max_entries:
break
html_str += '</tbody></table>\n' + html_footer()
return html_str
def fitness_thread(base_dir: str, fitness: {}):
"""Thread used to save fitness function scores
"""
fitness_filename = base_dir + '/accounts/fitness.json'
while True:
# every 10 mins
time.sleep(60 * 10)
save_json(fitness, fitness_filename)