glue rules with stripped BitPar labels

1 files changed, 9 insertions, 2 deletions

diff --git a/phrase-extract/extract-ghkm/ExtractGHKM.cpp b/phrase-extract/extract-ghkm/ExtractGHKM.cpp
index 1394707a7..a9a0bffc2 100644
--- a/phrase-extract/extract-ghkm/ExtractGHKM.cpp
+++ b/phrase-extract/extract-ghkm/ExtractGHKM.cpp
@@ -931,11 +931,18 @@ void ExtractGHKM::StripBitParLabels(
   }
   for (std::map<std::string,int>::const_iterator it=topLabelSet.begin();
        it!=topLabelSet.end(); ++it) {
-    std::map<std::string, int>::iterator found=outTopLabelSet.find(it->first);
+    size_t pos = it->first.find('-');
+    std::string stripped;
+    if (pos == std::string::npos) {
+      stripped = it->first;
+    } else {
+      stripped = it->first.substr(0,pos);
+    }
+    std::map<std::string, int>::iterator found=outTopLabelSet.find(stripped);
     if (found != outTopLabelSet.end()) {
       found->second += it->second;
     } else {
-      outTopLabelSet.insert(std::pair<std::string,int>(it->first,it->second));
+      outTopLabelSet.insert(std::pair<std::string,int>(stripped,it->second));
     }
   }
 }