# 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 heterostructures
import numpy as np
import matplotlib.pyplot as plt
g = geometry.chain()
h = g.get_hamiltonian()
h1 = h.copy() # copy
h2 = h.copy() # copy
h1.add_swave(.0) # add electron hole symmetry

# create a topological superconductor
h2.add_rashba(1.0)
h2.add_zeeman(0.5)
h2.add_onsite(2.0)
h2.get_bands()
h2.add_swave(.1) # pairing gap of 0.1

ht = heterostructures.build(h1,h2) # create the junction
ht.scale_lc = 0.1 # scale the left coupling
ht.delta = 1e-6 # imaginary part of the energy
es = np.linspace(-.5,.5,1000) # grid of energies
ts = [ht.didv(e) for e in es] # calculate transmission
plt.plot(es,ts,marker="o")
plt.ylim([0,4.1])
plt.show()






