TB2J 0.9.3rc0__py3-none-any.whl → 0.9.5rc0__py3-none-any.whl

TB2J/MAE.py

@@ -11,8 +11,9 @@ import matplotlib.pyplot as plt
 from pathlib import Path
 from TB2J.mathutils.rotate_spin import spherical_to_cartesian
 from HamiltonIO.model.occupations import Occupations
-from TB2J.abacus.abacus_wrapper import AbacusSplitSOCParser
-from HamiltonIO.siesta import SislWrapper
+#from TB2J.abacus.abacus_wrapper import AbacusSplitSOCParser
+from HamiltonIO.abacus.abacus_wrapper import AbacusSplitSOCParser
+from HamiltonIO.siesta import SislParser, SiestaHamiltonian
 import tqdm
 
 
@@ -75,7 +76,10 @@ class MAE:
         # e,rho = self.model.get_band_energy(dm=True)
         # self.calc_ref()
         # thetas = np.linspace(*angle_range, npoints)
-        for i, (theta, phi) in tqdm.tqdm(enumerate(zip(thetas, phis))):
+        nangles = len(thetas)
+        for i in tqdm.trange(nangles):
+            theta = thetas[i]
+            phi = phis[i]
             self.model.set_Hsoc_rotation_angle([theta, phi])
             e = self.get_band_energy()
             es.append(e)
@@ -96,28 +100,27 @@ def abacus_get_MAE(
         outpath_nosoc=path_nosoc, outpath_soc=path_soc, binary=False
     )
     model = parser.parse()
-    model.nel = 16
     ham = MAE(model, kmesh, gamma=gamma)
     es = ham.get_band_energy_vs_angles(thetas, psis)
     if outfile:
         with open(outfile, "w") as f:
             f.write("#theta, psi, energy\n")
             for theta, psi, e in zip(thetas, psis, es):
-                f.write(f"{theta}, {psi}, {e}\n")
+                f.write(f"{theta:5.3f}, {psi:5.3f}, {e:10.9f}\n")
     return es
 
 
 def siesta_get_MAE(fdf_fname, kmesh, thetas, phis, gamma=True, outfile="MAE.txt"):
     """ """
-    parser = SiestaParser(fdf_fname=fdf_fname, read_H_soc=True)
-    model = parser.parse()
+    model= SislParser(fdf_fname=fdf_fname, read_H_soc=True).get_model()
     ham = MAE(model, kmesh, gamma=gamma)
     es = ham.get_band_energy_vs_angles(thetas, phis)
     if outfile:
         with open(outfile, "w") as f:
             f.write("#theta, psi, energy\n")
             for theta, psi, e in zip(thetas, phis, es):
-                f.write(f"{theta}, {psi}, {e}\n")
+                #f.write(f"{theta}, {psi}, {e}\n")
+                f.write(f"{theta:5.3f}, {psi:5.3f}, {e:10.9f}\n")
     return es
 
 

TB2J/abacus/abacus_wrapper.py

@@ -36,6 +36,7 @@ class AbacusWrapper(AbstractTB):
             self.set_HR_soc(HR_soc=HR_soc, HR_nosoc=HR_nosoc, HR_full=HR)
         self.soc_rotation_angle = 0.0
 
+
     def set_HR_soc(self, HR_soc=None, HR_nosoc=None, HR_full=None):
         self.split_soc = True
         self.HR_soc = HR_soc
@@ -156,6 +157,9 @@ class AbacusParser:
         # read the information
         self.read_atoms()
         self.efermi = self.read_efermi()
+        self.nel = self.read_nel()
+        print(f"efermi: {self.efermi}")
+        print(f"nel: {self.nel}")
         self.read_basis()
 
     def read_spin(self):
@@ -254,10 +258,10 @@ class AbacusParser:
         nel = None
         with open(fname, "r") as myfile:
             for line in myfile:
-                if "NELECT" in line:
-                    nel = float(line.split()[2])
+                if "number of electrons" in line:
+                    nel = float(line.split()[-1])
         if nel is None:
-            raise ValueError(f"NELECT not found in the {str(fname)}  file.")
+            raise ValueError(f"number of electron not found in the {str(fname)}  file.")
         return nel
 
     def get_basis(self):
@@ -306,6 +310,7 @@ class AbacusSplitSOCParser:
             nspin=2,
             HR_soc=HR_soc,
             HR_nosoc=HR_nosoc,
+            nel=self.parser_nosoc.nel,
         )
         model.efermi = self.parser_soc.efermi
         model.basis = self.parser_nosoc.basis

TB2J/abacus/gen_exchange_abacus.py

@@ -6,7 +6,8 @@ The main function to compute exchange interaction from abacus data
 
 import os
 from pathlib import Path
-from TB2J.abacus.abacus_wrapper import AbacusParser
+#from TB2J.abacus.abacus_wrapper import AbacusParser
+from HamiltonIO.abacus import AbacusParser
 from TB2J.exchange import ExchangeNCL, ExchangeCL
 from TB2J.exchangeCL2 import ExchangeCL2
 

TB2J/manager.py

@@ -7,11 +7,15 @@ from TB2J.exchange_qspace import ExchangeCLQspace
 from TB2J.utils import read_basis, auto_assign_basis_name
 from ase.io import read
 
-#from TB2J.sisl_wrapper import SislWrapper
+# from TB2J.sisl_wrapper import SislWrapper
+
+from HamiltonIO.siesta import SislParser
 try:
-    from HamiltonIO.siesta import SislWrapper
+    from HamiltonIO.siesta import SislParser
 except ImportError:
-    print("Cannot import SislWrapper from HamiltonIO.siesta. Please install HamiltonIO first.")
+    print(
+        "Cannot import SislWrapper from HamiltonIO.siesta. Please install HamiltonIO first."
+    )
     from TB2J.sisl_wrapper import SislWrapper
 from TB2J.gpaw_wrapper import GPAWWrapper
 from TB2J.wannier import parse_atoms
@@ -283,14 +287,16 @@ def gen_exchange_siesta(
             include_orbs[element] = None
     magnetic_elements = list(include_orbs.keys())
 
-    fdf = sisl.get_sile(fdf_fname)
+    #fdf = sisl.get_sile(fdf_fname)
     # geom = fdf.read_geometry()
-    H = fdf.read_hamiltonian()
-    geom = H.geometry
-    if H.spin.is_colinear:
+    #H = fdf.read_hamiltonian()
+    #geom = H.geometry
+    parser=SislParser(fdf_fname=fdf_fname, ispin=None, read_H_soc=False)
+    if parser.spin.is_colinear:
         print("Reading Siesta hamiltonian: colinear spin.")
-        tbmodel_up = SislWrapper(fdf_fname=None,sisl_hamiltonian=H, spin=0, geom=geom)
-        tbmodel_dn = SislWrapper(fdf_fname=None, sisl_hamiltonian=H, spin=1, geom=geom)
+        #tbmodel_up = SislWrapper(fdf_fname=None, sisl_hamiltonian=H, spin=0, geom=geom)
+        #tbmodel_dn = SislWrapper(fdf_fname=None, sisl_hamiltonian=H, spin=1, geom=geom)
+        tbmodel_up, tbmodel_dn = parser.get_model()
         basis = dict(zip(tbmodel_up.orbs, list(range(tbmodel_up.norb))))
         print("Starting to calculate exchange.")
         description = f""" Input from collinear Siesta data.
@@ -320,7 +326,7 @@ def gen_exchange_siesta(
         print("\n")
         print(f"All calculation finsihed. The results are in {output_path} directory.")
 
-    elif H.spin.is_colinear and False:
+    elif parser.spin.is_colinear and False:
         print(
             "Reading Siesta hamiltonian: colinear spin. Treat as non-colinear. For testing only."
         )
@@ -355,10 +361,10 @@ def gen_exchange_siesta(
         print("\n")
         print(f"All calculation finsihed. The results are in {output_path} directory.")
 
-    elif H.spin.is_spinorbit or H.spin.is_noncolinear:
+    elif parser.spin.is_spinorbit or H.spin.is_noncolinear:
         print("Reading Siesta hamiltonian: non-colinear spin.")
-        tbmodel = SislWrapper(fdf_fname=None, sisl_hamiltonian=H, spin=None, geom=geom)
-        basis = dict(zip(tbmodel.orbs, list(range(tbmodel.nbasis))))
+        model = parser.get_model()
+        basis = dict(zip(model.orbs, list(range(model.nbasis))))
         print("Starting to calculate exchange.")
         description = f""" Input from non-collinear Siesta data.
  working directory: {os.getcwd()}
@@ -368,8 +374,8 @@ Warning: The DMI component parallel to the spin orientation, the Jani which has
  If you need these component, try to do three calculations with spin along x, y, z,  or use structure with z rotated to x, y and z. And then use TB2J_merge.py to get the full set of parameters.
 \n"""
         exchange = ExchangeNCL(
-            tbmodels=tbmodel,
-            atoms=tbmodel.atoms,
+            tbmodels=model,
+            atoms=model.atoms,
             basis=basis,
             efermi=0.0,
             magnetic_elements=magnetic_elements,

TB2J/pauli.py

@@ -143,11 +143,39 @@ def pauli_block_all(M):
     return MI, Mx, My, Mz
 
 
-def gather_pauli_blocks(MI, Mx, My, Mz):
+def gather_pauli_blocks(MI, Mx, My, Mz, coeffs=[1.0, 1.0, 1.0, 1.0]):
     """
     Gather the I, x, y, z component of a matrix.
     """
-    return np.kron(MI, s0) + np.kron(Mx, s1) + np.kron(My, s2) + np.kron(Mz, s3)
+    cI, cx, cy, cz = coeffs
+    return cI*np.kron(MI, s0) + cx*np.kron(Mx, s1) + cy*np.kron(My, s2) + cz*np.kron(Mz, s3)
+
+def pauli_part(M, coeffs=[1.0, 1.0, 1.0, 1.0]):
+    """
+    Get the I, x, y, z part of a matrix.
+    """
+    MI, Mx, My, Mz = pauli_block_all(M)
+    return gather_pauli_blocks(MI, Mx, My, Mz, coeffs=coeffs)
+
+def chargepart(M):
+    """
+    Get the charge part of a matrix.
+    """
+    MI = (M[::2, ::2] + M[1::2, 1::2]) / 2.0
+    Mcopy = np.zeros_like(M)
+    Mcopy[::2, ::2]  = MI
+    Mcopy[1::2, 1::2] = MI
+    return Mcopy
+
+def spinpart(M):
+    """
+    Get the spin part of a matrix.
+    """
+    MI = (M[::2, ::2] + M[1::2, 1::2]) / 2.0
+    Mcopy = M.copy()
+    Mcopy[::2, ::2]  -= MI
+    Mcopy[1::2, 1::2] -= MI
+    return Mcopy
 
 
 def test_gather_pauli_blocks():

{TB2J-0.9.3rc0.dist-info → TB2J-0.9.5rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: TB2J
-Version: 0.9.3rc0
+Version: 0.9.5rc0
 Summary: TB2J: First principle to Heisenberg exchange J using tight-binding Green function method
 Author: Xu He
 Author-email: mailhexu@gmail.com
@@ -21,7 +21,6 @@ Requires-Dist: ase >=3.19
 Requires-Dist: tqdm
 Requires-Dist: pathos
 Requires-Dist: packaging >=20.0
-Requires-Dist: pre-commit
-Requires-Dist: HamiltonIO >=0.1.4
+Requires-Dist: HamiltonIO >=0.1.5
 
 TB2J is a Python package aimed to compute automatically the magnetic interactions (superexchange  and Dzyaloshinskii-Moriya) between atoms of magnetic crystals from DFT Hamiltonian based on Wannier functions or Linear combination of atomic orbitals. It uses the Green's function method and take the local rigid spin rotation as a perturbation. The package can take the output from Wannier90, which is interfaced with many density functional theory codes or from codes based on localised orbitals. A minimal user input is needed, which allows for an easily integration into a high-throughput workflows. 

{TB2J-0.9.3rc0.dist-info → TB2J-0.9.5rc0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 TB2J/Jdownfolder.py,sha256=Rmg6KfQ-Lkhei5daTJ2POzr0XL-R1WM-rzUnDcfoDhc,9595
 TB2J/Jtensor.py,sha256=t6OsqrSlYW6Im4H7ykVAW8Al_pFXN4C5yj2UEsV6r7g,3181
-TB2J/MAE.py,sha256=0yU11DNKwcW5fL2EWgangIqHdWW4Vp97IiXGdi70Ayo,7396
+TB2J/MAE.py,sha256=-KWqxjKJvPFuxsMjm0c3nESyUFDeFTqsV7QzPJN-Fxo,7579
 TB2J/Oiju.py,sha256=cNGv8N5uH_swGq7cnAt2OyiDfqtjLlLrwseGu0E4iaM,3383
 TB2J/Oiju_epc.py,sha256=oytM3NYW7nWmklrGgNlqwIpI_JYv_hb7ZnR4o9nYNog,6809
 TB2J/__init__.py,sha256=hcEWkag_UvLm1ZSbjsgcTWkGVlR3Bwmzg1QYAwsvf-g,24
@@ -19,12 +19,12 @@ TB2J/green.py,sha256=giWSPhrLKc3ZKOpwyY3snUKFWH08GMqBlsJWbFG9Qo8,13565
 TB2J/greentest.py,sha256=2ISSfhor9ecSEOi_E6b4Cv26wEIQlwlzca0ru8z44_E,1603
 TB2J/io_merge.py,sha256=t85k3L6IL9X5ys-PWK7CzResb3xJsyqM3LAlKPUe9vM,6825
 TB2J/kpoints.py,sha256=6XK2KqTncidEq3o9GuO6VEZRPNTRtWeXg9QfcV-9smI,532
-TB2J/manager.py,sha256=Ogb_yRCnroNmGttKPRcbF70ddF9UNV_1UytdKbNFN4E,15685
+TB2J/manager.py,sha256=w_zKrmsTtlNNiR0tO3kNoQqRf9aCKdbIa8s2cgvESwI,15826
 TB2J/mathutils.py,sha256=tHA6q3KPDpXLIbZHdDZ2NU5s886VVM_oEG490zQ6Ris,300
 TB2J/myTB.py,sha256=ok_B4my29bOIghMSZfx0Es6G8FaXaIiLP4gPxTdSj00,17659
 TB2J/orbmap.py,sha256=RCMJkOPGbfPrcZzcc5ia1ZMKBQWxGcyj8W1ve8BJaEw,6669
 TB2J/patch.py,sha256=Z3KZklID9U8zKuk6Ek1Tq95JcY3eT4770dsdcXz6mAA,1067
-TB2J/pauli.py,sha256=zOELm7Vgxw6SMaO5l7qVWx1pBKZt25RLnEpnVM3dz_0,4545
+TB2J/pauli.py,sha256=2UKVHyWSzNDL43KzbGYa26h550lfrlreksCEh_2tWe4,5246
 TB2J/pert.py,sha256=RaCJfewl0doht4cjAnzzGKe-uj2le4aqe0iPKFrq9fo,1192
 TB2J/plot.py,sha256=AnFIFWE2vlmj7Z6f_7-dX_O1stJN-qbuiurPj43dUCM,4104
 TB2J/rotate_atoms.py,sha256=Dwptn-wdDW4zYzjYb95yxTzuZOe9WPuLjh3d3-YcSs0,3277
@@ -38,8 +38,8 @@ TB2J/versioninfo.py,sha256=wZwS9QDFRVDe7rf8JyPDDI8UGdTQiO6Pb_sWv8GAegA,337
 TB2J/abacus/MAE.py,sha256=q9aSVDRZFAnZL3gHdNmde7sxj80oe-BRjwDO-ipyfew,12237
 TB2J/abacus/__init__.py,sha256=5sHiDnF2L-Y80QeET9zOiS83a5T_TQAXvnIhcYB6wNU,56
 TB2J/abacus/abacus_api.py,sha256=D_NyXW-Pno92d3RVHByx0l1HDPHQAvXsmQVt8cfIGR8,7267
-TB2J/abacus/abacus_wrapper.py,sha256=k0ZyTQ0jungp7jtlVCcdFQVK9YKeWokZvASCvfKCuKs,11880
-TB2J/abacus/gen_exchange_abacus.py,sha256=lKZqkWMnLQtaSwgn8O5Fzr-pV6tzwoMqfZU_vbET6gU,2973
+TB2J/abacus/abacus_wrapper.py,sha256=MCQt1qjweJ1-NnzxH7QBCVmQ7p9K9cgE3a5mldui9TA,12055
+TB2J/abacus/gen_exchange_abacus.py,sha256=xdas1BtjLlxWdxtJPqoUousfo-Gk1iOg9jIfkgUrYjU,3017
 TB2J/abacus/occupations.py,sha256=vaMVeZwldgzGDxjA7i3-2-V6akXjpgJwJFWKozJ-l2k,8947
 TB2J/abacus/orbital_api.py,sha256=l48Hn5z7TA0TH7Is4NDmz74b6B9L2ztYO4dRR37U4mQ,1562
 TB2J/abacus/stru_api.py,sha256=aBKKlZ2hvAZChVCfNxRdxH51rfHKqZn6kOlazY-yW8k,67888
@@ -76,18 +76,18 @@ TB2J/spinham/supercell.py,sha256=y17uUC6r3gQb278FhxIW4CABihfLTvKFj6flyXrCPR8,122
 TB2J/wannier/__init__.py,sha256=7ojCbM84PYv1X1Tbo4NHI-d3gWmQsZB_xiYqbfxVV1E,80
 TB2J/wannier/w90_parser.py,sha256=dbd63LuKyv2DVUzqRINGsbDzEsOxsQyE8_Ear_LQIRg,4620
 TB2J/wannier/w90_tb_parser.py,sha256=qt8pnuprmPp9iIAYwPkPbmEzk6ZPgMq2xognoQp7vwc,4610
-TB2J-0.9.3rc0.data/scripts/TB2J_downfold.py,sha256=i4BVqnpDdgrX_amookVWeLGefGBn-qeAutWiwuY9SfQ,2099
-TB2J-0.9.3rc0.data/scripts/TB2J_eigen.py,sha256=Qs9v2hnMm2Tpfoa4h53muUKty2dZjwx8948MBoQooNg,1128
-TB2J-0.9.3rc0.data/scripts/TB2J_magnon.py,sha256=q7UwAmorRcFNk4tfE7gl_ny05l6p7pbD9Wm_LkIpKEw,3101
-TB2J-0.9.3rc0.data/scripts/TB2J_magnon_dos.py,sha256=TMXQvD2dIbO5FZ4tUMmxJgCgH2O2hDAPUNfEKO4z-x4,110
-TB2J-0.9.3rc0.data/scripts/TB2J_merge.py,sha256=y834SF4rIRn1L1ptkhczvavQpC-8Px6DTmDOOSaq_DE,1854
-TB2J-0.9.3rc0.data/scripts/TB2J_rotate.py,sha256=zgiDFuYZNmzKK0rwDmTaYD2OpRlmKA_VGeBx83w2Xwc,873
-TB2J-0.9.3rc0.data/scripts/TB2J_rotateDM.py,sha256=kCvF7gotuqAX1VnJ06cwfVm7RrhrdtiV5v7d9P2Pn_E,567
-TB2J-0.9.3rc0.data/scripts/abacus2J.py,sha256=M4B07lvTCDczTPTqvnDh_PERzCARAd09TLKv4aIdSQM,4408
-TB2J-0.9.3rc0.data/scripts/siesta2J.py,sha256=hBzS7ZgoHM3oXlTCQd-xVA07Ks2FiIwyRpQWUFITRPE,4303
-TB2J-0.9.3rc0.data/scripts/wann2J.py,sha256=2t2hWwyELskYCwkGDziCgiIAnfr6odLLJ6cQBJ2RQwQ,5714
-TB2J-0.9.3rc0.dist-info/LICENSE,sha256=CbZI-jyRTjiqIcWa244cRSHJdjjtUNqGR4HeJkgEwJw,1332
-TB2J-0.9.3rc0.dist-info/METADATA,sha256=9kUPr5GCrvzdnzjbhCmByaSBj6qZ-991Hab3hJ7WUzE,1455
-TB2J-0.9.3rc0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-TB2J-0.9.3rc0.dist-info/top_level.txt,sha256=whYa5ByLYhl5XnTPBHSWr-IGD6VWmr5Ql2bye2qwV_s,5
-TB2J-0.9.3rc0.dist-info/RECORD,,
+TB2J-0.9.5rc0.data/scripts/TB2J_downfold.py,sha256=i4BVqnpDdgrX_amookVWeLGefGBn-qeAutWiwuY9SfQ,2099
+TB2J-0.9.5rc0.data/scripts/TB2J_eigen.py,sha256=Qs9v2hnMm2Tpfoa4h53muUKty2dZjwx8948MBoQooNg,1128
+TB2J-0.9.5rc0.data/scripts/TB2J_magnon.py,sha256=q7UwAmorRcFNk4tfE7gl_ny05l6p7pbD9Wm_LkIpKEw,3101
+TB2J-0.9.5rc0.data/scripts/TB2J_magnon_dos.py,sha256=TMXQvD2dIbO5FZ4tUMmxJgCgH2O2hDAPUNfEKO4z-x4,110
+TB2J-0.9.5rc0.data/scripts/TB2J_merge.py,sha256=y834SF4rIRn1L1ptkhczvavQpC-8Px6DTmDOOSaq_DE,1854
+TB2J-0.9.5rc0.data/scripts/TB2J_rotate.py,sha256=zgiDFuYZNmzKK0rwDmTaYD2OpRlmKA_VGeBx83w2Xwc,873
+TB2J-0.9.5rc0.data/scripts/TB2J_rotateDM.py,sha256=kCvF7gotuqAX1VnJ06cwfVm7RrhrdtiV5v7d9P2Pn_E,567
+TB2J-0.9.5rc0.data/scripts/abacus2J.py,sha256=M4B07lvTCDczTPTqvnDh_PERzCARAd09TLKv4aIdSQM,4408
+TB2J-0.9.5rc0.data/scripts/siesta2J.py,sha256=hBzS7ZgoHM3oXlTCQd-xVA07Ks2FiIwyRpQWUFITRPE,4303
+TB2J-0.9.5rc0.data/scripts/wann2J.py,sha256=2t2hWwyELskYCwkGDziCgiIAnfr6odLLJ6cQBJ2RQwQ,5714
+TB2J-0.9.5rc0.dist-info/LICENSE,sha256=CbZI-jyRTjiqIcWa244cRSHJdjjtUNqGR4HeJkgEwJw,1332
+TB2J-0.9.5rc0.dist-info/METADATA,sha256=FGkO-SIi13BZYGhx7szEosm6WT73xTVPxTyYxWNns7U,1429
+TB2J-0.9.5rc0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+TB2J-0.9.5rc0.dist-info/top_level.txt,sha256=whYa5ByLYhl5XnTPBHSWr-IGD6VWmr5Ql2bye2qwV_s,5
+TB2J-0.9.5rc0.dist-info/RECORD,,