casa6

#include <imageanalysis/ImageAnalysis/ImageFFTer.h>

​

#include <imageanalysis/ImageAnalysis/ImageFFT.h>

​

namespace casa {

​

template<class T> ImageFFTer<T>::ImageFFTer(

    const SPCIIT image,

    const casacore::Record *const region,

    const casacore::String& maskInp, const casacore::Vector<casacore::uInt>& axes

) : ImageTask<T>(

        image, "", region, "", "", "",

        maskInp, "", false

    ),

    _axes(axes), _real(), _imag(), _amp(),

    _phase(), _complex() {

    this->_construct();

}

​

template<class T> void ImageFFTer<T>::fft() const {

    *this->_getLog() << casacore::LogOrigin(getClass(), __FUNCTION__);

    ThrowIf(

        _real.empty() && _imag.empty() && _amp.empty()

        && _phase.empty() && _complex.empty(),

        "No output image names have been specified"

    );

    _checkExists(_real);

    _checkExists(_imag);

    _checkExists(_amp);

    _checkExists(_phase);

    _checkExists(_complex);

    auto subImage = SubImageFactory<T>::createSubImageRO(

        *this->_getImage(), *this->_getRegion(), this->_getMask(),

        this->_getLog().get(), casacore::AxesSpecifier(), this->_getStretch()

    );

    ImageFFT<T> fft;

    if (_axes.size() == 0) {

        *this->_getLog() << casacore::LogIO::NORMAL

            << "FFT the direction coordinate" << casacore::LogIO::POST;

        fft.fftsky(*subImage);

    }

    else {

        // Set vector of bools specifying axes

        casacore::Vector<casacore::Bool> which(subImage->ndim(), false);

        for(casacore::uInt i: _axes) {

            which(_axes(i)) = true;

        }

        *this->_getLog() << casacore::LogIO::NORMAL << "FFT zero-based axes "

            << _axes << casacore::LogIO::POST;

        fft.fft(*subImage, which);

    }

​

    // casacore::String maskName("");

    _createOutputImages(*subImage, fft);

}

​

template <class T>

void ImageFFTer<T>::_createOutputImages(

    const casacore::SubImage<T>& subImage, const ImageFFT<T>& fft

) const {

    if (

        ! (_real.empty() && _imag.empty() && _amp.empty() && _phase.empty())

    ) {