attempted fix for #49. Uses new select_set() function

LanguageMachines · Oct 7, 2022 · 46f8f31 · 46f8f31
1 parent bd43a98
commit 46f8f31
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/folia_impl.cxx b/src/folia_impl.cxx
@@ -2298,9 +2298,17 @@ namespace folia {
       cerr << "no space? s: " << s << endl;
     }
     if ( s ){
-      vector<Word*> words = s->select<Word>(false);
-      if ( !words.empty() ){
-	Word *last = words.back();
+      set<ElementType> wanted = {Word_t,Correction_t};
+      set<ElementType> excl;
+      vector<FoliaElement*> elts = s->select_set( wanted,
+						  "",
+						  excl,
+						  SELECT_FLAGS::LOCAL );
+      if ( !elts.empty() ){
+	if ( debug ){
+	  cerr << "found some mixed stuff: " << elts << endl;
+	}
+	FoliaElement *last = elts.back();
 	if ( debug ){
 	  cerr << "no space? last: " << last << endl;
 	}