copy_fill.h

Go to the documentation of this file.

/*
    Copyright (C) 2020, 2024 University College London
    This file is part of STIR.

    SPDX-License-Identifier: Apache-2.0

    See STIR/LICENSE.txt for details
*/
#ifndef __stir_copy_fill__H__
#define __stir_copy_fill__H__

#include "stir/Array.h"
#include "stir/ProjDataInMemory.h"

START_NAMESPACE_STIR

template <typename T>
struct CopyFill
{
  template <typename iterT>
  static iterT copy_to(const T& stir_object, iterT iter)
  {
    return std::copy(stir_object.begin_all(), stir_object.end_all(), iter);
  }
  template <typename iterT>
  static void fill_from(T& stir_object, iterT iter, iterT iter_end)
  {
    std::copy(iter, iter_end, stir_object.begin_all());
  }
};


template <int num_dimensions, class elemT>
struct CopyFill<Array<num_dimensions, elemT>>
{
  typedef Array<num_dimensions, elemT> Array_type;

  template <typename iterT>
  static iterT copy_to(const Array_type& stir_array, iterT iter)
  {
    if (stir_array.is_contiguous())
      {
        // std::cerr << "Using 1D std::copy for copy_to\n";
        auto beg = stir_array.get_const_full_data_ptr();
        auto ret = std::copy(beg, beg + stir_array.size_all(), iter);
        stir_array.release_const_full_data_ptr();
        return ret;
      }
    else
      {
        // std::cerr<<"Using normal std::copy for copy_to\n";
        return std::copy(stir_array.begin_all(), stir_array.end_all(), iter);
      }
  }

  template <typename iterT>
  static void fill_from(Array_type& stir_array, iterT iter, iterT iter_end)
  {
    if (stir_array.is_contiguous())
      {
        // std::cerr << "Using 1D std::copy for fill_from\n";
        std::copy(iter, iter_end, stir_array.get_full_data_ptr());
        stir_array.release_full_data_ptr();
      }
    else
      {
        // std::cerr<<"Using normal std::copy for fill_from\n";
        std::copy(iter, iter_end, stir_array.begin_all());
      }
  }
};


template <>
struct CopyFill<ProjData>
{
  template <typename iterT>
  static iterT copy_to(const ProjData& stir_object, iterT iter)
  {
#if 1
    if (auto pdm_ptr = dynamic_cast<ProjDataInMemory const*>(&stir_object))
      {
        // std::cerr<<"Using stir::copy_to\n";
        return CopyFill<ProjDataInMemory>::copy_to(*pdm_ptr, iter);
      }
    else
#endif
      {
        // std::cerr<<"Using member copy_to\n";
        return stir_object.copy_to(iter);
      }
  }

  template <typename iterT>
  static void fill_from(ProjData& stir_object, iterT iter, iterT iter_end)
  {
    if (auto pdm_ptr = dynamic_cast<ProjDataInMemory*>(&stir_object))
      CopyFill<ProjDataInMemory>::fill_from(*pdm_ptr, iter, iter_end);
    else
      stir_object.fill_from(iter);
  }
};


template <typename T, typename iterT>
inline iterT
copy_to(const T& stir_object, iterT iter)
{
  return CopyFill<T>::copy_to(stir_object, iter);
}


template <typename T, typename iterT>
inline void
fill_from(T& stir_object, iterT iter, iterT iter_end)
{
  // return
  CopyFill<T>::fill_from(stir_object, iter, iter_end);
}


END_NAMESPACE_STIR

#endif