from copy import deepcopy
from lxml import etree
import matplotlib.pyplot as plt
import numpy as np

# Wind power installed capacity and generation in Denmark
# multilingual
# Production data from  https://fr.wikipedia.org/wiki/Énergie_éolienne_au_Danemark
# installed  data from  https://de.wikipedia.org/wiki/Windenergie_in_Dänemark

data=[
# show_X, year, installed (MW), Production (GWh)
['x',2005 ,3127 , 6614 ] ,
[' ',2006 ,3136 , 6108 ] ,
[' ',2007 ,3124 , 7171 ] ,
[' ',2008 ,3158 , 6928 ] ,
[' ',2009 ,3468 , 6721 ] ,
['x',2010 ,3801 , 7809 ] ,
[' ',2011 ,3956 , 9774 ] ,
[' ',2012 ,4161 ,10270 ] ,
[' ',2013 ,4807 ,11123 ] ,
[' ',2014 ,4883 ,13079 ] ,
['x',2015 ,5064 ,14133 ] ,
[' ',2016 ,5227 ,12782 ] ,
[' ',2017 ,5476 ,14780 ] ,
[' ',2018 ,5751 ,13902 ] ,
[' ',2019 ,6128 ,16150 ] ,
['x',2020 ,6180 ,16330 ] ,
[' ',2021 ,7014 ,16054 ] ,
[' ',2022 ,7282 ,18873 ] ,
['x',2023 ,7562 ,19540 ] 
]
# please update in future.

show_yesno = np.array([row[0] for row in data])
all_years  = np.array([row[1] for row in data])
bar_labels = []

# show axis-labels only the years marked with 'x'
for i, yesNo in enumerate(show_yesno):
    if (yesNo == 'x'):
        bar_labels.append(all_years[i])
    else:
        bar_labels.append("")
        
data_installed      = np.array([row[2] for row in data])  # installed
data_generated      = np.array([row[3] for row in data])  # generated
# percent =  np.array([row[4] for row in data])   future use

fig = plt.figure()
ax1 = plt.gca()
ax2 = ax1.twinx()

plt.xticks(range(len(bar_labels)), bar_labels  )

plt.rcParams.update({'text.usetex':False, "svg.fonttype":"none", "font.size":13})

filename = "Wind_power_installed_capacity_and_generation_in_Denmark.svg"
delta= "  "+ str(min(all_years)) + "-" + str(max(all_years))

labels = {'title': {"en":"Wind power in Denmark " + delta,
                    "de":"Windenergie in Dänemark" + delta,
                    "fr":"Énergie éolienne au Danemark" + delta,
                    "it":"Energia eolica in Danimarca" + delta,
                    "pl":"Energetyka wiatrowa w Danii" + delta,
                    "pt":"Energia eólica na Dinamarca" + delta,
                    "tr":"Danimarka'da rüzgâr enerjisi" + delta,
                    "es":"Energía eólica en Dinamarca" + delta},
          "installed": {"en":"installed capacity (MW)",
                        "de":"installierte Leistung (MW)",
                        "es":"capacidad instalada (MW)",
                        "it":"Capacità installata (MW)",
                        "pl":"Moc zainstalowana (MW)",
                        "pt":"capacidade instalada (MW)",
                        "tr":"Kurulu kapasite (MW)" ,
                        "fr":"Puissance installée (MW)"},
          "generation": {"en":"generation (GWh)             ",
                         "de":"erzeugt (GWh)",
                         "es":"producción (GWh)",
                         "it":"Produzione (GWh)",
                         "pl":"Produkcja energii (GWh)"  ,
                         "pt":"produção energética (GWh)",
                         "tr":"Elektrik üretimi (GWh)" ,
                         "fr":"Production (GWh)"}
            }

width =0.3
ax1.bar(np.arange(len(data_installed))- width/2,  data_installed, width=width)
ax2.bar([0], [0],  width=width, label=labels["installed"]["en"]  )
ax2.bar(np.arange(len(data_generated))+ width/2, data_generated, width=width, color='red', label=labels["generation"]["en"])

plt.title(labels["title"]["en"])
ax1.set_ylabel(labels["installed"]["en"])
ax2.set_ylabel(labels["generation"]["en"])

ax1.set_xlabel("Source: https://de.wikipedia.org/wiki/Windenergie_in_Dänemark \n+ https://fr.wikipedia.org/wiki/Énergie_éolienne_au_Danemark ", fontsize='xx-small')

plt.legend(loc='upper left', borderaxespad=1.5)
plt.tight_layout()

plt.savefig(filename)
#plt.show()

# create multilingual labels with the svg switch element
with open(filename, mode="r",encoding='utf-8') as svgfile:
    svg = etree.parse(svgfile, etree.XMLParser()).getroot()
for label, values in labels.items():
    for el in svg.findall(".//{" + svg.nsmap[None] + "}text"):
        if el.text == values["en"]:
            switch = etree.SubElement(el.getparent(), "switch")
            for lang, text in values.items():
                el2 = deepcopy(el)
                el2.set("systemLanguage", lang)
                el2.text = text
                switch.append(el2)
            switch.append(el)
with open(filename, mode='w', encoding='utf-8') as svgfile:
    svgfile.write(etree.tostring(svg, pretty_print=True, encoding="unicode"))