EMAN::RotationalAligner Class Reference
[a function or class that is CUDA enabled]

rotational alignment using angular correlation More...

#include <aligner.h>

Inheritance diagram for EMAN::RotationalAligner:

[legend]

Collaboration diagram for EMAN::RotationalAligner:

[legend]

List of all members.

Public Member Functions
virtual EMData *	align (EMData *this_img, EMData *to_img, const string &cmp_name="dot", const Dict &cmp_params=Dict()) const
	To align 'this_img' with another image passed in through its parameters.
virtual EMData *	align (EMData *this_img, EMData *to_img) const
virtual string	get_name () const
	Get the Aligner's name.
virtual string	get_desc () const
virtual TypeDict	get_param_types () const
Static Public Member Functions
static Aligner *	NEW ()
static EMData *	align_180_ambiguous (EMData *this_img, EMData *to_img, int rfp_mode=0)
Static Public Attributes
static const string	NAME = "rotational"

---

Detailed Description

rotational alignment using angular correlation

Parameters:

rfp_mode

Either 0,1 or 2. A temporary flag for testing the rotational foot print. O is the original eman1 way. 1 is just using calc_ccf without padding. 2 is using calc_mutual_correlation without padding

Definition at line 217 of file aligner.h.

---

Member Function Documentation

EMData * RotationalAligner::align	(	EMData *	this_img,
		EMData *	to_img,
		const string &	cmp_name = "dot",
		const Dict &	cmp_params = Dict()
	)			const [virtual]

To align 'this_img' with another image passed in through its parameters.

The alignment uses a user-given comparison method to compare the two images. If none is given, a default one is used.

Parameters:

	this_img	The image to be compared.
	to_img	'this_img" is aligned with 'to_img'.
	cmp_name	The comparison method to compare the two images.
	cmp_params	The parameter dictionary for comparison method.

Returns:

The aligned image.

Implements EMAN::Aligner.

Definition at line 254 of file aligner.cpp.

References align_180_ambiguous(), EMAN::EMData::cmp(), EMAN::EMData::get_attr(), EMAN::Transform::get_rotation(), InvalidParameterException, EMAN::Aligner::params, EMAN::EMData::process(), EMAN::EMData::set_attr(), and EMAN::Dict::set_default().

Referenced by align().

{
        if (!to) throw InvalidParameterException("Can not rotational align - the image to align to is NULL");

        // Perform 180 ambiguous alignment
        int rfp_mode = params.set_default("rfp_mode",0);
        EMData* rot_aligned = RotationalAligner::align_180_ambiguous(this_img,to,rfp_mode);
        Transform * tmp = rot_aligned->get_attr("xform.align2d");
        Dict rot = tmp->get_rotation("2d");
        float rotate_angle_solution = rot["alpha"];
        delete tmp;

        EMData *rot_align_180 = rot_aligned->process("math.rotate.180");

        // Generate the comparison metrics for both rotational candidates
        float rot_cmp = rot_aligned->cmp(cmp_name, to, cmp_params);
        float rot_180_cmp = rot_align_180->cmp(cmp_name, to, cmp_params);
        // Decide on the result
        float score = 0.0;
        EMData* result = NULL;
        if (rot_cmp < rot_180_cmp){
                result = rot_aligned;
                score = rot_cmp;
                delete rot_align_180; rot_align_180 = 0;
        } else {
                result = rot_align_180;
                score = rot_180_cmp;
                delete rot_aligned; rot_aligned = 0;
                rotate_angle_solution = rotate_angle_solution-180.0f;
        }

//      Transform* t = get_align_attr("xform.align2d",result);
//      t->set_rotation(Dict("type","2d","alpha",rotate_angle_solution));
        Transform tmp2(Dict("type","2d","alpha",rotate_angle_solution));
        result->set_attr("xform.align2d",&tmp2);
        return result;
}

virtual EMData* EMAN::RotationalAligner::align	(	EMData *	this_img,
		EMData *	to_img
	)			const [inline, virtual]

Implements EMAN::Aligner.

Definition at line 223 of file aligner.h.

References align().

                {
                        return align(this_img, to_img, "", Dict());
                }

virtual string EMAN::RotationalAligner::get_name

(

)

const [inline, virtual]

Get the Aligner's name.

Each Aligner is identified by a unique name.

Returns:

The Aligner's name.

Implements EMAN::Aligner.

Definition at line 228 of file aligner.h.

References NAME.

                {
                        return NAME;
                }

virtual string EMAN::RotationalAligner::get_desc

(

)

const [inline, virtual]

Implements EMAN::Aligner.

Definition at line 233 of file aligner.h.

                {
                        return "Performs rotational alignment,works accurately if the image is precentered, normally called internally in combination with translational and flip alignment";
                }

static Aligner* EMAN::RotationalAligner::NEW

(

)

[inline, static]

Definition at line 238 of file aligner.h.

                {
                        return new RotationalAligner();
                }

EMData * RotationalAligner::align_180_ambiguous	(	EMData *	this_img,
		EMData *	to_img,
		int	rfp_mode = 0
	)			[static]

Definition at line 207 of file aligner.cpp.

References EMAN::EMData::calc_ccfx(), data, EMAN::Util::find_max(), EMAN::EMData::get_data(), EMAN::EMData::get_xsize(), EMAN::EMData::get_ysize(), InvalidParameterException, EMAN::EMData::make_rotational_footprint(), EMAN::EMData::make_rotational_footprint_cmc(), EMAN::EMData::make_rotational_footprint_e1(), EMAN::EMData::process(), and EMAN::EMData::set_attr().

Referenced by EMAN::RotateTranslateAligner::align(), and align().

                                                                                            {

        // Make translationally invariant rotational footprints
        EMData* this_img_rfp, * to_rfp;
        if (rfp_mode == 0) {
                this_img_rfp = this_img->make_rotational_footprint_e1();
                to_rfp = to->make_rotational_footprint_e1();
        } else if (rfp_mode == 1) {
                this_img_rfp = this_img->make_rotational_footprint();
                to_rfp = to->make_rotational_footprint();
        } else if (rfp_mode == 2) {
                this_img_rfp = this_img->make_rotational_footprint_cmc();
                to_rfp = to->make_rotational_footprint_cmc();
        } else {
                throw InvalidParameterException("rfp_mode must be 0,1 or 2");
        }
        int this_img_rfp_nx = this_img_rfp->get_xsize();

        // Do row-wise correlation, returning a sum.
        EMData *cf = this_img_rfp->calc_ccfx(to_rfp, 0, this_img->get_ysize());

        // Delete them, they're no longer needed
        delete this_img_rfp; this_img_rfp = 0;
        delete to_rfp; to_rfp = 0;

        // Now solve the rotational alignment by finding the max in the column sum
        float *data = cf->get_data();
        float peak = 0;
        int peak_index = 0;
        Util::find_max(data, this_img_rfp_nx, &peak, &peak_index);

        if( cf ) {
                delete cf;
                cf = 0;
        }
        float rot_angle = (float) (peak_index * 180.0f / this_img_rfp_nx);

        // Return the result
        Transform tmp(Dict("type","2d","alpha",rot_angle));
        cf=this_img->process("xform",Dict("transform",(Transform*)&tmp));
//      Transform* t = get_set_align_attr("xform.align2d",cf,this_img);
//      Dict d("type","2d","alpha",rot_angle);
//      t->set_rotation(d);
        cf->set_attr("xform.align2d",&tmp);
        return cf;
}

virtual TypeDict EMAN::RotationalAligner::get_param_types

(

)

const [inline, virtual]

Implements EMAN::Aligner.

Definition at line 245 of file aligner.h.

References EMAN::EMObject::INT, and EMAN::TypeDict::put().

                {
                        TypeDict d;
                        d.put("rfp_mode", EMObject::INT,"Either 0,1 or 2. A temporary flag for testing the rotational foot print. O is the original eman1 way. 1 is just using calc_ccf without padding. 2 is using calc_mutual_correlation without padding.");
                        return d;
                }

---

Member Data Documentation

const string RotationalAligner::NAME = "rotational" [static]

Definition at line 252 of file aligner.h.

Referenced by get_name().

---

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

• aligner.h
• aligner.cpp