Skip to content
Snippets Groups Projects
diamond_chain-checkpoint.py 4.46 KiB
Newer Older
import kwant
import numpy as np


def diamond_chain_system(N_c, system_params, semi_infinite = False, leads = False):
    '''
    Create a diamond chain of trimer unit cells. Each atom of the trimer has two orbital angular momentum states, + and -
          _______
    ...  | C_i_+ |
         | C_i_- |         ...
        / ------- \ ______ /
    ...           | A_i_+ | ...
    ...  | B_i_+ |         ...
         | B_i_- |
          ------- 
    
    The Φ represents an out-of-plane magnetic field. The phase is added along the // bond in each unit cell 
    
    :param int N_c: number of unit cells to include in the cell
    :param dict system_params: parameters
    :param bool semi_infinite: whether to make a semi-infinite chain or not
    :param bool leads: whether to include leads 
    
    :rtype kwant.system.FiniteSystem:
    '''
    
    # make lattices and sublattices
    lat = kwant.lattice.Polyatomic(prim_vecs = [[1,0],[0,1]], basis = [[1,0],[1,0], [0,-1],[0,-1], [0,1],[0,1]], norbs = 1)
    a_lat_pos, a_lat_neg, b_lat_pos, b_lat_neg, c_lat_pos, c_lat_neg = lat.sublattices
    
    #make builder and populate with onsite and hoppings
    if semi_infinite == True:
        syst = kwant.Builder(symmetry=kwant.lattice.TranslationalSymmetry([1,0]))
    else:
        syst = kwant.Builder()
    
    
    for i in range(N_c):
        
        #staggered point
        syst[a_lat_pos(sp,0)] = system_params['mu_a_pos']
        syst[a_lat_neg(sp,0)] = system_params['mu_a_neg']
        syst[b_lat_pos(sp,0)] = system_params['mu_b_pos']
        syst[b_lat_neg(sp,0)] = system_params['mu_b_neg']
        syst[c_lat_pos(sp,0)] = system_params['mu_c_pos']
        syst[c_lat_neg(sp,0)] = system_params['mu_c_neg']
            syst[c_lat_pos(sp+1,0)] = system_params['mu_c_pos']
            syst[b_lat_pos(sp+1,0)] = system_params['mu_b_pos']
            syst[c_lat_neg(sp+1,0)] = system_params['mu_c_neg']
            syst[b_lat_neg(sp+1,0)] = system_params['mu_b_neg']
            syst[a_lat_pos(sp,0), c_lat_pos(sp+1,0)] = system_params['j2']
            syst[a_lat_pos(sp,0), b_lat_pos(sp+1,0)] = system_params['j2']
            syst[a_lat_neg(sp,0), c_lat_neg(sp+1,0)] = system_params['j2']
            syst[a_lat_neg(sp,0), b_lat_neg(sp+1,0)] = system_params['j2']
            syst[a_lat_pos(sp,0), b_lat_neg(sp+1,0)] = system_params['j3']
            syst[a_lat_neg(sp,0), b_lat_pos(sp+1,0)] = system_params['j3']

            # + <--> - hopping with phase
            syst[a_lat_pos(sp,0), c_lat_neg(sp+1,0)] = system_params['j3']*np.exp(1j*2*system_params['phi']) #phase e^i phi
            syst[a_lat_neg(sp,0), c_lat_pos(sp+1,0)] = system_params['j3']*np.exp(1j*2*system_params['phi']) #phase e^i phi
        syst[a_lat_pos(sp,0), c_lat_pos(sp,0)] = system_params['j2']
        if i == 0:
            added_phase = 1
        else:
            added_phase = np.exp(1j*2*system_params['phi_d'])
        syst[a_lat_pos(sp,0), b_lat_pos(sp,0)] = system_params['j2']*added_phase #phase e^i phi_d due to mag field
        syst[a_lat_neg(sp,0), c_lat_neg(sp,0)] = system_params['j2']
        syst[a_lat_neg(sp,0), b_lat_neg(sp,0)] = system_params['j2']*added_phase #phase e^i phi_d due to mag field
        syst[a_lat_pos(sp,0), c_lat_neg(sp,0)] = system_params['j3']
        syst[a_lat_neg(sp,0), c_lat_pos(sp,0)] = system_params['j3']
        # + <--> - hopping with phase
        syst[a_lat_pos(sp,0), b_lat_neg(sp,0)] = system_params['j3']*np.exp(1j*2*system_params['phi'])*added_phase #phase e^i phi_d due to mag field #phase e^i phi
        syst[a_lat_neg(sp,0), b_lat_pos(sp,0)] = system_params['j3']*np.exp(1j*2*system_params['phi'])*added_phase #phase e^i phi_d due to mag field #phase e^i phi
    if leads:
        
        lead_syst = kwant.Builder(symmetry=kwant.lattice.TranslationalSymmetry([1,0]))
        
        
        lead_syst[a_lat_pos(0,0)] = 0
        lead_syst[a_lat_neg(0,0)] = 0
        lead_syst[b_lat_pos(0,0)] = 0
        lead_syst[b_lat_neg(0,0)] = 0
        lead_syst[c_lat_pos(0,0)] = 0
        lead_syst[c_lat_neg(0,0)] = 0
        
        lead_syst[lat.neighbors(n=1)] = 1
        
        syst.attach_lead(lead_syst)
        syst.attach_lead(lead_syst.reversed())