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


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


from . import maxes
from . import maxes3d

__all__ = ['qhull_under_density', 'plane_under_density', 'dhull_vs_complete']

def qhull_under_density(fig, sample_box, space):
    from . import _axis3d_margins_patch, _axes3d
    ax3d = fig.add_subplot(111, projection='3d')
    ax3d.sample_box = sample_box
    ax3d.space = space
    maxes3d.qhull_under_density(ax3d)

def plane_under_density(fig, sample_box, space):
    from . import _axis3d_margins_patch, _axes3d
    ax3d = fig.add_subplot(111, projection='3d')
    ax3d.sample_box = sample_box
    ax3d.space = space
    maxes3d.plane_under_density(ax3d)


def dhull_vs_complete(fig, sample_box, space):
    from .. import sball
    rand_dir = sball.get_random_directions(8, 2)
    
    ax = fig.add_subplot(121)
    ax.sample_box = sample_box
    ax.space = space
    maxes.dhull_random_plot(ax, rand_dir=rand_dir)
    ax = fig.add_subplot(122)
    ax.sample_box = sample_box
    ax.space = space
    maxes.completehull_plot(ax, rand_dir=rand_dir)
    

    
    

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', candidates=False, 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
    
    if candidates:
        #č ax.scatter posílá parameter cmap Collections třídě.
        #č Třída mimo jiného dědí cm.ScalarMappable,
        #č která inicializaci deleguje funkci cm.get_cmap() ve (svém) modulu cm.
        # cmap='viridis_r' #cmap='plasma',
        mart.scatter_candidates(ax, s=5, marker='.', cmap='plasma_r',\
         alpha=None, linewidths=None, edgecolors=None, plotnonfinite=False)
        mart.plot_the_best_candidate(ax, "^", color='#3D0D5B')
    
    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()