SolutionHandle.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 "SystemBase.h"
#include "MooseVariableFieldBase.h"
#include "libmesh/system.h"
#include "libmesh/enum_norm_type.h"
#include "libmesh/petsc_vector.h"

class SolutionHandle
{
public:
  SolutionHandle(const MooseVariableFieldBase & variable)
    : _var(const_cast<MooseVariableFieldBase &>(variable)),
      _soln(_var.sys().solution()),
      _soln_old(_var.sys().solutionOld())
  {
  }

  Number operator()(const Node * node) const
  {
    // The 0 assumes linear Lagrange (I think)
    dof_id_type dof = node->dof_number(_var.sys().number(), _var.number(), 0);
    return _soln(dof);
  }
  Number old(const Node * node) const
  {
    // The 0 assumes linear Lagrange (I think)
    dof_id_type dof = node->dof_number(_var.sys().number(), _var.number(), 0);
    return _soln_old(dof);
  }
  void set(const Node * node, Number val)
  {
    dof_id_type dof = node->dof_number(_var.sys().number(), _var.number(), 0);
    _soln.set(dof, val);
  }

  Real L2norm() const
  {
    return _var.sys().system().calculate_norm(_soln, _var.number(), DISCRETE_L2);
  }

  void close() { _soln.close(); }

private:
  MooseVariableFieldBase & _var;
  NumericVector<Number> & _soln;
  const NumericVector<Number> & _soln_old;
};