Alignment::IterativeAlignmentEstimator< T > Class Template Reference

#include <iterative_alignment_estimator.h>

Inheritance diagram for Alignment::IterativeAlignmentEstimator< T >:

Public Member Functions
	IterativeAlignmentEstimator (ARotationEstimator< T > &rot_estimator, AShiftEstimator< T > &shift_estimator, BSplineGeoTransformer< T > &interpolator, AMeritComputer< T > &meritComputer, ctpl::thread_pool &threadPool)
void	loadReference (const T *ref)
AlignmentEstimation	compute (const T *others, unsigned iters=3)

Static Protected Member Functions
static void	sApplyTransform (ctpl::thread_pool &pool, const Dimensions &dims, const AlignmentEstimation &estimation, const T *orig, T *copy, bool hasSingleOrig)

Detailed Description

template<typename T>
class Alignment::IterativeAlignmentEstimator< T >

Definition at line 46 of file iterative_alignment_estimator.h.

Constructor & Destructor Documentation

IterativeAlignmentEstimator()

template<typename T>

Alignment::IterativeAlignmentEstimator< T >::IterativeAlignmentEstimator ( ARotationEstimator< T > &  rot_estimator, AShiftEstimator< T > &  shift_estimator, BSplineGeoTransformer< T > &  interpolator, AMeritComputer< T > &  meritComputer, ctpl::thread_pool &  threadPool  )

inline

Definition at line 48 of file iterative_alignment_estimator.h.

                                         :
                m_rot_est(rot_estimator),
                m_shift_est(shift_estimator),
                m_meritComputer(meritComputer),
                m_threadPool(threadPool),
                m_transformer(interpolator) {
        m_sameEstimators = ((void*)&m_shift_est == (void*)&m_rot_est);
        this->check();
    }

Member Function Documentation

compute()

template<typename T>

AlignmentEstimation Alignment::IterativeAlignmentEstimator< T >::compute	(	const T *	others,
		unsigned	iters = 3
	)

loadReference()

template<typename T >

void Alignment::IterativeAlignmentEstimator< T >::loadReference

(

const T *

ref

)

Definition at line 142 of file iterative_alignment_estimator.cpp.

                      {
    m_meritComputer.loadReference(ref);
    m_shift_est.load2DReferenceOneToN(ref);
    if ( ! m_sameEstimators) {
        m_rot_est.loadReference(ref);
    }
}

sApplyTransform()

template<typename T>

void Alignment::IterativeAlignmentEstimator< T >::sApplyTransform ( ctpl::thread_pool &  pool, const Dimensions &  dims, const AlignmentEstimation &  estimation, const T *  orig, T *  copy, bool  hasSingleOrig  )

staticprotected

Definition at line 64 of file iterative_alignment_estimator.cpp.

                                                                                {
    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*>(orig + (hasSingleOrig ? 0 : offset))); // removing const, but data should not be changed
        auto out = MultidimArray<T>(1, z, y, x, copy + offset);
        in.setXmippOrigin();
        out.setXmippOrigin();
        // compensate the movement
        applyGeometry(xmipp_transformation::LINEAR, out, in, estimation.poses.at(signalId), false, xmipp_transformation::DONT_WRAP);
    };

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

---

The documentation for this class was generated from the following files:

• xmipp/libraries/reconstruction/iterative_alignment_estimator.h
• xmipp/libraries/reconstruction/iterative_alignment_estimator.cpp