]> git.donarmstrong.com Git - rsem.git/blobdiff - NoiseProfile.h
Modified the acknowledgement section of README.md
[rsem.git] / NoiseProfile.h
index 9c3dd2230dec94e1a515473fb741d4eecc16ca8e..5662e0b216b473f3f3cea0440457c5b97d2e4fd2 100644 (file)
@@ -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) {