DistributedWorker.h

Go to the documentation of this file.

/*
    Copyright (C) 2007 - 2011-02-23, 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_DistributedWorker_h__
#  define __stir_recon_buildblock_DistributedWorker_h__

#  include "stir/shared_ptr.h"
#  include "stir/TimedObject.h"
//#include "stir/ParsingObject.h"
#  include "stir/ProjData.h"
#  include "stir/recon_buildblock/ProjectorByBinPair.h"
#  include "stir/recon_buildblock/distributable.h"
#  include <string>
#  include <vector>

START_NAMESPACE_STIR

class ExamInfo;

template <class TargetT>
class DistributedWorker : public TimedObject //, public ParsingObject
{
private:
  double* log_likelihood_ptr;
  bool zero_seg0_end_planes;
  shared_ptr<ProjectorByBinPair> proj_pair_sptr;
  shared_ptr<ExamInfo> exam_info_sptr;
  shared_ptr<const ProjDataInfo> proj_data_info_sptr;
  shared_ptr<TargetT> target_sptr;

  int image_buffer_size; // to save the image_size

  // cache variables
  bool cache_enabled;
  shared_ptr<ProjData> proj_data_ptr;
  shared_ptr<ProjData> binwise_correction;
  shared_ptr<ProjData> mult_proj_data_sptr;

  int my_rank; // rank of the worker

public:
  // Default constructor
  DistributedWorker();

  // Default destructor
  ~DistributedWorker();

  void start();

protected:
  void set_defaults();

  void setup_distributable_computation();
  void distributable_computation(RPC_process_related_viewgrams_type* RPC_process_related_viewgrams);
};

END_NAMESPACE_STIR

#endif
// __DistributedWorker_h__