alsk/plot/mkplot
2021-05-10 18:14:13 +02:00

142 lines
2.7 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os, sys
sys.path.append(os.path.abspath(os.path.dirname(os.path.abspath(__file__))+"/../scripts"))
import fileinput
import numpy as np
import matplotlib as mpl
show = len(sys.argv) > 1 and sys.argv[1] == 'show'
if not show:
mpl.use('pgf')
import matplotlib.pyplot as plt
import matplotlib.lines as lines
from common import set_size, config_plt
config_plt()
def loadData(fn):
xs = []
ys = []
content = []
cst = 0
inf = 0
sup = 0
with open(fn) as f:
cst, inf, sup = [float(x) for x in f.readline().split(',')]
content = f.readlines()
for line in content:
tok = line.split(',')
xs.append(float(tok[0]))
ys.append(float(tok[1]))
return [[cst, inf, sup], xs, ys]
def drawLine(plt, b, e, label, color):
x = np.arange(b, e, 1)
plt.plot(x, x, label=label, color=color);
def drawHLine(plt, b, e, v, color):
x = np.arange(b, e, 1)
plt.plot(x, [v]*(e-b), color=color);
def drawCloud(plt, data, label, color):
x, y = data
plt.scatter(x, y, label=label, s=3, linewidth=0)
def figCores(n, lblA, lblB):
name = 'fig'+str(n)
fig = plt.figure()
data = loadData('plot_data/'+name+'a')
cst, inf, sup = data[0]
# plt.title(u"nombre de cœurs fixe = " + str(int(cst)))
plt.xlabel(u'tâches')
plt.ylabel(u'PRNG')
drawLine(plt, inf, sup, u"sans optimisation", 'red')
drawCloud(plt, data[1:], lblA, 'blue')
data = loadData('plot_data/'+name+'b')
drawCloud(plt, data[1:], lblB, 'green')
fig.set_size_inches(set_size(455.24408, .8))
fig.legend()
if not show:
fig.savefig('plots/'+name+'.pdf', format='pdf', bbox_inches='tight')
return plt
def figTasks(n):
name = 'fig'+str(n)
fig = plt.figure()
width = 3.487
height = width/1.618
data = loadData('plot_data/'+name+'a')
cst, inf, sup = data[0]
plt.title(u"nombre de tâches fixe = " + str(int(cst)))
plt.xlabel(u'cœurs')
plt.ylabel(u'PRNG')
drawCloud(plt, data[1:], 'equilibrated distribution', 'blue')
data = loadData('plot_data/'+name+'b')
drawCloud(plt, data[1:], 'greedy distribution', 'green')
fig.set_size_inches(set_size(455.24408, .8))
fig.legend()
if not show:
fig.savefig('plots/'+name+'.pdf', format='pdf', bbox_inches='tight')
return plt
def fig1():
return figCores(1, u"politique d'exécution 1", u"politique d'exécution 2")
def fig2():
return figCores(2, u"politique d'exécution 1", u"politique d'exécution 2")
def fig3():
return figCores(3, u"$\mathbf{T} = \{1, 2, 3, 4, 5, 6, ..., 64\}$", u"$\mathbf{T} = \{1, 2, 4, 8, 16, 32, 64\}$")
def fig4():
return figTasks(4)
def fig5():
return figTasks(5)
def fig6():
return figTasks(6)
plts = []
plts.append(fig1())
plts.append(fig2())
plts.append(fig3())
plts.append(fig4())
plts.append(fig5())
plts.append(fig6())
if show:
for plt in plts:
plt.show()