import matplotlib.pyplot as plt

from cellcomplex.property_topomesh.example_topomesh import square_grid_topomesh
from cellcomplex.property_topomesh.utils.matplotlib_tools import mpl_draw_topomesh, mpl_draw_incidence_graph

topomesh = square_grid_topomesh(1)

figure = plt.figure(0)
figure.clf()

figure.add_subplot(1,2,1)
figure.gca().axis('equal')
mpl_draw_topomesh(topomesh,figure,2,color='g',plot_ids=True)
mpl_draw_topomesh(topomesh,figure,1,color='b',plot_ids=True)
mpl_draw_topomesh(topomesh,figure,0,color='m',plot_ids=True)
figure.gca().axis('off')

figure.add_subplot(1,2,2)
figure.gca().axis('equal')
mpl_draw_incidence_graph(topomesh,figure,plot_ids=True)
figure.gca().axis('off')

figure.set_size_inches(10,5)
figure.tight_layout()