#!/usr/bin/env python
# coding: utf-8


# nazvy proměnných jsou v angličtině
# Ale komenty teda ne)


import matplotlib
import matplotlib.pyplot as plt
matplotlib.rcParams['mathtext.fontset'] = 'cm' # Computer Modern (TeX)
matplotlib.rcParams['font.family'] = 'STIXGeneral'

from . import mart


def subplot3D(sample_box, ax3d, space='R'):
    Fails = getattr(sample_box.failure_samples, space)
    Succeses = getattr(sample_box.success_samples, space)
    xyz = [0,1,2]
    ax3d.scatter(Fails[:,xyz[0]], Fails[:,xyz[1]], Fails[:,xyz[2]], c='r', marker='.')
    ax3d.scatter(Succeses[:,xyz[0]], Succeses[:,xyz[1]], Succeses[:,xyz[2]], c='g', marker='.')
    
    ax3d.set_xlabel('X')
    ax3d.set_ylabel('Y')
    ax3d.set_zlabel('Z')
    


def plot3D(sample_box, space='R', filename='', format='pdf'):
    if not filename:
        filename = 'store/%s_%s_%s_plot3D'%(sample_box.gm_signature, space, sample_box.nsim)
    fig = plt.figure()
    ax3d = fig.add_subplot(111, projection='3d')
    subplot3D(sample_box, ax3d, space)
    
    try: # jen pro formu zkusíme, vždyť musí funkce formálně něco dělat, žejo?
        fig.savefig(filename + "." + format)#, dpi=300)
    except:
        pass # nic se neděje
        
        # vracím plt místo figury, protože nechcem sa trapiť ivent lupem
        # return plt insted of just fig to bypass event loop issues
    return plt.show()



# nikdo mi neuvěří, že by tohle postačílo a nebylo by nutné tohlensto furt úpravovat
def plot2D(sample_box, space='R', filename=None, format='pdf'):
    if filename is None:
        try:
            filename = sample_box.filename
        except:
            filename = 'store/%s_%s_%s_plot2D'%(sample_box.gm_signature, space, sample_box.nsim)
            
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.space = space
    ax.sample_box = sample_box
    
    mart.setup(ax)
    mart.curly(ax)
    mart.plot_boundaries(ax, linewidth=0.7)
    mart.scatter_points(ax)
    try:
        mart.triplot(ax, color="grey", linewidth=0.4)
    except:
        pass
    
    try: # jen pro formu zkusíme, vždyť musí funkce formálně něco dělat, žejo?
        fig.savefig(filename + "." + format, dpi=300)
    except:
        pass # nic se neděje
        
        # vracím plt místo figury, protože nechcem sa trapiť ivent lupem
        # return plt insted of just fig to bypass event loop issues
    return plt.show()