cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-09 21:12:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix the issue with obj view padding in Bragg2DPtycho not working with a probe fourier crop

Browse Source
This commit is contained in:
levita_a
2024-12-23 10:10:40 +01:00
parent a22ae1b476
commit c03d4284f2
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/cdtools/models/bragg_2d_ptycho.py
+9 -3
View File
@@ -371,11 +371,17 @@ class Bragg2DPtycho(CDIModel):
obj = t.exp(1j*(randomize_ang * (t.rand(obj_size)-0.5)))
pfc = (probe_fourier_crop if probe_fourier_crop else 0)
pfc = (probe_fourier_crop if probe_fourier_crop else [0,0])
if obj_view_crop is None:
obj_view_crop = min(probe.shape[-2], probe.shape[-1]) // 2 + pfc
obj_view_crop = min(
probe.shape[-2] // 2 + pfc[0],
probe.shape[-1] // 2 + pfc[1]
)
if obj_view_crop < 0:
obj_view_crop += min(probe.shape[-2], probe.shape[-1]) // 2 + pfc
obj_view_crop += min(
probe.shape[-2] // 2 + pfc[0],
probe.shape[-1] // 2 + pfc[1]
)
obj_view_crop += obj_padding