quant-met 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

quant_met/mean_field/quantum_metric.py

@@ -60,7 +60,7 @@ def quantum_metric(
 
 
 def quantum_metric_bdg(
-    h: BaseHamiltonian, k_grid: npt.NDArray[np.float64], band: int
+    h: BaseHamiltonian, k_grid: npt.NDArray[np.float64], bands: list[int]
 ) -> npt.NDArray[np.float64]:
     """Calculate the quantum metric in the BdG state.
 
@@ -85,24 +85,25 @@ def quantum_metric_bdg(
 
     quantum_geom_tensor = np.zeros(shape=(2, 2), dtype=np.complex64)
 
-    for i, direction_1 in enumerate(["x", "y"]):
-        h_derivative_direction_1 = h.bdg_hamiltonian_derivative(k=k_grid, direction=direction_1)
-        for j, direction_2 in enumerate(["x", "y"]):
-            h_derivative_direction_2 = h.bdg_hamiltonian_derivative(k=k_grid, direction=direction_2)
-            for k_index in range(len(k_grid)):
-                for n in [i for i in range(h.number_of_bands) if i != band]:
-                    quantum_geom_tensor[i, j] += (
-                        (
-                            bdg_functions[k_index][:, band].conjugate()
-                            @ h_derivative_direction_1[k_index]
-                            @ bdg_functions[k_index][:, n]
-                        )
-                        * (
-                            bdg_functions[k_index][:, n].conjugate()
-                            @ h_derivative_direction_2[k_index]
-                            @ bdg_functions[k_index][:, band]
+    for band in bands:
+        for i, direction_1 in enumerate(["x", "y"]):
+            h_derivative_dir_1 = h.bdg_hamiltonian_derivative(k=k_grid, direction=direction_1)
+            for j, direction_2 in enumerate(["x", "y"]):
+                h_derivative_dir_2 = h.bdg_hamiltonian_derivative(k=k_grid, direction=direction_2)
+                for k_index in range(len(k_grid)):
+                    for n in [i for i in range(h.number_of_bands) if i != band]:
+                        quantum_geom_tensor[i, j] += (
+                            (
+                                bdg_functions[k_index][:, band].conjugate()
+                                @ h_derivative_dir_1[k_index]
+                                @ bdg_functions[k_index][:, n]
+                            )
+                            * (
+                                bdg_functions[k_index][:, n].conjugate()
+                                @ h_derivative_dir_2[k_index]
+                                @ bdg_functions[k_index][:, band]
+                            )
+                            / (energies[k_index][band] - energies[k_index][n]) ** 2
                         )
-                        / (energies[k_index][band] - energies[k_index][n]) ** 2
-                    )
 
     return np.real(quantum_geom_tensor) / number_k_points

quant_met/mean_field/superfluid_weight.py

@@ -71,34 +71,8 @@ def _current_operator(
     return j
 
 
-def _w_matrix(
-    h: BaseHamiltonian, k: npt.NDArray[np.float64]
-) -> tuple[npt.NDArray[np.complex64], npt.NDArray[np.complex64]]:
-    _, bloch = h.diagonalize_nonint(k=k)
-    _, bdg_functions = h.diagonalize_bdg(k=k)
-
-    w_plus = np.zeros((2 * h.number_of_bands, h.number_of_bands), dtype=np.complex64)
-    for i in range(2 * h.number_of_bands):
-        for m in range(h.number_of_bands):
-            w_plus[i, m] = (
-                np.tensordot(bloch[:, m], np.array([1, 0]), axes=0).reshape(-1)
-                @ bdg_functions[:, i]
-            )
-
-    w_minus = np.zeros((2 * h.number_of_bands, h.number_of_bands), dtype=np.complex64)
-    for i in range(2 * h.number_of_bands):
-        for m in range(h.number_of_bands):
-            w_minus[i, m] = (
-                np.tensordot(bloch[:, m].conjugate(), np.array([0, 1]), axes=0).reshape(-1)
-                @ bdg_functions[:, i]
-            )
-
-    return w_plus, w_minus
-
-
 def _c_factor(h: BaseHamiltonian, k: npt.NDArray[np.float64]) -> npt.NDArray[np.complex64]:
-    bdg_energies, _ = h.diagonalize_bdg(k)
-    w_plus, w_minus = _w_matrix(h, k)
+    bdg_energies, bdg_functions = h.diagonalize_bdg(k)
     c_mnpq = np.zeros(
         shape=(
             h.number_of_bands,
@@ -124,10 +98,10 @@ def _c_factor(h: BaseHamiltonian, k: npt.NDArray[np.float64]) -> npt.NDArray[np.
                                 c_tmp -= _fermi_dirac_derivative()
 
                             c_tmp *= (
-                                w_minus[i, m].conjugate()
-                                * w_plus[j, n]
-                                * w_minus[j, p].conjugate()
-                                * w_minus[i, q]
+                                bdg_functions[i, m].conjugate()
+                                * bdg_functions[j, n]
+                                * bdg_functions[j, p].conjugate()
+                                * bdg_functions[i, q].conjugate()
                             )
 
                     c_mnpq[m, n, p, q] = 2 * c_tmp

{quant_met-0.0.4.dist-info → quant_met-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quant-met
-Version: 0.0.4
+Version: 0.0.5
 Summary: Calculate superconductivity in flat-band systems.
 Author: Tjark Sievers
 Author-email: tsievers@physnet.uni-hamburg.de
@@ -9,7 +9,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: h5py (>=3.11.0,<4.0.0)
-Requires-Dist: matplotlib (>=3.9.1,<4.0.0)
+Requires-Dist: matplotlib (>=3.9.2,<4.0.0)
 Requires-Dist: numpy (>=2.0.0,<3.0.0)
 Requires-Dist: pandas (>=2.2.2,<3.0.0)
 Requires-Dist: scipy (>=1.14.0,<2.0.0)

{quant_met-0.0.4.dist-info → quant_met-0.0.5.dist-info}/RECORD

@@ -5,13 +5,13 @@ quant_met/mean_field/base_hamiltonian.py,sha256=7rNBbkoSSaIQVg4GiKDI1WaSZxJiGR26
 quant_met/mean_field/eg_x.py,sha256=y_DWBoyRaHVIof_itAgHaoaFEEssY_Q9mhvsKC7DxdM,5286
 quant_met/mean_field/free_energy.py,sha256=FSGCHoBO1myHGwGQ8CqGu7_08whH0Ot3ikZhBu27tyM,3444
 quant_met/mean_field/graphene.py,sha256=rKD2UjB0blN4ALePk4bQlg0XahHoe_3mCqRAvEUGiqI,4162
-quant_met/mean_field/quantum_metric.py,sha256=B2QDjA4XpyWPhmElA_uqVn9pfllTvTOgqfy5yHV-nro,3820
-quant_met/mean_field/superfluid_weight.py,sha256=jKzGsNSzDTach9viQ57iV16DNkIfgPOJqmVZJGIMIAE,4891
+quant_met/mean_field/quantum_metric.py,sha256=y-ky4bU566TNBxldaYWyloqojfpTODo9gbn4TPe6-4A,3902
+quant_met/mean_field/superfluid_weight.py,sha256=840Fe3aHOVz1W7hi5GrX69_QxQ3vmdy3H0_B852dZqA,3971
 quant_met/plotting/__init__.py,sha256=QRQ3TNb0PNQi2lWXY0LHKgYSRuegM1N3dVVs9146Zug,457
 quant_met/plotting/plotting.py,sha256=iVTFZ9tQz_GalzqbQhxCiNWOhYHJM4wiZPTjXaXnApM,7326
 quant_met/utils.py,sha256=Tvw_YfqjIWx0FPGSReikSnw9xfN-T2dpQZN-KPMa69A,1709
-quant_met-0.0.4.dist-info/LICENSE.txt,sha256=QO_duPQihSJlaxSLxPAXo52X3esROP5wBkhxqBd1Z4E,1104
-quant_met-0.0.4.dist-info/LICENSES/MIT.txt,sha256=QO_duPQihSJlaxSLxPAXo52X3esROP5wBkhxqBd1Z4E,1104
-quant_met-0.0.4.dist-info/METADATA,sha256=dsJwfAESdbjfAWZtzzwASpHi9eb9_lQEDAf3f18PU8s,2598
-quant_met-0.0.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-quant_met-0.0.4.dist-info/RECORD,,
+quant_met-0.0.5.dist-info/LICENSE.txt,sha256=QO_duPQihSJlaxSLxPAXo52X3esROP5wBkhxqBd1Z4E,1104
+quant_met-0.0.5.dist-info/LICENSES/MIT.txt,sha256=QO_duPQihSJlaxSLxPAXo52X3esROP5wBkhxqBd1Z4E,1104
+quant_met-0.0.5.dist-info/METADATA,sha256=Kh99UUBpu1y3cOU1wYektey2RuA1a5Xd_acXxMQnsyc,2598
+quant_met-0.0.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+quant_met-0.0.5.dist-info/RECORD,,