baroness/app/plot.py

#import matplotlib #speed?
#matplotlib.use('GTKAgg')
from matplotlib import pyplot as plt
from matplotlib.dates import WeekdayLocator, DayLocator, HourLocator, DateFormatter, drange, MONDAY
import numpy as np
from user import User
from database import *
import multiprocessing as mp


def plot_all_thread(user=None):
    if user is not None:
        proc = mp.Process(target=plot_all, args=(user,))
    else:
        proc = mp.Process(target=plot_all, args=())
    proc.start()


def plot_all(user=None):
    if user is not None:
        plot_total(user)
    plot_total()
    plot_list(4)
    print 'plot_all'


def plot_total(user=None):
    print "start plot_total " + str(datetime.datetime.now())
    print 'plot_total'
    today = datetime.date.today()
    delta = datetime.timedelta(days=1)
    begin = today - datetime.timedelta(weeks=2)
    dates = drange(begin, today + delta, delta)

    #print begin
    #print today
    #print dates

    products = get_products()
    allconsumptions = [[0 for x in range(len(dates))] for product in products]

    consumed = get_consumed()
    for consumption in consumed:
        if user == None or consumption.consumer == user.id:
            i = 1
            for consumptions in allconsumptions:
                if consumption.prodnr == i:
                    if consumption.time.date() > begin:
                        consumptions[(consumption.time.date() - begin).days] += 1
                i += 1
    plt.xkcd()

    #print allconsumptions
    fig, ax = plt.subplots()

    i = 0
    for consumptions in allconsumptions:
        ax.plot(dates, consumptions, linestyle='-', marker='', label=products[i].name)
        i += 1

    ax.legend(loc=2,prop={'size':15})

   # ax.axes.margins = 1  # x margin.  See `axes.Axes.margins`
   # ax.axes.ymargin = 1  # y margin See `axes.Axes.margins`

    plt.xticks(rotation='vertical')

    ax.spines['right'].set_visible(False)
    ax.spines['top'].set_visible(False)

    ax.yaxis.set_ticks_position('left')
    ax.xaxis.set_ticks_position('bottom')

    plt.tick_params(which='minor', length=4)
    plt.tick_params(which='major', length=5)

    #ax.xaxis.set_major_locator(WeekdayLocator(MONDAY))
    #ax.xaxis.set_major_locator(DayLocator())
    ax.xaxis.set_major_formatter(DateFormatter('%d.%m'))
    #ax.xaxis.set_minor_formatter(DateFormatter('%d.%m'))

    #ax.fmt_xdata = DateFormatter('%d.%m')
    fig.autofmt_xdate()

    plt.xlabel('Datum')
    plt.ylabel('Konsumate')

    if user == None:
        tit = "Bierkonsum FET"
        fils = "app/static/total.png"
        fill = "app/static/total_big.png"
    else:
        tit = "Bierkonsum %s" % user.name
        fils = "app/static/total%03d.png" % user.id
        fill = "app/static/total%03d_big.png" % user.id
    plt.title(tit)
    print "plot plot_total " + str(datetime.datetime.now())
    #480x320
    fig.set_size_inches(4.8, 3.2)
    plt.savefig(fils, dpi=100)
    print "end plot_total " + str(datetime.datetime.now())
    #fig.set_size_inches(4.8, 3.2)
    #plt.savefig(fill, dpi=400)


def plot_list(duration):
    print "start plot_list " + str(datetime.datetime.now())
    today = datetime.datetime.today()
    begin = today - datetime.timedelta(weeks=duration)

    users = get_users()
    products = get_products()
    #consumptions = [0 for user in users]
    allconsumptions = [[0 for user in users] for product in products]

    consumed = get_consumed(startdate=begin)

    for consumption in consumed:
        allconsumptions[consumption.prodnr-1][consumption.consumer-1] += 1

    #cumulate consumptions for cumulative bar graph
    i = 0
    for consumptions in allconsumptions:
        if i > 0:
            j = 0
            for consumption in consumptions:
                allconsumptions[i][j] += allconsumptions[i-1][j]
                j += 1
        i += 1

    plt.xkcd()

    fig, ax = plt.subplots()
    ax.grid(True, linewidth=0.5)
    ax.grid(True, linewidth=0.5, which='minor')
    ax.yaxis.grid(False, which='major')
    ax.yaxis.grid(True, which='minor')

    colors = ['blue', 'green', 'red', 'yellow', 'orange' , 'black']

    #plot reversed to print longest bar lowest
    i = len(allconsumptions) - 1
    for consumptions in reversed(allconsumptions):
        x = list()
        y = list()
        for j, consumption in enumerate(consumptions):
            x += list(np.arange(1, consumption+1)-0.5)
            y += list(np.multiply(j, np.ones(consumption)))
        ax.scatter(x, y, marker='x', s=70, color = colors[i], label=products[i].name)
        #ax.barh(np.arange(len(consumptions)), consumptions, label=products[i].name, align='center', height=(0.5), color=colors[i])
        i -= 1

    names = list()

    for user in users:
        names.append(user.longname)

    #ticks
    plt.yticks(np.arange(len(names)), names)
    ax.set_yticks(np.arange(len(names))-0.5, minor=True)
    #limits
    plt.ylim(-0.5, len(names)-0.5)
    plt.xlim(0)
    plt.gca().set_xlim(right=30)
    #legend = auto
    ax.legend(loc=0)

    ax.spines['right'].set_visible(False)
    ax.spines['top'].set_visible(False)

    ax.yaxis.set_ticks_position('none')
    ax.xaxis.set_ticks_position('none')

    plt.subplots_adjust(left=0.15)
    #plt.tick_params(which='minor', length=4)
    #plt.tick_params(which='major', length=5)

    plt.xlabel('Konsumate')
    #plt.ylabel('Konsumate')

    plt.title("Bierliste ("+ str(duration) + " Wochen)")

    print "plot plot_list " + str(datetime.datetime.now())
    #800x600
    fig.set_size_inches(15, 10)
    plt.savefig('app/static/bierliste_small.png', dpi=72, bbox_inches='tight')
    #1024x768
    #fig.set_size_inches(10.24, 7.68)
    #plt.savefig('app/static/bierliste.png', dpi=100)
    print "end plot_list " + str(datetime.datetime.now())