Merge pull request #596 from trueagi-io/escape

add escape seq to fast_load
trueagi-io · Feb 22, 2024 · 02fc0c2 · 02fc0c2
2 parents 40088e1 + f5de873
commit 02fc0c2
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/python/hyperonpy.cpp b/python/hyperonpy.cpp
@@ -1011,10 +1011,16 @@ PYBIND11_MODULE(hyperonpy, m) {
         std::vector<atom_t> children;
         std::string str = "";
         int depth = 0;
+        bool isescape = false;
         while(f) {
             char c = f.get();
             bool isspace = std::isspace(c);
-            if(isspace or c == '(' or c == ')') {
+            if(isescape) {
+                str += c;
+                isescape = false;
+                continue;
+            }
+            if((isspace or c == '(' or c == ')')) {
                 if(c == ')' and depth == 0) {
                     // ignore for now --> exception
                     continue;
@@ -1066,7 +1072,11 @@ PYBIND11_MODULE(hyperonpy, m) {
                     }
                 }
             } else {
-                str += c;
+                if(c == '\\') {
+                    isescape = true;
+                } else {
+                    str += c;
+                }
             }
         }
         return true;