VectorPacker.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#include "VectorPacker.h"
#include "SerializerGuard.h"
#include "libmesh/parallel_algebra.h"
#include "libmesh/parallel_sync.h"

namespace libMesh
{
namespace Parallel
{

unsigned int
Packing<std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>>>::packable_size(
    const std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>> & object,
    const void *)
{
  unsigned int total_size = 0;

  // This package will contain a number for the global index, a number for the
  // variable index and a number for the size of the DenseVector. Then it contains
  // every item in the DenseVector as well.
  total_size += 3;
  total_size += std::get<2>(object)->size();

  return total_size;
}

unsigned int
Packing<std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>>>::packed_size(
    typename std::vector<Real>::const_iterator in)
{
  unsigned int total_size = 0;

  // The size of the dense vector
  const unsigned int vector_size = *in++;

  // Adding spaces for the vector size, global sample index and variable index.
  total_size += 3;

  // Adding spaces for the elements of the vector.
  total_size += vector_size;

  return total_size;
}

template <>
void
Packing<std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>>>::pack(
    const std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>> & object,
    std::back_insert_iterator<std::vector<Real>> data_out,
    const void *)
{
  // Encoding the size of the vector
  const auto & dense_vector = std::get<2>(object);
  data_out = dense_vector->size();

  // Encoding the global index and the variable index
  data_out = std::get<0>(object);
  data_out = std::get<1>(object);

  // Encoding the vector elements
  const auto & vector = dense_vector->get_values();
  for (std::size_t i = 0; i < dense_vector->size(); ++i)
    data_out = vector[i];
}

template <>
std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>>
Packing<std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>>>::unpack(
    std::vector<Real>::const_iterator in, void *)
{
  // Decoding the vector length
  const std::size_t data_size = *in++;
  std::tuple<unsigned int, unsigned int, std::shared_ptr<DenseVector<Real>>> object;

  // Decoding and filling the global sample index and variable index
  std::get<0>(object) = *in++;
  std::get<1>(object) = *in++;

  // Decoding and filling the values in the vector.
  auto & dense_vector = std::get<2>(object);
  dense_vector = std::make_shared<DenseVector<Real>>(data_size);
  auto & vector_values = dense_vector->get_values();
  for (std::size_t i = 0; i < data_size; ++i)
    vector_values[i] = *in++;
  return object;
}

} // namespace Parallel
} // namespace libMesh