nabu 2024.1.0rc3__py3-none-any.whl → 2024.1.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/build/lib/doc/conf.py +137 -0
- build/lib/build/lib/doc/create_conf_doc.py +38 -0
- build/lib/build/lib/doc/get_mathjax.py +34 -0
- build/lib/build/lib/nabu/__init__.py +17 -0
- build/lib/build/lib/nabu/app/__init__.py +0 -0
- build/lib/build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/build/lib/nabu/app/generate_header.py +261 -0
- build/lib/build/lib/nabu/app/histogram.py +197 -0
- build/lib/build/lib/nabu/app/multicor.py +90 -0
- build/lib/build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/build/lib/nabu/app/rotate.py +162 -0
- build/lib/build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/build/lib/nabu/app/stitching.py +85 -0
- build/lib/build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/build/lib/nabu/app/utils.py +33 -0
- build/lib/build/lib/nabu/app/validator.py +110 -0
- build/lib/build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/build/lib/nabu/cuda/fft.py +4 -0
- build/lib/build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/build/lib/nabu/cuda/padding.py +6 -0
- build/lib/build/lib/nabu/cuda/processing.py +75 -0
- build/lib/build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/build/lib/nabu/cuda/utils.py +331 -0
- build/lib/build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/build/lib/nabu/estimation/focus.py +362 -0
- build/lib/build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/build/lib/nabu/estimation/translation.py +193 -0
- build/lib/build/lib/nabu/estimation/utils.py +39 -0
- build/lib/build/lib/nabu/io/__init__.py +2 -0
- build/lib/build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/build/lib/nabu/io/reader.py +628 -0
- build/lib/build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/build/lib/nabu/io/utils.py +263 -0
- build/lib/build/lib/nabu/io/writer.py +892 -0
- build/lib/build/lib/nabu/misc/__init__.py +0 -0
- build/lib/build/lib/nabu/misc/binning.py +98 -0
- build/lib/build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/build/lib/nabu/misc/filters.py +24 -0
- build/lib/build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/build/lib/nabu/misc/histogram.py +6 -0
- build/lib/build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/build/lib/nabu/misc/padding.py +83 -0
- build/lib/build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/build/lib/nabu/misc/rotation.py +6 -0
- build/lib/build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/build/lib/nabu/misc/transpose.py +6 -0
- build/lib/build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/build/lib/nabu/misc/utils.py +117 -0
- build/lib/build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/build/lib/nabu/opencl/fft.py +6 -0
- build/lib/build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/build/lib/nabu/opencl/padding.py +6 -0
- build/lib/build/lib/nabu/opencl/processing.py +62 -0
- build/lib/build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/opencl/utils.py +292 -0
- build/lib/build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/build/lib/nabu/pipeline/config.py +265 -0
- build/lib/build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/build/lib/nabu/pipeline/params.py +163 -0
- build/lib/build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/build/lib/nabu/preproc/phase.py +425 -0
- build/lib/build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/build/lib/nabu/preproc/shift.py +95 -0
- build/lib/build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/build/lib/nabu/processing/__init__.py +0 -0
- build/lib/build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/build/lib/nabu/processing/histogram.py +286 -0
- build/lib/build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/build/lib/nabu/processing/muladd.py +29 -0
- build/lib/build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/build/lib/nabu/processing/rotation.py +63 -0
- build/lib/build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/build/lib/nabu/processing/transpose.py +126 -0
- build/lib/build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/build/lib/nabu/resources/__init__.py +0 -0
- build/lib/build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/build/lib/nabu/resources/cor.py +5 -0
- build/lib/build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/build/lib/nabu/resources/gpu.py +169 -0
- build/lib/build/lib/nabu/resources/logger.py +130 -0
- build/lib/build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/build/lib/nabu/resources/utils.py +174 -0
- build/lib/build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/build/lib/nabu/stitching/config.py +1286 -0
- build/lib/build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/build/lib/nabu/stitching/utils.py +563 -0
- build/lib/build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/build/lib/nabu/tests.py +51 -0
- build/lib/build/lib/nabu/testutils.py +375 -0
- build/lib/build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/build/lib/nabu/utils.py +836 -0
- build/lib/doc/conf.py +137 -0
- build/lib/doc/create_conf_doc.py +38 -0
- build/lib/doc/get_mathjax.py +34 -0
- build/lib/nabu/__init__.py +17 -0
- build/lib/nabu/app/__init__.py +0 -0
- build/lib/nabu/app/bootstrap.py +88 -0
- build/lib/nabu/app/bootstrap_stitching.py +69 -0
- build/lib/nabu/app/cast_volume.py +272 -0
- build/lib/nabu/app/cli_configs.py +626 -0
- build/lib/nabu/app/compare_volumes.py +94 -0
- build/lib/nabu/app/composite_cor.py +144 -0
- build/lib/nabu/app/correct_rot.py +70 -0
- build/lib/nabu/app/create_distortion_map_from_poly.py +153 -0
- build/lib/nabu/app/diag_to_pix.py +358 -0
- build/lib/nabu/app/diag_to_rot.py +449 -0
- build/lib/nabu/app/double_flatfield.py +143 -0
- build/lib/nabu/app/generate_header.py +261 -0
- build/lib/nabu/app/histogram.py +197 -0
- build/lib/nabu/app/multicor.py +90 -0
- build/lib/nabu/app/nx_z_splitter.py +125 -0
- build/lib/nabu/app/parse_reconstruction_log.py +151 -0
- build/lib/nabu/app/prepare_weights_double.py +163 -0
- build/lib/nabu/app/reconstruct.py +120 -0
- build/lib/nabu/app/reconstruct_helical.py +116 -0
- build/lib/nabu/app/reduce_dark_flat.py +181 -0
- build/lib/nabu/app/rotate.py +162 -0
- build/lib/nabu/app/shrink_dataset.py +98 -0
- build/lib/nabu/app/stitching.py +85 -0
- build/lib/nabu/app/tests/test_reduce_dark_flat.py +78 -0
- build/lib/nabu/app/utils.py +33 -0
- build/lib/nabu/app/validator.py +110 -0
- build/lib/nabu/cuda/__init__.py +0 -0
- build/lib/nabu/cuda/convolution.py +8 -0
- build/lib/nabu/cuda/fft.py +4 -0
- build/lib/nabu/cuda/kernel.py +97 -0
- build/lib/nabu/cuda/medfilt.py +6 -0
- build/lib/nabu/cuda/padding.py +6 -0
- build/lib/nabu/cuda/processing.py +75 -0
- build/lib/nabu/cuda/tests/__init__.py +1 -0
- build/lib/nabu/cuda/utils.py +331 -0
- build/lib/nabu/estimation/__init__.py +12 -0
- build/lib/nabu/estimation/alignment.py +578 -0
- build/lib/nabu/estimation/cor.py +1621 -0
- build/lib/nabu/estimation/cor_sino.py +183 -0
- build/lib/nabu/estimation/distortion.py +112 -0
- build/lib/nabu/estimation/focus.py +362 -0
- build/lib/nabu/estimation/tests/__init__.py +0 -0
- build/lib/nabu/estimation/tests/test_alignment.py +70 -0
- build/lib/nabu/estimation/tests/test_cor.py +472 -0
- build/lib/nabu/estimation/tests/test_focus.py +105 -0
- build/lib/nabu/estimation/tests/test_tilt.py +41 -0
- build/lib/nabu/estimation/tests/test_translation.py +78 -0
- build/lib/nabu/estimation/tilt.py +214 -0
- build/lib/nabu/estimation/translation.py +193 -0
- build/lib/nabu/estimation/utils.py +39 -0
- build/lib/nabu/io/__init__.py +2 -0
- build/lib/nabu/io/cast_volume.py +371 -0
- build/lib/nabu/io/detector_distortion.py +305 -0
- build/lib/nabu/io/reader.py +628 -0
- build/lib/nabu/io/reader_helical.py +114 -0
- build/lib/nabu/io/tests/__init__.py +0 -0
- build/lib/nabu/io/tests/test_cast_volume.py +339 -0
- build/lib/nabu/io/tests/test_detector_distortion.py +178 -0
- build/lib/nabu/io/tests/test_writers.py +187 -0
- build/lib/nabu/io/tiffwriter_zmm.py +99 -0
- build/lib/nabu/io/utils.py +263 -0
- build/lib/nabu/io/writer.py +892 -0
- build/lib/nabu/misc/__init__.py +0 -0
- build/lib/nabu/misc/binning.py +98 -0
- build/lib/nabu/misc/fftshift.py +6 -0
- build/lib/nabu/misc/filters.py +24 -0
- build/lib/nabu/misc/fourier_filters.py +152 -0
- build/lib/nabu/misc/histogram.py +6 -0
- build/lib/nabu/misc/histogram_cuda.py +8 -0
- build/lib/nabu/misc/kernel_base.py +4 -0
- build/lib/nabu/misc/padding.py +83 -0
- build/lib/nabu/misc/padding_base.py +6 -0
- build/lib/nabu/misc/processing_base.py +4 -0
- build/lib/nabu/misc/rotation.py +6 -0
- build/lib/nabu/misc/rotation_cuda.py +6 -0
- build/lib/nabu/misc/tests/__init__.py +1 -0
- build/lib/nabu/misc/tests/test_binning.py +48 -0
- build/lib/nabu/misc/tests/test_interpolation.py +70 -0
- build/lib/nabu/misc/transpose.py +6 -0
- build/lib/nabu/misc/unsharp.py +4 -0
- build/lib/nabu/misc/unsharp_cuda.py +6 -0
- build/lib/nabu/misc/unsharp_opencl.py +8 -0
- build/lib/nabu/misc/utils.py +117 -0
- build/lib/nabu/opencl/__init__.py +0 -0
- build/lib/nabu/opencl/fft.py +6 -0
- build/lib/nabu/opencl/kernel.py +135 -0
- build/lib/nabu/opencl/memcpy.py +35 -0
- build/lib/nabu/opencl/padding.py +6 -0
- build/lib/nabu/opencl/processing.py +62 -0
- build/lib/nabu/opencl/tests/__init__.py +0 -0
- build/lib/nabu/opencl/utils.py +292 -0
- build/lib/nabu/pipeline/__init__.py +0 -0
- build/lib/nabu/pipeline/config.py +265 -0
- build/lib/nabu/pipeline/config_validators.py +589 -0
- build/lib/nabu/pipeline/datadump.py +164 -0
- build/lib/nabu/pipeline/dataset_validator.py +211 -0
- build/lib/nabu/pipeline/detector_distortion_provider.py +20 -0
- build/lib/nabu/pipeline/estimators.py +1012 -0
- build/lib/nabu/pipeline/fallback_utils.py +149 -0
- build/lib/nabu/pipeline/fullfield/__init__.py +0 -0
- build/lib/nabu/pipeline/fullfield/chunked.py +849 -0
- build/lib/nabu/pipeline/fullfield/chunked_cuda.py +152 -0
- build/lib/nabu/pipeline/fullfield/computations.py +260 -0
- build/lib/nabu/pipeline/fullfield/dataset_validator.py +63 -0
- build/lib/nabu/pipeline/fullfield/nabu_config.py +590 -0
- build/lib/nabu/pipeline/fullfield/processconfig.py +825 -0
- build/lib/nabu/pipeline/fullfield/reconstruction.py +851 -0
- build/lib/nabu/pipeline/helical/__init__.py +0 -0
- build/lib/nabu/pipeline/helical/dataset_validator.py +19 -0
- build/lib/nabu/pipeline/helical/fbp.py +147 -0
- build/lib/nabu/pipeline/helical/filtering.py +246 -0
- build/lib/nabu/pipeline/helical/gridded_accumulator.py +575 -0
- build/lib/nabu/pipeline/helical/helical_chunked_regridded.py +1670 -0
- build/lib/nabu/pipeline/helical/helical_chunked_regridded_cuda.py +119 -0
- build/lib/nabu/pipeline/helical/helical_reconstruction.py +598 -0
- build/lib/nabu/pipeline/helical/helical_utils.py +42 -0
- build/lib/nabu/pipeline/helical/nabu_config.py +175 -0
- build/lib/nabu/pipeline/helical/processconfig.py +62 -0
- build/lib/nabu/pipeline/helical/span_strategy.py +483 -0
- build/lib/nabu/pipeline/helical/tests/__init__.py +0 -0
- build/lib/nabu/pipeline/helical/tests/test_accumulator.py +157 -0
- build/lib/nabu/pipeline/helical/tests/test_pipeline_elements_full.py +355 -0
- build/lib/nabu/pipeline/helical/tests/test_strategy.py +61 -0
- build/lib/nabu/pipeline/helical/utils.py +51 -0
- build/lib/nabu/pipeline/helical/weight_balancer.py +101 -0
- build/lib/nabu/pipeline/params.py +163 -0
- build/lib/nabu/pipeline/processconfig.py +210 -0
- build/lib/nabu/pipeline/tests/test_chunk_reader.py +78 -0
- build/lib/nabu/pipeline/tests/test_estimators.py +149 -0
- build/lib/nabu/pipeline/utils.py +116 -0
- build/lib/nabu/pipeline/writer.py +193 -0
- build/lib/nabu/pipeline/xrdct/__init__.py +0 -0
- build/lib/nabu/preproc/__init__.py +7 -0
- build/lib/nabu/preproc/alignment.py +11 -0
- build/lib/nabu/preproc/ccd.py +161 -0
- build/lib/nabu/preproc/ccd_cuda.py +169 -0
- build/lib/nabu/preproc/ctf.py +361 -0
- build/lib/nabu/preproc/ctf_cuda.py +142 -0
- build/lib/nabu/preproc/distortion.py +92 -0
- build/lib/nabu/preproc/double_flatfield.py +212 -0
- build/lib/nabu/preproc/double_flatfield_cuda.py +174 -0
- build/lib/nabu/preproc/double_flatfield_variable_region.py +68 -0
- build/lib/nabu/preproc/flatfield.py +448 -0
- build/lib/nabu/preproc/flatfield_cuda.py +144 -0
- build/lib/nabu/preproc/flatfield_variable_region.py +79 -0
- build/lib/nabu/preproc/phase.py +425 -0
- build/lib/nabu/preproc/phase_cuda.py +143 -0
- build/lib/nabu/preproc/shift.py +95 -0
- build/lib/nabu/preproc/shift_cuda.py +102 -0
- build/lib/nabu/preproc/tests/__init__.py +1 -0
- build/lib/nabu/preproc/tests/test_ccd_corr.py +59 -0
- build/lib/nabu/preproc/tests/test_ctf.py +250 -0
- build/lib/nabu/preproc/tests/test_double_flatfield.py +86 -0
- build/lib/nabu/preproc/tests/test_flatfield.py +626 -0
- build/lib/nabu/preproc/tests/test_paganin.py +67 -0
- build/lib/nabu/preproc/tests/test_vshift.py +73 -0
- build/lib/nabu/processing/__init__.py +0 -0
- build/lib/nabu/processing/convolution_cuda.py +375 -0
- build/lib/nabu/processing/fft_base.py +163 -0
- build/lib/nabu/processing/fft_cuda.py +256 -0
- build/lib/nabu/processing/fft_opencl.py +54 -0
- build/lib/nabu/processing/fftshift.py +134 -0
- build/lib/nabu/processing/histogram.py +286 -0
- build/lib/nabu/processing/histogram_cuda.py +103 -0
- build/lib/nabu/processing/kernel_base.py +126 -0
- build/lib/nabu/processing/medfilt_cuda.py +159 -0
- build/lib/nabu/processing/muladd.py +29 -0
- build/lib/nabu/processing/muladd_cuda.py +68 -0
- build/lib/nabu/processing/padding_base.py +71 -0
- build/lib/nabu/processing/padding_cuda.py +75 -0
- build/lib/nabu/processing/padding_opencl.py +77 -0
- build/lib/nabu/processing/processing_base.py +123 -0
- build/lib/nabu/processing/roll_opencl.py +64 -0
- build/lib/nabu/processing/rotation.py +63 -0
- build/lib/nabu/processing/rotation_cuda.py +66 -0
- build/lib/nabu/processing/tests/__init__.py +0 -0
- build/lib/nabu/processing/tests/test_fft.py +268 -0
- build/lib/nabu/processing/tests/test_fftshift.py +71 -0
- build/lib/nabu/processing/tests/test_histogram.py +55 -0
- build/lib/nabu/processing/tests/test_medfilt.py +77 -0
- build/lib/nabu/processing/tests/test_muladd.py +54 -0
- build/lib/nabu/processing/tests/test_padding.py +247 -0
- build/lib/nabu/processing/tests/test_roll.py +63 -0
- build/lib/nabu/processing/tests/test_rotation.py +83 -0
- build/lib/nabu/processing/tests/test_transpose.py +72 -0
- build/lib/nabu/processing/tests/test_unsharp.py +106 -0
- build/lib/nabu/processing/transpose.py +126 -0
- build/lib/nabu/processing/unsharp.py +79 -0
- build/lib/nabu/processing/unsharp_cuda.py +53 -0
- build/lib/nabu/processing/unsharp_opencl.py +75 -0
- build/lib/nabu/reconstruction/__init__.py +3 -0
- build/lib/nabu/reconstruction/cone.py +241 -0
- build/lib/nabu/reconstruction/fbp.py +124 -0
- build/lib/nabu/reconstruction/fbp_base.py +391 -0
- build/lib/nabu/reconstruction/fbp_opencl.py +78 -0
- build/lib/nabu/reconstruction/filtering.py +220 -0
- build/lib/nabu/reconstruction/filtering_cuda.py +103 -0
- build/lib/nabu/reconstruction/filtering_opencl.py +111 -0
- build/lib/nabu/reconstruction/projection.py +240 -0
- build/lib/nabu/reconstruction/reconstructor.py +174 -0
- build/lib/nabu/reconstruction/reconstructor_cuda.py +44 -0
- build/lib/nabu/reconstruction/rings.py +248 -0
- build/lib/nabu/reconstruction/rings_cuda.py +322 -0
- build/lib/nabu/reconstruction/sinogram.py +467 -0
- build/lib/nabu/reconstruction/sinogram_cuda.py +250 -0
- build/lib/nabu/reconstruction/sinogram_opencl.py +40 -0
- build/lib/nabu/reconstruction/tests/__init__.py +1 -0
- build/lib/nabu/reconstruction/tests/test_cone.py +308 -0
- build/lib/nabu/reconstruction/tests/test_deringer.py +220 -0
- build/lib/nabu/reconstruction/tests/test_fbp.py +252 -0
- build/lib/nabu/reconstruction/tests/test_filtering.py +133 -0
- build/lib/nabu/reconstruction/tests/test_halftomo.py +100 -0
- build/lib/nabu/reconstruction/tests/test_projector.py +164 -0
- build/lib/nabu/reconstruction/tests/test_reconstructor.py +93 -0
- build/lib/nabu/reconstruction/tests/test_sino_normalization.py +85 -0
- build/lib/nabu/resources/__init__.py +0 -0
- build/lib/nabu/resources/cli/__init__.py +0 -0
- build/lib/nabu/resources/cor.py +5 -0
- build/lib/nabu/resources/dataset_analyzer.py +469 -0
- build/lib/nabu/resources/gpu.py +169 -0
- build/lib/nabu/resources/logger.py +130 -0
- build/lib/nabu/resources/nxflatfield.py +217 -0
- build/lib/nabu/resources/templates/__init__.py +0 -0
- build/lib/nabu/resources/tests/__init__.py +0 -0
- build/lib/nabu/resources/tests/test_nxflatfield.py +104 -0
- build/lib/nabu/resources/tests/test_units.py +47 -0
- build/lib/nabu/resources/utils.py +174 -0
- build/lib/nabu/stitching/__init__.py +0 -0
- build/lib/nabu/stitching/alignment.py +184 -0
- build/lib/nabu/stitching/config.py +1286 -0
- build/lib/nabu/stitching/definitions.py +6 -0
- build/lib/nabu/stitching/frame_composition.py +155 -0
- build/lib/nabu/stitching/overlap.py +373 -0
- build/lib/nabu/stitching/sample_normalization.py +60 -0
- build/lib/nabu/stitching/slurm_utils.py +204 -0
- build/lib/nabu/stitching/tests/__init__.py +0 -0
- build/lib/nabu/stitching/tests/test_alignment.py +99 -0
- build/lib/nabu/stitching/tests/test_config.py +226 -0
- build/lib/nabu/stitching/tests/test_frame_composition.py +180 -0
- build/lib/nabu/stitching/tests/test_overlap.py +98 -0
- build/lib/nabu/stitching/tests/test_sample_normalization.py +49 -0
- build/lib/nabu/stitching/tests/test_slurm_utils.py +146 -0
- build/lib/nabu/stitching/tests/test_utils.py +21 -0
- build/lib/nabu/stitching/tests/test_z_stitching.py +1183 -0
- build/lib/nabu/stitching/utils.py +563 -0
- build/lib/nabu/stitching/z_stitching.py +2279 -0
- build/lib/nabu/tests.py +51 -0
- build/lib/nabu/testutils.py +375 -0
- build/lib/nabu/thirdparty/__init__.py +0 -0
- build/lib/nabu/thirdparty/algotom_convert_sino.py +276 -0
- build/lib/nabu/thirdparty/pore3d_deringer_munch.py +108 -0
- build/lib/nabu/thirdparty/tomocupy_remove_stripe.py +586 -0
- build/lib/nabu/thirdparty/tomopy_phase.py +239 -0
- build/lib/nabu/thirdparty/tomwer_load_flats_darks.py +162 -0
- build/lib/nabu/utils.py +836 -0
- nabu/__init__.py +1 -1
- nabu/estimation/cor.py +2 -2
- nabu/io/cast_volume.py +10 -2
- nabu/pipeline/estimators.py +13 -7
- nabu/resources/dataset_analyzer.py +15 -6
- nabu/stitching/config.py +2 -2
- nabu/stitching/z_stitching.py +3 -1
- {nabu-2024.1.0rc3.dist-info → nabu-2024.1.1.dist-info}/METADATA +1 -1
- nabu-2024.1.1.dist-info/RECORD +2375 -0
- nabu-2024.1.1.dist-info/top_level.txt +4 -0
- doc/doc_config.py +0 -32
- nabu-2024.1.0rc3.dist-info/RECORD +0 -296
- nabu-2024.1.0rc3.dist-info/top_level.txt +0 -2
- {nabu-2024.1.0rc3.dist-info → nabu-2024.1.1.dist-info}/LICENSE +0 -0
- {nabu-2024.1.0rc3.dist-info → nabu-2024.1.1.dist-info}/WHEEL +0 -0
- {nabu-2024.1.0rc3.dist-info → nabu-2024.1.1.dist-info}/entry_points.txt +0 -0
@@ -0,0 +1,130 @@
|
|
1
|
+
import logging
|
2
|
+
import logging.config
|
3
|
+
|
4
|
+
|
5
|
+
class Logger(object):
|
6
|
+
def __init__(self, loggername, level="DEBUG", logfile="logger.log", console=True):
|
7
|
+
"""
|
8
|
+
Configure a Logger object.
|
9
|
+
|
10
|
+
Parameters
|
11
|
+
-----------
|
12
|
+
loggername: str
|
13
|
+
Logger name.
|
14
|
+
level: str, optional
|
15
|
+
Logging level. Can be "debug", "info", "warning", "error", "critical".
|
16
|
+
Default is "debug".
|
17
|
+
logfile: str, optional
|
18
|
+
File where the logs are written.
|
19
|
+
If set to None, the logs are not written in a file.
|
20
|
+
Default is "logger.log".
|
21
|
+
console: bool, optional
|
22
|
+
If set to True, the logs are (also) written in stdout/stderr.
|
23
|
+
Default is True.
|
24
|
+
"""
|
25
|
+
self.loggername = loggername
|
26
|
+
self.level = level
|
27
|
+
self.logfile = logfile
|
28
|
+
self.console = console
|
29
|
+
self._configure_logger()
|
30
|
+
|
31
|
+
def _configure_logger(self):
|
32
|
+
conf = self._get_default_config_dict()
|
33
|
+
for handler in conf["handlers"].keys():
|
34
|
+
conf["handlers"][handler]["level"] = self.level.upper()
|
35
|
+
conf["loggers"][self.loggername]["level"] = self.level.upper()
|
36
|
+
if not (self.console):
|
37
|
+
conf["loggers"][self.loggername]["handlers"].remove("console")
|
38
|
+
|
39
|
+
self.config = conf
|
40
|
+
logging.config.dictConfig(conf)
|
41
|
+
self.logger = logging.getLogger(self.loggername)
|
42
|
+
|
43
|
+
def _get_default_config_dict(self):
|
44
|
+
conf = {
|
45
|
+
"version": 1,
|
46
|
+
"formatters": {
|
47
|
+
"default": {"format": "%(asctime)s - %(levelname)s - %(message)s", "datefmt": "%d-%m-%Y %H:%M:%S"},
|
48
|
+
"console": {"format": "%(message)s"},
|
49
|
+
},
|
50
|
+
"handlers": {
|
51
|
+
"console": {
|
52
|
+
"level": "DEBUG",
|
53
|
+
"class": "logging.StreamHandler",
|
54
|
+
"formatter": "console",
|
55
|
+
"stream": "ext://sys.stdout",
|
56
|
+
},
|
57
|
+
"file": {
|
58
|
+
"level": "DEBUG",
|
59
|
+
"class": "logging.FileHandler",
|
60
|
+
"formatter": "default",
|
61
|
+
"filename": self.logfile,
|
62
|
+
},
|
63
|
+
},
|
64
|
+
"loggers": {
|
65
|
+
self.loggername: {
|
66
|
+
"level": "DEBUG",
|
67
|
+
"handlers": ["console", "file"],
|
68
|
+
# This logger inherits from root logger - dont propagate !
|
69
|
+
"propagate": False,
|
70
|
+
}
|
71
|
+
},
|
72
|
+
"disable_existing_loggers": False,
|
73
|
+
}
|
74
|
+
return conf
|
75
|
+
|
76
|
+
def info(self, msg):
|
77
|
+
return self.logger.info(msg)
|
78
|
+
|
79
|
+
def debug(self, msg):
|
80
|
+
return self.logger.debug(msg)
|
81
|
+
|
82
|
+
def warning(self, msg):
|
83
|
+
return self.logger.warning(msg)
|
84
|
+
|
85
|
+
warn = warning
|
86
|
+
|
87
|
+
def error(self, msg):
|
88
|
+
return self.logger.error(msg)
|
89
|
+
|
90
|
+
def fatal(self, msg):
|
91
|
+
return self.logger.fatal(msg)
|
92
|
+
|
93
|
+
def critical(self, msg):
|
94
|
+
return self.logger.critical(msg)
|
95
|
+
|
96
|
+
|
97
|
+
def LoggerOrPrint(logger):
|
98
|
+
"""
|
99
|
+
Logger that is either a "true" logger object, or a fall-back to "print".
|
100
|
+
"""
|
101
|
+
if logger is None:
|
102
|
+
return PrinterLogger()
|
103
|
+
return logger
|
104
|
+
|
105
|
+
|
106
|
+
class PrinterLogger(object):
|
107
|
+
def __init__(self):
|
108
|
+
methods = [
|
109
|
+
"debug",
|
110
|
+
"warn",
|
111
|
+
"warning",
|
112
|
+
"info",
|
113
|
+
"error",
|
114
|
+
"fatal",
|
115
|
+
"critical",
|
116
|
+
]
|
117
|
+
for method in methods:
|
118
|
+
self.__setattr__(method, print)
|
119
|
+
|
120
|
+
|
121
|
+
LogLevel = {
|
122
|
+
"notset": logging.NOTSET,
|
123
|
+
"debug": logging.DEBUG,
|
124
|
+
"info": logging.INFO,
|
125
|
+
"warn": logging.WARN,
|
126
|
+
"warning": logging.WARNING,
|
127
|
+
"error": logging.ERROR,
|
128
|
+
"critical": logging.CRITICAL,
|
129
|
+
"fatal": logging.FATAL,
|
130
|
+
}
|
@@ -0,0 +1,217 @@
|
|
1
|
+
import os
|
2
|
+
import numpy as np
|
3
|
+
from silx.io.url import DataUrl
|
4
|
+
from tomoscan.io import HDF5File
|
5
|
+
from tomoscan.esrf.scan.nxtomoscan import NXtomoScan
|
6
|
+
from ..utils import check_supported, is_writeable
|
7
|
+
|
8
|
+
|
9
|
+
def get_frame_possible_urls(dataset_info, user_dir, output_dir, frame_type):
|
10
|
+
"""
|
11
|
+
Return a list with the possible location of reduced dark/flat frames.
|
12
|
+
|
13
|
+
Parameters
|
14
|
+
----------
|
15
|
+
dataset_info: DatasetAnalyzer object
|
16
|
+
DatasetAnalyzer object: data structure containing information on the parsed dataset
|
17
|
+
user_dir: str or None
|
18
|
+
User-provided directory location for the reduced frames.
|
19
|
+
output_dir: str or None
|
20
|
+
Output processing directory
|
21
|
+
frame_type: str
|
22
|
+
Frame type, can be "flats" or "darks".
|
23
|
+
"""
|
24
|
+
check_supported(frame_type, ["flats", "darks"], "frame type")
|
25
|
+
|
26
|
+
h5scan = dataset_info.dataset_scanner # tomoscan object
|
27
|
+
if frame_type == "flats":
|
28
|
+
dataurl_default_template = h5scan.REDUCED_FLATS_DATAURLS[0]
|
29
|
+
else:
|
30
|
+
dataurl_default_template = h5scan.REDUCED_DARKS_DATAURLS[0]
|
31
|
+
|
32
|
+
def make_dataurl(dirname):
|
33
|
+
# The template formatting should be done by tomoscan in principle, but this complicates logging.
|
34
|
+
rel_file_path = dataurl_default_template.file_path().format(
|
35
|
+
scan_prefix=dataset_info.dataset_scanner.get_dataset_basename()
|
36
|
+
)
|
37
|
+
return DataUrl(
|
38
|
+
file_path=os.path.join(dirname, rel_file_path),
|
39
|
+
data_path=dataurl_default_template.data_path(),
|
40
|
+
data_slice=dataurl_default_template.data_slice(), # not sure if needed
|
41
|
+
scheme="silx",
|
42
|
+
)
|
43
|
+
|
44
|
+
urls = {"user": None, "dataset": None, "output": None}
|
45
|
+
|
46
|
+
if user_dir is not None:
|
47
|
+
urls["user"] = make_dataurl(user_dir)
|
48
|
+
|
49
|
+
# tomoscan.esrf.scan.hdf5scan.REDUCED_{DARKS|FLATS}_DATAURLS.file_path() is a relative path
|
50
|
+
# Create a absolute path instead
|
51
|
+
urls["dataset"] = make_dataurl(os.path.dirname(h5scan.master_file))
|
52
|
+
|
53
|
+
if output_dir is not None:
|
54
|
+
urls["output"] = make_dataurl(output_dir)
|
55
|
+
|
56
|
+
return urls
|
57
|
+
|
58
|
+
|
59
|
+
def get_metadata_url(url, frame_type):
|
60
|
+
"""
|
61
|
+
Return the url of the metadata stored alongside flats/darks
|
62
|
+
"""
|
63
|
+
check_supported(frame_type, ["flats", "darks"], "frame type")
|
64
|
+
template_url = getattr(NXtomoScan, "REDUCED_%s_METADATAURLS" % frame_type.upper())[0]
|
65
|
+
return DataUrl(
|
66
|
+
file_path=url.file_path(),
|
67
|
+
data_path=template_url.data_path(),
|
68
|
+
scheme="silx",
|
69
|
+
)
|
70
|
+
|
71
|
+
|
72
|
+
def tomoscan_load_reduced_frames(dataset_info, frame_type, url):
|
73
|
+
tomoscan_method = getattr(dataset_info.dataset_scanner, "load_reduced_%s" % frame_type)
|
74
|
+
return tomoscan_method(
|
75
|
+
inputs_urls=[url],
|
76
|
+
return_as_url=True,
|
77
|
+
return_info=True,
|
78
|
+
metadata_input_urls=[get_metadata_url(url, frame_type)],
|
79
|
+
)
|
80
|
+
|
81
|
+
|
82
|
+
def tomoscan_save_reduced_frames(dataset_info, frame_type, url, frames, info):
|
83
|
+
tomoscan_method = getattr(dataset_info.dataset_scanner, "save_reduced_%s" % frame_type)
|
84
|
+
kwargs = {"%s_infos" % frame_type: info, "overwrite": True}
|
85
|
+
return tomoscan_method(
|
86
|
+
frames, output_urls=[url], metadata_output_urls=[get_metadata_url(url, frame_type)], **kwargs
|
87
|
+
)
|
88
|
+
|
89
|
+
|
90
|
+
# pylint: disable=E1136
|
91
|
+
def update_dataset_info_flats_darks(dataset_info, flatfield_mode, output_dir=None, darks_flats_dir=None):
|
92
|
+
"""
|
93
|
+
Update a DatasetAnalyzer object with reduced flats/darks (hereafter "reduced frames").
|
94
|
+
|
95
|
+
How the reduced frames are loaded/computed/saved will depend on the "flatfield_mode" parameter.
|
96
|
+
|
97
|
+
The principle is the following:
|
98
|
+
(1) Attempt at loading already-computed reduced frames (XXX_darks.h5 and XXX_flats.h5):
|
99
|
+
- First check files in the user-defined directory 'darks_flats_dir'
|
100
|
+
- Then try to load from files located alongside the .nx dataset (dataset directory)
|
101
|
+
- Then try to load from output_dir, if provided
|
102
|
+
(2) If loading fails, or flatfield_mode == "force_compute", compute the reduced frames.
|
103
|
+
(3) Save these reduced frames
|
104
|
+
- Save in darks_flats_dir, if provided by user
|
105
|
+
- Otherwise, save in the data directory (next to the .nx file), if write access OK
|
106
|
+
- Otherwise, save in output directory
|
107
|
+
"""
|
108
|
+
if flatfield_mode is False:
|
109
|
+
return
|
110
|
+
logger = dataset_info.logger
|
111
|
+
frames_types = ["darks", "flats"]
|
112
|
+
|
113
|
+
reduced_frames_urls = {}
|
114
|
+
for frame_type in frames_types:
|
115
|
+
reduced_frames_urls[frame_type] = get_frame_possible_urls(dataset_info, darks_flats_dir, output_dir, frame_type)
|
116
|
+
|
117
|
+
reduced_frames = dict.fromkeys(frames_types, None)
|
118
|
+
|
119
|
+
#
|
120
|
+
# Try to load frames
|
121
|
+
#
|
122
|
+
def load_reduced_frame(url, frame_type, frames_loaded, reduced_frames):
|
123
|
+
if frames_loaded[frame_type]:
|
124
|
+
return
|
125
|
+
frames, info = tomoscan_load_reduced_frames(dataset_info, frame_type, url)
|
126
|
+
if frames not in (None, {}):
|
127
|
+
dataset_info.logger.info("Loaded %s from %s" % (frame_type, url.file_path()))
|
128
|
+
frames_loaded[frame_type] = True
|
129
|
+
reduced_frames[frame_type] = frames, info
|
130
|
+
else:
|
131
|
+
msg = "Could not load %s from %s" % (frame_type, url.file_path())
|
132
|
+
logger.error(msg)
|
133
|
+
|
134
|
+
frames_loaded = dict.fromkeys(frames_types, False)
|
135
|
+
if flatfield_mode != "force-compute":
|
136
|
+
for load_from in ["user", "dataset", "output"]: # in that order
|
137
|
+
for frame_type in frames_types:
|
138
|
+
url = reduced_frames_urls[frame_type][load_from]
|
139
|
+
if url is None:
|
140
|
+
continue # cannot load from this source (eg. undefined folder)
|
141
|
+
load_reduced_frame(url, frame_type, frames_loaded, reduced_frames)
|
142
|
+
if all(frames_loaded.values()):
|
143
|
+
break
|
144
|
+
|
145
|
+
if not all(frames_loaded.values()) and flatfield_mode == "force-load":
|
146
|
+
raise ValueError("Could not load darks/flats (using 'force-load')")
|
147
|
+
|
148
|
+
#
|
149
|
+
# COMPAT. Keep DataUrl - won't be needed in future versions when pipeline will use FlatField
|
150
|
+
# instead of FlatFieldDataUrl
|
151
|
+
frames_urls = reduced_frames.copy()
|
152
|
+
#
|
153
|
+
|
154
|
+
# Compute reduced frames, if needed
|
155
|
+
#
|
156
|
+
if reduced_frames["flats"] is None:
|
157
|
+
reduced_frames["flats"] = dataset_info.dataset_scanner.compute_reduced_flats(return_info=True)
|
158
|
+
if reduced_frames["darks"] is None:
|
159
|
+
reduced_frames["darks"] = dataset_info.dataset_scanner.compute_reduced_darks(return_info=True)
|
160
|
+
|
161
|
+
if reduced_frames["darks"][0] == {} or reduced_frames["flats"][0] == {}:
|
162
|
+
raise ValueError(
|
163
|
+
"Could not get any reduced flat/dark. This probably means that no already-reduced flats/darks were found and that the dataset itself does not have any flat/dark"
|
164
|
+
)
|
165
|
+
|
166
|
+
#
|
167
|
+
# Save reduced frames
|
168
|
+
#
|
169
|
+
|
170
|
+
def save_reduced_frame(url, frame_type, frames_saved):
|
171
|
+
frames, info = reduced_frames[frame_type]
|
172
|
+
tomoscan_save_reduced_frames(dataset_info, frame_type, url, frames, info)
|
173
|
+
dataset_info.logger.info("Saved reduced %s to %s" % (frame_type, url.file_path()))
|
174
|
+
frames_saved[frame_type] = True
|
175
|
+
|
176
|
+
frames_saved = dict.fromkeys(frames_types, False)
|
177
|
+
if not all(frames_loaded.values()):
|
178
|
+
for save_to in ["user", "dataset", "output"]: # in that order
|
179
|
+
for frame_type in frames_types:
|
180
|
+
if frames_loaded[frame_type]:
|
181
|
+
continue # already loaded
|
182
|
+
url = reduced_frames_urls[frame_type][save_to]
|
183
|
+
if url is None:
|
184
|
+
continue # cannot load from this source (eg. undefined folder)
|
185
|
+
if not is_writeable(os.path.dirname(url.file_path())):
|
186
|
+
continue
|
187
|
+
save_reduced_frame(url, frame_type, frames_saved)
|
188
|
+
# COMPAT.
|
189
|
+
if frames_urls[frame_type] is None:
|
190
|
+
frames_urls[frame_type] = tomoscan_load_reduced_frames(dataset_info, frame_type, url)
|
191
|
+
#
|
192
|
+
if all(frames_saved.values()):
|
193
|
+
break
|
194
|
+
|
195
|
+
dataset_info.flats = frames_urls["flats"][0] # reduced_frames["flats"] # in future versions
|
196
|
+
dataset_info.flats_srcurrent = frames_urls["flats"][1].machine_electric_current
|
197
|
+
# This is an extra check to avoid having more than 1 (reduced) dark.
|
198
|
+
# FlatField only works with exactly 1 (reduced) dark (having more than 1 series of darks makes little sense)
|
199
|
+
# This is normally prevented by tomoscan HDF5FramesReducer, but let's add this extra check
|
200
|
+
darks_ = frames_urls["darks"][0] # reduced_frames["darks"] # in future versions
|
201
|
+
if len(darks_) > 1:
|
202
|
+
dark_idx = sorted(darks_.keys())[0]
|
203
|
+
dataset_info.logger.error("Found more that one series of darks. Keeping only the first one")
|
204
|
+
darks_ = {dark_idx: darks_[dark_idx]}
|
205
|
+
#
|
206
|
+
dataset_info.darks = darks_
|
207
|
+
dataset_info.darks_srcurrent = frames_urls["darks"][1].machine_electric_current
|
208
|
+
|
209
|
+
|
210
|
+
# tomoscan "compute_reduced_XX" is quite slow. If needed, here is an alternative implementation
|
211
|
+
def my_reduce_flats(di):
|
212
|
+
res = {}
|
213
|
+
with HDF5File(di.dataset_hdf5_url.file_path(), "r") as f:
|
214
|
+
for data_slice in di.get_data_slices("flats"):
|
215
|
+
data = f[di.dataset_hdf5_url.data_path()][data_slice.start : data_slice.stop]
|
216
|
+
res[data_slice.start] = np.median(data, axis=0)
|
217
|
+
return res
|
File without changes
|
File without changes
|
@@ -0,0 +1,104 @@
|
|
1
|
+
from os import path
|
2
|
+
from tempfile import mkdtemp
|
3
|
+
from shutil import rmtree
|
4
|
+
import pytest
|
5
|
+
import numpy as np
|
6
|
+
from silx.io import get_data
|
7
|
+
from nxtomo.nxobject.nxdetector import ImageKey
|
8
|
+
from nabu.testutils import generate_nx_dataset
|
9
|
+
from nabu.resources.nxflatfield import update_dataset_info_flats_darks
|
10
|
+
from nabu.resources.dataset_analyzer import HDF5DatasetAnalyzer
|
11
|
+
|
12
|
+
|
13
|
+
test_nxflatfield_scenarios = [
|
14
|
+
{
|
15
|
+
"name": "simple",
|
16
|
+
"flats_pos": [slice(1, 6)],
|
17
|
+
"darks_pos": [slice(0, 1)],
|
18
|
+
"output_dir": None,
|
19
|
+
},
|
20
|
+
{
|
21
|
+
"name": "simple_with_save",
|
22
|
+
"flats_pos": [slice(1, 6)],
|
23
|
+
"darks_pos": [slice(0, 1)],
|
24
|
+
"output_dir": None,
|
25
|
+
},
|
26
|
+
{
|
27
|
+
"name": "multiple_with_save",
|
28
|
+
"flats_pos": [slice(0, 10), slice(30, 40)],
|
29
|
+
"darks_pos": [slice(95, 100)],
|
30
|
+
"output_dir": path.join("{tempdir}", "output_reduced"),
|
31
|
+
},
|
32
|
+
]
|
33
|
+
|
34
|
+
|
35
|
+
# parametrize with fixture and "params=" will launch a new class for each scenario.
|
36
|
+
# the attributes set to "cls" will remain for all the tests done in this class
|
37
|
+
# with the current scenario.
|
38
|
+
@pytest.fixture(scope="class", params=test_nxflatfield_scenarios)
|
39
|
+
def bootstrap(request):
|
40
|
+
cls = request.cls
|
41
|
+
cls.n_projs = 265
|
42
|
+
cls.params = request.param
|
43
|
+
cls.tempdir = mkdtemp(prefix="nabu_")
|
44
|
+
yield
|
45
|
+
rmtree(cls.tempdir)
|
46
|
+
|
47
|
+
|
48
|
+
@pytest.mark.usefixtures("bootstrap")
|
49
|
+
class TestNXFlatField:
|
50
|
+
_reduction_func = {"flats": np.median, "darks": np.mean}
|
51
|
+
|
52
|
+
def get_nx_filename(self):
|
53
|
+
return path.join(self.tempdir, "dataset_" + self.params["name"] + ".nx")
|
54
|
+
|
55
|
+
def get_image_key(self):
|
56
|
+
keys = np.zeros(self.n_projs, np.int32)
|
57
|
+
for what, val in [("flats_pos", ImageKey.FLAT_FIELD.value), ("darks_pos", ImageKey.DARK_FIELD.value)]:
|
58
|
+
for pos in self.params[what]:
|
59
|
+
keys[pos.start : pos.stop] = val
|
60
|
+
return keys
|
61
|
+
|
62
|
+
@staticmethod
|
63
|
+
def check_image_key(dataset_info, frame_type, expected_slices):
|
64
|
+
data_slices = dataset_info.get_data_slices(frame_type)
|
65
|
+
assert set(map(str, data_slices)) == set(map(str, expected_slices))
|
66
|
+
|
67
|
+
def test_nxflatfield(self):
|
68
|
+
dataset_fname = self.get_nx_filename()
|
69
|
+
image_key = self.get_image_key()
|
70
|
+
generate_nx_dataset(dataset_fname, image_key)
|
71
|
+
|
72
|
+
dataset_info = HDF5DatasetAnalyzer(dataset_fname)
|
73
|
+
|
74
|
+
# When parsing a "raw" dataset, flats/darks are a series of images.
|
75
|
+
# dataset_info.flats is a dictionary where each key is the index of the frame.
|
76
|
+
# For example dataset_info.flats.keys() = [10, 11, 12, 13, ..., 19, 1200, 1201, 1202, ..., 1219]
|
77
|
+
for frame_type in ["darks", "flats"]:
|
78
|
+
self.check_image_key(dataset_info, frame_type, self.params[frame_type + "_pos"])
|
79
|
+
|
80
|
+
output_dir = self.params.get("output_dir", None)
|
81
|
+
if output_dir is not None:
|
82
|
+
output_dir = output_dir.format(tempdir=self.tempdir)
|
83
|
+
update_dataset_info_flats_darks(dataset_info, True, output_dir=output_dir)
|
84
|
+
# After reduction (median/mean), the flats/darks are located in another file.
|
85
|
+
# median(series_1) goes to entry/flats/idx1, mean(series_2) goes to entry/flats/idx2, etc.
|
86
|
+
assert set(dataset_info.flats.keys()) == set(s.start for s in self.params["flats_pos"])
|
87
|
+
assert set(dataset_info.darks.keys()) == set(s.start for s in self.params["darks_pos"])
|
88
|
+
|
89
|
+
# Check that the computations were correct
|
90
|
+
# Loads the entire volume in memory ! So keep the data volume small for the tests
|
91
|
+
data_volume = get_data(dataset_info.dataset_hdf5_url)
|
92
|
+
expected_flats = {}
|
93
|
+
for s in self.params["flats_pos"]:
|
94
|
+
expected_flats[s.start] = self._reduction_func["flats"](data_volume[s.start : s.stop], axis=0)
|
95
|
+
expected_darks = {}
|
96
|
+
for s in self.params["darks_pos"]:
|
97
|
+
expected_darks[s.start] = self._reduction_func["darks"](data_volume[s.start : s.stop], axis=0)
|
98
|
+
|
99
|
+
flats = {idx: get_data(dataset_info.flats[idx]) for idx in dataset_info.flats.keys()}
|
100
|
+
for idx in flats.keys():
|
101
|
+
assert np.allclose(flats[idx], expected_flats[idx])
|
102
|
+
darks = {idx: get_data(dataset_info.darks[idx]) for idx in dataset_info.darks.keys()}
|
103
|
+
for idx in darks.keys():
|
104
|
+
assert np.allclose(darks[idx], expected_darks[idx])
|
@@ -0,0 +1,47 @@
|
|
1
|
+
import pytest
|
2
|
+
from nabu.utils import compare_dicts
|
3
|
+
from nabu.resources.utils import get_quantities_and_units
|
4
|
+
|
5
|
+
|
6
|
+
class TestUnits:
|
7
|
+
expected_results = {
|
8
|
+
"distance = 1 m ; pixel_size = 2.0 um": {"distance": 1.0, "pixel_size": 2e-6},
|
9
|
+
"distance = 1 m ; pixel_size = 2.6 micrometer": {"distance": 1.0, "pixel_size": 2.6e-6},
|
10
|
+
"distance = 10 m ; pixel_size = 2e-6 m": {"distance": 10, "pixel_size": 2e-6},
|
11
|
+
"distance = .5 m ; pixel_size = 2.6e-4 centimeter": {"distance": 0.5, "pixel_size": 2.6e-6},
|
12
|
+
"distance = 10 cm ; pixel_size = 2.5 micrometer ; energy = 1 ev": {
|
13
|
+
"distance": 0.1,
|
14
|
+
"pixel_size": 2.5e-6,
|
15
|
+
"energy": 1.0e-3,
|
16
|
+
},
|
17
|
+
"distance = 10 cm ; pixel_size = 9.0e-3 millimeter ; energy = 19 kev": {
|
18
|
+
"distance": 0.1,
|
19
|
+
"pixel_size": 9e-6,
|
20
|
+
"energy": 19.0,
|
21
|
+
},
|
22
|
+
}
|
23
|
+
expected_failures = {
|
24
|
+
# typo ("ke" instead of "kev")
|
25
|
+
"distance = 10 cm ; energy = 10 ke": ValueError("Cannot convert: ke"),
|
26
|
+
# No units
|
27
|
+
"distance = 10 ; energy = 10 kev": ValueError("not enough values to unpack (expected 2, got 1)"),
|
28
|
+
# Unit not separated by space
|
29
|
+
"distance = 10m; energy = 10 kev": ValueError("not enough values to unpack (expected 2, got 1)"),
|
30
|
+
# Invalid separator
|
31
|
+
"distance = 10 m, energy = 10 kev": ValueError("too many values to unpack (expected 2)"),
|
32
|
+
}
|
33
|
+
|
34
|
+
def test_conversion(self):
|
35
|
+
for test_str, expected_result in self.expected_results.items():
|
36
|
+
res = get_quantities_and_units(test_str)
|
37
|
+
err_msg = str(
|
38
|
+
"Something wrong with quantities/units extraction from '%s': expected %s, got %s"
|
39
|
+
% (test_str, str(expected_result), str(res))
|
40
|
+
)
|
41
|
+
assert compare_dicts(res, expected_result) is None, err_msg
|
42
|
+
|
43
|
+
def test_valid_input(self):
|
44
|
+
for test_str, expected_failure in self.expected_failures.items():
|
45
|
+
with pytest.raises(type(expected_failure)) as e_info:
|
46
|
+
get_quantities_and_units(test_str)
|
47
|
+
assert e_info.value.args[0] == str(expected_failure)
|
@@ -0,0 +1,174 @@
|
|
1
|
+
from ast import literal_eval
|
2
|
+
import numpy as np
|
3
|
+
from psutil import virtual_memory, cpu_count
|
4
|
+
from pyunitsystem.metricsystem import MetricSystem
|
5
|
+
from pyunitsystem.energysystem import EnergySI
|
6
|
+
|
7
|
+
|
8
|
+
def get_values_from_file(fname, n_values=None, shape=None, sep=None, any_size=False):
|
9
|
+
"""
|
10
|
+
Read a text file and scan the values inside.
|
11
|
+
This function expects one value per line, or values separated with a separator
|
12
|
+
defined with the `sep` parameter.
|
13
|
+
|
14
|
+
Parameters
|
15
|
+
----------
|
16
|
+
fname: str
|
17
|
+
Path of the text file
|
18
|
+
n_values: int, optional
|
19
|
+
If set to a value, this function will check that it scans exactly this
|
20
|
+
number of values.
|
21
|
+
Ignored if `shape` is provided
|
22
|
+
shape: tuple, optional
|
23
|
+
Generalization of n_values for higher dimensions.
|
24
|
+
sep: str, optional
|
25
|
+
Separator between values. Default is white space
|
26
|
+
any_size: bool, optional
|
27
|
+
If set to True, then the parameters 'n_values' and 'shape' are ignored.
|
28
|
+
|
29
|
+
Returns
|
30
|
+
--------
|
31
|
+
arr: numpy.ndarray
|
32
|
+
An array containing the values scanned from the text file
|
33
|
+
"""
|
34
|
+
if not (any_size) and not ((n_values is not None) ^ (shape is not None)):
|
35
|
+
raise ValueError("Please provide either n_values or shape")
|
36
|
+
arr = np.loadtxt(fname, ndmin=1)
|
37
|
+
if (n_values is not None) and (arr.shape[0] != n_values):
|
38
|
+
if any_size:
|
39
|
+
arr = arr[:n_values]
|
40
|
+
else:
|
41
|
+
raise ValueError("Expected %d values, but could get %d values" % (n_values, arr.shape[0]))
|
42
|
+
if (shape is not None) and (arr.shape != shape):
|
43
|
+
if any_size:
|
44
|
+
arr = arr[: shape[0], : shape[1]] # TODO handle more dimensions ?
|
45
|
+
else:
|
46
|
+
raise ValueError("Expected shape %s, but got shape %s" % (shape, arr.shape))
|
47
|
+
return arr
|
48
|
+
|
49
|
+
|
50
|
+
def get_memory_per_node(max_mem, is_percentage=True):
|
51
|
+
"""
|
52
|
+
Get the available memory per node in GB.
|
53
|
+
|
54
|
+
Parameters
|
55
|
+
----------
|
56
|
+
max_mem: float
|
57
|
+
If is_percentage is False, then number is interpreted as an absolute
|
58
|
+
number in GigaBytes.
|
59
|
+
Otherwise, it should be a number between 0 and 100 and is interpreted
|
60
|
+
as a percentage.
|
61
|
+
is_percentage: bool
|
62
|
+
A boolean indicating whether the parameter max_mem is to be interpreted
|
63
|
+
as a percentage of available system memory.
|
64
|
+
"""
|
65
|
+
sys_avail_mem = virtual_memory().available / 1e9
|
66
|
+
if is_percentage:
|
67
|
+
return (max_mem / 100.0) * sys_avail_mem
|
68
|
+
else:
|
69
|
+
return min(max_mem, sys_avail_mem)
|
70
|
+
|
71
|
+
|
72
|
+
def get_threads_per_node(max_threads, is_percentage=True):
|
73
|
+
"""
|
74
|
+
Get the available memory per node in GB.
|
75
|
+
|
76
|
+
Parameters
|
77
|
+
----------
|
78
|
+
max_threads: float
|
79
|
+
If is_percentage is False, then number is interpreted as an absolute
|
80
|
+
number of threads.
|
81
|
+
Otherwise, it should be a number between 0 and 100 and is interpreted
|
82
|
+
as a percentage.
|
83
|
+
is_percentage: bool
|
84
|
+
A boolean indicating whether the parameter max_threads is to be interpreted
|
85
|
+
as a percentage of available system memory.
|
86
|
+
"""
|
87
|
+
sys_n_threads = cpu_count(logical=True)
|
88
|
+
if is_percentage:
|
89
|
+
return (max_threads / 100.0) * sys_n_threads
|
90
|
+
else:
|
91
|
+
return min(max_threads, sys_n_threads)
|
92
|
+
|
93
|
+
|
94
|
+
def extract_parameters(params_str, sep=";"):
|
95
|
+
"""
|
96
|
+
Extract the named parameters from a string.
|
97
|
+
|
98
|
+
Example
|
99
|
+
--------
|
100
|
+
The function can be used as follows:
|
101
|
+
|
102
|
+
>>> extract_parameters("window_width=None; median_filt_shape=(3,3); padding_mode='wrap'")
|
103
|
+
... {'window_width': None, 'median_filt_shape': (3, 3), 'padding_mode': 'wrap'}
|
104
|
+
"""
|
105
|
+
if params_str in ("", None):
|
106
|
+
return {}
|
107
|
+
params_list = params_str.strip(sep).split(sep)
|
108
|
+
res = {}
|
109
|
+
for param_str in params_list:
|
110
|
+
param_name, param_val_str = param_str.strip().split("=")
|
111
|
+
param_name = param_name.strip()
|
112
|
+
param_val_str = param_val_str.strip()
|
113
|
+
param_val = literal_eval(param_val_str)
|
114
|
+
res[param_name] = param_val
|
115
|
+
return res
|
116
|
+
|
117
|
+
|
118
|
+
def compact_parameters(params_dict, sep=";"):
|
119
|
+
"""
|
120
|
+
Compact the parameters from a dict into a string.
|
121
|
+
|
122
|
+
This is the inverse of extract_parameters. It can be used for example in tomwer to convert
|
123
|
+
parameters into a string, for example for cor_options, prior to calling a nabu method which
|
124
|
+
is expecting an argument to be in the form of a string containing options.
|
125
|
+
|
126
|
+
Example
|
127
|
+
--------
|
128
|
+
The function can be used as follows:
|
129
|
+
|
130
|
+
>>> compact_parameters( {"side":"near", "near_pos":300 } )
|
131
|
+
... "side=near; nearpos= 300;"
|
132
|
+
"""
|
133
|
+
if params_dict in ({}, None):
|
134
|
+
return ""
|
135
|
+
|
136
|
+
res = ""
|
137
|
+
for key, val in params_dict.items():
|
138
|
+
res = res + "{key} = {val} " + sep
|
139
|
+
return res
|
140
|
+
|
141
|
+
|
142
|
+
def is_hdf5_extension(ext):
|
143
|
+
return ext.lower() in ["h5", "hdf5", "nx"]
|
144
|
+
|
145
|
+
|
146
|
+
def get_quantities_and_units(string, sep=";"):
|
147
|
+
"""
|
148
|
+
Return a dictionary with quantities as keys, and values in SI.
|
149
|
+
|
150
|
+
Example
|
151
|
+
-------
|
152
|
+
get_quantities_and_units("pixel_size = 1.2 um ; distance = 1 m")
|
153
|
+
Will return
|
154
|
+
{"pixel_size": 1.2e-6, "distance": 1}
|
155
|
+
"""
|
156
|
+
|
157
|
+
result = {}
|
158
|
+
quantities = string.split(sep)
|
159
|
+
for quantity in quantities:
|
160
|
+
quantity_name, value_and_unit = quantity.split("=")
|
161
|
+
quantity_name = quantity_name.strip()
|
162
|
+
value_and_unit = value_and_unit.strip()
|
163
|
+
value, unit = value_and_unit.split()
|
164
|
+
val = float(value)
|
165
|
+
# Convert to SI
|
166
|
+
try:
|
167
|
+
# handle metrics
|
168
|
+
conversion_factor = MetricSystem.from_str(unit).value
|
169
|
+
except ValueError:
|
170
|
+
# handle energies
|
171
|
+
conversion_factor = EnergySI.from_str(unit).value / EnergySI.KILOELECTRONVOLT.value
|
172
|
+
|
173
|
+
result[quantity_name] = val * conversion_factor
|
174
|
+
return result
|
File without changes
|