Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
ift
NIFTy
Commits
c63f8f58
Commit
c63f8f58
authored
Feb 05, 2015
by
ultimanet
Browse files
implemented fft factory
parent
b590b3e7
Changes
2
Hide whitespace changes
Inline
Side-by-side
__init__.py
View file @
c63f8f58
...
...
@@ -36,6 +36,7 @@ except(ImportError):
try
:
from
lm
import
*
except
(
ImportError
):
print
'asdf'
pass
from
demos
import
*
...
...
rg/nifty_rg.py
View file @
c63f8f58
...
...
@@ -44,11 +44,15 @@ from nifty.nifty_core import about, \
field
import
nifty.smoothing
as
gs
import
powerspectrum
as
gp
'''
try:
import gfft as gf
except(ImportError):
about.infos.cprint('INFO: "plain" gfft version 0.1.0')
import gfft_rg as gf
'''
import
fft_rg
##-----------------------------------------------------------------------------
...
...
@@ -198,6 +202,8 @@ class rg_space(space):
self
.
vol
=
np
.
real
(
dist
)[::
-
1
]
self
.
fourier
=
bool
(
fourier
)
self
.
my_fft_object
=
fft_rg
.
fft_factory
()
##+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
...
...
@@ -808,22 +814,26 @@ class rg_space(space):
naxes
=
(
np
.
size
(
self
.
para
)
-
1
)
//
2
## select machine
if
(
np
.
all
(
np
.
absolute
(
self
.
para
[:
naxes
]
*
self
.
vol
*
codomain
.
vol
-
1
)
<
self
.
epsilon
)):
## Use the codomain information only for the rescaling. The direction
## of transformation is infered from the fourier attribute of the
## supplied space
if
(
codomain
.
fourier
):
ftmachine
=
"fft"
#
ftmachine = "fft"
## correct for 'fft'
x
=
self
.
calc_weight
(
x
,
power
=
1
)
else
:
ftmachine
=
"ifft"
#
ftmachine = "ifft"
## correct for 'ifft'
x
=
self
.
calc_weight
(
x
,
power
=
1
)
x
*=
self
.
dim
(
split
=
False
)
else
:
ftmachine
=
"none"
## TODO: Is this error correct?
raise
ValueError
(
about
.
_errors
.
cstring
(
"ERROR: unsupported transformation."
))
#ftmachine = "none"
## transform
if
(
self
.
datatype
==
np
.
float64
):
Tx
=
gf
.
gfft
(
x
.
astype
(
np
.
complex128
),
in_ax
=
[],
out_ax
=
[],
ftmachine
=
ftmachine
,
in_zero_center
=
self
.
para
[
-
naxes
:].
astype
(
np
.
bool
).
tolist
(),
out_zero_center
=
codomain
.
para
[
-
naxes
:].
astype
(
np
.
bool
).
tolist
(),
enforce_hermitian_symmetry
=
bool
(
codomain
.
para
[
naxes
]
==
1
),
W
=-
1
,
alpha
=-
1
,
verbose
=
False
)
else
:
Tx
=
gf
.
gfft
(
x
,
in_ax
=
[],
out_ax
=
[],
ftmachine
=
ftmachine
,
in_zero_center
=
self
.
para
[
-
naxes
:].
astype
(
np
.
bool
).
tolist
(),
out_zero_center
=
codomain
.
para
[
-
naxes
:].
astype
(
np
.
bool
).
tolist
(),
enforce_hermitian_symmetry
=
bool
(
codomain
.
para
[
naxes
]
==
1
),
W
=-
1
,
alpha
=-
1
,
verbose
=
False
)
Tx
=
self
.
my_fft_object
(
x
,
self
,
codomain
)
## check complexity
if
(
not
codomain
.
para
[
naxes
]):
## purely real
## check imaginary part
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment