[util] Add graph library
Add a library for drawing horizontal and vertical bar graphs (thanks to @somospocos for contributing!)
This commit is contained in:
parent
3496fa622a
commit
b2064142d0
1 changed files with 111 additions and 0 deletions
111
util/graph.py
Normal file
111
util/graph.py
Normal file
|
@ -0,0 +1,111 @@
|
||||||
|
MAX_PERCENTS = 100.
|
||||||
|
|
||||||
|
class Bar(object):
|
||||||
|
"""superclass"""
|
||||||
|
bars = None
|
||||||
|
|
||||||
|
def __init__(self, value):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
|
||||||
|
value (float): value between 0. and 100. meaning percents
|
||||||
|
"""
|
||||||
|
self.value = value
|
||||||
|
|
||||||
|
class HBar(Bar):
|
||||||
|
"""horizontal bar (1 char)"""
|
||||||
|
bars = [
|
||||||
|
u"\u2581",
|
||||||
|
u"\u2582",
|
||||||
|
u"\u2583",
|
||||||
|
u"\u2584",
|
||||||
|
u"\u2585",
|
||||||
|
u"\u2586",
|
||||||
|
u"\u2587",
|
||||||
|
u"\u2588"
|
||||||
|
]
|
||||||
|
|
||||||
|
def __init__(self, value):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
|
||||||
|
value (float): value between 0. and 100. meaning percents
|
||||||
|
"""
|
||||||
|
super(HBar, self).__init__(value)
|
||||||
|
self.step = MAX_PERCENTS / len(HBar.bars)
|
||||||
|
|
||||||
|
def get_char(self):
|
||||||
|
"""
|
||||||
|
Decide which char to draw
|
||||||
|
|
||||||
|
Return: str
|
||||||
|
"""
|
||||||
|
for i in range(len(HBar.bars)):
|
||||||
|
left = i * self.step
|
||||||
|
right = (i + 1) * self.step
|
||||||
|
if left <= self.value < right:
|
||||||
|
return self.bars[i]
|
||||||
|
return self.bars[-1]
|
||||||
|
|
||||||
|
def hbar(value):
|
||||||
|
"""wrapper function"""
|
||||||
|
return HBar(value).get_char()
|
||||||
|
|
||||||
|
|
||||||
|
class VBar(Bar):
|
||||||
|
"""vertical bar (can be more than 1 char)"""
|
||||||
|
bars = [
|
||||||
|
u"\u258f",
|
||||||
|
u"\u258e",
|
||||||
|
u"\u258d",
|
||||||
|
u"\u258c",
|
||||||
|
u"\u258b",
|
||||||
|
u"\u258a",
|
||||||
|
u"\u2589",
|
||||||
|
u"\u2588"
|
||||||
|
]
|
||||||
|
|
||||||
|
def __init__(self, value, width=1):
|
||||||
|
"""
|
||||||
|
Args:
|
||||||
|
|
||||||
|
value (float): value between 0. and 100. meaning percents
|
||||||
|
|
||||||
|
width (int): width
|
||||||
|
"""
|
||||||
|
super(VBar, self).__init__(value)
|
||||||
|
self.step = MAX_PERCENTS / (len(VBar.bars) * width)
|
||||||
|
self.width = width
|
||||||
|
|
||||||
|
def get_chars(self):
|
||||||
|
"""
|
||||||
|
Decide which char to draw
|
||||||
|
|
||||||
|
Return: str
|
||||||
|
"""
|
||||||
|
if self.value == 100:
|
||||||
|
return self.bars[-1] * self.width
|
||||||
|
if self.width == 1:
|
||||||
|
for i in range(len(VBar.bars)):
|
||||||
|
left = i * self.step
|
||||||
|
right = (i + 1) * self.step
|
||||||
|
if left <= self.value < right:
|
||||||
|
return self.bars[i]
|
||||||
|
else:
|
||||||
|
full_parts = int(self.value // (self.step * len(Vbar.bars)))
|
||||||
|
remainder = self.value - full_parts * self.step * CHARS
|
||||||
|
empty_parts = self.width - full_parts
|
||||||
|
if remainder >= 0:
|
||||||
|
empty_parts -= 1
|
||||||
|
part_vbar = VBar(remainder * self.width) # scale to width
|
||||||
|
chars = self.bars[-1] * full_parts
|
||||||
|
chars += part_vbar.get_chars()
|
||||||
|
chars += " " * empty_parts
|
||||||
|
return chars
|
||||||
|
|
||||||
|
|
||||||
|
def vbar(value, width):
|
||||||
|
"""wrapper function"""
|
||||||
|
return VBar(value, width).get_chars()
|
||||||
|
|
||||||
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|
Loading…
Reference in a new issue