X-Git-Url: https://git.donarmstrong.com/?p=rsem.git;a=blobdiff_plain;f=NoiseProfile.h;h=5662e0b216b473f3f3cea0440457c5b97d2e4fd2;hp=9c3dd2230dec94e1a515473fb741d4eecc16ca8e;hb=636b82d9f60ebcbec7ef1b73ba23bbbacfd8b36a;hpb=a95154919f950f86de9104b2b9dcf1f0c7e83387

diff --git a/NoiseProfile.h b/NoiseProfile.h
index 9c3dd22..5662e0b 100644
--- a/NoiseProfile.h
+++ b/NoiseProfile.h
@@ -5,6 +5,7 @@
 #include<cstdio>
 #include<cstring>
 #include<string>
+#include<cassert>
 
 #include "utils.h"
 #include "RefSeq.h"
@@ -80,7 +81,7 @@ void NoiseProfile::finish() {
 	logp = 0.0;
 	sum = 0.0;
 	for (int i = 0; i < NCODES; i++) sum += (p[i] + c[i]);
-	if (sum <= 0.0) return;
+	if (sum <= EPSILON) return;
 	for (int i = 0; i < NCODES; i++) {
 		p[i] = (p[i] + c[i]) / sum;
 		if (c[i] > 0.0) { logp += c[i] * log(p[i]); }
@@ -119,10 +120,10 @@ void NoiseProfile::read(FILE *fi) {
 	int tmp_ncodes;
 
 	memset(c, 0, sizeof(c));
-	fscanf(fi, "%d", &tmp_ncodes);
+	assert(fscanf(fi, "%d", &tmp_ncodes) == 1);
 	assert(tmp_ncodes == NCODES);
 	for (int i = 0; i < NCODES; i++)
-		fscanf(fi, "%lf", &p[i]);
+	  assert(fscanf(fi, "%lf", &p[i]) == 1);
 }
 
 void NoiseProfile::write(FILE *fo) {