SamplerTransientMultiApp.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

// MOOSE includes
#include "TransientMultiApp.h"
#include "SamplerInterface.h"

#include "StochasticToolsTypes.h"

class Sampler;
class StochasticToolsTransfer;

class SamplerTransientMultiApp : public TransientMultiApp, public SamplerInterface
{
public:
  static InputParameters validParams();

  SamplerTransientMultiApp(const InputParameters & parameters);

  virtual bool solveStep(Real dt, Real target_time, bool auto_advance = true) override;

  virtual void initialSetup() override;

protected:
  Sampler & _sampler;

  const StochasticTools::MultiAppMode _mode;

  dof_id_type _local_batch_app_index;

  virtual std::vector<std::string> getCommandLineArgs(const unsigned int local_app) override;

private:
  bool solveStepBatch(Real dt, Real target_time, bool auto_advance = true);

  void updateRowData(dof_id_type local_index);

  std::vector<std::shared_ptr<StochasticToolsTransfer>>
  getActiveStochasticToolsTransfers(Transfer::DIRECTION direction);

  const dof_id_type _number_of_sampler_rows;
  std::vector<std::vector<std::unique_ptr<Backup>>> _batch_backup;

  std::vector<Real> _row_data;
  dof_id_type _local_row_index = std::numeric_limits<dof_id_type>::max();
};