36 lines
771 B
Python
36 lines
771 B
Python
|
import qpms
|
||
|
|
import numpy as np
|
||
|
|
from numpy import newaxis as nx
|
||
|
|
import math
|
||
|
|
import cmath
|
||
|
|
import os
|
||
|
|
from scipy.constants import c, e as eV, hbar
|
||
|
|
s3 = math.sqrt(3)
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("omega")
|
||
|
|
#parser.add_argument("maxlayer")
|
||
|
|
args = parser.parse_args()
|
||
|
|
omega_eV = float(args.omega)
|
||
|
|
|
||
|
|
print(omega_eV)
|
||
|
|
|
||
|
|
epsilon_b = 2.3104
|
||
|
|
hexside = 375e-9
|
||
|
|
lMax = 3
|
||
|
|
maxlayer = 222
|
||
|
|
my, ny = qpms.get_mn_y(lMax)
|
||
|
|
nelem = len(my)
|
||
|
|
|
||
|
|
omega = omega_eV * eV / hbar
|
||
|
|
|
||
|
|
k_0 = omega * math.sqrt(epsilon_b) / c
|
||
|
|
|
||
|
|
output_prefix = './diracpoints-newdata/%d/' % maxlayer
|
||
|
|
|
||
|
|
os.makedirs(output_prefix, exist_ok=True)
|
||
|
|
qpms.hexlattice_precalc_AB_save(file=output_prefix+str(omega_eV), lMax=lMax, k_hexside=k_0*hexside,
|
||
|
|
maxlayer=maxlayer, savepointinfo=True)
|