fixing a crash when the chromosome name only has one word

smithlabcode · Jul 3, 2021 · 17d21f3 · 17d21f3
1 parent 9a070f4
commit 17d21f3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/smithlab_os.cpp b/smithlab_os.cpp
@@ -390,7 +390,11 @@ read_fasta_file_short_names(const string &filename,
   string line;
   while (getline(in, line)) {
     if (line[0] == '>') {
-      names.push_back(string(begin(line) + 1,
+      const auto first_space = line.find_first_of(" \t", 1);
+      if (first_space == string::npos)
+        names.push_back(line.substr(1));
+      else
+        names.push_back(string(begin(line) + 1,
                              begin(line) + line.find_first_of(" \t", 1)));
       sequences.push_back(string());
     }