Fix createdb support for db input

Author: matchy233

src/util/createdb.cpp

@@ -98,9 +98,12 @@ int createdb(int argc, const char **argv, const Command& command) {
 
     size_t fileCount = filenames.size();
     DBReader<unsigned int>* reader = NULL;
+    DBReader<unsigned int>* hdrReader = nullptr;
     if (dbInput == true) {
         reader = new DBReader<unsigned int>(par.db1.c_str(), par.db1Index.c_str(), 1, DBReader<unsigned int>::USE_DATA | DBReader<unsigned int>::USE_INDEX | DBReader<unsigned int>::USE_LOOKUP);
         reader->open(DBReader<unsigned int>::LINEAR_ACCCESS);
+        hdrReader = new DBReader<unsigned int>((par.db1 + "_h").c_str(), (par.db1 + "_h.index").c_str(), 1, DBReader<unsigned int>::USE_DATA | DBReader<unsigned int>::USE_INDEX);
+        hdrReader->open(DBReader<unsigned int>::LINEAR_ACCCESS);
         fileCount = reader->getSize();
     }
 
@@ -126,8 +129,11 @@ int createdb(int argc, const char **argv, const Command& command) {
         }
 
         KSeqWrapper* kseq = NULL;
+        std::string seq = ">";
         if (dbInput == true) {
-            kseq = new KSeqBuffer(reader->getData(fileIdx, 0), reader->getEntryLen(fileIdx) - 1);
+            seq.append(hdrReader->getData(fileIdx, 0));
+            seq.append(reader->getData(fileIdx, 0));
+            kseq = new KSeqBuffer(seq.c_str(), seq.length());
         } else {
             kseq = KSeqFactory(filenames[fileIdx].c_str());
         }
@@ -260,6 +266,8 @@ int createdb(int argc, const char **argv, const Command& command) {
     if (dbInput == true) {
         reader->close();
         delete reader;
+        hdrReader->close();
+        delete hdrReader;
     }
 
     if (entries_num == 0) {