Merge branch 'add_harmonic_beam' of github.com:lucafedeli88/WarpX into add_harmonic_beam

1 files changed, 42 insertions, 0 deletions

diff --git a/Source/Utils/WarpXUtil.cpp b/Source/Utils/WarpXUtil.cpp
index e8533a382..e9fb958fd 100644
--- a/Source/Utils/WarpXUtil.cpp
+++ b/Source/Utils/WarpXUtil.cpp
@@ -163,3 +163,45 @@ namespace WarpXUtilIO{
         return  of.good();
     }
 }
+
+void Store_parserString(amrex::ParmParse& pp, std::string query_string,
+                        std::string& stored_string)
+{
+
+    char cstr[query_string.size()+1];
+    strcpy(cstr, query_string.c_str());
+
+    std::vector<std::string> f;
+    pp.getarr(cstr, f);
+    stored_string.clear();
+    for (auto const& s : f) {
+        stored_string += s;
+    }
+    f.clear();
+
+}
+
+
+WarpXParser makeParser (std::string const& parse_function)
+{
+    WarpXParser parser(parse_function);
+    parser.registerVariables({"x","y","z"});
+    ParmParse pp("my_constants");
+    std::set<std::string> symbols = parser.symbols();
+    symbols.erase("x");
+    symbols.erase("y");
+    symbols.erase("z");
+    for (auto it = symbols.begin(); it != symbols.end(); ) {
+        Real v;
+        if (pp.query(it->c_str(), v)) {
+           parser.setConstant(*it, v);
+           it = symbols.erase(it);
+        } else {
+           ++it;
+        }
+    }
+    for (auto const& s : symbols) {
+        amrex::Abort("makeParser::Unknown symbol "+s);
+    }
+    return parser;
+}