minor change (tried using similar instead, but CUDA assigment does not work)

Author: RainerHeintzmann

src/fourier_shear.jl

@@ -46,6 +46,7 @@ function shear!(arr::TA, Δ, shear_dir_dim=1, shear_dim=2; fix_nyquist=false, as
 
     # stores the maximum amount of shift
     TR = real_arr_type(TA)
+    # shift = similar(arr, real(eltype(arr)), select_sizes(arr, shear_dir_dim))
     # shift = TR(reshape(fftfreq(size(arr, shear_dir_dim)), NDTools.select_sizes(arr, shear_dir_dim)))
     shift = TR(reorient(fftfreq(size(arr, shear_dir_dim)),shear_dir_dim, Val(N)))
 
@@ -65,12 +66,13 @@ function shear!(arr::TA, Δ, shear_dir_dim=1, shear_dim=2; fix_nyquist=false, as
 
     # stores the maximum amount of shift
     TR = real_arr_type(TA)
+    # shift = similar(arr, real(eltype(arr_ft)), select_sizes(arr_ft, shear_dir_dim))
     # shift = TR(reshape(rfftfreq(size(arr, shear_dir_dim)), NDTools.select_sizes(arr_ft, shear_dir_dim)))
     shift = TR(reorient(rfftfreq(size(arr, shear_dir_dim)),shear_dir_dim, Val(N)))
 
     apply_shift_strength!(arr_ft, arr, shift, shear_dir_dim, shear_dim, Δ, fix_nyquist)
     # go back to real space
-     
+
     # overwrites arr in-place
     ldiv!(arr, p, arr_ft)
     if assign_wrap
@@ -110,9 +112,9 @@ end
 
 function apply_shift_strength!(arr::TA, arr_orig, shift, shear_dir_dim, shear_dim, Δ, fix_nyquist=false) where {T, N, TA<:AbstractArray{T, N}}
     #applies the strength to each slice
-    # shift_strength = reshape(fftpos(1, size(arr, shear_dim), CenterFT), NDTools.select_sizes(arr, shear_dim))
-    # TR = real_arr_type(typeof(collect(arr[1:1]))) # There is a problem with circshifted arrays and this way of finding the type.
-    shift_strength = reorient(fftpos(1, size(arr, shear_dim), CenterFT), shear_dim, Val(N))
+    # The TR trick does not seem to work for the code below due to a call with a PaddedArray.
+    shift_strength = similar(arr, real(eltype(arr)), select_sizes(arr, shear_dim))
+    shift_strength .= (real(eltype(TA))).(reorient(fftpos(1, size(arr, shear_dim), CenterFT), shear_dim, Val(N)))
 
     # do the exp multiplication in place
     e = cispi.(2 .* Δ .* shift .* shift_strength)

src/fourier_shifting.jl

@@ -88,6 +88,7 @@ function shift_by_1D_FT!(arr::TA, shifts; soft_fraction=0, take_real=false, fix_
         end
         # better use reorient from NDTools here?
         TR = real_arr_type(TA)
+        # freqs = similar(arr, real(eltype(arr)), select_sizes(arr, d))
         freqs = TR(reorient(fftfreq(size(arr, d)),d, Val(N)))
         # freqs = TR(reshape(fftfreq(size(arr, d)), ntuple(i -> 1, Val(d-1))..., size(arr,d)))
         # @show size(freqs)
@@ -142,8 +143,10 @@ function shift_by_1D_RFT!(arr::TA, shifts; soft_fraction=0, fix_nyquist_frequenc
         s = size(arr_ft,d); # s1[d]
         # @show size(arr, d) ÷ 2 + 1
         # freqs = TR(reshape(fftfreq(size(arr, d))[1:s], ntuple(i -> 1, d-1)..., s1))
+
         TR = real_arr_type(TA)
-        freqs = TR(reorient(fftfreq(size(arr, d))[1:s],d, Val(N)))
+        # freqs = similar(arr, real(eltype(arr_ft)), select_sizes(arr_ft, d))
+        freqs = TR(reorient(fftfreq(size(arr, d))[1:s], d, Val(N)))
         if iszero(soft_fraction)
             ϕ = cispi.(-freqs .* 2 .* shift)
         else