/usr/include/relion-1.3/src/ml_optimiser_mpi.h is in librelion+mpi-dev 1.3+dfsg-2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 | /***************************************************************************
*
* Author: "Sjors H.W. Scheres"
* MRC Laboratory of Molecular Biology
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* This complete copyright notice must be included in any revised version of the
* source code. Additional authorship citations may be added, but existing
* author citations must be preserved.
***************************************************************************/
#ifndef ML_OPTIMISER_MPI_H_
#define ML_OPTIMISER_MPI_H_
#include "src/mpi.h"
#include "src/ml_optimiser.h"
#define MPITAG_JOB_REQUEST 0
#define MPITAG_JOB_REPLY 1
#define MPITAG_METADATA 2
#define MPITAG_RANDOMSEED 3
#define MPITAG_IMAGE 4
#define MPITAG_PACK 5
#define MPITAG_DOUBLE 6
#define MPITAG_INT 7
#ifdef TIMING
int TIMING_MPIPACK, TIMING_MPIWAIT;
#endif
class MlOptimiserMpi: public MlOptimiser
{
private:
MpiNode *node;
public:
#ifdef TIMINGMPI
int MPIR_PACK, MPIR_ALLREDUCE, MPIR_UNPACK, MPIR_EXP, MPIR_MAX, MPIR_BCAST;
#endif
// Name of the directory to write temporary files to
FileName fn_scratch;
/** Destructor, calls MPI_Finalize */
~MlOptimiserMpi()
{
delete node;
}
/** Read
* This could take care of mpi-parallelisation-dependent variables
*/
void read(int argc, char **argv);
/** Finalise
* Free some memory
*/
void finalise();
void initialise();
/** Initialise the work load: divide images equally over all nodes
* Also initialise the same random seed for all nodes
*/
void initialiseWorkLoad();
/** Perform individual power spectra calculation in parallel */
void calculateSumOfPowerSpectraAndAverageImage(MultidimArray<double> &Mavg);
/** Expectation
* This cares care of gathering all weighted sums after the expectation
*/
void expectation();
/** After expectation combine all weighted sum arrays across all nodes
* Use read/write to temporary files instead of MPI
*/
void combineAllWeightedSumsViaFile();
/** Join the sums from two random halves
* Use read/write to temporary files instead of MPI
*/
void combineWeightedSumsTwoRandomHalvesViaFile();
/** After expectation combine all weighted sum arrays across all nodes
*/
void combineAllWeightedSums();
/** Join the sums from two random halves
*/
void combineWeightedSumsTwoRandomHalves();
/** Maximization
* This takes care of the parallel reconstruction of the classes
*/
void maximization();
/**
* Write temporary data and weight arrays from the backprojector to disc to allow unregularized reconstructions
*/
void writeTemporaryDataAndWeightArrays();
/**
* Read temporary data and weight arrays from disc and perform unregularized reconstructions
* Also write the unregularized reconstructions to disc.
*/
void readTemporaryDataAndWeightArraysAndReconstruct(int iclass, int ihalf);
/**
* Join two independent reconstructions ate the lowest frequencies to avoid convergence in distinct orientations
*/
void joinTwoHalvesAtLowResolution();
/** When refining two random halves separately, the master receives both models, calculates FSC and the power of their difference
* and sends these curves, together with new tau2_class estimates to all slaves...
*/
void compareTwoHalves();
/** Do the real work
* Expectation is split in image subsets over all nodes, each reconstruction is done on a separate node
*/
void iterate();
};
#endif /* ML_OPTIMISER_MPI_H_ */
|