image_operate.cpp File Reference

#include "image_operate.h"
#include "core/transformations.h"
#include "core/xmipp_fft.h"
#include "data/numerical_tools.h"
#include "core/utils/memory_utils.h"

Include dependency graph for image_operate.cpp:

Go to the source code of this file.

Classes
class	MinusAdjustedPrm

Functions
void	minus (Image< double > &op1, const Image< double > &op2)
double	minusAdjusted_L1 (double *x, void *_prm)
void	minusAdjusted (Image< double > &op1, const Image< double > &op2)
void	imageDotProduct (Image< double > &op1, const Image< double > &op2)
void	dropOut (Image< double > &op1)
void	plus (Image< double > &op1, const Image< double > &op2)
void	mult (Image< double > &op1, const Image< double > &op2)
void	divide (Image< double > &op1, const Image< double > &op2)
void	min (Image< double > &op1, const Image< double > &op2)
void	max (Image< double > &op1, const Image< double > &op2)
void	compare (Image< double > &op1, const Image< double > &op2)
void	eq (Image< double > &op1, const Image< double > &op2)
	Be careful with integer images for relational operations...due to double comparisons. More...
void	ne (Image< double > &op1, const Image< double > &op2)
void	lt (Image< double > &op1, const Image< double > &op2)
void	le (Image< double > &op1, const Image< double > &op2)
void	gt (Image< double > &op1, const Image< double > &op2)
void	ge (Image< double > &op1, const Image< double > &op2)
void	sqrt (Image< double > &op)
void	abs (Image< double > &op)
void	log (Image< double > &op)
void	log10 (Image< double > &op)
void	power (Image< double > &op)
void	getSlice (Image< double > &op)
void	radialAvg (Image< double > &op)
void	psdRadialAvg (Image< double > &op)
void	reset (Image< double > &op)

Variables
double	pDropout
double	powerExp = 2
int	nSlice
char	axis
FileName	fnOut

Function Documentation

abs()

void abs

(

Image< double > &

op

)

Definition at line 219 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp)
    {
        dAi(mOp, n) = ABS(dAi(mOp, n));
    }
}

compare()

void compare	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 138 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) == dAi(mOp2, n) ? 0 : (dAi(mOp1, n) < dAi(mOp2, n) ? -1 : 1 );
    }
}

divide()

void divide	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 113 of file image_operate.cpp.

{
    op1() /= op2();
}

dropOut()

void dropOut

(

Image< double > &

op1

)

Definition at line 92 of file image_operate.cpp.

{
    if (pDropout>0.0 && pDropout<1.0)
    {
        MultidimArray<double> &mOp1 = op1();
        FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
            if (rnd_unif()<pDropout)
                dAi(mOp1, n) = 0.0;
    }
}

eq()

void eq	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Be careful with integer images for relational operations...due to double comparisons.

Definition at line 149 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) == dAi(mOp2, n) ? 1 : 0;
    }
}

ge()

void ge	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 199 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) >= dAi(mOp2, n) ? 1 : 0;
    }
}

getSlice()

void getSlice

(

Image< double > &

op

)

Definition at line 258 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    MultidimArray<double> imAux;

    mOp.getSlice(nSlice, imAux, axis);
    mOp = imAux;
}

gt()

void gt	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 189 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) > dAi(mOp2, n) ? 1 : 0;
    }
}

imageDotProduct()

void imageDotProduct	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 85 of file image_operate.cpp.

{
    double dot=op1().dotProduct(op2());
    std::cout << "<" << op1.name() << "," << op2.name() << ">=" << dot << std::endl;
}

le()

void le	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 179 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) <= dAi(mOp2, n) ? 1 : 0;
    }
}

log()

void log

(

Image< double > &

op

)

Definition at line 228 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp)
    {
        dAi(mOp, n) = log(dAi(mOp, n));
    }
}

log10()

void log10

(

Image< double > &

op

)

Definition at line 237 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp)
    {
        dAi(mOp, n) = log10(dAi(mOp, n));
    }
}

lt()

void lt	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 169 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) < dAi(mOp2, n) ? 1 : 0;
    }
}

max()

void max	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 128 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = XMIPP_MAX(dAi(mOp1, n), dAi(mOp2, n));
    }
}

min()

void min	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 118 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = XMIPP_MIN(dAi(mOp1, n), dAi(mOp2, n));
    }
}

minus()

void minus	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 33 of file image_operate.cpp.

{
    op1() -= op2();
}

minusAdjusted()

void minusAdjusted	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 60 of file image_operate.cpp.

{
    MultidimArray<double> &pI1=op1();
    const MultidimArray<double> &pI2=op2();

    MinusAdjustedPrm prm;
    prm.I1=&op1();
    prm.I2=&op2();

    Matrix1D<double> p(2);
    Matrix1D<double> steps(2);
    p(0)=1; // a in I'=a*I+b
    p(1)=0; // b in I'=a*I+b
    steps.initConstant(1);
    double cost;
    int iter;
    powellOptimizer(p, 1, 2, &minusAdjusted_L1, &prm, 0.01, cost, iter, steps, false);

    double a=p(0);
    double b=p(1);
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(pI1)
        DIRECT_MULTIDIM_ELEM(pI1,n)-=(a*DIRECT_MULTIDIM_ELEM(pI2,n)+b);
}

minusAdjusted_L1()

double minusAdjusted_L1	(	double *	x,
		void *	_prm
	)

Definition at line 45 of file image_operate.cpp.

{
    double a=x[1];
    double b=x[2];

    double retval=0;
    auto *prm = (MinusAdjustedPrm *) _prm;
    const MultidimArray<double> &pI1=*(prm->I1);
    const MultidimArray<double> &pI2=*(prm->I2);
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(pI1)
        retval+=fabs(DIRECT_MULTIDIM_ELEM(pI1,n)-(a*DIRECT_MULTIDIM_ELEM(pI2,n)+b));
    return retval;
}

mult()

void mult	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 108 of file image_operate.cpp.

{
    op1() *= op2();
}

ne()

void ne	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 159 of file image_operate.cpp.

{
    MultidimArray<double> &mOp1 = op1();
    const MultidimArray<double> &mOp2 = op2();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp1)
    {
        dAi(mOp1, n) = dAi(mOp1, n) != dAi(mOp2, n) ? 1 : 0;
    }
}

plus()

void plus	(	Image< double > &	op1,
		const Image< double > &	op2
	)

Definition at line 103 of file image_operate.cpp.

{
    op1() += op2();
}

power()

void power

(

Image< double > &

op

)

Definition at line 247 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp)
    {
        dAi(mOp, n) = pow(dAi(mOp, n), powerExp);
    }
}

psdRadialAvg()

void psdRadialAvg

(

Image< double > &

op

)

Definition at line 288 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    CenterFFT(mOp,true);
    mOp.selfLog10();
    mOp.setXmippOrigin();
    Matrix1D<int> center(3);
    center.initZeros();
    MultidimArray<double> radial_mean;
    MultidimArray<int> radial_count;
    radialAverage(mOp, center, radial_mean, radial_count);
    radial_mean.write((fnOut.withoutExtension()).addExtension("txt"));

    int my_rad;
    FOR_ALL_ELEMENTS_IN_ARRAY3D(mOp)
    {
        my_rad = (int)floor(sqrt((double)(i * i + j * j + k * k)));
        op(k, i, j) = radial_mean(my_rad);
    }
}

reset()

void reset

(

Image< double > &

op

)

Definition at line 309 of file image_operate.cpp.

{
    op().initZeros();
}

sqrt()

void sqrt

(

Image< double > &

op

)

Definition at line 210 of file image_operate.cpp.

{
    MultidimArray<double> &mOp = op();
    FOR_ALL_DIRECT_ELEMENTS_IN_MULTIDIMARRAY(mOp)
    {
        dAi(mOp, n) = sqrt(dAi(mOp, n));
    }
}

Variable Documentation

axis

char axis

Definition at line 257 of file image_operate.cpp.

fnOut

FileName fnOut

Definition at line 268 of file image_operate.cpp.

nSlice

int nSlice

Definition at line 256 of file image_operate.cpp.

pDropout

double pDropout

Definition at line 91 of file image_operate.cpp.

powerExp

double powerExp = 2

Definition at line 246 of file image_operate.cpp.