1 files changed, 150 insertions, 0 deletions
diff --git a/Source/Parser/WarpXParser.cpp b/Source/Parser/WarpXParser.cpp
new file mode 100644
index 000000000..8c800215f
--- /dev/null
+++ b/Source/Parser/WarpXParser.cpp
@@ -0,0 +1,150 @@
+
+#include "WarpXParser.H"
+
+WarpXParser::WarpXParser (std::string const& func_body)
+{
+    define(func_body);
+}
+
+void
+WarpXParser::define (std::string const& func_body)
+{
+    clear();
+
+    m_expression = func_body;
+    std::string f = m_expression + "\n";
+
+#ifdef _OPENMP
+
+    int nthreads = omp_get_max_threads();
+    m_variables.resize(nthreads);
+    m_parser.resize(nthreads);
+    m_parser[0] = wp_c_parser_new(f.c_str());
+#pragma omp parallel
+    {
+        int tid = omp_get_thread_num();
+        if (tid > 0) {
+            m_parser[tid] = wp_parser_dup(m_parser[0]);
+        }
+    }
+
+#else
+
+    m_parser = wp_c_parser_new(f.c_str());
+
+#endif
+}
+
+WarpXParser::~WarpXParser ()
+{
+    clear();
+}
+
+void
+WarpXParser::clear ()
+{
+    m_expression.clear();
+
+#ifdef _OPENMP
+
+    if (!m_parser.empty())
+    {
+#pragma omp parallel
+        {
+            wp_parser_delete(m_parser[omp_get_thread_num()]);
+        }
+    }
+    m_parser.clear();
+    m_variables.clear();
+
+#else
+
+    if (m_parser) wp_parser_delete(m_parser);
+    m_parser = nullptr;
+
+#endif
+}
+
+void
+WarpXParser::registerVariable (std::string const& name, double& var)
+{
+    // We assume this is called inside OMP parallel region
+#ifdef _OPENMP
+    wp_parser_regvar(m_parser[omp_get_thread_num()], name.c_str(), &var);
+#else
+    wp_parser_regvar(m_parser, name.c_str(), &var);
+#endif
+}
+
+void
+WarpXParser::registerVariables (std::vector<std::string> const& names)
+{
+#ifdef _OPENMP
+
+// This must be called outside OpenMP parallel region.
+#pragma omp parallel
+    {
+        const int tid = omp_get_thread_num();
+        struct wp_parser* p = m_parser[tid];
+        auto& v = m_variables[tid];
+        for (int j = 0; j < names.size(); ++j) {
+            wp_parser_regvar(p, names[j].c_str(), &(v[j]));
+        }
+    }
+
+#else
+
+    for (int j = 0; j < names.size(); ++j) {
+        wp_parser_regvar(m_parser, names[j].c_str(), &(m_variables[j]));
+    }
+
+#endif
+}
+
+void
+WarpXParser::setConstant (std::string const& name, double c)
+{
+#ifdef _OPENMP
+
+    bool in_parallel = omp_in_parallel();
+    // We don't know if this is inside OMP parallel region or not
+#pragma omp parallel if (!in_parallel)
+    {
+        wp_parser_setconst(m_parser[omp_get_thread_num()], name.c_str(), c);
+    }
+
+#else
+
+    wp_parser_setconst(m_parser, name.c_str(), c);
+
+#endif
+}
+
+void
+WarpXParser::print () const
+{
+#ifdef _OPENMP
+#pragma omp critical(warpx_parser_pint)
+    wp_ast_print(m_parser[omp_get_thread_num()]->ast);
+#else
+    wp_ast_print(m_parser->ast);
+#endif
+}
+
+std::string const&
+WarpXParser::expr () const
+{
+    return m_expression;
+}
+
+std::set<std::string>
+WarpXParser::symbols () const
+{
+    std::set<std::string> results;
+#ifdef _OPENMP
+    wp_ast_get_symbols(m_parser[omp_get_thread_num()]->ast, results);
+#else
+    wp_ast_get_symbols(m_parser->ast, results);
+#endif
+    return results;
+}