# coding=utf-8
import matplotlib.pyplot as pl
import matplotlib
matplotlib.rcParams['svg.fonttype'] = 'none'
import numpy as np
# pl.ion()


#color codes as in http://en.wikipedia.org/wiki/Template:Canadian_politics/party_colours, but more saturated to enhance visibility
colorcodes = {'Liberal Party of Canada': '#ff7171',
              'Conservative Party of Canada':'#5290ff',
              '(Progressive) Conservative Party':'#9999FF',
              'Reform Party/Canadian Alliance':'#3CB371',
              'New Democratic Party':'#ffa355',
              'Co-operative Commonwealth':'#ffff00',
              'Bloc Québécois':'#82cfff',
              'Social Credit Party':'#7fff7f',
              'Ralliement des créditistes':'#7fff7f',
              'Other':'#000000',
              'Green Party of Canada':'#2F873E',
              'New Democracy (Includes votes for the Social Credit)':'#7fff7f',
              'Liberal Labour Party*':'#FFFFFF',
              'Unionist Party':'#9999FF'}

#Results of elections
#data source: http://www.parl.gc.ca/parlinfo/compilations/electionsandridings/ResultsParty.aspx
#the following dictionary contains for each party a list of years and four lists of the corresponding results.
#results[party] = [years, [Elected_Members,	Number_of_Candidates, Number_of_Valid_Votes, Percentage_of_Popular_Vote]]
results = \
{'Ralliement des créditistes': [[1968, 1965], [[14, 71, 361045, 4.4], [9, 77, 359438, 4.7]]], \
'Reform Party/Canadian Alliance': [[2000, 1997, 1993], [[66, 298, 3276929, 25.5], [60, 227, 2513070, 19.4], [52, 207, 2559245, 18.7]]], \
'New Democratic Party': [[2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004, 2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962], [[7, 308, 1237295, 6.3], [25, 308, 3036346, 17.83], [24, 308, 2903722, 15.98], [44, 308, 3441409, 19.71], [103, 308, 4512411, 30.6], [37, 308, 2515288, 18.2], [29, 308, 2589597, 17.5], [19, 308, 2127403, 15.7], [13, 298, 1093868, 8.5], [21, 301, 1434509, 11.0], [9, 294, 939575, 6.9], [43, 295, 2685308, 20.38], [30, 282, 2359915, 18.8], [32, 282, 2164987, 19.78], [26, 282, 2048779, 17.88], [16, 262, 1467748, 15.44], [31, 251, 1713528, 17.72], [22, 263, 1378260, 17.0], [21, 255, 1381658, 17.9], [17, 232, 1037857, 13.1], [19, 218, 1036853, 13.4]]], \
'New Democracy (Includes votes for the Social Credit)': [[1940], [[10, 30, 123443, 2.7]]], \
'(Progressive) Conservative Party': [[2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1940, 1935, 1930, 1926, 1925, 1921, 1917, 1911, 1908, 1904, 1900, 1896, 1891, 1887, 1882, 1878, 1874, 1872, 1867], [[12, 291, 1566998, 12.2], [20, 301, 2446705, 18.8], [2, 295, 2186422, 16.0], [169, 295, 5667563, 43.02], [211, 282, 6278697, 50.03], [103, 282, 3552994, 32.45], [136, 282, 4111559, 35.89], [95, 264, 3369335, 35.44], [107, 264, 3383530, 35.0], [72, 263, 2554880, 31.4], [97, 265, 2499913, 32.4], [95, 265, 2591614, 32.9], [116, 265, 2868295, 37.3], [208, 265, 3910852, 53.7], [112, 256, 2578045, 39.0], [51, 248, 1751215, 31.0], [41, 250, 1742276, 29.7], [67, 203, 1455453, 27.7], [39, 213, 1416257, 30.6], [39, 232, 1311459, 29.8], [137, 233, 1909955, 49.0], [91, 233, 1504855, 46.2], [116, 234, 1466324, 46.6], [50, 208, 945681, 30.3], [153, 0, 1074918, 57.0], [133, 215, 669594, 51.2], [85, 212, 552134, 47.0], [75, 205, 483177, 46.9], [81, 204, 450790, 47.4], [88, 209, 416640, 46.3], [123, 217, 404599, 52.0], [123, 209, 367191, 50.7], [139, 0, 275537, 53.38], [137, 0, 282966, 53.2], [73, 0, 0, 45.4], [103, 0, 0, 49.9], [101, 0, 0, 50.0]]], \
'Liberal Labour Party*': [[1958], [[1, 1, 11956, 0.2]]], \
'Other': [[2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004, 2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1940, 1935, 1930, 1926, 1925, 1921, 1917, 1911, 1908, 1904, 1900, 1896, 1891, 1887, 1882, 1878, 1874, 1872, 1867], [[0, 0, 232355, 1.19], [0, 0, 995253, 5.84], [1, 0, 432566, 2.38], [0, 0, 141453, 0.80], [0, 285, 129703, 1.0], [2, 604, 1096761, 7.8], [1, 635, 820875, 5.5], [1, 686, 755472, 5.6], [0, 545, 290220, 2.3], [1, 467, 211482, 1.6], [1, 989, 488453, 3.6], [0, 681, 719841, 4.65], [1, 552, 376964, 3.0], [0, 574, 190533, 1.74], [0, 475, 173441, 1.52], [1, 267, 84818, 0.9], [2, 173, 362773, 1.19], [1, 76, 71910, 0.9], [2, 63, 90334, 1.1], [0, 38, 30112, 0.4], [1, 40, 14336, 0.2], [0, 50, 35630, 0.4], [4, 66, 87188, 1.3], [5, 143, 127961, 2.3], [8, 132, 255079, 4.4], [12, 216, 582393, 11.1], [10, 85, 151116, 3.3], [11, 251, 566041, 12.8], [17, 79, 274180, 7.1], [38, 77, 329849, 10.2], [30, 130, 408612, 13.0], [69, 225, 905562, 29.0], [0, 0, 55325, 2.9], [2, 26, 12831, 1.0], [3, 31, 28299, 2.4], [0, 28, 6784, 0.7], [0, 25, 5546, 0.6], [7, 61, 76900, 8.6], [0, 27, 12662, 1.6], [0, 37, 2708, 0.4], [0, 0, 187, 0.04], [0, 0, 8817, 1.7], [0, 0, 0, 0.8], [0, 0, 0, 1.0], [0, 0, 0, 1.0]]], \
'Bloc Québécois': [[2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004, 2000, 1997, 1993], [[23, 0, 1232853, 6.3], [32, 0, 1301598, 7.6], [32, 0, 1387030, 7.6], [10, 0, 818652, 4.7], [4, 75, 891425, 6.1], [49, 75, 1379991, 10.0], [51, 75, 1553201, 10.5], [54, 75, 1680109, 12.4], [38, 75, 1377727, 10.7], [44, 75, 1385821, 10.7], [54, 75, 1846024, 13.5]]], \
'Liberal Party of Canada': [[2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004, 2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1940, 1935, 1930, 1926, 1925, 1921, 1917, 1911, 1908, 1904, 1900, 1896, 1891, 1887, 1882, 1878, 1874, 1872, 1867], [[168, 0, 8556629, 43.7], [160, 0, 5556629, 32.6], [157, 0, 6018728, 33.12], [184, 0, 6928055, 39.47], [34, 308, 2783076, 18.9], [77, 307, 3633185, 26.3], [103, 308, 4479415, 30.2], [135, 308, 4982220, 36.7], [172, 301, 5252031, 40.8], [155, 301, 4994377, 38.5], [177, 295, 5647952, 41.3], [83, 294, 4205072, 31.92], [40, 282, 3516486, 28.03], [147, 282, 4853914, 44.34], [114, 282, 4594319, 40.1], [141, 264, 4102776, 43.16], [109, 263, 3718258, 38.46], [155, 263, 3696945, 45.5], [131, 265, 3099519, 40.2], [129, 264, 3293790, 41.7], [99, 263, 2874076, 37.4], [48, 264, 2447744, 33.6], [105, 264, 2796039, 42.3], [171, 264, 2819813, 50.0], [190, 258, 2929391, 50.1], [125, 233, 2170625, 41.4], [178, 244, 2536214, 54.9], [171, 244, 1955727, 44.4], [91, 234, 1714860, 43.9], [116, 220, 1421804, 43.6], [99, 215, 1269401, 40.4], [116, 202, 1272660, 40.7], [82, 0, 755086, 40.1], [86, 222, 625103, 47.8], [133, 216, 594270, 50.6], [139, 211, 540827, 52.5], [132, 207, 494427, 52.0], [118, 193, 405506, 45.1], [92, 192, 361238, 46.4], [92, 193, 353919, 48.9], [71, 0, 240420, 46.58], [69, 0, 240065, 45.1], [133, 0, 0, 53.8], [97, 0, 0, 49.1], [80, 0, 0, 49.0]]], \
'Social Credit Party': [[1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1935], [[0, 9, 3407, 0.03], [0, 51, 16659, 0.13], [0, 84, 185486, 1.69], [6, 103, 527604, 4.61], [11, 152, 481231, 5.06], [15, 166, 737972, 7.63], [0, 31, 62956, 0.8], [5, 86, 282454, 3.7], [24, 224, 940703, 11.9], [30, 230, 896574, 11.7], [0, 82, 188717, 2.6], [19, 114, 437049, 6.6], [15, 72, 305973, 5.4], [10, 28, 139815, 2.4], [13, 95, 214998, 4.1], [17, 46, 182767, 4.1]]], \
'Total': [[2011, 2008, 2006, 2004, 2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1940, 1935, 1930, 1926, 1925, 1921, 1917, 1911, 1908, 1904, 1900, 1896, 1891, 1887, 1882, 1878, 1874, 1872, 1867], [[308, 1587, 14723980, 100.0], [308, 1601, 13834294, 100.0], [308, 1634, 14817159, 100.0], [308, 1685, 13564702, 100.0], [301, 1808, 12857773, 100.0], [301, 1672, 12985964, 100.0], [295, 2155, 13667671, 100.0], [295, 1574, 13281191, 100.0], [282, 1449, 12548721, 100.0], [282, 1504, 10947914, 100.0], [282, 1424, 11455702, 100.0], [264, 1209, 9505908, 100.0], [264, 1117, 9916061, 100.0], [264, 967, 8125996, 100.0], [265, 1011, 7713316, 100.0], [265, 1023, 7894076, 100.0], [265, 1016, 7690134, 100.0], [265, 831, 7287297, 100.0], [265, 862, 6605980, 100.0], [265, 897, 5641272, 100.0], [262, 848, 5848971, 100.0], [245, 952, 5246130, 100.0], [245, 668, 4620260, 100.0], [245, 892, 4406854, 100.0], [245, 546, 3898995, 100.0], [245, 530, 3256508, 100.0], [245, 579, 3144337, 100.0], [235, 635, 3123903, 100.0], [235, 470, 1885329, 100.0], [221, 463, 1307528, 100.0], [221, 459, 1174703, 100.0], [214, 444, 1030788, 100.1], [213, 436, 950763, 100.0], [213, 463, 899046, 100.0], [215, 436, 778499, 100.0], [215, 439, 723818, 100.0], [210, 413, 516144, 100.0], [206, 412, 531848, 100.0], [206, 364, 0, 100.0], [200, 337, 0, 100.0], [181, 0, 0, 100.0]]], \
'Conservative Party of Canada': [[2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004], [[144, 0, 8089941, 41.27], [119, 0, 5747410, 33.74], [121, 0, 6239227, 34.34], [99, 0, 5578101, 31.89], [166, 307, 5835270, 39.6], [143, 307, 5209069, 37.7], [124, 308, 5374071, 36.3], [99, 308, 4019498, 29.6]]], \
'Co-operative Commonwealth': [[1958, 1957, 1953, 1949, 1945, 1940, 1935], [[8, 169, 692398, 9.5], [25, 162, 707659, 10.8], [23, 170, 636310, 11.3], [13, 180, 782410, 13.4], [28, 205, 822661, 15.7], [8, 96, 393230, 8.5], [7, 119, 390860, 8.9]]], \
'Green Party of Canada': [[2025, 2021, 2019, 2015, 2011], [[1, 0, 244963, 1.31], [2, 0, 397014, 2.33], [3, 0, 1189607, 6.55], [1, 0, 605637, 3.45], [1, 304, 572095, 3.9]]]}

#years of elections
years = [2028, 2025, 2021, 2019, 2015, 2011, 2008, 2006, 2004, 2000, 1997, 1993, 1988, 1984, 1980, 1979, 1974, 1972, 1968, 1965, 1963, 1962, 1958, 1957, 1953, 1949, 1945, 1940, 1935, 1930, 1926, 1925, 1921, 1917, 1911, 1908, 1904, 1900, 1896, 1891, 1887, 1882, 1878, 1874, 1872, 1867]

#governments after the elections
#source https://de.wikipedia.org/wiki/Liste_der_kanadischen_Unterhauswahlen
governments = ['Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Conservative Party of Canada', 'Conservative Party of Canada', 'Conservative Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', '(Progressive) Conservative Party', '(Progressive) Conservative Party', 'Liberal Party of Canada', '(Progressive) Conservative Party', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', '(Progressive) Conservative Party', '(Progressive) Conservative Party', '(Progressive) Conservative Party', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', '(Progressive) Conservative Party', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Unionist Party', '(Progressive) Conservative Party', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', 'Liberal Party of Canada', '(Progressive) Conservative Party', '(Progressive) Conservative Party', '(Progressive) Conservative Party', '(Progressive) Conservative Party', 'Liberal Party of Canada', '(Progressive) Conservative Party', '(Progressive) Conservative Party']


#add some small parties to 'Others'
def add_to_others(party):
    for i,year in enumerate(results[party][0]):
        j = results['Other'][0].index(year)
        results['Other'][1][j] = [a+b for a,b in zip(results['Other'][1][j],results[party][1][i])]
    del results[party]
add_to_others('Liberal Labour Party*')
add_to_others('New Democracy (Includes votes for the Social Credit)')
add_to_others('Green Party of Canada')
add_to_others('Ralliement des créditistes')

#plot it
fig = pl.figure(figsize=(15.96,6.125))
for party in ['(Progressive) Conservative Party', 'Conservative Party of Canada', 'Liberal Party of Canada','New Democratic Party', 'Bloc Québécois', 'Reform Party/Canadian Alliance', 'Social Credit Party', 'Co-operative Commonwealth', 'Other']:
    result = results[party]
    pl.plot(result[0],[r[3] for r in result[1]],color=colorcodes[party],linewidth=3,label=party)
pl.xticks([years[i] for i in range(len(years)) if i%3==1 or i==len(years)-1]) #Label every 3 elections
pl.xlim(min(years),max(years))
pl.ylim(0,60)
pl.yticks(pl.yticks()[0],[str(int(p))+'%' for p in pl.yticks()[0]])
ax = pl.gca()
for i in range(len(governments)):
    ax.add_patch(pl.Rectangle((years[i],0),years[i+1]-years[i],60,color=colorcodes[governments[i]],alpha=0.35))
pl.legend(fontsize=12,loc='center left',bbox_to_anchor=(0, 0.4))
pl.savefig("election_plot.svg", format="svg")
pl.show(block=True)