Fixed a bug

deweylab · Nov 11, 2015 · 25a74f0 · 25a74f0
1 parent 308e5b6
commit 25a74f0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/Transcript.h b/Transcript.h
@@ -126,7 +126,7 @@ void Transcript::read(std::ifstream& fin) {
 	strin>> transcript_id>> transcript_name;
 
 	getline(fin, tmp);
-	strin.str(tmp);
+	strin.clear(); strin.str(tmp);
 	strin>> gene_id>> gene_name;
 
 	getline(fin, seqname);

diff --git a/WriteResults.h b/WriteResults.h
@@ -232,7 +232,11 @@ void writeResultsEM(int M, const char* refName, const char* imdName, Transcripts
 	  }
 	  for (int i = 1; i <= M; i++) {
 	    const Transcript& transcript = transcripts.getTranscriptAt(i);
-	    fprintf(fo, "%s%c", transcript.getGeneID().c_str(), (i < M ? '\t' : '\n'));
+
+	    fprintf(fo, "%s", transcript.getGeneID().c_str());
+	    if (appendNames && transcript.getGeneName() != "")
+	      fprintf(fo, "_%s", transcript.getGeneName().c_str());
+	    fprintf(fo, "%c", (i < M ? '\t' : '\n'));
 	  }
 	  for (int i = 1; i <= M; i++)
 	    fprintf(fo, "%d%c", tlens[i], (i < M ? '\t' : '\n'));
@@ -313,7 +317,8 @@ void writeResultsEM(int M, const char* refName, const char* imdName, Transcripts
 		const Transcript& transcript = transcripts.getTranscriptAt(gi.spAt(i));
 
 		fprintf(fo, "%s", transcript.getGeneID().c_str());
-		if (appendNames && transcript.getGeneName() != "") fprintf(fo, "_%s", transcript.getGeneName().c_str());
+		if (appendNames && transcript.getGeneName() != "")
+		  fprintf(fo, "_%s", transcript.getGeneName().c_str());
 		fprintf(fo, "%c", (i < m - 1 ? '\t' : '\n'));
 	}
 	for (int i = 0; i < m; i++) {