pk_plots.py

# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rcParams, gridspec

do_plots = True

party_colors = {
'АБВ'      : '#6c1d45',
'АТАКА'    : '#00563f',
'ББЦ'      : '#ffff00',
'БДЦ'      : '#ffff00', # maybe change
'БДЦНС'    : '#ffff00', # maybe change
'БНС'      : '#ca7725',
'БСП'      : '#d71921',
'БСПЗБ'    : '#d71921', # maybe change
'БСПЛБ'    : '#d71921', # maybe change
'ВОЛЯ'     : '#00718f',
'ГЕРБ'     : '#0056a7',
'ДПС'      : '#0066b7',
'ДСБ'      : '#02528a',
'КБ'       : '#6c1d45', # maybe change
'НДСВ'     : '#f7cb41',
'ОДС'      : '#002d7b',
'ОП'       : '#000000',
'ПФ'       : '#088a85',
'РБ'       : '#0070bb',
'РЗС'      : '#f67700',
'СК'       : '#02528a', # maybe change
'независим': '#dddddd'
}

##############################################################################
# Matplotlib configuration.
##############################################################################
rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = ['FreeSans']
rcParams['font.size'] = 8
rcParams['figure.figsize'] = (5, 3)
figsize_square = (5, 5)
figsize_square_small = (3, 3)
figsize_long = (10, 3)
rcParams['savefig.dpi'] = 90
rcParams['legend.fontsize'] = 'medium'
rcParams['text.antialiased'] = True
rcParams['patch.antialiased'] = True
rcParams['lines.antialiased'] = True


##############################################################################
# Per stenogram plots.
##############################################################################
def registration_figure(date, names, reg_presences, reg_expected):
    """Barchart: x:name vs y:stacked presences and absences."""
    if not do_plots: return
    datestr = date.strftime('%Y%m%d')
    datestr_human = date.strftime('%d/%m/%Y')
    absences = reg_expected - reg_presences

    pos = np.arange(len(names))
    width = 0.35
    f = plt.figure()
    f.suptitle(u'Регистрирани Депутати %s'%datestr_human)
    gs = gridspec.GridSpec(3,5)
    main = f.add_subplot(gs[:,0:-1])
    p1 = main.bar(pos, reg_presences, width, color='g')
    p2 = main.bar(pos, absences, width, color='r', bottom=reg_presences)
    main.set_ylabel(u'Брой Депутати')
    main.set_xticks(pos+width/2.)
    main.set_xticklabels(names)
    main.tick_params(axis='x', length=0)
    main.set_xlim(-0.5*width, pos[-1]+1.5*width)
    main.legend((p1[0], p2[0]), (u'Присъстващи', u'Отсъстващи'), loc='upper left', bbox_to_anchor=(1,1))
    summ = f.add_subplot(gs[-1,-1])
    pie_array = [np.sum(reg_presences), np.sum(absences)]
    summ.pie(pie_array, colors=['g', 'r'])
    summ.set_title(u'Общо')
    f.savefig('generated_html/registration%s.png' % datestr)
    plt.close()

def absences_figure(date, names, vote_absences, vote_absences_percent):
    """Time series chart: x:session nb vs y:absences, color:name"""
    if not do_plots: return
    datestr = date.strftime('%Y%m%d')
    datestr_human = date.strftime('%d/%m/%Y')
    vote_absences = np.column_stack(vote_absences)
    vote_absences_percent = np.column_stack(vote_absences_percent)

    f = plt.figure()
    f.suptitle(u'Отсъствия по Време на Гласуване %s'%datestr_human)
    gs = gridspec.GridSpec(2,5)
    su = f.add_subplot(gs[0,:-1])
    su.plot(vote_absences, alpha=0.8)
    su.set_ylabel(u'Брой Депутати')
    su.set_ylim(0)
    su.set_xticks([])
    su.legend(names, loc='upper left', bbox_to_anchor=(1,1))
    sd = f.add_subplot(gs[1,:-1], sharex=su)
    sd.plot(vote_absences_percent, alpha=0.8)
    sd.set_ylabel(u'% от Партията')
    sd.set_xlabel(u'хронологичен ред на гласуванията')
    sd.set_ylim(0, 100)
    sd.set_xticks([])
    sd.set_yticks([25, 50, 75])
    f.savefig('generated_html/absences%s.png' % datestr)
    plt.close()


##############################################################################
# Per stenogram session plots.
##############################################################################
f = plt.figure()                      # XXX "ugly tweak"
gs = gridspec.GridSpec(3,5)           # These were made global and reused,
main = f.add_subplot(gs[:,0:-1])      # giving 2x speedup. The `remove` calls
summ = f.add_subplot(gs[-1,-1])       # below are part of the same tweak.
title = f.suptitle('')
main.set_ylabel(u'Брой Депутати')
summ.set_title(u'Общо')

def session_votes_by_party_figure(date, i, party_names, yes, no, abstain, absences):
    if not do_plots: return
    datestr = date.strftime('%Y%m%d')
    datestr_human = date.strftime('%d/%m/%Y')
    pos = np.arange(len(party_names))
    width = 0.35

    global f, main, summ, title

    title.set_text(u'Гласували Депутати %s гл.%d' % (datestr_human, i+1))
    p1 = main.bar(pos, yes, width, color='g')
    p2 = main.bar(pos, no, width, color='r', bottom=yes)
    p3 = main.bar(pos, abstain, width, color='c', bottom=yes+no)
    p4 = main.bar(pos, absences, width, color='k', bottom=yes+no+abstain)
    main.set_xticks(pos+width/2.)
    main.set_xticklabels(party_names)
    main.tick_params(axis='x', length=0)
    main.set_xlim(-0.5*width, pos[-1]+1.5*width)
    main.legend((p1[0], p2[0], p3[0], p4[0]), (u'За', u'Против', u'Въздържали се', u'Отсъстващи'), loc='upper left', bbox_to_anchor=(1,1))
    pie_array = [np.sum(yes), np.sum(no), np.sum(abstain), np.sum(absences)]
    wedges, texts = summ.pie(pie_array, colors=['g', 'r', 'c', 'k'])
    f.savefig('generated_html/session%svotes%s.png' % (datestr, i+1))

    for o in [p1, p2, p3, p4] + wedges + texts:
        o.remove()


##############################################################################
# Pie plots for MPs.
##############################################################################
def alltime_regs(present, absent, manual):
    if not do_plots: return
    f = plt.figure(figsize=figsize_square)
    f.suptitle(u'Обобщение на Всички Регистрации.')
    s = f.add_subplot(1,1,1)
    pie_array = [present, absent, manual]
    s.pie(pie_array, colors=['g', 'r', 'c'], labels=[u'Присъствия', u'Отсъствия', u'Ръчно\nзаписани'])
    f.savefig('generated_html/alltimeregs.png')
    plt.close()

def alltime_votes(y, n, abst, absent):
    if not do_plots: return
    f = plt.figure(figsize=figsize_square)
    f.suptitle(u'Обобщение на Всички Гласове.')
    s = f.add_subplot(1,1,1)
    pie_array = [y, n, abst, absent]
    s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'За', u'Против', u'Въздържали се', u'Отсъстващи'])
    f.savefig('generated_html/alltimevotes.png')
    plt.close()

def alltime_regs_singleMP((present, absent, manual), name, asciiname):
    if not do_plots: return
    f = plt.figure(figsize=figsize_square_small)
    f.suptitle(u'Обобщение на Всички Регистрации на\n%s'%name)
    s = f.add_subplot(1,1,1)
    pie_array = [present, absent, manual]
    s.pie(pie_array, colors=['g', 'r', 'c'], labels=[u'Присъствия', u'Отсъствия', u'Ръчно\nзаписани'])
    f.savefig('generated_html/alltimeregs_%s.png'%asciiname)
    plt.close()

def alltime_votes_singleMP_compare_all((w, a), (abst, absent), name, asciiname):
    if not do_plots: return
    f = plt.figure(figsize=figsize_square_small)
    f.suptitle(u'Обобщение на Всички Гласове на\n%s'%name)
    s = f.add_subplot(1,1,1)
    pie_array = [w, a, abst, absent]
    s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'Съгласие с\nмнозинството', u'\n\n Противоречие с\n   мнозинството', u'Въздържал се\n', u'Отсъстващ'])
    f.savefig('generated_html/alltimevotes_compare_all_%s.png'%asciiname)
    plt.close()

def alltime_votes_singleMP_compare_party((w, a), (abst, absent), name, asciiname):
    if not do_plots: return
    f = plt.figure(figsize=figsize_square_small)
    f.suptitle(u'Обобщение на Всички Гласове на\n%s'%name)
    s = f.add_subplot(1,1,1)
    pie_array = [w, a, abst, absent]
    s.pie(pie_array, colors=['g', 'r', 'c', 'k'], labels=[u'Съгласие с\nпартията', u'\n\n Противоречие с\n    партията', u'Въздържал се\n', u'Отсъстващ'])
    f.savefig('generated_html/alltimevotes_compare_party_%s.png'%asciiname)
    plt.close()

def evolution_of_votes_singleMP(dates, votes, wa_all, wa_party, name, asciiname):
    if not do_plots: return
    f = plt.figure(figsize=figsize_long)
    f.suptitle(u'Гласове и отсъствия на %s през годините.'%name)
    absences = f.add_subplot(3,1,3)
    with_all = f.add_subplot(3,1,1, sharex=absences)
    with_party = f.add_subplot(3,1,2, sharex=absences)

    all_votes_no_abs = np.sum(votes[:,:3], 1)
    all_votes = np.sum(votes, 1)
    mask_no_abs = np.logical_not(all_votes_no_abs)
    mask = np.logical_not(all_votes)
    with_all_array = np.ma.masked_array(100*wa_all[:,0], mask=mask_no_abs)/all_votes_no_abs
    with_party_array = np.ma.masked_array(100*wa_party[:,0], mask=mask_no_abs)/all_votes_no_abs
    absences_array = np.ma.masked_array(100*votes[:,3], mask=mask)/all_votes

    with_all.plot(dates, with_all_array, '.-', alpha=0.3, linewidth=0.1)
    with_all.legend([u'% съгласие с мнозинството (без отсъствия)'])
    with_party.plot(dates, with_party_array, '.-', alpha=0.3, linewidth=0.1)
    with_party.legend([u'% съгласие с партията (без отсъствия)'])
    absences.plot(dates, absences_array, '.-', alpha=0.3, linewidth=0.1)
    absences.legend([u'% отсъствия'])

    with_all.set_yticks([25, 50, 75])
    with_party.set_yticks([25, 50, 75])
    absences.set_yticks([25, 50, 75])
    with_all.set_ylim(0, 100)
    with_party.set_ylim(0, 100)
    absences.set_ylim(0, 100)
    absences.set_xlim(dates[0], dates[-1])
    f.autofmt_xdate()
    f.savefig('generated_html/vote_evol_%s.png'%asciiname)
    plt.close()