ode_elements.h

Go to the documentation of this file.

// LIC// ====================================================================
// LIC// This file forms part of oomph-lib, the object-oriented,
// LIC// multi-physics finite-element library, available
// LIC// at http://www.oomph-lib.org.
// LIC//
// LIC// Copyright (C) 2006-2022 Matthias Heil and Andrew Hazel
// LIC//
// LIC// This library is free software; you can redistribute it and/or
// LIC// modify it under the terms of the GNU Lesser General Public
// LIC// License as published by the Free Software Foundation; either
// LIC// version 2.1 of the License, or (at your option) any later version.
// LIC//
// LIC// This library is distributed in the hope that it will be useful,
// LIC// but WITHOUT ANY WARRANTY; without even the implied warranty of
// LIC// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// LIC// Lesser General Public License for more details.
// LIC//
// LIC// You should have received a copy of the GNU Lesser General Public
// LIC// License along with this library; if not, write to the Free Software
// LIC// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
// LIC// 02110-1301  USA.
// LIC//
// LIC// The authors may be contacted at oomph-lib@maths.man.ac.uk.
// LIC//
// LIC//====================================================================
#ifndef OOMPH_ODE_ELEMENTS_H
#define OOMPH_ODE_ELEMENTS_H
 
#include "../generic/oomph_definitions.h"
#include "../generic/oomph_utilities.h"
 
#include "../generic/matrices.h"
#include "../generic/Vector.h"
#include "../generic/elements.h"
#include "../generic/timesteppers.h"
 
namespace oomph
{
  class ODEElement : public GeneralisedElement
  {
  public:
    ODEElement()
    {
      Exact_solution_pt = 0;
 
      Use_fd_jacobian = false;
    }
 
    ODEElement(TimeStepper* time_stepper_pt,
               SolutionFunctorBase* exact_solution_pt)
    {
      build(time_stepper_pt, exact_solution_pt);
    }
 
    void build(TimeStepper* time_stepper_pt,
               SolutionFunctorBase* exact_solution_pt)
    {
      Exact_solution_pt = exact_solution_pt;
 
      Vector<double> exact = this->exact_solution(0);
      unsigned nvalue = exact.size();
 
      add_internal_data(new Data(time_stepper_pt, nvalue));
 
      Use_fd_jacobian = false;
    }
 
 
    virtual ~ODEElement() {}
 
    unsigned nvalue() const
    {
      return internal_data_pt(0)->nvalue();
    }
 
    virtual void fill_in_contribution_to_residuals(Vector<double>& residuals)
    {
      // Get pointer to one-and-only internal data object
      Data* dat_pt = internal_data_pt(0);
 
      // Get it's values
      Vector<double> u(nvalue(), 0.0);
      dat_pt->value(u);
 
      // Get time stepper
      TimeStepper* time_stepper_pt = dat_pt->time_stepper_pt();
 
      // Get continuous time
      double t = time_stepper_pt->time();
 
      Vector<double> deriv = derivative_function(t, u);
      for (unsigned j = 0, nj = deriv.size(); j < nj; j++)
      {
        // Get dudt approximation from time stepper
        double dudt = time_stepper_pt->time_derivative(1, dat_pt, j);
 
        // Residual is difference between the exact derivative and our
        // time stepper's derivative estimate.
        residuals[j] = deriv[j] - dudt;
      }
    }
 
    virtual void fill_in_contribution_to_jacobian(Vector<double>& residuals,
                                                  DenseMatrix<double>& jacobian)
    {
      // Get residuals
      fill_in_contribution_to_residuals(residuals);
 
      if (Exact_solution_pt->have_jacobian() && !Use_fd_jacobian)
      {
        // get df/du jacobian
        double t = internal_data_pt(0)->time_stepper_pt()->time();
        Vector<double> dummy, u(nvalue(), 0.0);
        internal_data_pt(0)->value(u);
        Exact_solution_pt->jacobian(t, dummy, u, jacobian);
 
        // We need jacobian of residual = f - dudt so subtract diagonal
        // (dudt)/du term.
        const double a = internal_data_pt(0)->time_stepper_pt()->weight(1, 0);
        const unsigned n = nvalue();
        for (unsigned i = 0; i < n; i++)
        {
          jacobian(i, i) -= a;
        }
      }
      else
      {
        // Use FD for jacobian
        GeneralisedElement::fill_in_jacobian_from_internal_by_fd(
          residuals, jacobian, true);
      }
    }
 
    virtual void fill_in_contribution_to_mass_matrix(Vector<double>& residuals,
                                                     DenseMatrix<double>& mm)
    {
      fill_in_contribution_to_residuals(residuals);
      for (unsigned j = 0, nj = nvalue(); j < nj; j++)
      {
        mm(j, j) = 1;
      }
    }
 
    Vector<double> exact_solution(const double& t) const
    {
#ifdef PARANOID
      if (Exact_solution_pt == 0)
      {
        throw OomphLibError("No exact solution function",
                            OOMPH_EXCEPTION_LOCATION,
                            OOMPH_CURRENT_FUNCTION);
      }
#endif
      Vector<double> dummy_x;
      return (*Exact_solution_pt)(t, dummy_x);
    }
 
    Vector<double> derivative_function(const double& t, const Vector<double>& u)
    {
#ifdef PARANOID
      if (Exact_solution_pt == 0)
      {
        throw OomphLibError("No derivative function",
                            OOMPH_EXCEPTION_LOCATION,
                            OOMPH_CURRENT_FUNCTION);
      }
#endif
      Vector<double> dummy_x;
      return Exact_solution_pt->derivative(t, dummy_x, u);
    }
 
    SolutionFunctorBase* Exact_solution_pt;
 
    bool Use_fd_jacobian;
  };
 
} // namespace oomph
 
#endif