Xmipp  v3.23.11-Nereus
Public Member Functions | List of all members
MpiProgAngularContinuousAssign2 Class Reference
Inheritance diagram for MpiProgAngularContinuousAssign2:
Inheritance graph
[legend]
Collaboration diagram for MpiProgAngularContinuousAssign2:
Collaboration graph
[legend]

Public Member Functions

void defineParams ()
 
void readParams ()
 
void read (int argc, char **argv, bool reportErrors=true)
 
void preProcess ()
 
void startProcessing ()
 
void showProgress ()
 
virtual bool getImageToProcess (size_t &objId, size_t &objIndex) override
 
void finishProcessing ()
 
void wait ()
 Wait for the distributor to finish. More...
 
- Public Member Functions inherited from ProgAngularContinuousAssign2
 ProgAngularContinuousAssign2 ()
 Empty constructor. More...
 
 ~ProgAngularContinuousAssign2 ()
 Destructor. More...
 
void readParams ()
 Read argument from command line. More...
 
void show ()
 Show. More...
 
void defineParams ()
 Define parameters. More...
 
void startProcessing ()
 
void preProcess ()
 
void processImage (const FileName &fnImg, const FileName &fnImgOut, const MDRow &rowIn, MDRow &rowOut)
 
void updateCTFImage (double defocusU, double defocusV, double angle)
 
void postProcess ()
 
- Public Member Functions inherited from XmippMetadataProgram
MetaDatagetInputMd ()
 
MetaDataVecgetOutputMd ()
 
 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)
 
ParamDefgetParamDef (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 ()
 
MpiMetadataProgramoperator= (const MpiMetadataProgram &)=delete
 
MpiMetadataProgramoperator= (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 ProgAngularContinuousAssign2
FileName fnVol
 
FileName fnResiduals
 
FileName fnProjections
 
double maxShift
 
double maxScale
 
double maxAngularChange
 
double maxResol
 
double maxDefocusChange
 
double maxA
 
double maxB
 
double Ts
 
int Rmax
 
int pad
 
bool optimizeGrayValues
 
bool optimizeShift
 
bool optimizeScale
 
bool optimizeAngles
 
bool optimizeDefocus
 
bool ignoreCTF
 
String originalImageLabel
 
bool phaseFlipped
 
bool sameDefocus
 
int rank
 
MultidimArray< int > mask2D
 
double iMask2Dsum
 
FourierProjectorprojector
 
size_t Xdim
 
Image< double > I
 
Image< double > Ip
 
Image< double > E
 
Image< double > Ifiltered
 
Image< double > Ifilteredp
 
Projection P
 
FourierFilter filter
 
Matrix2D< double > A
 
double old_rot
 
double old_tilt
 
double old_psi
 
double old_shiftX
 
double old_shiftY
 
bool old_flip
 
double old_grayA
 
double old_grayB
 
bool hasCTF
 
double old_defocusU
 
double old_defocusV
 
double old_defocusAngle
 
CTFDescription ctf
 
Matrix2D< double > C0
 
Matrix2D< double > C
 
double Istddev
 
int contCost
 
double currentDefocusU
 
double currentDefocusV
 
double currentAngle
 
MultidimArray< double > * ctfImage
 
std::unique_ptr< MultidimArray< double > > ctfEnvelope
 
FourierTransformer fftTransformer
 
MultidimArray< std::complex< double > > fftE
 
- 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
 
ProgramDefprogDef
 Program definition and arguments parser. More...
 
std::map< String, CommentListdefaultComments
 
int argc
 Original command line arguments. More...
 
const char ** argv
 
- Protected Attributes inherited from MpiMetadataProgram
int blockSize
 
std::vector< size_t > imgsId
 
MpiTaskDistributordistributor =nullptr
 
size_t first
 
size_t last
 
- Protected Attributes inherited from XmippMpiProgram
std::shared_ptr< MpiNodenode
 
size_t nProcs
 
size_t numberOfJobs
 
MPI_Status status
 

Detailed Description

Definition at line 31 of file mpi_angular_continuous_assign.cpp.

Member Function Documentation

◆ defineParams()

void MpiProgAngularContinuousAssign2::defineParams ( )
inlinevirtual

Function in which the param of each Program are defined.

Reimplemented from XmippMetadataProgram.

Definition at line 35 of file mpi_angular_continuous_assign.cpp.

◆ finishProcessing()

void MpiProgAngularContinuousAssign2::finishProcessing ( )
inlinevirtual

finishProcessing

Reimplemented from MpiMetadataProgram.

Definition at line 106 of file mpi_angular_continuous_assign.cpp.

107  {
108  node->gatherMetadatas(getOutputMd(), fn_out);
109  MetaDataVec MDaux;
110  MDaux.sort(getOutputMd(), MDL_GATHER_ID);
111  MDaux.removeLabel(MDL_GATHER_ID);
112  getOutputMd()=MDaux;
113  if (node->isMaster())
115  }
void sort(const MetaDataVec &MDin, const MDLabel sortLabel, bool asc=true, int limit=-1, int offset=0)
bool removeLabel(const MDLabel label) override
std::shared_ptr< MpiNode > node
Definition: xmipp_mpi.h:164

◆ getImageToProcess()

virtual bool MpiProgAngularContinuousAssign2::getImageToProcess ( size_t &  objId,
size_t &  objIndex 
)
inlineoverridevirtual

Reimplemented from XmippMetadataProgram.

Definition at line 102 of file mpi_angular_continuous_assign.cpp.

103  {
104  return getTaskToProcess(objId, objIndex);
105  }
bool getTaskToProcess(size_t &objId, size_t &objIndex)
Definition: xmipp_mpi.cpp:316

◆ preProcess()

void MpiProgAngularContinuousAssign2::preProcess ( )
inlinevirtual

Preprocess

Reimplemented from MpiMetadataProgram.

Definition at line 49 of file mpi_angular_continuous_assign.cpp.

50  {
51  rank=node->rank;
53  node->barrierWait();
54 
55  // Get the volume padded size from rank 0
56  int realSize, origin;
57  if (node->rank==0)
58  {
59  realSize = XSIZE(projector->VfourierRealCoefs);
61  }
62  MPI_Bcast(&realSize, 1, MPI_INT, 0, MPI_COMM_WORLD);
63  MPI_Bcast(&origin, 1, MPI_INT, 0, MPI_COMM_WORLD);
64  MPI_Bcast(&(projector->volumePaddedSize), 1, MPI_INT, 0, MPI_COMM_WORLD);
65  MPI_Bcast(&(projector->volumeSize), 1, MPI_INT, 0, MPI_COMM_WORLD);
66 
67  if (rank!=0)
68  {
69  projector->VfourierRealCoefs.resizeNoCopy(realSize,realSize,realSize);
70  projector->VfourierImagCoefs.resizeNoCopy(realSize,realSize,realSize);
73  }
74  MPI_Bcast(MULTIDIM_ARRAY(projector->VfourierRealCoefs), MULTIDIM_SIZE(projector->VfourierRealCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
75  MPI_Bcast(MULTIDIM_ARRAY(projector->VfourierImagCoefs), MULTIDIM_SIZE(projector->VfourierImagCoefs), MPI_DOUBLE, 0, MPI_COMM_WORLD);
76  if (rank!=0)
78 
79  MetaData &mdIn = *getInputMd();
80  mdIn.addLabel(MDL_GATHER_ID);
81  mdIn.fillLinear(MDL_GATHER_ID,1,1);
83  }
#define MULTIDIM_SIZE(v)
void resizeNoCopy(const MultidimArray< T1 > &v)
#define MULTIDIM_ARRAY(v)
#define STARTINGX(v)
void produceSideInfoProjection()
Prepare projection space.
#define STARTINGY(v)
MultidimArray< double > VfourierRealCoefs
void createTaskDistributor(MetaData &mdIn, size_t blockSize=0)
Definition: xmipp_mpi.cpp:302
MultidimArray< double > VfourierImagCoefs
#define XSIZE(v)
std::shared_ptr< MpiNode > node
Definition: xmipp_mpi.h:164
virtual bool addLabel(const MDLabel label, int pos=-1)=0
virtual void fillLinear(MDLabel label, double initial, double step)=0
#define STARTINGZ(v)

◆ read()

void MpiProgAngularContinuousAssign2::read ( int  argc,
char **  argv,
bool  reportErrors = true 
)
inlinevirtual

Read the command line arguments A convenience wrapper

Reimplemented from XmippProgram.

Definition at line 45 of file mpi_angular_continuous_assign.cpp.

46  {
48  }
int argc
Original command line arguments.
Definition: xmipp_program.h:86
void read(int argc, char **argv)
Definition: xmipp_mpi.cpp:284
const char ** argv
Definition: xmipp_program.h:87

◆ readParams()

void MpiProgAngularContinuousAssign2::readParams ( )
inlinevirtual

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 40 of file mpi_angular_continuous_assign.cpp.

◆ showProgress()

void MpiProgAngularContinuousAssign2::showProgress ( )
inlinevirtual

Reimplemented from XmippMetadataProgram.

Definition at line 93 of file mpi_angular_continuous_assign.cpp.

94  {
95  if (node->rank==1)
96  {
99  }
100  }
std::shared_ptr< MpiNode > node
Definition: xmipp_mpi.h:164

◆ startProcessing()

void MpiProgAngularContinuousAssign2::startProcessing ( )
inlinevirtual

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

Reimplemented from XmippMetadataProgram.

Definition at line 84 of file mpi_angular_continuous_assign.cpp.

85  {
86  if (node->rank==1)
87  {
88  verbose=1;
90  }
91  node->barrierWait();
92  }
std::shared_ptr< MpiNode > node
Definition: xmipp_mpi.h:164
int verbose
Verbosity level.

◆ wait()

void MpiProgAngularContinuousAssign2::wait ( )
inlinevirtual

Wait for the distributor to finish.

Reimplemented from XmippMetadataProgram.

Definition at line 116 of file mpi_angular_continuous_assign.cpp.

117  {
118  distributor->wait();
119  }
MpiTaskDistributor * distributor
Definition: xmipp_mpi.h:190

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