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

quant_met/utils.py

@@ -1,3 +1,22 @@
+# SPDX-FileCopyrightText: 2024 Tjark Sievers
+#
+# SPDX-License-Identifier: MIT
+
+"""
+Utility functions (:mod:`quant_met.utils`)
+==========================================
+
+.. currentmodule:: quant_met.utils
+
+Functions
+---------
+
+.. autosummary::
+   :toctree: generated/
+
+    generate_uniform_grid - Generate a uniform grid of points in 2D.
+"""  # noqa: D205, D400
+
 import numpy as np
 import numpy.typing as npt
 
@@ -9,10 +28,34 @@ def generate_uniform_grid(
     corner_2: npt.NDArray[np.float64],
     origin: npt.NDArray[np.float64],
 ) -> npt.NDArray[np.float64]:
+    """
+    Generate a uniform grid of points in 2D.
+
+    Parameters
+    ----------
+        ncols : int
+            Number of columns
+        nrows : int
+            Number of rows
+        corner_1 : :py:class:`numpy.ndarray`
+            First corner vector
+        corner_2 : :py:class:`numpy.ndarray`
+            Second corner vector
+        origin : :py:class:`numpy.ndarray`
+            Origin point
+
+    Returns
+    -------
+        :py:class:`numpy.ndarray`
+            Grid
+
+    """
     if ncols <= 1 or nrows <= 1:
-        raise ValueError("Number of columns and rows must be greater than 1.")
+        msg = "Number of columns and rows must be greater than 1."
+        raise ValueError(msg)
     if np.linalg.norm(corner_1) == 0 or np.linalg.norm(corner_2) == 0:
-        raise ValueError("Vectors to the corners cannot be zero.")
+        msg = "Vectors to the corners cannot be zero."
+        raise ValueError(msg)
 
     grid: npt.NDArray[np.float64] = np.concatenate(
         [

quant_met-0.0.4.dist-info/LICENSES/MIT.txt

@@ -0,0 +1,9 @@
+MIT License
+
+Copyright (c) 2024-present Tjark <tsievers@physnet.uni-hamburg.de>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

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

@@ -1,22 +1,26 @@
 Metadata-Version: 2.1
 Name: quant-met
-Version: 0.0.2
+Version: 0.0.4
 Summary: Calculate superconductivity in flat-band systems.
 Author: Tjark Sievers
 Author-email: tsievers@physnet.uni-hamburg.de
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.11,<4.0
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 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.8.4,<4.0.0)
-Requires-Dist: numpy (>=1.26.4,<2.0.0)
+Requires-Dist: matplotlib (>=3.9.1,<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.13.0,<2.0.0)
-Requires-Dist: sympy (>=1.12,<2.0)
+Requires-Dist: scipy (>=1.14.0,<2.0.0)
 Description-Content-Type: text/markdown
 
+<!--
+SPDX-FileCopyrightText: 2024 Tjark Sievers
+
+SPDX-License-Identifier: MIT
+-->
+
 # quant-met
 
 [![Test](https://github.com/Ruberhauptmann/quant-met/actions/workflows/test.yml/badge.svg)](https://github.com/Ruberhauptmann/quant-met/actions/workflows/test.yml)

quant_met-0.0.4.dist-info/RECORD

@@ -0,0 +1,17 @@
+quant_met/__init__.py,sha256=ZO1UFz1awUYTI7B9ZkBwucvDz7GMGXnLLUGnEwLBhkc,155
+quant_met/mean_field/__init__.py,sha256=msHp5Y5cuHjqr_EdC9jS9JKpeAi6CBHggtxrG_psDRk,1182
+quant_met/mean_field/_utils.py,sha256=plkx6eYjyYV3CT3BWwlulqW7L-Q0t1TzZTLR4k7u0dg,666
+quant_met/mean_field/base_hamiltonian.py,sha256=7rNBbkoSSaIQVg4GiKDI1WaSZxJiGR26bdidh9uygzw,8812
+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/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/hamiltonians/__init__.py

@@ -1,14 +0,0 @@
-from ._base_hamiltonian import BaseHamiltonian
-from ._eg_x import EGXHamiltonian
-from ._free_energy import free_energy, free_energy_uniform_pairing
-from ._graphene import GrapheneHamiltonian
-from ._superfluid_weight import calculate_superfluid_weight
-
-__all__ = [
-    "BaseHamiltonian",
-    "GrapheneHamiltonian",
-    "EGXHamiltonian",
-    "calculate_superfluid_weight",
-    "free_energy",
-    "free_energy_uniform_pairing",
-]

quant_met/hamiltonians/_base_hamiltonian.py

@@ -1,172 +0,0 @@
-import pathlib
-from abc import ABC, abstractmethod
-
-import h5py
-import numpy as np
-import numpy.typing as npt
-import pandas as pd
-
-
-class BaseHamiltonian(ABC):
-    """Base class for Hamiltonians."""
-
-    @property
-    @abstractmethod
-    def number_of_bands(self) -> int:
-        raise NotImplementedError
-
-    @property
-    @abstractmethod
-    def coloumb_orbital_basis(self) -> npt.NDArray[np.float64]:
-        raise NotImplementedError
-
-    @property
-    def delta_orbital_basis(self) -> npt.NDArray[np.float64]:
-        raise NotImplementedError
-
-    @delta_orbital_basis.setter
-    @abstractmethod
-    def delta_orbital_basis(self, new_delta: npt.NDArray[np.float64]) -> None:
-        raise NotImplementedError
-
-    @abstractmethod
-    def _hamiltonian_one_point(
-        self, k_point: npt.NDArray[np.float64]
-    ) -> npt.NDArray[np.complex64]:
-        raise NotImplementedError
-
-    @abstractmethod
-    def _hamiltonian_derivative_one_point(
-        self, k_point: npt.NDArray[np.float64], directions: str
-    ) -> npt.NDArray[np.complex64]:
-        raise NotImplementedError
-
-    def _bdg_hamiltonian_one_point(
-        self, k_point: npt.NDArray[np.float64]
-    ) -> npt.NDArray[np.complex64]:
-        delta_matrix: npt.NDArray[np.complex64] = np.zeros(
-            shape=(self.number_of_bands, self.number_of_bands), dtype=np.complex64
-        )
-        np.fill_diagonal(delta_matrix, self.delta_orbital_basis)
-
-        h = np.block(
-            [
-                [self.hamiltonian(k_point), delta_matrix],
-                [np.conjugate(delta_matrix), -np.conjugate(self.hamiltonian(-k_point))],
-            ]
-        )
-        return h
-
-    def save(self, filename: pathlib.Path) -> None:
-        with h5py.File(f"{filename}", "a") as f:
-            f.create_dataset("delta", data=self.delta_orbital_basis)
-            for key, value in vars(self).items():
-                if not key.startswith("_"):
-                    f.attrs[key] = value
-
-    @classmethod
-    def from_file(cls, filename: pathlib.Path) -> "BaseHamiltonian":
-        config_dict = {}
-        with h5py.File(f"{filename}", "r") as f:
-            config_dict["delta"] = f["delta"][()]
-            for key, value in f.attrs.items():
-                config_dict[key] = value
-
-        return cls(**config_dict)
-
-    def bdg_hamiltonian(self, k: npt.NDArray[np.float64]) -> npt.NDArray[np.complex64]:
-        if np.isnan(k).any() or np.isinf(k).any():
-            raise ValueError("k is NaN or Infinity")
-        if k.ndim == 1:
-            h = self._bdg_hamiltonian_one_point(k)
-        else:
-            h = np.array([self._bdg_hamiltonian_one_point(k) for k in k])
-        return h
-
-    def hamiltonian(self, k: npt.NDArray[np.float64]) -> npt.NDArray[np.complex64]:
-        if np.isnan(k).any() or np.isinf(k).any():
-            raise ValueError("k is NaN or Infinity")
-        if k.ndim == 1:
-            h = self._hamiltonian_one_point(k)
-        else:
-            h = np.array([self._hamiltonian_one_point(k) for k in k])
-        return h
-
-    def hamiltonian_derivative(
-        self, k: npt.NDArray[np.float64], direction: str
-    ) -> npt.NDArray[np.complex64]:
-        if np.isnan(k).any() or np.isinf(k).any():
-            raise ValueError("k is NaN or Infinity")
-        if k.ndim == 1:
-            h = self._hamiltonian_derivative_one_point(k, direction)
-        else:
-            h = np.array(
-                [self._hamiltonian_derivative_one_point(k, direction) for k in k]
-            )
-        return h
-
-    def diagonalize_nonint(
-        self, k: npt.NDArray[np.float64]
-    ) -> tuple[npt.NDArray[np.float64], npt.NDArray[np.float64]]:
-        k_point_matrix = self.hamiltonian(k)
-
-        if k.ndim == 1:
-            band_energies, bloch_wavefunctions = np.linalg.eigh(k_point_matrix)
-        else:
-            bloch_wavefunctions = np.zeros(
-                (len(k), self.number_of_bands, self.number_of_bands),
-                dtype=complex,
-            )
-            band_energies = np.zeros((len(k), self.number_of_bands))
-
-            for i, k in enumerate(k):
-                band_energies[i], bloch_wavefunctions[i] = np.linalg.eigh(
-                    k_point_matrix[i]
-                )
-
-        return band_energies, bloch_wavefunctions
-
-    def diagonalize_bdg(
-        self, k: npt.NDArray[np.float64]
-    ) -> tuple[npt.NDArray[np.float64], npt.NDArray[np.complex64]]:
-        bdg_matrix = self.bdg_hamiltonian(k)
-
-        if k.ndim == 1:
-            bdg_energies, bdg_wavefunctions = np.linalg.eigh(bdg_matrix)
-        else:
-            bdg_wavefunctions = np.zeros(
-                (len(k), 2 * self.number_of_bands, 2 * self.number_of_bands),
-                dtype=np.complex64,
-            )
-            bdg_energies = np.zeros((len(k), 2 * self.number_of_bands))
-
-            for i, k in enumerate(k):
-                bdg_energies[i], bdg_wavefunctions[i] = np.linalg.eigh(bdg_matrix[i])
-
-        return bdg_energies, bdg_wavefunctions
-
-    def calculate_bandstructure(
-        self,
-        k: npt.NDArray[np.float64],
-        overlaps: tuple[npt.NDArray[np.float64], npt.NDArray[np.float64]] | None = None,
-    ) -> pd.DataFrame:
-        k_point_matrix = self.hamiltonian(k)
-
-        results = pd.DataFrame(
-            index=range(len(k)),
-            dtype=float,
-        )
-
-        for i, k in enumerate(k):
-            energies, eigenvectors = np.linalg.eigh(k_point_matrix[i])
-
-            for band_index in range(self.number_of_bands):
-                results.at[i, f"band_{band_index}"] = energies[band_index]
-
-                if overlaps is not None:
-                    results.at[i, f"wx_{band_index}"] = (
-                        np.abs(np.dot(eigenvectors[:, band_index], overlaps[0])) ** 2
-                        - np.abs(np.dot(eigenvectors[:, band_index], overlaps[1])) ** 2
-                    )
-
-        return results

quant_met/hamiltonians/_eg_x.py

@@ -1,124 +0,0 @@
-import numpy as np
-import numpy.typing as npt
-
-from ._base_hamiltonian import BaseHamiltonian
-from ._utils import _check_valid_float
-
-
-class EGXHamiltonian(BaseHamiltonian):
-    def __init__(
-        self,
-        t_gr: float,
-        t_x: float,
-        V: float,
-        a: float,
-        mu: float,
-        U_gr: float,
-        U_x: float,
-        delta: npt.NDArray[np.float64] | None = None,
-    ):
-        self.t_gr = _check_valid_float(t_gr, "Hopping graphene")
-        self.t_x = _check_valid_float(t_x, "Hopping impurity")
-        self.V = _check_valid_float(V, "Hybridisation")
-        self.a = _check_valid_float(a, "Lattice constant")
-        self.mu = _check_valid_float(mu, "Chemical potential")
-        self.U_gr = _check_valid_float(U_gr, "Coloumb interaction graphene")
-        self.U_x = _check_valid_float(U_x, "Coloumb interaction impurity")
-        if delta is None:
-            self._delta_orbital_basis = np.zeros(3)
-        else:
-            self._delta_orbital_basis = delta
-
-    @property
-    def coloumb_orbital_basis(self) -> npt.NDArray[np.float64]:
-        return np.array([self.U_gr, self.U_gr, self.U_x])
-
-    @property
-    def delta_orbital_basis(self) -> npt.NDArray[np.float64]:
-        return self._delta_orbital_basis
-
-    @delta_orbital_basis.setter
-    def delta_orbital_basis(self, new_delta: npt.NDArray[np.float64]) -> None:
-        self._delta_orbital_basis = new_delta
-
-    @property
-    def number_of_bands(self) -> int:
-        return 3
-
-    def _hamiltonian_derivative_one_point(
-        self, k: npt.NDArray[np.float64], direction: str
-    ) -> npt.NDArray[np.complex64]:
-        assert direction in ["x", "y"]
-
-        t_gr = self.t_gr
-        t_x = self.t_x
-        a = self.a
-
-        h = np.zeros((self.number_of_bands, self.number_of_bands), dtype=np.complex64)
-
-        if direction == "x":
-            h[0, 1] = (
-                t_gr
-                * a
-                * np.exp(-0.5j * a / np.sqrt(3) * k[1])
-                * np.sin(0.5 * a * k[0])
-            )
-            h[1, 0] = h[0, 1].conjugate()
-            h[2, 2] = (
-                2
-                * a
-                * t_x
-                * (
-                    np.sin(a * k[0])
-                    + np.sin(0.5 * a * k[0]) * np.cos(0.5 * np.sqrt(3) * a * k[1])
-                )
-            )
-        else:
-            h[0, 1] = (
-                -t_gr
-                * 1j
-                * a
-                / np.sqrt(3)
-                * (
-                    np.exp(1j * a / np.sqrt(3) * k[1])
-                    - np.exp(-0.5j * a / np.sqrt(3) * k[1]) * np.cos(0.5 * a * k[0])
-                )
-            )
-            h[1, 0] = h[0, 1].conjugate()
-            h[2, 2] = np.sqrt(3) * a * t_x * np.cos(0.5 * np.sqrt(3) * a * k[1])
-
-        return h
-
-    def _hamiltonian_one_point(
-        self, k: npt.NDArray[np.float64]
-    ) -> npt.NDArray[np.complex64]:
-        t_gr = self.t_gr
-        t_x = self.t_x
-        a = self.a
-        # a_0 = a / np.sqrt(3)
-        V = self.V
-        mu = self.mu
-
-        h = np.zeros((self.number_of_bands, self.number_of_bands), dtype=np.complex64)
-
-        h[0, 1] = -t_gr * (
-            np.exp(1j * k[1] * a / np.sqrt(3))
-            + 2 * np.exp(-0.5j * a / np.sqrt(3) * k[1]) * (np.cos(0.5 * a * k[0]))
-        )
-
-        h[1, 0] = h[0, 1].conjugate()
-
-        h[2, 0] = V
-        h[0, 2] = V
-
-        h[2, 2] = (
-            -2
-            * t_x
-            * (
-                np.cos(a * k[0])
-                + 2 * np.cos(0.5 * a * k[0]) * np.cos(0.5 * np.sqrt(3) * a * k[1])
-            )
-        )
-        h -= mu * np.eye(3, dtype=np.complex64)
-
-        return h

quant_met/hamiltonians/_free_energy.py

@@ -1,39 +0,0 @@
-import numpy as np
-import numpy.typing as npt
-
-from ._base_hamiltonian import BaseHamiltonian
-
-
-def free_energy(
-    delta_vector: npt.NDArray[np.float64],
-    hamiltonian: BaseHamiltonian,
-    k_points: npt.NDArray[np.float64],
-) -> float:
-    number_k_points = len(k_points)
-    hamiltonian.delta_orbital_basis = delta_vector
-    bdg_energies, _ = hamiltonian.diagonalize_bdg(k_points)
-
-    k_array = np.array(
-        [
-            np.sum(np.abs(bdg_energies[k_index][0 : hamiltonian.number_of_bands]))
-            for k_index in range(number_k_points)
-        ]
-    )
-
-    integral: float = -np.sum(k_array, axis=-1) / number_k_points + np.sum(
-        np.power(np.abs(delta_vector), 2) / hamiltonian.coloumb_orbital_basis
-    )
-
-    return integral
-
-
-def free_energy_uniform_pairing(
-    delta: float,
-    hamiltonian: BaseHamiltonian,
-    k_points: npt.NDArray[np.float64],
-) -> float:
-    delta_vector = np.ones(hamiltonian.number_of_bands) * delta
-
-    return free_energy(
-        delta_vector=delta_vector, hamiltonian=hamiltonian, k_points=k_points
-    )

quant_met/hamiltonians/_graphene.py

@@ -1,93 +0,0 @@
-import numpy as np
-import numpy.typing as npt
-
-from ._base_hamiltonian import BaseHamiltonian
-from ._utils import _check_valid_float
-
-
-class GrapheneHamiltonian(BaseHamiltonian):
-    def __init__(
-        self,
-        t_nn: float,
-        a: float,
-        mu: float,
-        coulomb_gr: float,
-        delta: npt.NDArray[np.float64] | None = None,
-    ):
-        self.t_nn = _check_valid_float(t_nn, "Hopping")
-        if a <= 0:
-            raise ValueError("Lattice constant must be positive")
-        self.a = _check_valid_float(a, "Lattice constant")
-        self.mu = _check_valid_float(mu, "Chemical potential")
-        self.coulomb_gr = _check_valid_float(coulomb_gr, "Coloumb interaction")
-        if delta is None:
-            self._delta_orbital_basis = np.zeros(2)
-        else:
-            self._delta_orbital_basis = delta
-
-    @property
-    def coloumb_orbital_basis(self) -> npt.NDArray[np.float64]:
-        return np.array([self.coulomb_gr, self.coulomb_gr])
-
-    @property
-    def number_of_bands(self) -> int:
-        return 2
-
-    @property
-    def delta_orbital_basis(self) -> npt.NDArray[np.float64]:
-        return self._delta_orbital_basis
-
-    @delta_orbital_basis.setter
-    def delta_orbital_basis(self, new_delta: npt.NDArray[np.float64]) -> None:
-        self._delta_orbital_basis = new_delta
-
-    def _hamiltonian_derivative_one_point(
-        self, k: npt.NDArray[np.float64], direction: str
-    ) -> npt.NDArray[np.complex64]:
-        assert direction in ["x", "y"]
-
-        t_nn = self.t_nn
-        a = self.a
-
-        h = np.zeros((self.number_of_bands, self.number_of_bands), dtype=np.complex64)
-
-        if direction == "x":
-            h[0, 1] = (
-                t_nn
-                * a
-                * np.exp(-0.5j * a / np.sqrt(3) * k[1])
-                * np.sin(0.5 * a * k[0])
-            )
-            h[1, 0] = h[0, 1].conjugate()
-        else:
-            h[0, 1] = (
-                -t_nn
-                * 1j
-                * a
-                / np.sqrt(3)
-                * (
-                    np.exp(1j * a / np.sqrt(3) * k[1])
-                    - np.exp(-0.5j * a / np.sqrt(3) * k[1]) * np.cos(0.5 * a * k[0])
-                )
-            )
-            h[1, 0] = h[0, 1].conjugate()
-
-        return h
-
-    def _hamiltonian_one_point(
-        self, k: npt.NDArray[np.float64]
-    ) -> npt.NDArray[np.complex64]:
-        t_nn = self.t_nn
-        a = self.a
-        mu = self.mu
-
-        h = np.zeros((self.number_of_bands, self.number_of_bands), dtype=np.complex64)
-
-        h[0, 1] = -t_nn * (
-            np.exp(1j * k[1] * a / np.sqrt(3))
-            + 2 * np.exp(-0.5j * a / np.sqrt(3) * k[1]) * (np.cos(0.5 * a * k[0]))
-        )
-        h[1, 0] = h[0, 1].conjugate()
-        h -= mu * np.eye(2)
-
-        return h

quant_met/hamiltonians/_superfluid_weight.py

@@ -1,130 +0,0 @@
-import numpy as np
-import numpy.typing as npt
-
-from ._base_hamiltonian import BaseHamiltonian
-
-
-def calculate_current_operator(
-    h: BaseHamiltonian, direction: str, k: npt.NDArray[np.float64]
-) -> npt.NDArray[np.complex64]:
-    j = np.zeros(shape=(h.number_of_bands, h.number_of_bands), dtype=np.complex64)
-
-    _, bloch = h.diagonalize_nonint(k=k)
-
-    for m in range(h.number_of_bands):
-        for n in range(h.number_of_bands):
-            j[m, n] = (
-                np.conjugate(bloch[:, m])
-                @ h.hamiltonian_derivative(direction=direction, k=k)
-                @ bloch[:, n]
-            )
-
-    return j
-
-
-def calculate_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(
-                    np.conjugate(bloch[:, m]), np.array([0, 1]), axes=0
-                ).reshape(-1)
-                @ bdg_functions[:, i]
-            )
-
-    return w_plus, w_minus
-
-
-def calculate_c_factor(
-    h: BaseHamiltonian, k: npt.NDArray[np.float64]
-) -> npt.NDArray[np.complex64]:
-    bdg_energies, _ = h.diagonalize_bdg(k)
-    w_plus, w_minus = calculate_w_matrix(h, k)
-    C_mnpq = np.zeros(
-        shape=(
-            h.number_of_bands,
-            h.number_of_bands,
-            h.number_of_bands,
-            h.number_of_bands,
-        ),
-        dtype=np.complex64,
-    )
-
-    for m in range(h.number_of_bands):
-        for n in range(h.number_of_bands):
-            for p in range(h.number_of_bands):
-                for q in range(h.number_of_bands):
-                    C_tmp: float = 0
-                    for i in range(2 * h.number_of_bands):
-                        for j in range(2 * h.number_of_bands):
-                            if bdg_energies[i] != bdg_energies[j]:
-                                C_tmp += (
-                                    fermi_dirac(bdg_energies[i])
-                                    - fermi_dirac(bdg_energies[j])
-                                ) / (bdg_energies[j] - bdg_energies[i])
-                            else:
-                                C_tmp -= fermi_dirac_derivative()
-
-                            C_tmp *= (
-                                np.conjugate(w_minus[i, m])
-                                * w_plus[j, n]
-                                * np.conjugate(w_minus[j, p])
-                                * w_minus[i, q]
-                            )
-
-                    C_mnpq[m, n, p, q] = 2 * C_tmp
-
-    return C_mnpq
-
-
-def fermi_dirac_derivative() -> float:
-    return 0
-
-
-def fermi_dirac(energy: np.float64) -> np.float64:
-    if energy > 0:
-        return np.float64(0)
-    else:
-        return np.float64(1)
-
-
-def calculate_superfluid_weight(
-    h: BaseHamiltonian,
-    k_grid: npt.NDArray[np.float64],
-    direction_1: str,
-    direction_2: str,
-) -> tuple[float, float]:
-    # number_k_points = len(k_grid)
-
-    s_weight_conv = 0
-    s_weight_geom = 0
-
-    for k in k_grid:
-        C_mnpq = calculate_c_factor(h, k)
-        j_up = calculate_current_operator(h, direction_1, k)
-        j_down = calculate_current_operator(h, direction_2, -k)
-        for m in range(h.number_of_bands):
-            for n in range(h.number_of_bands):
-                for p in range(h.number_of_bands):
-                    for q in range(h.number_of_bands):
-                        s_weight = C_mnpq[m, n, p, q] * j_up[m, n] * j_down[q, p]
-                        if m == n and p == q:
-                            s_weight_conv += s_weight
-                        else:
-                            s_weight_geom += s_weight
-
-    return s_weight_conv, s_weight_geom