ParallelSolutionStorage.h

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

#pragma once

#include "GeneralReporter.h"

class ParallelSolutionStorage : public GeneralReporter
{
public:
  static InputParameters validParams();
  ParallelSolutionStorage(const InputParameters & parameters);
  virtual void initialSetup() override;
  virtual void initialize() override {}
  virtual void execute() override {}
  virtual void finalize() override {}

  void
  addEntry(const VariableName & vname, unsigned int global_i, const DenseVector<Real> & solution);
  std::unordered_map<unsigned int, std::vector<DenseVector<Real>>> &
  getStorage(const VariableName & variable);

  const std::unordered_map<unsigned int, std::vector<DenseVector<Real>>> &
  getStorage(const VariableName & variable) const;

  std::map<VariableName, std::unordered_map<unsigned int, std::vector<DenseVector<Real>>>> &
  getStorage() const
  {
    return _distributed_solutions;
  }

  bool hasGlobalSample(unsigned int global_sample_i, const VariableName & variable) const;

  const std::vector<DenseVector<Real>> & getGlobalSample(unsigned int global_sample_i,
                                                         const VariableName & variable) const;

  unsigned int totalNumberOfStoredSolutions(const VariableName & vname) const;

  const std::vector<VariableName> & variableNames() const { return _variable_names; }

protected:
  std::map<VariableName, std::unordered_map<unsigned int, std::vector<DenseVector<Real>>>> &
      _distributed_solutions;

private:
  const std::vector<VariableName> & _variable_names;
};

void to_json(
    nlohmann::json & json,
    const std::map<VariableName, std::unordered_map<unsigned int, std::vector<DenseVector<Real>>>> &
        solution_storage);