distributed_functions.h

Go to the documentation of this file.

/*
    Copyright (C) 2007 - 2011/03/02, Hammersmith Imanet Ltd
    Copyright (C) 2014, University College London
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/

#ifndef __stir_recon_buildblock_DistributedFunctions_h__
#define __stir_recon_buildblock_DistributedFunctions_h__

#ifdef DOXYGEN_SKIP
// need to define the following to get the documentation
#  define STIR_MPI
#  define STIR_MPI_TIMINGS
#endif

#include "mpi.h"
#include "stir/shared_ptr.h"
#include "stir/recon_buildblock/ProjectorByBinPair.h"
#include "stir/DiscretisedDensity.h"
#include "stir/RelatedViewgrams.h"
#include "stir/Viewgram.h"
#include "stir/VoxelsOnCartesianGrid.h"
#include "stir/ProjDataInfo.h"

namespace stir
{
class ExamInfo;
}

namespace distributed
{
extern int num_processors;

extern bool first_iteration;

extern bool test;

// for timings
extern bool rpc_time;                
extern bool test_send_receive_times; 

extern double total_rpc_time; 
extern double
    total_rpc_time_2; 
extern double total_rpc_time_slaves; 
extern double min_threshold;         

//----------------------Send operations----------------------------------

void send_int_value(int value, int destination);

void send_string(const std::string& str, int tag, int destination);

void send_bool_value(bool value, int tag, int destination);

void send_int_values(int* values, int count, int tag, int destination);

void send_double_values(double* values, int count, int tag, int destination);

void send_view_segment_numbers(const stir::ViewSegmentNumbers& vs_num, int tag, int destination);

void send_projectors(const stir::shared_ptr<stir::ProjectorByBinPair>& proj_pair_sptr, int destination);

void send_image_parameters(const stir::DiscretisedDensity<3, float>* input_image_ptr, int tag, int destination);

void send_image_estimate(const stir::DiscretisedDensity<3, float>* input_image_ptr, int destination);

void send_exam_and_proj_data_info(const stir::ExamInfo& exam_info, const stir::ProjDataInfo& proj_data_info, int destination);

void send_related_viewgrams(stir::RelatedViewgrams<float>* viewgrams, int destination);

void send_viewgram(const stir::Viewgram<float>& viewgram, int destination);

//----------------------Receive operations----------------------------------

int receive_int_value(int source);

std::string receive_string(int tag, int source);

void receive_and_initialize_projectors(stir::shared_ptr<stir::ProjectorByBinPair>& projector_pair_ptr, int source);

bool receive_bool_value(int tag, int source);

MPI_Status receive_int_values(int* values, int count, int tag);

MPI_Status receive_double_values(double* values, int count, int tag);

MPI_Status receive_view_segment_numbers(stir::ViewSegmentNumbers& vs_num, int tag);

void receive_and_set_image_parameters(stir::shared_ptr<stir::DiscretisedDensity<3, float>>& image_ptr,
                                      int& buffer,
                                      int tag,
                                      int source);

MPI_Status receive_image_values_and_fill_image_ptr(stir::shared_ptr<stir::DiscretisedDensity<3, float>>& image_ptr,
                                                   int buffer_size,
                                                   int source);

void receive_and_construct_exam_and_proj_data_info_ptr(stir::shared_ptr<stir::ExamInfo>& exam_info_sptr,
                                                       stir::shared_ptr<stir::ProjDataInfo>& proj_data_info_sptr,
                                                       int source);

void receive_and_construct_related_viewgrams(stir::RelatedViewgrams<float>*& viewgrams,
                                             const stir::shared_ptr<stir::ProjDataInfo>& proj_data_info_ptr,
                                             const stir::shared_ptr<stir::DataSymmetriesForViewSegmentNumbers> symmetries_sptr,
                                             int source);

void receive_and_construct_viewgram(stir::Viewgram<float>*& viewgram,
                                    const stir::shared_ptr<stir::ProjDataInfo>& proj_data_info_ptr,
                                    int source);

//-----------------------reduce operations-------------------------------------

void reduce_received_output_image(stir::DiscretisedDensity<3, float>* output_image_ptr, int destination);

void reduce_output_image(stir::shared_ptr<stir::DiscretisedDensity<3, float>>& output_image_ptr,
                         int image_buffer_size,
                         int my_rank,
                         int destination);

const int INT_TAG = 7;
const int ARBITRARY_TAG = 8; 
const int STIR_MPI_CONF_TAG = 9;
const int IMAGE_ESTIMATE_TAG = 23;
const int IMAGE_PARAMETER_TAG = 24;
const int VIEWGRAM_DIMENSIONS_TAG = 27;
const int VIEWGRAM_TAG = 28;
const int VIEWGRAM_COUNT_TAG = 29;
const int PROJECTION_DATA_INFO_TAG = 30;
const int PARAMETER_INFO_TAG = 21;
const int REGISTERED_NAME_TAG = 25;
} // namespace distributed

#endif