Commit 2b070f69 authored by theos's avatar theos
Browse files

Code deduplication.

parent 06e2d7fa
...@@ -172,7 +172,7 @@ class FFTW(Transform): ...@@ -172,7 +172,7 @@ class FFTW(Transform):
(101 * codomain.__hash__()) ^ (101 * codomain.__hash__()) ^
(211 * transform_shape.__hash__()) ^ (211 * transform_shape.__hash__()) ^
(131 * is_local.__hash__()) (131 * is_local.__hash__())
) )
# generate the plan_and_info object if not already there # generate the plan_and_info object if not already there
if temp_id not in self.info_dict: if temp_id not in self.info_dict:
...@@ -449,17 +449,17 @@ class FFTWTransformInfo(object): ...@@ -449,17 +449,17 @@ class FFTWTransformInfo(object):
if pyfftw is None: if pyfftw is None:
raise ImportError("The module pyfftw is needed but not available.") raise ImportError("The module pyfftw is needed but not available.")
self.cmask_domain = fftw_context.get_centering_mask( shape = (local_shape if axes is None else
domain.zerocenter, [y for x, y in enumerate(local_shape) if x in axes])
local_shape if axes is None else
[y for x, y in enumerate(local_shape) if x in axes], self.cmask_domain = fftw_context.get_centering_mask(domain.zerocenter,
local_offset_Q) shape,
local_offset_Q)
self.cmask_codomain = fftw_context.get_centering_mask( self.cmask_codomain = fftw_context.get_centering_mask(
codomain.zerocenter, codomain.zerocenter,
local_shape if axes is None else shape,
[y for x, y in enumerate(local_shape) if x in axes], local_offset_Q)
local_offset_Q)
# If both domain and codomain are zero-centered the result, # If both domain and codomain are zero-centered the result,
# will get a global minus. Store the sign to correct it. # will get a global minus. Store the sign to correct it.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment