#include <mpi_subtract_projection.h>

Inheritance diagram for MpiProgSubtractProjection:

Collaboration diagram for MpiProgSubtractProjection:

Public Member Functions
void	defineParams () override

void	readParams () override

void	read (int argc, char **argv, bool reportErrors=true) override

void	preProcess () override

void	startProcessing () override

void	showProgress () override

bool	getImageToProcess (size_t &objId, size_t &objIndex) override

void	finishProcessing () override

void	wait () override
	Wait for the distributor to finish. More...

Public Member Functions inherited from ProgSubtractProjection
void	readParticle (const MDRow &rowIn)
	Read and write methods. More...

void	writeParticle (MDRow &rowOut, FileName, Image< double > &, double, double, double)

void	createMask (const FileName &, Image< double > &, Image< double > &)
	Processing methods. More...

Image< double >	binarizeMask (Projection &) const

Image< double >	invertMask (const Image< double > &)

Image< double >	applyCTF (const MDRow &, Projection &)

void	processParticle (const MDRow &rowIn, int, FourierTransformer &, FourierTransformer &)

MultidimArray< std::complex< double > >	computeEstimationImage (const MultidimArray< double > &, const MultidimArray< double > &, FourierTransformer &)

double	evaluateFitting (const MultidimArray< std::complex< double > > &, const MultidimArray< std::complex< double > > &) const

Matrix1D< double >	checkBestModel (MultidimArray< std::complex< double > > &, const MultidimArray< std::complex< double > > &, const MultidimArray< std::complex< double > > &, const MultidimArray< std::complex< double > > &) const

	ProgSubtractProjection ()
	Empty constructor. More...

	~ProgSubtractProjection ()
	Destructor. More...

void	readParams () override
	Read argument from command line. More...

void	show () const override
	Show. More...

void	defineParams () override
	Define parameters. More...

void	preProcess () override

void	processImage (const FileName &fnImg, const FileName &fnImgOut, const MDRow &rowIn, MDRow &rowOut) override

void	postProcess () override

Public Member Functions inherited from XmippMetadataProgram
MetaData *	getInputMd ()

MetaDataVec &	getOutputMd ()

	XmippMetadataProgram ()
	Empty constructor. More...

virtual int	tryRead (int argc, const char **argv, bool reportErrors=true)

virtual void	init ()

virtual void	setup (MetaData *md, const FileName &o="", const FileName &oroot="", bool applyGeo=false, MDLabel label=MDL_IMAGE)

virtual	~XmippMetadataProgram ()

void	setMode (WriteModeMetaData _mode)

void	setupRowOut (const FileName &fnImgIn, const MDRow &rowIn, const FileName &fnImgOut, MDRow &rowOut) const
	Prepare rowout. More...

virtual void	checkPoint ()
	For very long programs, it may be needed to write checkpoints. More...

virtual void	run ()
	Run over all images. More...

Public Member Functions inherited from XmippProgram
const char *	getParam (const char *param, int arg=0)

const char *	getParam (const char param, const char subparam, int arg=0)

int	getIntParam (const char *param, int arg=0)

int	getIntParam (const char param, const char subparam, int arg=0)

double	getDoubleParam (const char *param, int arg=0)

double	getDoubleParam (const char param, const char subparam, int arg=0)

float	getFloatParam (const char *param, int arg=0)

float	getFloatParam (const char param, const char subparam, int arg=0)

void	getListParam (const char *param, StringVector &list)

int	getCountParam (const char *param)

bool	checkParam (const char *param)

bool	existsParam (const char *param)

void	addParamsLine (const String &line)

void	addParamsLine (const char *line)

ParamDef *	getParamDef (const char *param) const

virtual void	quit (int exit_code=0) const

void	initProgress (size_t total, size_t stepBin=60)

void	setProgress (size_t value=0)

void	endProgress ()

void	processDefaultComment (const char param, const char left)

void	setDefaultComment (const char param, const char comment)

void	setProgramName (const char *name)

void	addUsageLine (const char *line, bool verbatim=false)

void	clearUsage ()

void	addExampleLine (const char *example, bool verbatim=true)

void	addSeeAlsoLine (const char *seeAlso)

void	addKeywords (const char *keywords)

const char *	name () const

virtual void	usage (int verb=0) const

virtual void	usage (const String &param, int verb=2)

int	version () const

virtual void	read (int argc, const char **argv, bool reportErrors=true)

void	read (const String &argumentsLine)

	XmippProgram ()

	XmippProgram (int argc, const char **argv)

virtual	~XmippProgram ()

Public Member Functions inherited from MpiMetadataProgram
	MpiMetadataProgram ()

	MpiMetadataProgram (const MpiMetadataProgram &)=delete

	MpiMetadataProgram (const MpiMetadataProgram &&)=delete

	~MpiMetadataProgram ()

MpiMetadataProgram &	operator= (const MpiMetadataProgram &)=delete

MpiMetadataProgram &	operator= (const MpiMetadataProgram &&)=delete

void	read (int argc, char **argv)

void	defineParams ()

void	readParams ()

void	createTaskDistributor (MetaData &mdIn, size_t blockSize=0)

bool	getTaskToProcess (size_t &objId, size_t &objIndex)

Public Member Functions inherited from XmippMpiProgram
void	read (int argc, char **argv)

virtual int	tryRun ()

Additional Inherited Members
Public Attributes inherited from ProgSubtractProjection
FileName	fnVolR

FileName	fnParticles

FileName	fnImgI

FileName	fnOut

FileName	fnMaskVol

FileName	fnMask

FileName	fnProj

double	sampling

double	padFourier

double	maxResol

double	cirmaskrad

int	sigma

int	limitfreq

int	maxwiIdx

bool	nonNegative

bool	boost

bool	subtract

MultidimArray< int >	wi

Image< double >	V

Image< double >	vM

Image< double >	ivM

Image< double >	M

Image< double >	I

Image< double >	Pctf

Image< double >	iM

Image< double >	Mfinal

Image< double >	Idiff

Image< double >	cirmask

Projection	P

Projection	Pmask

Projection	PmaskVol

FourierFilter	FilterG

const MultidimArray< double > *	ctfImage = nullptr

FourierTransformer	transformerP

FourierTransformer	transformerI

MultidimArray< std::complex< double > >	IFourier

MultidimArray< std::complex< double > >	PFourier

MultidimArray< std::complex< double > >	PFourier0

MultidimArray< std::complex< double > >	PFourier1

MultidimArray< std::complex< double > >	IiMFourier

MultidimArray< std::complex< double > >	PiMFourier

FourierTransformer	transformerIiM

FourierTransformer	transformerPiM

CTFDescription	ctf

FourierFilter	FilterCTF

Image< double >	padp

Image< double >	PmaskI

Image< double >	ImgiM

MultidimArray< std::complex< double > >	ImgiMFourier

MetaDataVec	mdParticles

MDRowVec	row

Matrix1D< double >	roffset

struct Angles	part_angles

bool	disable

int	rank

FourierProjector *	projector

FourierProjector *	projectorMask

Public Attributes inherited from XmippMetadataProgram
FileName	fn_in
	Filenames of input and output Metadata. More...

FileName	fn_out

FileName	baseName

FileName	pathBaseName

FileName	oextBaseName

bool	apply_geo
	Apply geo. More...

size_t	ndimOut
	Output dimensions. More...

size_t	zdimOut

size_t	ydimOut

size_t	xdimOut

DataType	datatypeOut

size_t	mdInSize
	Number of input elements. More...

Public Attributes inherited from XmippProgram
bool	doRun

bool	runWithoutArgs

int	verbose
	Verbosity level. More...

int	debug

Protected Member Functions inherited from XmippMetadataProgram
virtual void	initComments ()

void	show () const override

virtual void	writeOutput ()

virtual void	defineLabelParam ()

Protected Member Functions inherited from XmippProgram
void	defineCommons ()

Protected Member Functions inherited from XmippMpiProgram
void	setNode (const std::shared_ptr< MpiNode > &node)

Protected Attributes inherited from XmippMetadataProgram
WriteModeMetaData	mode
	Metadata writing mode: OVERWRITE, APPEND. More...

FileName	oext
	Output extension and root. More...

FileName	oroot

MDLabel	image_label
	MDLabel to be used to read/write images, usually will be MDL_IMAGE. More...

bool	produces_an_output
	Indicate that a unique final output is produced. More...

bool	produces_a_metadata
	Indicate that the unique final output file is a Metadata. More...

bool	each_image_produces_an_output
	Indicate that an output is produced for each image in the input. More...

bool	allow_apply_geo

bool	decompose_stacks
	Input Metadata will treat a stack file as a set of images instead of a unique file. More...

bool	delete_output_stack
	Delete previous output stack file prior to process images. More...

bool	get_image_info
	Get the input image file dimensions to further operations. More...

bool	save_metadata_stack
	Save the associated output metadata when output file is a stack. More...

bool	track_origin
	Include the original input image filename in the output stack. More...

bool	keep_input_columns
	Keep input metadata columns. More...

bool	remove_disabled
	Remove disabled images from the input selfile. More...

bool	allow_time_bar
	Show process time bar. More...

bool	input_is_metadata
	Input is a metadata. More...

bool	single_image
	Input is a single image. More...

bool	input_is_stack
	Input is a stack. More...

bool	output_is_stack
	Output is a stack. More...

bool	create_empty_stackfile

bool	delete_mdIn

size_t	time_bar_step
	Some time bar related counters. More...

size_t	time_bar_size

size_t	time_bar_done

Protected Attributes inherited from XmippProgram
int	errorCode

ProgramDef *	progDef
	Program definition and arguments parser. More...

std::map< String, CommentList >	defaultComments

int	argc
	Original command line arguments. More...

const char **	argv

Protected Attributes inherited from MpiMetadataProgram
int	blockSize

std::vector< size_t >	imgsId

MpiTaskDistributor *	distributor =nullptr

size_t	first

size_t	last

Protected Attributes inherited from XmippMpiProgram
std::shared_ptr< MpiNode >	node

size_t	nProcs

size_t	numberOfJobs

MPI_Status	status

Detailed Description

Definition at line 31 of file mpi_subtract_projection.h.

Member Function Documentation

◆ defineParams()

void MpiProgSubtractProjection::defineParams ( )

overridevirtual

Function in which the param of each Program are defined.

Reimplemented from XmippMetadataProgram.

Definition at line 28 of file mpi_subtract_projection.cpp.

 {
     ProgSubtractProjection::defineParams();
     MpiMetadataProgram::defineParams();
 }

◆ finishProcessing()

void MpiProgSubtractProjection::finishProcessing ( )

overridevirtual

finishProcessing

Reimplemented from MpiMetadataProgram.

Definition at line 116 of file mpi_subtract_projection.cpp.

 {
     node->gatherMetadatas(getOutputMd(), fn_out);
     MetaDataVec MDaux;
     MDaux.sort(getOutputMd(), MDL_GATHER_ID);
     MDaux.removeLabel(MDL_GATHER_ID);
     getOutputMd() = MDaux;
     if (node->isMaster())
         ProgSubtractProjection::finishProcessing();
 }

◆ getImageToProcess()

bool MpiProgSubtractProjection::getImageToProcess	(	size_t &	objId,
		size_t &	objIndex
	)

inlineoverridevirtual

Reimplemented from XmippMetadataProgram.

Definition at line 41 of file mpi_subtract_projection.h.

     {
         return getTaskToProcess(objId, objIndex);
     }

◆ preProcess()

void MpiProgSubtractProjection::preProcess ( )

overridevirtual

Preprocess

Reimplemented from MpiMetadataProgram.

Definition at line 42 of file mpi_subtract_projection.cpp.

 {
     rank = (int)node->rank;
     ProgSubtractProjection::preProcess();
     // Get the volume padded size from rank 0
     int realSize;
     int origin;
     int realSizeMask;
     int originMask;
     if (node->rank == 0)
     {
         realSize = (int)XSIZE(projector->VfourierRealCoefs);
         origin = STARTINGX(projector->VfourierRealCoefs);
         realSizeMask = (int)XSIZE(projectorMask->VfourierRealCoefs);
         originMask = STARTINGX(projectorMask->VfourierRealCoefs);
     }
 
     MPI_Bcast(&realSize, 1, MPI_INT, 0, MPI_COMM_WORLD);
     MPI_Bcast(&origin, 1, MPI_INT, 0, MPI_COMM_WORLD);
     MPI_Bcast(&(projector->volumePaddedSize), 1, MPI_INT, 0, MPI_COMM_WORLD); 
     MPI_Bcast(&projector->volumeSize, 1, MPI_INT, 0, MPI_COMM_WORLD);
 
     MPI_Bcast(&realSizeMask, 1, MPI_INT, 0, MPI_COMM_WORLD);
     MPI_Bcast(&originMask, 1, MPI_INT, 0, MPI_COMM_WORLD);
     MPI_Bcast(&(projectorMask->volumePaddedSize), 1, MPI_INT, 0, MPI_COMM_WORLD);
     MPI_Bcast(&projectorMask->volumeSize, 1, MPI_INT, 0, MPI_COMM_WORLD);
 
     if (rank != 0)
     {
         projector->VfourierRealCoefs.resizeNoCopy(realSize,realSize,realSize);
         projector->VfourierImagCoefs.resizeNoCopy(realSize,realSize,realSize);
         STARTINGX(projector->VfourierRealCoefs)=STARTINGY(projector->VfourierRealCoefs)=STARTINGZ(projector->VfourierRealCoefs)=origin;
         STARTINGX(projector->VfourierImagCoefs)=STARTINGY(projector->VfourierImagCoefs)=STARTINGZ(projector->VfourierImagCoefs)=origin;
 
         projectorMask->VfourierRealCoefs.resizeNoCopy(realSizeMask,realSizeMask,realSizeMask);
         projectorMask->VfourierImagCoefs.resizeNoCopy(realSizeMask,realSizeMask,realSizeMask);
         STARTINGX(projectorMask->VfourierRealCoefs)=STARTINGY(projectorMask->VfourierRealCoefs)=STARTINGZ(projectorMask->VfourierRealCoefs)=originMask;
         STARTINGX(projectorMask->VfourierImagCoefs)=STARTINGY(projectorMask->VfourierImagCoefs)=STARTINGZ(projectorMask->VfourierImagCoefs)=originMask;
     }
 
     MPI_Bcast(MULTIDIM_ARRAY(projector->VfourierRealCoefs), (int)MULTIDIM_SIZE(projector->VfourierRealCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
     MPI_Bcast(MULTIDIM_ARRAY(projector->VfourierImagCoefs), (int)MULTIDIM_SIZE(projector->VfourierImagCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
     MPI_Bcast(MULTIDIM_ARRAY(projectorMask->VfourierRealCoefs), (int)MULTIDIM_SIZE(projectorMask->VfourierRealCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
     MPI_Bcast(MULTIDIM_ARRAY(projectorMask->VfourierImagCoefs), (int)MULTIDIM_SIZE(projectorMask->VfourierImagCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
 
     if (rank != 0)
     {
         projector->produceSideInfoProjection();
         projectorMask->produceSideInfoProjection();
     }
 
     MetaData &mdIn = *getInputMd();
     mdIn.addLabel(MDL_GATHER_ID);
     mdIn.fillLinear(MDL_GATHER_ID, 1, 1);
     createTaskDistributor(mdIn, blockSize);
 }

◆ read()

void MpiProgSubtractProjection::read	(	int	argc,
		char **	argv,
		bool	reportErrors = `true`
	)

overridevirtual

Read the command line arguments A convenience wrapper

Reimplemented from XmippProgram.

Definition at line 38 of file mpi_subtract_projection.cpp.

 {
     MpiMetadataProgram::read(argc, argv);
 }

◆ readParams()

void MpiProgSubtractProjection::readParams ( )

overridevirtual

Function in which each program will read parameters that it need. If some error occurs the usage will be printed out.

Reimplemented from XmippMetadataProgram.

Definition at line 33 of file mpi_subtract_projection.cpp.

 {
     MpiMetadataProgram::readParams();
     ProgSubtractProjection::readParams();
 }

◆ showProgress()

void MpiProgSubtractProjection::showProgress ( )

overridevirtual

Reimplemented from XmippMetadataProgram.

Definition at line 107 of file mpi_subtract_projection.cpp.

 {
     if (node->rank == 1)
     {
         time_bar_done = first + 1;
         ProgSubtractProjection::showProgress();
     }
 }

◆ startProcessing()

void MpiProgSubtractProjection::startProcessing ( )

overridevirtual

Do some stuff before starting processing in a parallel environment usually this only be executed by master.

Reimplemented from XmippMetadataProgram.

Definition at line 98 of file mpi_subtract_projection.cpp.

 {
     if (node->rank == 1)
     {
         verbose = 1;
         ProgSubtractProjection::startProcessing();
     }
     node->barrierWait();
 }

◆ wait()

void MpiProgSubtractProjection::wait ( )

overridevirtual

Wait for the distributor to finish.

Reimplemented from XmippMetadataProgram.

Definition at line 126 of file mpi_subtract_projection.cpp.

 {
     distributor->wait();
 }

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

xmipp/libraries/parallel/mpi_subtract_projection.h
xmipp/libraries/parallel/mpi_subtract_projection.cpp

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ defineParams()

◆ finishProcessing()

◆ getImageToProcess()

◆ preProcess()

◆ read()

◆ readParams()

◆ showProgress()

◆ startProcessing()

◆ wait()