tensor_function_parser.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (C) 2005 - 2020 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------
 
#ifndef dealii_tensor_function_parser_h
#define dealii_tensor_function_parser_h
 
 
#include <deal.II/base/config.h>
 
#include <deal.II/base/exceptions.h>
#include <deal.II/base/point.h>
#include <deal.II/base/tensor.h>
#include <deal.II/base/tensor_function.h>
#include <deal.II/base/thread_local_storage.h>
 
#include <map>
#include <memory>
#include <vector>
 
namespace mu
{
  class Parser;
}
 
DEAL_II_NAMESPACE_OPEN
 
// Forward declaration
#ifndef DOXYGEN
template <typename>
class Vector;
#endif
 
template <int rank, int dim, typename Number = double>
class TensorFunctionParser : public TensorFunction<rank, dim, Number>
{
public:
  TensorFunctionParser(const double initial_time = 0.0);
 
  TensorFunctionParser(
    const std::string &expression,
    const std::string &constants      = "",
    const std::string &variable_names = default_variable_names() + ",t");
 
  TensorFunctionParser(const TensorFunctionParser &) = delete;
 
  TensorFunctionParser(TensorFunctionParser &&) = delete;
 
  virtual ~TensorFunctionParser() override;
 
  TensorFunctionParser &
  operator=(const TensorFunctionParser &) = delete;
 
  TensorFunctionParser &
  operator=(TensorFunctionParser &&) = delete;
 
  using ConstMap = std::map<std::string, double>;
 
 
  void
  initialize(const std::string &             vars,
             const std::vector<std::string> &expressions,
             const ConstMap &                constants,
             const bool                      time_dependent = false);
 
  void
  initialize(const std::string &vars,
             const std::string &expression,
             const ConstMap &   constants,
             const bool         time_dependent = false);
 
  static std::string
  default_variable_names();
 
  virtual Tensor<rank, dim, Number>
  value(const Point<dim> &p) const override;
 
  virtual void
  value_list(const std::vector<Point<dim>> &         p,
             std::vector<Tensor<rank, dim, Number>> &values) const override;
 
  const std::vector<std::string> &
  get_expressions() const;
 
  DeclException2(ExcParseError,
                 int,
                 std::string,
                 << "Parsing Error at Column " << arg1
                 << ". The parser said: " << arg2);
 
  DeclException2(ExcInvalidExpressionSize,
                 int,
                 int,
                 << "The number of components (" << arg1
                 << ") is not equal to the number of expressions (" << arg2
                 << ").");
 
 
private:
#ifdef DEAL_II_WITH_MUPARSER
 
  mutable Threads::ThreadLocalStorage<std::vector<double>> vars;
 
  mutable Threads::ThreadLocalStorage<std::vector<std::unique_ptr<mu::Parser>>>
    tfp;
 
  std::map<std::string, double> constants;
 
  std::vector<std::string> var_names;
 
  void
  init_muparser() const;
#endif
 
  std::vector<std::string> expressions;
 
  bool initialized;
 
  unsigned int n_vars;
 
  unsigned int n_components;
};
 
 
template <int rank, int dim, typename Number>
std::string
TensorFunctionParser<rank, dim, Number>::default_variable_names()
{
  switch (dim)
    {
      case 1:
        return "x";
      case 2:
        return "x,y";
      case 3:
        return "x,y,z";
      default:
        Assert(false, ExcNotImplemented());
    }
  return "";
}
 
 
 
DEAL_II_NAMESPACE_CLOSE
 
#endif