from latticeTB import *
from plotTB import *
from math import pi
nx, ny = 5, 5
ri = [[0, 0], [0.5*sqrt(3), 0.5], [0.25*sqrt(3), 0.25], [0.75*sqrt(3), 0.25], [0.5*sqrt(3), 1.]]
tags = [b'a', b'b', b'c', b'd', b'e']
hexa_lc = latticeTB(tags=tags, ri=ri, nor=sqrt(3), ang=pi/3)
hexa_lc.get_lattice(nx=nx, ny=ny)
fig_hexa_lc = hexa_lc.plt_lattice(colors=['b', 'b', 'r', 'r', 'r'], ms=10)
hexa_lc.remove_dangling(nor_bond=0.5)
fig_hexa_lc_dang = hexa_lc.plt_lattice(colors=['b', 'b', 'r', 'r', 'r'], ms=10)
plt.show()
save_hlc = saveFigTB(sys=hexa_lc, dir_name='hexa_lc')
save_hlc.save_fig_lat(fig_hexa_lc, 'lat')
save_hlc.save_fig_lat(fig_hexa_lc_dang, 'lat_dang')