TransientDriver Class Reference

The driver for a transient initial-value problem. More...

Detailed Description

The driver for a transient initial-value problem.

#include <TransientDriver.h>

Inheritance diagram for TransientDriver:

Public Member Functions
	TransientDriver (const OptionSet &options)
	Construct a new TransientDriver object.
void	diagnose (std::vector< Diagnosis > &) const override
	Check for common problems.
bool	run () override
	Let the driver run, return true upon successful completion, and return false otherwise.
const Model &	model () const
virtual std::string	save_as_path () const
	The destination file/path to save the results.
virtual torch::nn::ModuleDict	result () const
	The results (input and output) from all time steps.
Public Member Functions inherited from Driver
	Driver (const OptionSet &options)
	Construct a new Driver object.
Public Member Functions inherited from NEML2Object
	NEML2Object ()=delete
	NEML2Object (NEML2Object &&)=delete
	NEML2Object (const NEML2Object &)=delete
NEML2Object &	operator= (const NEML2Object &)=delete
NEML2Object &	operator= (NEML2Object &&)=delete
virtual	~NEML2Object ()=default
	NEML2Object (const OptionSet &options)
	Construct a new NEML2Object object.
const OptionSet &	input_options () const
virtual void	setup ()
	Setup this object.
const std::string &	name () const
	A readonly reference to the object's name.
const std::string &	type () const
	A readonly reference to the object's type.
const std::string &	path () const
	A readonly reference to the object's path.
const std::string &	doc () const
	A readonly reference to the object's docstring.
template<typename T = NEML2Object>
const T *	host () const
	Get a readonly pointer to the host.
template<typename T = NEML2Object>
T *	host ()
	Get a writable pointer to the host.
Public Member Functions inherited from DiagnosticsInterface
	DiagnosticsInterface ()=delete
	DiagnosticsInterface (NEML2Object *object)
	DiagnosticsInterface (DiagnosticsInterface &&)=delete
	DiagnosticsInterface (const DiagnosticsInterface &)=delete
DiagnosticsInterface &	operator= (const DiagnosticsInterface &)=delete
DiagnosticsInterface &	operator= (DiagnosticsInterface &&)=delete
virtual	~DiagnosticsInterface ()=default
template<typename... Args>
void	diagnostic_assert (std::vector< Diagnosis > &diagnoses, bool assertion, Args &&... args) const
void	diagnostic_assert_state (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_assert_old_state (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_assert_force (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_assert_old_force (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_assert_residual (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_check_input_variable (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const
void	diagnostic_check_output_variable (std::vector< Diagnosis > &diagnoses, const VariableBase &v) const

Static Public Member Functions
static OptionSet	expected_options ()
Static Public Member Functions inherited from Driver
static OptionSet	expected_options ()
Static Public Member Functions inherited from NEML2Object
static OptionSet	expected_options ()

Protected Member Functions
virtual bool	solve ()
	Solve the initial value problem.
virtual void	advance_step ()
	Advance in time: the state becomes old state, and forces become old forces.
virtual void	update_forces ()
	Update the driving forces for the current time step.
virtual void	apply_ic ()
	Apply the initial conditions.
virtual void	apply_predictor ()
	Apply the predictor to calculate the initial guess for the current time step.
virtual void	solve_step ()
	Perform the constitutive update for the current time step.
virtual void	store_input ()
	Save the input of the current time step.
virtual void	output () const
	Save the results into the destination file/path.

Protected Attributes
Model &	_model
	The model which the driver uses to perform constitutive updates.
const torch::Device	_device
	The device on which to evaluate the model.
const VariableName	_time_name
	VariableName for the time.
Scalar	_time
	The current time.
Size	_step_count
	The current step count.
const Size	_nsteps
	Total number of steps.
ValueMap	_in
	The input to the constitutive model.
const EnumSelection	_predictor
	The predictor used to set the initial guess.
std::string	_save_as
	The destination file name or file path.
const bool	_show_params
	Set to true to list all the model parameters at the beginning.
const bool	_show_input
	Set to true to show model's input axis at the beginning.
const bool	_show_output
	Set to true to show model's output axis at the beginning.
std::vector< ValueMap >	_result_in
	Inputs from all time steps.
std::vector< ValueMap >	_result_out
	Outputs from all time steps.
Protected Attributes inherited from Driver
bool	_verbose
	Whether to print out additional (debugging) information during the execution.

Constructor & Destructor Documentation

TransientDriver()

TransientDriver

(

const OptionSet &

options

)

Construct a new TransientDriver object.

Parameters

options

The options extracted from the input file

Member Function Documentation

advance_step()

void advance_step ( )

protectedvirtual

Advance in time: the state becomes old state, and forces become old forces.

apply_ic()

void apply_ic ( )

protectedvirtual

Apply the initial conditions.

apply_predictor()

void apply_predictor ( )

protectedvirtual

Apply the predictor to calculate the initial guess for the current time step.

Reimplemented in LDISolidMechanicsDriver.

diagnose()

void diagnose ( std::vector< Diagnosis > & diagnoses ) const

overridevirtual

Check for common problems.

This method serves as the entry point for diagnosing common problems in object setup. The idea behind this method is that while some errors could be detected at construction time, i.e., when the object's constructor is called, it doesn't hinder other objects' creation. We therefore would like to defer the detection of errors until after all objects have been created, collect all errors at once, and present the user with a complete understanding of all errors encountered.

Note, however, if an error could interfere with other objects' creation, it should be raised right away inside the constructor, instead of inside this method.

Parameters

diagnoses

A vector of exceptions of type Diagnosis for each of the detected problem.

Reimplemented from Driver.

expected_options()

OptionSet expected_options ( )

static

model()

const Model & model ( ) const

inline

output()

void output ( ) const

protectedvirtual

Save the results into the destination file/path.

result()

torch::nn::ModuleDict result ( ) const

virtual

The results (input and output) from all time steps.

Returns

torch::nn::ModuleDict The results (input and output) from all time steps. The keys of the dict are "input" and "output". The values are torch::nn::ModuleList with length equal to the number of time steps. Each ModuleList contains the input/output variables at each time step in the form of torch::nn::ModuleDict whose keys are variable names and values are variable values.

run()

bool run ( )

overridevirtual

Let the driver run, return true upon successful completion, and return false otherwise.

Implements Driver.

save_as_path()

std::string save_as_path ( ) const

virtual

The destination file/path to save the results.

solve()

bool solve ( )

protectedvirtual

Solve the initial value problem.

solve_step()

void solve_step ( )

protectedvirtual

Perform the constitutive update for the current time step.

store_input()

void store_input ( )

protectedvirtual

Save the input of the current time step.

update_forces()

void update_forces ( )

protectedvirtual

Update the driving forces for the current time step.

Reimplemented in LDISolidMechanicsDriver, and SolidMechanicsDriver.

Member Data Documentation

_device

const torch::Device _device

protected

The device on which to evaluate the model.

_in

ValueMap _in

protected

The input to the constitutive model.

_model

Model& _model

protected

The model which the driver uses to perform constitutive updates.

_nsteps

const Size _nsteps

protected

Total number of steps.

_predictor

const EnumSelection _predictor

protected

The predictor used to set the initial guess.

_result_in

std::vector<ValueMap> _result_in

protected

Inputs from all time steps.

_result_out

std::vector<ValueMap> _result_out

protected

Outputs from all time steps.

_save_as

std::string _save_as

protected

The destination file name or file path.

_show_input

const bool _show_input

protected

Set to true to show model's input axis at the beginning.

_show_output

const bool _show_output

protected

Set to true to show model's output axis at the beginning.

_show_params

const bool _show_params

protected

Set to true to list all the model parameters at the beginning.

_step_count

Size _step_count

protected

The current step count.

_time

Scalar _time

protected

The current time.

_time_name

const VariableName _time_name

protected

VariableName for the time.