# Add the root path of the pygra library
import os
import sys
sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../../../src")
from pygra import geometry
import numpy as np
from pygra import films
from pygra import algebra
algebra.accelerate = True
g = geometry.diamond_lattice_minimal(
) # get the three dimensional diamond lattice
g = films.geometry_film(g, nz=20) # create a film
h = g.get_hamiltonian() # create hamiltonian
def get_hamiltonian():
"""Hamiltonian for parameter p"""
h = g.get_hamiltonian(is_multicell=True,
is_sparse=False) # create hamiltonian
def step(z, width=0.00001):
"""This will yield 0 for z<0 and 1 for z>0"""
out = (-np.tanh(z / width) + 1.0) / 2. # this is the interpolator
return out
h.add_antiferromagnetism(lambda r: 0.5 * step(r[2]))
h.shift_fermi(lambda r: 0.7 * (-step(r[2], width=0.0001) + 1.0))
h.add_swave(lambda r: 0.4 * (-step(r[2]) + 1.0))
h.add_kane_mele(0.1)
# Add the root path of the pygra library import os ; import sys ; sys.path.append(os.environ['PYGRAROOT']) import numpy as np from pygra import geometry from pygra import films g = geometry.pyrochlore_lattice() g = films.geometry_film(g,nz=10) h = g.get_hamiltonian() h.add_kane_mele(0.1) h.turn_dense() h.get_bands(operator="zposition")
# Add the root path of the pygra library import os import sys sys.path.append(os.path.dirname(os.path.realpath(__file__)) + "/../../../src") from pygra import geometry from pygra import films g = geometry.diamond_lattice_minimal() g = films.geometry_film(g, nz=20) h = g.get_hamiltonian() h.get_bands()
