From 1c6d8695a0af2c3920de6012853c9c1f432097b5 Mon Sep 17 00:00:00 2001
From: Johanna <johanna@zijderveld.de>
Date: Thu, 11 Apr 2024 23:07:42 +0200
Subject: [PATCH] move ifftn to model.py to remove one line function
---
codes/model.py | 10 +++++-----
codes/tb/transforms.py | 4 ----
2 files changed, 5 insertions(+), 9 deletions(-)
diff --git a/codes/model.py b/codes/model.py
index 9606a2e..69e1a4f 100644
--- a/codes/model.py
+++ b/codes/model.py
@@ -1,12 +1,13 @@
# %%
from codes.tb.tb import addTb
-from codes.tb.transforms import tb2kham, kdens2tbFFT, ifftn2tb
+from codes.tb.transforms import tb2kham, ifftn2tb
from codes.mf import (
densityMatrix,
fermiOnGrid,
meanField,
)
import numpy as np
+from scipy.fftpack import ifftn
class Model:
@@ -37,7 +38,9 @@ class Model:
def makeDensityMatrixTb(self, mf_model, nK=200):
self.kham = tb2kham(addTb(self.h_0, mf_model), nK=nK, ndim=self._ndim)
self.calculateEF()
- return ifftn2tb(kdens2tbFFT(densityMatrix(self.kham, self.EF), self._ndim))
+ return ifftn2tb(
+ ifftn(densityMatrix(self.kham, self.EF), axes=np.arange(self._ndim))
+ )
def mfield(self, mf_model, nK=200):
densityMatrixTb = self.makeDensityMatrixTb(mf_model, nK=nK)
@@ -45,6 +48,3 @@ class Model:
meanField(densityMatrixTb, self.h_int, n=self._ndim),
{self._localKey: -self.EF * np.eye(self._size)},
)
-
-
-# %%
diff --git a/codes/tb/transforms.py b/codes/tb/transforms.py
index 45c1abc..20e7a4a 100644
--- a/codes/tb/transforms.py
+++ b/codes/tb/transforms.py
@@ -134,7 +134,3 @@ def kfunc2tb(kfunc, nSamples, ndim=1):
raise NotImplementedError("n > 2 not implemented")
ifftnArray = ifftn(kfuncOnGrid, axes=np.arange(ndim))
return ifftn2tb(ifftnArray)
-
-
-def kdens2tbFFT(kdens, ndim=1):
- return ifftn(kdens, axes=np.arange(ndim))
--
GitLab