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

# Axes level functions (for Matplotlib)

#č Založme nový modul pro funkce (procedury),
#č které sice stejně jako mart pracují na urovni axes,
#č ale na rozdil od atomických funkcí mart modulu
#č udělaj z prazdných os hotový obrazek.
#č mplot.show2D() může použivat tuhle nabídku obrázků.

from . import mart

# it is mostly for qt_plot, it offers availiable options to user
__all__ = ['base_drawing', 'candidates_drawing']


def base_drawing(ax):
    mart.setup(ax)
    mart.curly(ax)
    try:
        mart.triplot(ax, color="grey", linewidth=0.4)
    except:
        pass
    mart.plot_boundaries(ax, linewidth=0.7)
    mart.scatter_points(ax)
    
    
def candidates_drawing(ax):
    try:
        mart.triplot(ax, color="grey", linewidth=0.4)
    except:
        pass
    mart.plot_boundaries(ax, linewidth=0.7)
    
    #č 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')
    
    mart.scatter_points(ax)