misc/iht-saving.py

   1 import qpms
   2 import numpy as np
   3 from numpy import newaxis as nx
   4 import math
   5 import cmath
   6 import os
   7 from scipy.constants import c, e as eV, hbar
   8 s3 = math.sqrt(3)
   9
  10 import argparse
  11
  12 parser = argparse.ArgumentParser()
  13 parser.add_argument("omega")
  14 #parser.add_argument("maxlayer")
  15 args = parser.parse_args()
  16 omega_eV = float(args.omega)
  17
  18 print(omega_eV)
  19
  20 epsilon_b = 2.3104
  21 hexside = 375e-9
  22 lMax = 3
  23 maxlayer = 222
  24 my, ny = qpms.get_mn_y(lMax)
  25 nelem = len(my)
  26
  27 omega = omega_eV * eV / hbar
  28
  29 k_0 = omega * math.sqrt(epsilon_b) / c
  30
  31 output_prefix = '/tmp/diracpoints-newdata2/%d/' % maxlayer
  32
  33 os.makedirs(output_prefix, exist_ok=True)
  34 qpms.hexlattice_precalc_AB_save(file=output_prefix+str(omega_eV), lMax=lMax, k_hexside=k_0*hexside,
  35         maxlayer=maxlayer, savepointinfo=True)