Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F85876285
fft_engine_base.hh
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Wed, Oct 2, 17:17
Size
4 KB
Mime Type
text/x-c
Expires
Fri, Oct 4, 17:17 (2 d)
Engine
blob
Format
Raw Data
Handle
21286051
Attached To
rMUSPECTRE µSpectre
fft_engine_base.hh
View Options
/**
* file fft_engine_base.hh
*
* @author Till Junge <till.junge@epfl.ch>
*
* @date 01 Dec 2017
*
* @brief Interface for FFT engines
*
* @section LICENCE
*
* Copyright © 2017 Till Junge
*
* µSpectre is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation, either version 3, or (at
* your option) any later version.
*
* µSpectre is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Emacs; see the file COPYING. If not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifndef FFT_ENGINE_BASE_H
#define FFT_ENGINE_BASE_H
#include "common/common.hh"
#include "common/field_collection.hh"
namespace
muSpectre
{
enum
class
FFT_PlanFlags
{
estimate
,
measure
,
patient
};
template
<
Dim_t
DimS
,
Dim_t
DimM
>
class
FFT_Engine_base
{
public
:
constexpr
static
Dim_t
sdim
{
DimS
};
constexpr
static
Dim_t
mdim
{
DimM
};
using
Ccoord
=
Ccoord_t
<
DimS
>
;
using
Rcoord
=
std
::
array
<
Real
,
DimS
>
;
using
GFieldCollection_t
=
FieldCollection
<
DimS
,
DimM
,
true
>
;
using
LFieldCollection_t
=
FieldCollection
<
DimS
,
DimM
,
false
>
;
using
Field_t
=
TensorField
<
GFieldCollection_t
,
Real
,
2
,
DimM
>
;
using
Workspace_t
=
TensorField
<
LFieldCollection_t
,
Complex
,
2
,
DimM
>
;
using
iterator
=
typename
LFieldCollection_t
::
iterator
;
//! Default constructor
FFT_Engine_base
()
=
delete
;
//! Constructor with system resolutions
FFT_Engine_base
(
Ccoord
resolutions
,
Rcoord
lengths
);
//! Copy constructor
FFT_Engine_base
(
const
FFT_Engine_base
&
other
)
=
delete
;
//! Move constructor
FFT_Engine_base
(
FFT_Engine_base
&&
other
)
noexcept
=
default
;
//! Destructor
virtual
~
FFT_Engine_base
()
noexcept
=
default
;
//! Copy assignment operator
FFT_Engine_base
&
operator
=
(
const
FFT_Engine_base
&
other
)
=
delete
;
//! Move assignment operator
FFT_Engine_base
&
operator
=
(
FFT_Engine_base
&&
other
)
noexcept
=
default
;
// compute the plan, etc
virtual
void
initialise
(
FFT_PlanFlags
/*plan_flags*/
);
//! forward transform (dummy for interface)
virtual
Workspace_t
&
fft
(
Field_t
&
/*field*/
)
=
0
;
//! inverse transform (dummy for interface)
virtual
void
ifft
(
Field_t
&
/*field*/
)
const
=
0
;
/**
* iterators over only thos pixels that exist in frequency space
* (i.e. about half of all pixels, see rfft)
*/
inline
iterator
begin
()
{
return
this
->
work_space_container
.
begin
();}
inline
iterator
end
()
{
return
this
->
work_space_container
.
end
();}
//! nb of pixels (mostly for debugging)
size_t
size
()
const
;
size_t
workspace_size
()
const
;
//!
const
Ccoord
&
get_resolutions
()
const
{
return
this
->
resolutions
;}
const
Rcoord
&
get_lengths
()
const
{
return
this
->
lengths
;}
LFieldCollection_t
&
get_field_collection
()
{
return
this
->
work_space_container
;}
//! factor by which to multiply projection before inverse transform (this is
//! typically 1/nb_pixels for so-called unnormalized transforms (see,
//! e.g. http://www.fftw.org/fftw3_doc/Multi_002dDimensional-DFTs-of-Real-Data.html#Multi_002dDimensional-DFTs-of-Real-Data
//! or https://docs.scipy.org/doc/numpy-1.13.0/reference/routines.fft.html
//! . Rather than scaling the inverse transform (which would cost one more
//! loop), FFT engines provide this value so it can be used in the
//! projection operator (where no additional loop is required)
inline
Real
normalisation
()
const
{
return
norm_factor
;};
protected
:
LFieldCollection_t
work_space_container
{};
const
Ccoord
resolutions
;
const
Rcoord
lengths
;
Workspace_t
&
work
;
const
Real
norm_factor
;
private
:
};
}
// muSpectre
#endif
/* FFT_ENGINE_BASE_H */
Event Timeline
Log In to Comment