pybinding周期性边缘态可视化

可惜图不是很好看

import pybinding as pb
import numpy as np
import matplotlib.pyplot as plt
from math import sqrt, pi
pb.pltutils.use_style()

def SU4():
    t = 1      # [eV] nearest neighbour hopping
 
    lat = pb.Lattice(a1=[2*sqrt(3), 0],
                     a2=[sqrt(3)/2, -1.5])
            
                     
    lat.add_sublattices(('A', [0, 0]),
                        ('B', [sqrt(3)/2, -0.5]),
                        ('C', [sqrt(3), 0]),
                        ('D', [sqrt(3)*3/2, -0.5]))
    
    lat.add_hoppings(
        # inside the main cell
        ([0,  0], 'A', 'B', t),
        ([0,  0], 'B', 'C', t),
        ([0,  0], 'C', 'D', t),
        # between neighboring cells
        ([0, -1], 'A', 'B', t),
        ([0, -1], 'C', 'D', -t),
        ([-1, 0], 'A', 'D', t),
    )
    return lat

lat = SU4()

def vacancy(position, radius):
    @pb.site_state_modifier
    def modifier(state, x, y):
        x0, y0 = position
        state[(x-x0)**2 + (y-y0)**2 < radius**2] = False
        return state
    return modifier

model = pb.Model(
    lat,
    pb.rectangle(x=20, y=10),
    vacancy(position=[-0.83 + 2*sqrt(3),  4.64], radius=0.5),
    vacancy(position=[-0.83,  4.64], radius=0.5),
    vacancy(position=[-0.83-2*sqrt(3),  4.64], radius=0.5),
    pb.translational_symmetry(a1=6*sqrt(3), a2=False),
)
model.plot()
plt.show()
solver = pb.solver.lapack(model)
solver.set_wave_vector(k=pi)
ldos = solver.calc_spatial_ldos(energy=sqrt(3), broadening=0.1)  
plt.ylim(0, 5)
ldos.plot(site_radius=(0.01, 0.2))
pb.pltutils.colorbar()
plt.show()