mpi_reconstruct_fourier_gpu.h

Go to the documentation of this file.

/***************************************************************************
 *
 * Authors:     David Strelak (davidstrelak@gmail.com)
 *
 * Unidad de  Bioinformatica of Centro Nacional de Biotecnologia , CSIC
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 * 02111-1307  USA
 *
 *  All comments concerning this program package may be sent to the
 *  e-mail address 'xmipp@cnb.csic.es'
 ***************************************************************************/
#ifndef MPI_RECONSTRUCT_FOURIER_GPU_H_
#define MPI_RECONSTRUCT_FOURIER_GPU_H_

#include <parallel/xmipp_mpi.h>
#include <reconstruction_adapt_cuda/reconstruct_fourier_gpu.h>

#include <core/args.h>
#include <data/projection.h>
#include <cstring>
#include <cstdlib>
#include <core/xmipp_funcs.h>
#include <core/matrix2d.h>
#include <sys/time.h>
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>

#define TAG_WORKFORWORKER    0
#define TAG_STOP     1
#define TAG_TRANSFER   2
#define TAG_FREEWORKER    3
#define TAG_SETVERBOSE  5

//TODO (MARIANA) Please give more documentation and in a good structure e.g. @name



class ProgMPIRecFourierGPU: public ProgRecFourierGPU, public XmippMpiProgram
{
public:
    ProgMPIRecFourierGPU(){};

    /*  constructor ------------------------------------------------------- */
    ProgMPIRecFourierGPU(int argc, char *argv[]);

    /* constructor providing an MpiNode
     * this is useful for using this programs from others
     */
    ProgMPIRecFourierGPU(const std::shared_ptr<MpiNode> &node);

    void read(int argc, char** argv);

private:
    int mpi_job_size;

    int threadsPerGPU;

    int gpusPerNode;

    void readParams();

    void defineParams();

    void preRun();

    void run();


};
//end of class MPI reconstruct fourier gpu

#endif /* MPI_RECONSTRUCT_FOURIER_GPU_H_ */