def write(self, filename="", mode="w"):
assert mode in ["w", "a"], "Writing Mode can only be 'w' or 'a'!"
if filename == "":
filename = "PhaseDiag_Data_" + nt() + ".txt"
elif ((len(filename) <= 4) or (filename[-4:] != ".txt")):
filename = filename + ".txt"
d = deepcopy(self.info)
d["data"] = self.data
with open(filename, mode) as f:
f.write(dumps(d))
import numpy as np
from itertools import product
from functools import partial
from scipy import linalg
from PhaseDiagram import PhaseDiag
from _utils import nt, convertSpin, settings, settings_strict, CONST_HJZ, CONST_HZL
from TI_Film import Hamiltonian, plotLine
from TopoInvCalc import Calc, TopoOrder
from runfunc import Run_SpinZ
if __name__ == '__main__':
now = nt()
File = "PD_D1.594_HZL" + now
func = partial(Run_SpinZ, Delta=1.594, CONST=CONST_HZL, settings=settings)
PD = PhaseDiag().run(func, 6, 24, 19, 0, 0.02, 1, "N", "J")
PD.write(filename=File)
PD.draw(
title=
"PhaseDiag of N & J, Delta=$1.594\AA$ (6 site per QL), Spin-z\nParameters from Hai-Zhou Lu's paper.",
xlabel="# of Layers",
ylabel="$J(\\rm eV)$",
filename=File)