nabu 2024.1.8__tar.gz → 2024.1.8.post1__tar.gz

{nabu-2024.1.8/nabu.egg-info → nabu-2024.1.8.post1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nabu
-Version: 2024.1.8
+Version: 2024.1.8.post1
 Summary: Nabu - Tomography software
 Author-email: Pierre Paleo <pierre.paleo@esrf.fr>, Henri Payno <henri.payno@esrf.fr>, Alessandro Mirone <mirone@esrf.fr>, Jérôme Lesaint <jerome.lesaint@esrf.fr>
 Maintainer-email: Pierre Paleo <pierre.paleo@esrf.fr>
@@ -48,29 +48,9 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numpy<2,>1.9.0
-Requires-Dist: scipy
-Requires-Dist: h5py>=3.0
-Requires-Dist: silx>=0.15.0
-Requires-Dist: tomoscan>=2.0.4
-Requires-Dist: psutil
-Requires-Dist: pytest
-Requires-Dist: tifffile
 Provides-Extra: full
-Requires-Dist: scikit-image; extra == "full"
-Requires-Dist: PyWavelets; extra == "full"
-Requires-Dist: glymur; extra == "full"
-Requires-Dist: pycuda<2024.1.1; extra == "full"
-Requires-Dist: scikit-cuda; extra == "full"
-Requires-Dist: pycudwt; extra == "full"
-Requires-Dist: sluurp>=0.3; extra == "full"
-Requires-Dist: pyvkfft; extra == "full"
 Provides-Extra: doc
-Requires-Dist: sphinx; extra == "doc"
-Requires-Dist: cloud_sptheme; extra == "doc"
-Requires-Dist: myst-parser; extra == "doc"
-Requires-Dist: nbsphinx; extra == "doc"
+License-File: LICENSE
 
 # Nabu
 

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2024.1.8"
+__version__ = "2024.1.8.post1"
 __nabu_modules__ = [
     "app",
     "cuda",

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu/pipeline/fullfield/chunked.py

@@ -577,7 +577,7 @@ class ChunkedPipeline:
                 sample_detector_dist,
                 angles=-options["angles"],
                 rot_center=options["rotation_axis_position"],
-                axis_correction=-options["axis_correction"],
+                axis_correction=(-options["axis_correction"] if options["axis_correction"] is not None else None),
                 pixel_size=1,
                 scale_factor=1.0 / options["voxel_size_cm"][0],
             )

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu/reconstruction/tests/test_cone.py

@@ -281,7 +281,7 @@ class TestCone:
             # Reconstruct with horizontal shifts
             cone_reconstructor_with_correction = ConebeamReconstructor(
                 *reconstructor_args,
-                axis_corrections=shifts,
+                axis_correction=shifts,
                 **reconstructor_kwargs,
             )
 

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu/stitching/overlap.py

@@ -301,7 +301,7 @@ def check_overlaps(frames: Union[tuple, numpy.ndarray], positions: tuple, axis:
         if raise_error:
             raise ValueError(error_msg)
         else:
-            _logger.error(raise_error)
+            _logger.error(error_msg)
 
     # convert each frame to appropriate bounding box according to the axis
     def convert_to_bb(frame: numpy.ndarray, position: tuple, axis: int):

{nabu-2024.1.8 → nabu-2024.1.8.post1/nabu.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nabu
-Version: 2024.1.8
+Version: 2024.1.8.post1
 Summary: Nabu - Tomography software
 Author-email: Pierre Paleo <pierre.paleo@esrf.fr>, Henri Payno <henri.payno@esrf.fr>, Alessandro Mirone <mirone@esrf.fr>, Jérôme Lesaint <jerome.lesaint@esrf.fr>
 Maintainer-email: Pierre Paleo <pierre.paleo@esrf.fr>
@@ -48,29 +48,9 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numpy<2,>1.9.0
-Requires-Dist: scipy
-Requires-Dist: h5py>=3.0
-Requires-Dist: silx>=0.15.0
-Requires-Dist: tomoscan>=2.0.4
-Requires-Dist: psutil
-Requires-Dist: pytest
-Requires-Dist: tifffile
 Provides-Extra: full
-Requires-Dist: scikit-image; extra == "full"
-Requires-Dist: PyWavelets; extra == "full"
-Requires-Dist: glymur; extra == "full"
-Requires-Dist: pycuda<2024.1.1; extra == "full"
-Requires-Dist: scikit-cuda; extra == "full"
-Requires-Dist: pycudwt; extra == "full"
-Requires-Dist: sluurp>=0.3; extra == "full"
-Requires-Dist: pyvkfft; extra == "full"
 Provides-Extra: doc
-Requires-Dist: sphinx; extra == "doc"
-Requires-Dist: cloud_sptheme; extra == "doc"
-Requires-Dist: myst-parser; extra == "doc"
-Requires-Dist: nbsphinx; extra == "doc"
+License-File: LICENSE
 
 # Nabu
 

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu.egg-info/SOURCES.txt

@@ -3,7 +3,6 @@ README.md
 pyproject.toml
 doc/conf.py
 doc/create_conf_doc.py
-doc/doc_config.py
 doc/get_mathjax.py
 nabu/__init__.py
 nabu/tests.py
@@ -41,6 +40,7 @@ nabu/app/shrink_dataset.py
 nabu/app/stitching.py
 nabu/app/utils.py
 nabu/app/validator.py
+nabu/app/tests/__init__.py
 nabu/app/tests/test_reduce_dark_flat.py
 nabu/cuda/__init__.py
 nabu/cuda/convolution.py
@@ -296,35 +296,4 @@ nabu/thirdparty/algotom_convert_sino.py
 nabu/thirdparty/pore3d_deringer_munch.py
 nabu/thirdparty/tomocupy_remove_stripe.py
 nabu/thirdparty/tomopy_phase.py
-nabu/thirdparty/tomwer_load_flats_darks.py
-sandbox/align_test.py
-sandbox/app.py
-sandbox/binning_cython.py
-sandbox/circ_sm.py
-sandbox/composite_image.py
-sandbox/convert_id15.py
-sandbox/do_test_ctf.py
-sandbox/esrf_envs.py
-sandbox/fbp_polar.py
-sandbox/fbp_tilt.py
-sandbox/interleaved.py
-sandbox/interp_sinos_halftomo.py
-sandbox/linear_interp.py
-sandbox/merge_recs.py
-sandbox/moduleutils.py
-sandbox/nbreconstruct.py
-sandbox/pag_margin.py
-sandbox/parse.py
-sandbox/plot.py
-sandbox/proj3D.py
-sandbox/rec_bm05.py
-sandbox/rec_thread.py
-sandbox/shift_bilinear2.py
-sandbox/sinotilt.py
-sandbox/sysutils.py
-sandbox/test_mp_queue.py
-sandbox/tilt.py
-sandbox/utils.py
-sandbox/vo.py
-sandbox/workers.py
-sandbox/xrdrec_pyFAI_data.py
+nabu/thirdparty/tomwer_load_flats_darks.py

{nabu-2024.1.8 → nabu-2024.1.8.post1}/nabu.egg-info/top_level.txt

@@ -2,3 +2,4 @@ build
 dist
 doc
 nabu
+scripts

nabu-2024.1.8/doc/doc_config.py

@@ -1,32 +0,0 @@
-#!/usr/bin/env python
-
-from nabu.resources.nabu_config import nabu_config
-
-
-def generate(file_):
-    def write(content):
-        print(content, file=file_)
-    for section, values in nabu_config.items():
-        if section == "about":
-            continue
-        write("## %s\n" % section)
-        for key, val in values.items():
-            if val["type"] == "unsupported":
-                continue
-            write(val["help"] + "\n")
-            write(
-                "```ini\n%s = %s\n```"
-                % (key, val["default"])
-            )
-
-
-
-if __name__ == "__main__":
-
-    import sys, os
-    print(os.path.abspath(__file__))
-    exit(0)
-
-    fname = "/tmp/test.md"
-    with open(fname, "w") as f:
-        generate(f)

nabu-2024.1.8/sandbox/align_test.py

@@ -1,61 +0,0 @@
-import numpy as np
-from nabu.estimation.translation import DetectorTranslationAlongBeam
-from tomoscan.io import HDF5File
-
-
-def get_data(fname, h5_paths):
-    with HDF5File(fname, "r") as f:
-        align_data = f[h5_paths["align_data"]][()]
-        dark = f[h5_paths["dark"]][0]
-        pixel_size_um = f[h5_paths["pixel_size"]][()]
-        positions_mm = f[h5_paths["positions"]][()]
-
-    # Dark subtraction
-    align_data = align_data - dark.astype("f")
-
-    return align_data, pixel_size_um, positions_mm
-
-
-
-def get_alignment(align_data, pixel_size_um, positions_mm, plot=True):
-
-    tr_calc = DetectorTranslationAlongBeam()
-    if plot == True:
-        tr_calc.verbose=True
-
-    shifts_v, shifts_h = tr_calc.find_shift(align_data, positions_mm)
-
-    # pixel_size is in microns, motor position is in mm
-    tilt_v_deg = np.rad2deg(np.arctan(shifts_v * pixel_size_um / 1e3))
-    tilt_h_deg = np.rad2deg(np.arctan(shifts_h * pixel_size_um / 1e3))
-    print (f"\nVertical tilt  to be applied in deg (thy): {tilt_v_deg}")
-    print (f"Horizontal tilt to be applied in deg (thz): {tilt_h_deg}\n")
-
-    return align_data, shifts_v, shifts_h, positions_mm
-
-
-
-
-
-
-if __name__ == "__main__":
-    fname = "/data/id19/inhouse/id192201/id19/sample/sample_0001/sample_0001.h5"
-    h5_paths = {
-        "align_data": "11.1/measurement/pcolinux",
-        "dark": "12.1/measurement/pcolinux",
-        "pixel_size": "11.1/instrument/pcolinux/x_pixel_size",
-        "positions": "11.1/measurement/hrxc"
-    }
-    align_data, pixel_size_um, positions_mm = get_data(fname, h5_paths)
-    # align_data, shifts_v, shifts_h, positions_mm = get_alignment(align_data, pixel_size_um, positions_mm)
-
-    D = DetectorTranslationAlongBeam()
-    D. verbose = True
-    sv, sh = D.find_shift(align_data, positions_mm)
-    D.verbose = False
-    sv, sh = D.find_shift(align_data, positions_mm)
-    D.verbose = True
-    sv, sh = D.find_shift(align_data, positions_mm)
-
-
-

nabu-2024.1.8/sandbox/app.py

@@ -1,123 +0,0 @@
-from time import sleep
-import pycuda.autoinit
-import pycuda.gpuarray as garray
-from nabu.resources.processconfig import ProcessConfig
-from nabu.resources.tasks import build_processing_steps
-
-from nabu.app.logger import Logger
-
-from nabu.app.chunkreader import ChunkReaderComponent
-from nabu.app.flatfield import FlatFieldComponent
-from nabu.app.phase import PhaseRetrievalComponent
-from nabu.app.opmap import NegativeLogComponent
-from nabu.app.ccdfilter import CCDFilterComponent
-from nabu.app.unsharp import UnsharpMaskComponent
-from nabu.app.reconstructor import ReconstructorComponent
-
-
-if __name__ == "__main__":
-
-    conf = ProcessConfig("/home/pierre/workspace/data/nabu.conf")
-    steps, options = build_processing_steps(conf)
-    dataset_infos = conf.dataset_infos
-    SUB_REGION = (None, None, None, 50)
-
-    logger = Logger("nabu_processing", console=True)
-
-    # Read chunk
-    options["read_chunk"]["sub_region"] = SUB_REGION
-    options["read_chunk"]["convert_float"] = True
-    Ch = ChunkReaderComponent(options["read_chunk"], dataset_infos, logger=logger)
-    Ch.execute()
-    radios = Ch.chunk_reader.files_data
-    d_radios = garray.to_gpu(radios)
-
-    # Flat-field
-    options["flatfield"]["sub_region"] = SUB_REGION
-    options["flatfield"]["use_opencl"] = False
-    options["flatfield"]["use_cuda"] = True
-    F = FlatFieldComponent(d_radios, options["flatfield"], dataset_infos, logger=logger)
-    F.execute()
-
-    # CCD filter
-    options["ccd_correction"]["use_cuda"] = True
-    options["ccd_correction"]["use_opencl"] = False
-    CCD = CCDFilterComponent(d_radios, options["ccd_correction"], dataset_infos, logger=logger)
-    CCD.execute()
-
-
-    # Phase retrieval
-    options["phase"]["use_cuda"] = True
-    options["phase"]["use_opencl"] = False
-    P = PhaseRetrievalComponent(radios[0].shape, options["phase"], dataset_infos, logger=logger)
-    P.execute(d_radios)
-
-    # Unsharp
-    # ~ options["unsharp_mask"]["use_cuda"] = True
-    # ~ options["unsharp_mask"]["use_opencl"] = False
-    # ~ U = UnsharpMaskComponent(radios[0].shape, options["unsharp_mask"], dataset_infos, logger=logger)
-    # ~ U.execute(d_radios)
-
-    # -log()
-    options["take_log"]["use_cuda"] = True
-    options["take_log"]["use_opencl"] = False
-    L = NegativeLogComponent(d_radios, options["take_log"], dataset_infos, logger=logger)
-    L.execute()
-
-
-    # Test
-    # ~ from spire.utils import ims
-    # ~ r = d_radios.get()
-    # ~ ims(r[:, 10, :])
-
-
-    # Reconstruction
-    options["reconstruction"]["use_cuda"] = True
-
-    # Test ...
-    if 1:
-        options["reconstruction"]["start_x"] = 100
-        options["reconstruction"]["end_x"] = -100
-        options["reconstruction"]["start_y"] = 100
-        options["reconstruction"]["end_y"] = -100
-        options["reconstruction"]["start_z"] = 0
-        options["reconstruction"]["end_z"] = SUB_REGION[-1]-1
-        d_rec = garray.zeros((50, 1849, 1849), "f") # y
-    #
-    else:
-        d_rec = garray.zeros((SUB_REGION[-1], 2048, 2048), "f")
-    R = ReconstructorComponent(d_radios.shape, options["reconstruction"], dataset_infos, logger=logger)
-    R.execute(d_radios, output=d_rec)
-
-
-    # Write to file
-    rec = d_rec.get()
-    import numpy as np
-    np.save("/home/pierre/tmp/crayon/recs_nabu_vertical.npy", rec)
-
-    """from nabu.reconstruction.reconstructor_cuda import CudaReconstructor
-    R = CudaReconstructor(
-        radios.shape,
-        [0, 1],
-        vol_type="projections",
-        extra_options={"padding_mode": conf.nabu_config["reconstruction"]["padding_type"]}
-    )
-    d_recs = garray.zeros((1, 2048, 2048), "f")
-    R.reconstruct(d_radios, output=d_recs)
-    recs = d_recs.get()
-    ims(recs[0], cmap="gray")
-    # save
-    import numpy as np
-    np.save("/tmp/rec0.npy", recs[0])
-    """
-
-
-"""
-Notes
-  - If the cuda backend is used at step N, it should also be used at step N+1,
-
-
-otherwise we have to get() the data
-
-"""
-

nabu-2024.1.8/sandbox/binning_cython.py

@@ -1,37 +0,0 @@
-%%cython --compile-args=-fopenmp --link-args=-fopenmp -a
-#%%cython -a
-#cython: embedsignature=True, language_level=3, binding=True
-#cython: boundscheck=False, wraparound=False, cdivision=True, initializedcheck=False,
-## This is for developping:
-## cython: profile=True, warn.undeclared=True, warn.unused=True, warn.unused_result=False, warn.unused_arg=True
-
-import numpy as np
-from cython.parallel import prange
-"""
-def bin2_cython(float[:, ::1] img):
-    cdef:
-        float[:, ::1] res
-        int i, j, Ny, Nx
-          
-    shp = img.shape
-    Ny, Nx = (img.shape[0]//2, img.shape[1]//2)
-    res = np.zeros((Ny, Nx), dtype="f")
-    for i in prange(Ny, nogil=True): 
-        for j in range(Nx):
-            res[i, j] = 0.25 * (img[2*i, 2*j] + img[2*i+1, 2*j] + img[2*i, 2*j+1] + img[2*i+1, 2*j+1])
-    return np.asarray(res)
-"""
-
-
-def bin2_cython(unsigned short[:, ::1] img):
-    cdef:
-        float[:, ::1] res
-        int i, j, Ny, Nx
-          
-    shp = img.shape
-    Ny, Nx = (img.shape[0]//2, img.shape[1]//2)
-    res = np.zeros((Ny, Nx), dtype="f")
-    for i in prange(Ny, nogil=True): 
-        for j in range(Nx):
-            res[i, j] = 0.25 * (img[2*i, 2*j] + img[2*i+1, 2*j] + img[2*i, 2*j+1] + img[2*i+1, 2*j+1])
-    return np.asarray(res)

nabu-2024.1.8/sandbox/circ_sm.py

@@ -1,195 +0,0 @@
-import numpy as np
-from multiprocessing.shared_memory import SharedMemory
-from multiprocessing.pool import ThreadPool
-from multiprocessing import Pool
-from itertools import product
-from functools import partial
-
-# Constants
-# Miller index of reflection
-H = 0
-# Miller index of reflection
-K = 4
-# Number of unit cells per direction
-N = 32
-# Defines how many points are needed to describe a single Laue fringe (2 = Nyquist frequency)
-oversampling = 3
-
-# Radius of the crystal
-R = N/2
-
-# Maximum strain at surface
-e0 = 0.01
-# Width of the strain profile below the surface
-w = 5.
-
-# Generate real and reciprocal space coordinates
-n = np.arange(N)
-m = np.arange(N)
-h = np.arange(H-0.5, H+0.5, 1./(oversampling*N))
-k = np.arange(K-0.5, K+0.5, 1./(oversampling*N))
-
-
-# Displacement of atoms as function of the radius
-def delta(radius, crystal_radius, strain_width):
-    """Displacement of atoms as function of the radius"""
-    return 1 + np.tanh((radius - crystal_radius) / strain_width)
-
-
-
-def circ_numpy_2(N, h, k):
-    R, C = np.indices((N, N))
-    radius = np.sqrt((R - N/2.)**2 + (C - N/2.)**2)
-    support = radius**2 <= (N/2)**2
-    Delta = delta(radius, N/2, w)
-    delta_n = e0 * (n - N/2) * Delta
-    delta_n_n = n + delta_n
-    delta_m_m = delta_n_n.T
-
-    res = np.zeros((h.size, k.size))
-    for hk in product(enumerate(h), enumerate(k)):
-        (i_h, v_h), (i_k, v_k) = hk
-        res[i_h, i_k] = np.abs(np.sum(
-            support * np.exp(2j * np.pi * (v_h*delta_n_n + v_k*delta_m_m))
-        ))**2
-    return res
-
-
-def compute_image_part(my_part, N, h, k):
-    (start_x, end_x), (start_y, end_y) = my_part
-
-    # Retrieve shared arrays
-    my_s_support = SharedMemory(create=False, name="support")
-    my_support = np.ndarray((N, N), dtype=np.float64, buffer=my_s_support.buf)
-
-    my_s_delta_n_n = SharedMemory(create=False, name="delta_n_n")
-    my_delta_n_n = np.ndarray((N, N), dtype=np.float64, buffer=my_s_delta_n_n.buf)
-    my_delta_m_m = my_delta_n_n.T
-
-    my_s_result = SharedMemory(create=False, name="result")
-    my_result = np.ndarray((h.size, k.size), dtype=np.float64, buffer=my_s_result.buf)
-
-    # Compute on a partial set of reciprocal coordinates
-    for i_h, i_k in product(range(start_y, end_y), range(start_x, end_x)):
-        v_h = h[i_h]
-        v_k = k[i_k]
-        my_result[i_h, i_k] = np.abs(np.sum(my_support * np.exp(2j * np.pi * (v_h*my_delta_n_n + v_k*my_delta_m_m))))**2
-
-
-def circ_mp(N, h, k):
-    R, C = np.indices((N, N))
-    radius = np.sqrt((R - N/2.)**2 + (C - N/2.)**2)
-    # support = radius**2 <= (N/2)**2
-    Delta = delta(radius, N/2, w)
-    delta_n = e0 * (n - N/2) * Delta
-    # delta_n_n = n + delta_n
-    # delta_m_m = delta_n_n.T
-
-    parts_1D = [(0, h.size//2), (h.size//2, h.size)]
-    parts = list(product(parts_1D, parts_1D))
-
-    try:
-        # Create shared arrays
-        s_support = SharedMemory(create=True, size=N*N*np.dtype(np.float64).itemsize, name="support")
-        support = np.ndarray((N, N), dtype=np.float64, buffer=s_support.buf)
-        support[:] = radius**2 <= (N/2)**2
-
-        s_delta_n_n = SharedMemory(create=True, size=support.nbytes, name="delta_n_n")
-        delta_n_n = np.ndarray((N, N), dtype=np.float64, buffer=s_delta_n_n.buf)
-        delta_n_n[:] = n + delta_n
-
-        s_result = SharedMemory(
-            create=True, size=h.size * k.size * np.dtype(np.float64).itemsize, name="result"
-        )
-        result = np.ndarray((h.size, k.size), dtype=np.float64, buffer=s_result.buf)
-        result[:] = 0.
-
-        # Dispatch computations
-        with Pool(4) as p:
-            p.map(partial(compute_image_part, N=N, h=h, k=k), parts)
-
-    finally:
-        res = result.copy()
-        for s_array in [s_support, s_delta_n_n, s_result]:
-            s_array.unlink()
-
-    return res
-
-
-
-
-
-
-
-def circ_mt(N, h, k):
-    R, C = np.indices((N, N))
-    radius = np.sqrt((R - N/2.)**2 + (C - N/2.)**2)
-    support = radius**2 <= (N/2)**2
-    Delta = delta(radius, N/2, w)
-    delta_n = e0 * (n - N/2) * Delta
-    delta_n_n = n + delta_n
-    delta_m_m = delta_n_n.T
-
-    parts_1D = [(0, h.size//2), (h.size//2, h.size)]
-    parts = list(product(parts_1D, parts_1D))
-
-    result = np.zeros((h.size, k.size), dtype=np.float64)
-
-    def compute_image_part(my_part):
-        (start_x, end_x), (start_y, end_y) = my_part
-        # Compute on a partial set of reciprocal coordinates
-        for i_h, i_k in product(range(start_y, end_y), range(start_x, end_x)):
-            v_h = h[i_h]
-            v_k = k[i_k]
-            result[i_h, i_k] = np.abs(np.sum(
-                support * np.exp(2j * np.pi * (v_h*delta_n_n + v_k*delta_m_m))
-            ))**2
-
-    with ThreadPool(4) as tp:
-        tp.map(compute_image_part, parts)
-
-
-    return result
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-from matplotlib.pyplot import subplots
-from matplotlib.colors import LogNorm
-def display(result):
-    "Display the array"
-    fig, ax = subplots()
-    fig.suptitle("Bragg peak")
-    ax.imshow(result.T, extent=(h.min(), h.max(), k.min(), k.max()), norm=LogNorm(), origin = 'lower')
-    ax.set_xlabel('H');
-    ax.set_ylabel('K')
-    ax.set_title("Crystal")
-
-
-
-def circ_einsum(N, h, k):
-    R, C = np.indices((N, N))
-    radius = np.sqrt((R - N/2.)**2 + (C - N/2.)**2)
-    support = radius**2 <= (N/2)**2
-    Delta = delta(radius, N/2, w)
-    delta_n = e0 * (n - N/2) * Delta
-    delta_n_n = n + delta_n
-    delta_m_m = delta_n_n.T
-
-    tmp1 = np.exp(2j*pi*np.einsum("i,jk", h, delta_n_n))
-    tmp2 = np.exp(2j*pi*np.einsum("i,jk", k, delta_m_m))
-    res = np.abs(np.einsum("jk,ijk,ljk->il", support, tmp1, tmp2))**2
-
-    return res