forward_zernike_images_priors.cpp File Reference

#include "forward_zernike_images_priors.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 "fstream"

Include dependency graph for forward_zernike_images_priors.cpp:

Go to the source code of this file.

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

Function Documentation

continuousZernikePriorsCost()

double continuousZernikePriorsCost	(	double *	x,
		void *	_prm
	)

Definition at line 522 of file forward_zernike_images_priors.cpp.

{
    ProgForwardZernikeImagesPriors *prm=(ProgForwardZernikeImagesPriors *)_prm;
    int idx = prm->priors.size();
    // TODO: Optimize parameters for each image (not sharing)
    // prm->deltaDefocusU[0]=x[idx+6]; prm->deltaDefocusU[1]=x[idx+6]; 
    // prm->deltaDefocusV[0]=x[idx+7]; prm->deltaDefocusV[1]=x[idx+7]; 
    // prm->deltaDefocusAngle[0]=x[idx+8]; prm->deltaDefocusAngle[1]=x[idx+8];

    switch (prm->num_images)
    {
    case 2:
        prm->deltaX[0] = x[idx + 1];
        prm->deltaY[0] = x[idx + 3];
        prm->deltaRot[0] = x[idx + 5];
        prm->deltaTilt[0] = x[idx + 7];
        prm->deltaPsi[0] = x[idx + 9];
        // prm->deltaDefocusU[0]=x[idx + 11];
        // prm->deltaDefocusV[0]=x[idx + 13];
        // prm->deltaDefocusAngle[0]=x[idx + 15];

        prm->deltaX[1] = x[idx + 2];
        prm->deltaY[1] = x[idx + 4];
        prm->deltaRot[1] = x[idx + 6];
        prm->deltaTilt[1] = x[idx + 8];
        prm->deltaPsi[1] = x[idx + 10];
        // prm->deltaDefocusU[1]=x[idx + 12];
        // prm->deltaDefocusV[1]=x[idx + 14];
        // prm->deltaDefocusAngle[1]=x[idx + 16];

        MAT_ELEM(prm->A1, 0, 2) = prm->old_shiftX[0] + prm->deltaX[0];
        MAT_ELEM(prm->A1, 1, 2) = prm->old_shiftY[0] + prm->deltaY[0];
        MAT_ELEM(prm->A1, 0, 0) = 1;
        MAT_ELEM(prm->A1, 0, 1) = 0;
        MAT_ELEM(prm->A1, 1, 0) = 0;
        MAT_ELEM(prm->A1, 1, 1) = 1;

        MAT_ELEM(prm->A2, 0, 2) = prm->old_shiftX[1] + prm->deltaX[1];
        MAT_ELEM(prm->A2, 1, 2) = prm->old_shiftY[1] + prm->deltaY[1];
        MAT_ELEM(prm->A2, 0, 0) = 1;
        MAT_ELEM(prm->A2, 0, 1) = 0;
        MAT_ELEM(prm->A2, 1, 0) = 0;
        MAT_ELEM(prm->A2, 1, 1) = 1;
        break;

    case 3:
        prm->deltaX[0] = x[idx + 1];
        prm->deltaY[0] = x[idx + 4];
        prm->deltaRot[0] = x[idx + 7];
        prm->deltaTilt[0] = x[idx + 10];
        prm->deltaPsi[0] = x[idx + 13];
        // prm->deltaDefocusU[0]=x[idx + 16];
        // prm->deltaDefocusV[0]=x[idx + 19];
        // prm->deltaDefocusAngle[0]=x[idx + 22];

        prm->deltaX[1] = x[idx + 2];
        prm->deltaY[1] = x[idx + 5];
        prm->deltaRot[1] = x[idx + 8];
        prm->deltaTilt[1] = x[idx + 11];
        prm->deltaPsi[1] = x[idx + 14];
        // prm->deltaDefocusU[1]=x[idx + 17];
        // prm->deltaDefocusV[1]=x[idx + 20];
        // prm->deltaDefocusAngle[1]=x[idx + 23];

        prm->deltaX[2] = x[idx + 3];
        prm->deltaY[2] = x[idx + 6];
        prm->deltaRot[2] = x[idx + 9];
        prm->deltaTilt[2] = x[idx + 12];
        prm->deltaPsi[2] = x[idx + 15];
        // prm->deltaDefocusU[2]=x[idx + 18];
        // prm->deltaDefocusV[2]=x[idx + 21];
        // prm->deltaDefocusAngle[2]=x[idx + 24];

        MAT_ELEM(prm->A1, 0, 2) = prm->old_shiftX[0] + prm->deltaX[0];
        MAT_ELEM(prm->A1, 1, 2) = prm->old_shiftY[0] + prm->deltaY[0];
        MAT_ELEM(prm->A1, 0, 0) = 1;
        MAT_ELEM(prm->A1, 0, 1) = 0;
        MAT_ELEM(prm->A1, 1, 0) = 0;
        MAT_ELEM(prm->A1, 1, 1) = 1;

        MAT_ELEM(prm->A2, 0, 2) = prm->old_shiftX[1] + prm->deltaX[1];
        MAT_ELEM(prm->A2, 1, 2) = prm->old_shiftY[1] + prm->deltaY[1];
        MAT_ELEM(prm->A2, 0, 0) = 1;
        MAT_ELEM(prm->A2, 0, 1) = 0;
        MAT_ELEM(prm->A2, 1, 0) = 0;
        MAT_ELEM(prm->A2, 1, 1) = 1;

        MAT_ELEM(prm->A3, 0, 2) = prm->old_shiftX[2] + prm->deltaX[2];
        MAT_ELEM(prm->A3, 1, 2) = prm->old_shiftY[2] + prm->deltaY[2];
        MAT_ELEM(prm->A3, 0, 0) = 1;
        MAT_ELEM(prm->A3, 0, 1) = 0;
        MAT_ELEM(prm->A3, 1, 0) = 0;
        MAT_ELEM(prm->A3, 1, 1) = 1;
        break;

    
    default:
        prm->deltaX[0] = x[idx + 1];
        prm->deltaY[0] = x[idx + 2];
        prm->deltaRot[0] = x[idx + 3];
        prm->deltaTilt[0] = x[idx + 4];
        prm->deltaPsi[0] = x[idx + 5];
        prm->deltaDefocusU[0]=x[idx + 6];
        prm->deltaDefocusV[0]=x[idx + 7];
        prm->deltaDefocusAngle[0]=x[idx + 8];

        MAT_ELEM(prm->A1, 0, 2) = prm->old_shiftX[0] + prm->deltaX[0];
        MAT_ELEM(prm->A1, 1, 2) = prm->old_shiftY[0] + prm->deltaY[0];
        MAT_ELEM(prm->A1, 0, 0) = 1;
        MAT_ELEM(prm->A1, 0, 1) = 0;
        MAT_ELEM(prm->A1, 1, 0) = 0;
        MAT_ELEM(prm->A1, 1, 1) = 1;
        break;
    }

    return prm->transformImageSph(x);

}