quoracle/examples/plot_workload_distribution.py
Michael Whittaker 07a25f6ca9 Revised paper.
2021-02-04 20:53:42 -08:00

51 lines
1.5 KiB
Python

# See https://stackoverflow.com/a/19521297/3187068
import matplotlib
matplotlib.use('pdf')
font = {'size': 8}
matplotlib.rc('font', **font)
from quorums import *
import itertools
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
def main():
a = Node('a', write_capacity=100, read_capacity=200)
b = Node('b', write_capacity=100, read_capacity=200)
c = Node('c', write_capacity=50, read_capacity=100)
d = Node('d', write_capacity=50, read_capacity=100)
dist = {
0.00: 10 / 18,
0.25: 4 / 18,
0.50: 2 / 18,
0.75: 1 / 18,
1.00: 1 / 18,
}
qs = QuorumSystem(reads=a*c + b*d)
xs = np.arange(0, 1.01, 0.01)
markers = itertools.cycle(['o', 'v', '^', 'p', '*'])
fig, ax = plt.subplots(figsize=(3.25, 2.5))
for fr in dist.keys():
sigma = qs.strategy(read_fraction=fr)
ys = [sigma.capacity(read_fraction=x) for x in xs]
ax.plot(xs, ys, '--', label=str(f'$\sigma_{{{fr}}}$'), linewidth=1,
marker=next(markers), markevery=25, markersize=4, alpha=0.75)
sigma = qs.strategy(read_fraction=dist)
ys = [sigma.capacity(read_fraction=x) for x in xs]
ax.plot(xs, ys, label='$\sigma$', linewidth=1.5,
marker=next(markers), markevery=25, markersize=4)
ax.legend(ncol=3, loc='lower center', bbox_to_anchor=(0.5, 1.0))
ax.set_ylabel('Capacity')
ax.set_xlabel('Read Fraction')
ax.set_xticks([0, 0.25, 0.5, 0.75, 1])
fig.tight_layout()
fig.savefig(f'workload_distribution.pdf')
if __name__ == '__main__':
main()