def calc_edge_pol(self):
# Calculate bandstructure and highlight edge states
Lat = lattice('Ribbon', self.SelectEdge.currentText()+' '+self.SelectStack.currentText(), self.NUnit.value()) # Create lattice
H = Hamiltonian(Lat, self.Hoppingt.value(), self.Hoppingtprime.value())# Create Hamiltonian
if self.Magneticfield.value() != 0: # Add magnetic field
H.add_magnetic_field(self.Magneticfield.value(), self.SelectBBil.currentText())
if self.OnsiteV.value() != 0:# Add lattice imbalance
H.add_lattice_imbalance(self.OnsiteV.value())
P = Plot(H)
P.calc_layer_pol()
