DistributedCachingInformation.h

//
//
/*
    Copyright (C) 2007- 2011, Hammersmith Imanet Ltd
    This file is part of STIR.
 
    SPDX-License-Identifier: Apache-2.0
 
    See STIR/LICENSE.txt for details
*/
 
#ifndef __stir_recon_buildblock_DistributedCachingInformation_h__
#define __stir_recon_buildblock_DistributedCachingInformation_h__

 
#include "stir/ViewSegmentNumbers.h"
#include <vector>
#include <algorithm>
 
START_NAMESPACE_STIR

class DistributedCachingInformation
{
public:
  explicit DistributedCachingInformation(const int num_processors);

  virtual ~DistributedCachingInformation();

  void initialise();
  void initialise_new_subiteration(const std::vector<ViewSegmentNumbers>& vs_nums_to_process);

  bool get_unprocessed_vs_num(ViewSegmentNumbers& vs_num, int proc);
 
private:
  int num_workers;

  std::vector<ViewSegmentNumbers> vs_nums_to_process;

  std::vector<std::vector<ViewSegmentNumbers>> proc_vs_nums;


  std::vector<bool> still_to_process;

  int find_vs_num_position_in_list_to_process(const ViewSegmentNumbers& vs_num) const;

  int find_position_of_first_unprocessed() const;

  int get_num_remaining_cached_data_to_process(int proc) const;

  void add_vs_num_to_proc(int proc, const ViewSegmentNumbers& vs_num);

  bool get_oldest_unprocessed_vs_num(ViewSegmentNumbers& vs_num, int proc) const;

  ViewSegmentNumbers get_vs_num_of_proc_with_most_work_left(int proc) const;

  void set_processed(const ViewSegmentNumbers& vs_num);

  void set_all_vs_num_unprocessed();

  bool is_still_to_be_processed(const ViewSegmentNumbers& vs_num) const;
};
 
END_NAMESPACE_STIR
 
#endif /*__stir_recon_buildblock_DistributedCachingInformation_h__*/