# -*- coding: utf-8 -*-

"""

Plotting terminal based barplots

"""

from .utils.helpers import *

from .histogram import calc_bins, read_numbers

def plot_bar(f, height=20.0, bincount=None, binwidth=None, pch="o",

colour="default", title="", xlab=None, showSummary=False,

regular=False, return_str=False):

''' Plot bar.

Parameters

----------

f : list(number), numpy.ndarray, str(filepath)

input array

height : float

the height of the histogram in # of lines

bincount : int

number of bins in the histogram

binwidth : int

width of bins in the histogram

pch : str

shape of the bars in the plot, e.g 'o'

colour : str

white,aqua,pink,blue,yellow,green,red,grey,black,default,ENDC

title : str

title at the top of the plot, None = no title

xlab : boolean

whether or not to display x-axis labels

showSummary : boolean

whether or not to display a summary

regular : boolean

whether or not to start y-labels at 0

return_str : boolean

return string represent the plot or print it out, default: False

Example

-------

>>> y = np.random.rand(50)

>>> plot_bar(y, bincount=50, colour='red')

>>> 0.971|

>>> 0.923| o o o o

>>> 0.875| o o o o o oo

>>> 0.827| o o o o o o ooo

>>> 0.779| o o o o o o o o ooo

>>> 0.731| o o o o o o o o o ooo

>>> 0.683| oo oo o o o o o o o o o ooo

>>> 0.635| oo oo o oo o o o o o oo o ooo

>>> 0.587| oo oo o oo o ooo o o oo o ooo

>>> 0.539| oo oo o oooo ooo o o oo o ooo o

>>> 0.491| ooo ooo o oooo ooo o o ooo o ooo o

>>> 0.443| ooo ooo oo ooooo oooo o o ooo o ooo o

>>> 0.395| ooo ooo ooooooooo oooo o o ooo o ooo o

>>> 0.347| oooo ooo ooooooooo ooooo o o ooo o ooo o

>>> 0.299| oooo ooo ooooooooo ooooo o o ooo o ooo o

>>> 0.251| oooo ooo ooooooooo ooooo o o ooo o ooo o o

>>> 0.203| oooo ooo ooooooooo ooooo ooo ooo o ooo o o

>>> 0.155| oooo ooo oo ooooooooo ooooo ooo ooo ooooooo oo

>>> 0.107| ooooo ooo oo ooooooooo oooooooooo ooooooooooo ooo

>>> 0.059| ooooo oooooo ooooooooo ooooooooooooooooooooooo ooo

>>> 0.011| oooooooooooooooooooooo ooooooooooooooooooooooooooo

>>> --------------------------------------------------

'''

if pch is None:

pch = "o"

splot = ''

if isinstance(f, str):

f = open(f).readlines()

# ====== Create data ====== #

min_val, max_val = None, None

n, mean, sd = 0.0, 0.0, 0.0

# pick mode and get data

numbers = [i for i in read_numbers(f)]

int_mode = False

if numbers[0].is_integer():

int_mode = True

# rescale big enough to show on bars

min_orig = min(numbers) # original

max_orig = max(numbers)

numbers = [1000 * (i - min_orig) / (max_orig - min_orig + 1e-8) for i in numbers]

# statistics

n = len(numbers)

min_val = min(numbers)

max_val = max(numbers)

mean = sum(numbers) / n

sd = (sum([(mean - i)**2 for i in numbers]) / (n - 1)) ** 0.5

# bins is index

if bincount is not None:

bincount = min(bincount, n)

bins = list(calc_bins(n, 0., n + 0., bincount, binwidth))

bins = [int(i) for i in bins]

hist = dict((i, 0) for i in range(len(bins) - 1))

# hist is the mean value of array with indices within bin

for idx, (i, j) in enumerate(zip(bins, bins[1:])):

arr = numbers[i:j]

hist[idx] = sum(arr) / len(arr) # calculate mean

# ====== Start plot ====== #

min_y, max_y = min(hist.values()), max(hist.values())

start = max(min_y, 1)

stop = max_y + 1

if regular:

start = 1

if height is None:

height = stop - start

if height > 20:

height = 20

ys = list(drange(start, stop, float(stop - start) / height))

ys.reverse()

nlen = max(len(str(min_y)), len(str(max_y))) + 1

if title:

splot += print_return_str(

box_text(title, max(len(hist) * 2, len(title)), nlen),

return_str=return_str)

splot += print_return_str('', return_str=return_str)

used_labs = set()

for y in ys:

if int_mode:

ylab = '%d' % int(y * (max_orig - min_orig + 1e-8) / 1000 + min_orig)

else:

ylab = '%.3f' % float(y * (max_orig - min_orig + 1e-8) / 1000 + min_orig)

if ylab in used_labs:

ylab = ""

else:

used_labs.add(ylab)

ylab = " " * (nlen - len(ylab)) + ylab + "|"

splot += print_return_str(ylab, end=' ', return_str=return_str)

for i in range(len(hist)):

if int(y) <= hist[i]:

splot += printcolour(pch, True, colour, return_str)

else:

splot += printcolour(" ", True, colour, return_str)

splot += print_return_str('', return_str=return_str)

xs = hist.keys()

splot += print_return_str(" " * (nlen + 1) + "-" * len(xs),

return_str=return_str)

if xlab:

xlen = len(str(float((max_y) / height) + max_y))

for i in range(0, xlen):

splot += printcolour(" " * (nlen + 1), True, colour, return_str)

for x in range(0, len(hist)):

num = str(bins[x])

if x % 2 != 0:

pass

elif i < len(num):

splot += print_return_str(num[i], end=' ',

return_str=return_str)

else:

splot += print_return_str(" ", end=' ',

return_str=return_str)

splot += print_return_str('', return_str=return_str)

center = max(map(len, map(str, [n, min_val, mean, max_val])))

center += 15

if showSummary:

splot += print_return_str('', return_str=return_str)

splot += print_return_str("-" * (2 + center), return_str=return_str)

splot += print_return_str("|" + "Summary".center(center) + "|",

return_str=return_str)

splot += print_return_str("-" * (2 + center), return_str=return_str)

summary = "|" + ("observations: %d" % n).center(center) + "|\n"

summary += "|" + ("min value: %f" % min_val).center(center) + "|\n"

summary += "|" + ("mean : %f" % mean).center(center) + "|\n"

summary += "|" + ("sd : %f" % sd).center(center) + "|\n"

summary += "|" + ("max value: %f" % max_val).center(center) + "|\n"

summary += "-" * (2 + center)

splot += print_return_str(summary, return_str=return_str)

if return_str:

return splot