Allows to define duplicated user defined symbols

1 files changed, 15 insertions, 0 deletions

diff --git a/src/model_interface.cc b/src/model_interface.cc
index 656334f..255d1be 100644
--- a/src/model_interface.cc
+++ b/src/model_interface.cc
@@ -54,6 +54,21 @@ int PrefixMatcher::PrefixMatch(StringPiece w, bool *found) const {
   return mblen;
 }
 
+std::string PrefixMatcher::GlobalReplace(StringPiece w, StringPiece out) const {
+  std::string result;
+  while (!w.empty()) {
+    bool found = false;
+    const int mblen = PrefixMatch(w, &found);
+    if (found) {
+      result.append(out.data(), out.size());
+    } else {
+      result.append(w.data(), mblen);
+    }
+    w.remove_prefix(mblen);
+  }
+  return result;
+}
+
 ModelInterface::ModelInterface(const ModelProto &model_proto)
     : model_proto_(&model_proto), status_(util::OkStatus()) {}
 ModelInterface::~ModelInterface() {}