SnapshotContainerBase.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"
#include "libmesh/petsc_vector.h"
#include "libmesh/petsc_matrix.h"
#include "NonlinearSystemBase.h"

class SnapshotContainerBase : public GeneralReporter
{
public:
  static InputParameters validParams();
  SnapshotContainerBase(const InputParameters & parameters);

  class Snapshots : public UniqueStorage<NumericVector<Number>>
  {
  public:
    friend class SnapshotContainerBase;
  };

  virtual void initialSetup() override;
  virtual void initialize() override {}
  virtual void execute() override;
  virtual void finalize() override {}

  const Snapshots & getSnapshots() const { return _accumulated_data; }

  const NumericVector<Number> & getSnapshot(unsigned int local_i) const;

protected:
  virtual std::unique_ptr<NumericVector<Number>> collectSnapshot() = 0;

  Snapshots & _accumulated_data;

  const unsigned int _nonlinear_system_number;
};

void dataStore(std::ostream & stream, SnapshotContainerBase::Snapshots & v, void * context);
void dataLoad(std::istream & stream, SnapshotContainerBase::Snapshots & v, void * context);