FourierTraits.h

Go to the documentation of this file.

// -*- c++ -*-
/*
 * Copyright (c) 2010-2012, Jim Bosch
 * All rights reserved.
 *
 * ndarray is distributed under a simple BSD-like license;
 * see the LICENSE file that should be present in the root
 * of the source distribution, or alternately available at:
 * https://github.com/ndarray/ndarray
 */
#ifndef NDARRAY_FFT_FourierTraits_h_INCLUDED
#define NDARRAY_FFT_FourierTraits_h_INCLUDED
 
#include <complex>
#include <boost/shared_ptr.hpp>
 
#include "ndarray/fft_fwd.h"
 
namespace ndarray {
namespace detail {
 
template <typename T, bool IsConst>
struct FourierTraits {
    BOOST_STATIC_ASSERT(sizeof(T) < 0);
};
 
 
template <typename T>
struct FourierTraits<T,false> {
    typedef T ElementX;
    typedef T ValueX;
    typedef std::complex<T> ElementK;
    typedef std::complex<T> ValueK;
 
    static inline Size computeLastDimensionSize(Size n) { return n/2 + 1; }
};
 
template <typename T>
struct FourierTraits<T,true> {
    typedef T ElementX;
    typedef typename boost::remove_const<T>::type ValueX;
    typedef std::complex<ValueX> ValueK;
    typedef ValueK const ElementK;
 
    static inline Size computeLastDimensionSize(Size n) { return n/2 + 1; }
};
 
template <typename U>
struct FourierTraits<std::complex<U>,false> {
    typedef std::complex<U> ElementX;
    typedef std::complex<U> ElementK;
    typedef std::complex<U> ValueX;
    typedef std::complex<U> ValueK;
 
    static inline Size computeLastDimensionSize(Size n) { return n; }
};
 
template <typename U>
struct FourierTraits<std::complex<U> const,true> {
    typedef std::complex<U> const ElementX;
    typedef std::complex<U> const ElementK;
    typedef std::complex<U> ValueX;
    typedef std::complex<U> ValueK;
 
    static inline Size computeLastDimensionSize(Size n) { return n; }
};
 
 
} // namespace detail
} // namespace ndarray
 
#endif // !NDARRAY_FFT_FourierTraits_h_INCLUDED