BSpline Transform Settings

Collaboration diagram for BSpline Transform Settings:

Classes
class	BSplineTransformSettings< T >
class	BSplineGeoTransformer< T >

Functions
void	BSplineTransformSettings< T >::check () const override
	BSplineGeoTransformer< T >::BSplineGeoTransformer ()
	BSplineGeoTransformer< T >::BSplineGeoTransformer (const BSplineGeoTransformer &)=delete
	BSplineGeoTransformer< T >::BSplineGeoTransformer (const BSplineGeoTransformer &&)=delete
virtual	BSplineGeoTransformer< T >::~BSplineGeoTransformer ()
BSplineGeoTransformer &	BSplineGeoTransformer< T >::operator= (const BSplineGeoTransformer &)=delete
BSplineGeoTransformer &	BSplineGeoTransformer< T >::operator= (const BSplineGeoTransformer &&)=delete
virtual void	BSplineGeoTransformer< T >::setSrc (const T *data) override
virtual const T *	BSplineGeoTransformer< T >::getSrc () const
virtual T *	BSplineGeoTransformer< T >::getDest () const override
virtual void	BSplineGeoTransformer< T >::copySrcToDest () override
virtual T *	BSplineGeoTransformer< T >::interpolate (const std::vector< float > &matrices)
virtual void	BSplineGeoTransformer< T >::sum (T *dest, const std::vector< float > &weights, size_t firstN, T norm)
virtual void	BSplineGeoTransformer< T >::initialize (bool doAllocation) override
virtual void	BSplineGeoTransformer< T >::release ()
virtual void	BSplineGeoTransformer< T >::setDefault ()
virtual void	BSplineGeoTransformer< T >::check () override
virtual bool	BSplineGeoTransformer< T >::canBeReused (const BSplineTransformSettings< T > &s) const override

Variables
bool	BSplineTransformSettings< T >::keepSrcCopy

Detailed Description

Function Documentation

BSplineGeoTransformer() [1/3]

template<typename T>

BSplineGeoTransformer< T >::BSplineGeoTransformer ( )

inline

Definition at line 50 of file bspline_geo_transformer.h.

                            {
        setDefault();
    }

BSplineGeoTransformer() [2/3]

template<typename T>

BSplineGeoTransformer< T >::BSplineGeoTransformer ( const BSplineGeoTransformer< T > &  )

delete

BSplineGeoTransformer() [3/3]

template<typename T>

BSplineGeoTransformer< T >::BSplineGeoTransformer ( const BSplineGeoTransformer< T > &&  )

delete

canBeReused()

template<typename T >

bool BSplineGeoTransformer< T >::canBeReused ( const BSplineTransformSettings< T > &  s ) const

overrideprotectedvirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Definition at line 89 of file bspline_geo_transformer.cpp.

                                                                                     {
    bool result = true;
    if ( ! this->isInitialized()) {
        return false;
    }
    auto &sOrig = this->getSettings();
    result &= sOrig.dims.size() >= s.dims.size(); // previously, we needed more space
    result &= !(( ! sOrig.keepSrcCopy) && s.keepSrcCopy); // we have a problem if now we need to make a copy and before not

    return result;
}

check() [1/2]

template<typename T>

void BSplineTransformSettings< T >::check ( ) const

inlineoverridevirtual

Reimplemented from GeoTransformerSettings< T >.

Definition at line 41 of file bspline_geo_transformer.h.

                                {
        GeoTransformerSettings<T>::check();
    }

check() [2/2]

template<typename T >

void BSplineGeoTransformer< T >::check ( )

overrideprotectedvirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Definition at line 49 of file bspline_geo_transformer.cpp.

                                     {
    const auto &s = this->getSettings();
    if (InterpolationDegree::Linear != s.degree) {
        REPORT_ERROR(ERR_NOT_IMPLEMENTED, "Only linear interpolation is available");
    }
    if (s.doWrap) {
        REPORT_ERROR(ERR_NOT_IMPLEMENTED, "Wrapping is not yet implemented");
    }
    if (InterpolationType::NToN != s.type) {
        REPORT_ERROR(ERR_NOT_IMPLEMENTED, "Only NToN is currently implemented");
    }
}

copySrcToDest()

template<typename T >

void BSplineGeoTransformer< T >::copySrcToDest ( )

overridevirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 64 of file bspline_geo_transformer.cpp.

                                             {
    bool isReady = this->isInitialized()
            && this->isSrcSet();
    if ( ! isReady) {
        REPORT_ERROR(ERR_LOGIC_ERROR, "Instance is either not initialized or the 'src' has not been set.");
    }
    memcpy(m_dest.get(),
            m_src,
            this->getSettings().dims.size() * sizeof(T));
}

getDest()

template<typename T>

virtual T* BSplineGeoTransformer< T >::getDest ( ) const

inlineoverridevirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 71 of file bspline_geo_transformer.h.

                                        {
        return m_dest.get();
    }

getSrc()

template<typename T>

virtual const T* BSplineGeoTransformer< T >::getSrc ( ) const

inlinevirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 67 of file bspline_geo_transformer.h.

                                    {
        return m_src;
    }

initialize()

template<typename T >

void BSplineGeoTransformer< T >::initialize ( bool  doAllocation )

overrideprotectedvirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Definition at line 31 of file bspline_geo_transformer.cpp.

                                                           {
    const auto &s = this->getSettings();

    for (auto &hw : s.hw) {
        if ( ! dynamic_cast<CPU*>(hw)) {
            REPORT_ERROR(ERR_LOGIC_ERROR, "Instance of CPU is expected");
        }
    }
//    m_threadPool.resize(s.hw.size()); // FIXME DS set to requested number of thread
    m_threadPool.resize(CPU::findCores());

    if (doAllocation) {
        release();
        m_dest = std::unique_ptr<T[]>(new T[s.dims.size()]);
    }
}

interpolate()

template<typename T >

T * BSplineGeoTransformer< T >::interpolate ( const std::vector< float > &  matrices )

virtual

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 107 of file bspline_geo_transformer.cpp.

                                                                         {
    bool isReady = this->isInitialized()
            && this->isSrcSet();
    if ( ! isReady) {
        REPORT_ERROR(ERR_LOGIC_ERROR, "Instance is either not initialized or the 'original' has not been set.");
    }
    const Dimensions dims = this->getSettings().dims;
    const size_t n = dims.n();
    const size_t z = dims.z();
    const size_t y = dims.y();
    const size_t x = dims.x();

    auto futures = std::vector<std::future<void>>();

    auto workload = [&](int id, size_t signalId){
        size_t offset = signalId * dims.sizeSingle();
        auto in = MultidimArray<T>(1, z, y, x, const_cast<T*>(m_src + offset)); // removing const, but data should not be changed
        auto out = MultidimArray<T>(1, z, y, x, m_dest.get() + offset);
        in.setXmippOrigin();
        out.setXmippOrigin();
        // compensate the movement
        Matrix2D<double> m(3,3);
        const float *f = matrices.data() + (9 * signalId);
        for (int i = 0; i < 9; ++i) {
            m.mdata[i] = f[i];
        }
        applyGeometry(xmipp_transformation::LINEAR, out, in, m, true, xmipp_transformation::DONT_WRAP);
    };

    for (size_t i = 0; i < n; ++i) {
        futures.emplace_back(m_threadPool.push(workload, i));
    }
    for (auto &f : futures) {
        f.get();
    }
    return m_dest.get();
}

operator=() [1/2]

template<typename T>

BSplineGeoTransformer& BSplineGeoTransformer< T >::operator= ( const BSplineGeoTransformer< T > &  )

delete

operator=() [2/2]

template<typename T>

BSplineGeoTransformer& BSplineGeoTransformer< T >::operator= ( const BSplineGeoTransformer< T > &&  )

delete

release()

template<typename T >

void BSplineGeoTransformer< T >::release ( )

protectedvirtual

Definition at line 76 of file bspline_geo_transformer.cpp.

                                       {
    m_dest.release();
    setDefault();
}

setDefault()

template<typename T >

void BSplineGeoTransformer< T >::setDefault ( )

protectedvirtual

Definition at line 82 of file bspline_geo_transformer.cpp.

                                          {
    m_dest.reset();
    m_src = nullptr;
    m_threadPool.resize(1);
}

setSrc()

template<typename T>

virtual void BSplineGeoTransformer< T >::setSrc ( const T *  data )

inlineoverridevirtual

Implements AGeoTransformer< BSplineTransformSettings< T >, T >.

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 62 of file bspline_geo_transformer.h.

                                                {
        m_src = data;
        this->setIsSrcSet(nullptr != data);
    }

sum()

template<typename T >

void BSplineGeoTransformer< T >::sum ( T *  dest, const std::vector< float > &  weights, size_t  firstN, T  norm  )

virtual

Reimplemented in CudaBSplineGeoTransformer< T >.

Definition at line 102 of file bspline_geo_transformer.cpp.

                                                                                                  {
    REPORT_ERROR(ERR_NOT_IMPLEMENTED, "This functionality is not yet available.");
}

~BSplineGeoTransformer()

template<typename T>

virtual BSplineGeoTransformer< T >::~BSplineGeoTransformer ( )

inlinevirtual

Definition at line 56 of file bspline_geo_transformer.h.

                                     {
        release();
    }

Variable Documentation

keepSrcCopy

template<typename T>

bool BSplineTransformSettings< T >::keepSrcCopy

Definition at line 39 of file bspline_geo_transformer.h.