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