diff --git a/codes/model.py b/codes/model.py index 9606a2e69ac5c844f9fc57df6d8aca47bfa73a5a..69e1a4f8780c6f8fefdb835b8a9446c9295b228f 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 45c1abc16ed4db323668a934993645f4e9fdfd64..20e7a4ad0290373efdbe65cf2487ba0334fd1739 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))