angular_sph_alignment_gpu.cpp File Reference

#include "angular_sph_alignment_gpu.h"
#include "core/transformations.h"
#include "core/xmipp_image_extension.h"
#include "core/xmipp_image_generic.h"
#include "data/projection.h"
#include "data/mask.h"
#include "reconstruction_cuda/gpu.h"
#include "data/fourier_filter.h"

Include dependency graph for angular_sph_alignment_gpu.cpp:

Go to the source code of this file.

Functions
double	continuousSphCost (double *x, void *_prm)

Function Documentation

continuousSphCost()

double continuousSphCost	(	double *	x,
		void *	_prm
	)

Definition at line 297 of file angular_sph_alignment_gpu.cpp.

{
    auto *prm=(ProgAngularSphAlignmentGpu *)_prm;
    int idx = 3*(prm->vecSize);
    double deltax=x[idx+1];
    double deltay=x[idx+2];
    double deltaRot=x[idx+3];
    double deltaTilt=x[idx+4];
    double deltaPsi=x[idx+5];
    double deltaDefocusU=x[idx+6];
    double deltaDefocusV=x[idx+7];
    double deltaDefocusAngle=x[idx+8];
    if (prm->maxShift>0 && deltax*deltax+deltay*deltay>prm->maxShift*prm->maxShift)
        return 1e38;
    if (prm->maxAngularChange>0 && (fabs(deltaRot)>prm->maxAngularChange || fabs(deltaTilt)>prm->maxAngularChange || fabs(deltaPsi)>prm->maxAngularChange))
        return 1e38;

    MAT_ELEM(prm->A,0,2)=prm->old_shiftX+deltax;
    MAT_ELEM(prm->A,1,2)=prm->old_shiftY+deltay;
    MAT_ELEM(prm->A,0,0)=1;
    MAT_ELEM(prm->A,0,1)=0;
    MAT_ELEM(prm->A,1,0)=0;
    MAT_ELEM(prm->A,1,1)=1;

    return prm->tranformImageSph(x,prm->old_rot+deltaRot, prm->old_tilt+deltaTilt, prm->old_psi+deltaPsi,
            prm->A, deltaDefocusU, deltaDefocusV, deltaDefocusAngle);
}