
import sys
sys.path.append("../../../pygra")  # add pygra library

import geometry
import heterostructures
import numpy as np

g = geometry.chain()
h = g.get_hamiltonian()

ht = heterostructures.create_leads_and_central(h,h,h)

h.get_bands() # get bandstructure
es = np.linspace(-1.,1.,50)
ts = [ht.landauer(e) for e in es]
np.savetxt("TRANSPORT.OUT",np.matrix([es,ts]).T)


