Weigh the group_hit_counts by EC counts

tmaklin · tmaklin · commit d7dee7b248b8 · 2024-05-29T18:39:38.000+02:00
diff --git a/include/Likelihood.hpp b/include/Likelihood.hpp
@@ -117,7 +117,7 @@ class LL_WOR21 : public Likelihood<T> {
 	// Create mask identifying groups that have at least 1 alignment
 	for (size_t i = 0; i < num_ecs; ++i) {
 	    for (size_t j = 0; j < n_groups; ++j) {
-		group_hit_counts[j] += (alignment(j, i) > 0);
+		group_hit_counts[j] += (alignment(j, i) > 0) * alignment.reads_in_ec(i);
 	    }
 	}
 	for (size_t i = 0; i < n_groups; ++i) {

Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ class LL_WOR21 : public Likelihood<T> {`
`117`	`117`	`// Create mask identifying groups that have at least 1 alignment`
`118`	`118`	`for (size_t i = 0; i < num_ecs; ++i) {`
`119`	`119`	`for (size_t j = 0; j < n_groups; ++j) {`
`120`		`- group_hit_counts[j] += (alignment(j, i) > 0);`
	`120`	`+ group_hit_counts[j] += (alignment(j, i) > 0) * alignment.reads_in_ec(i);`
`121`	`121`	`}`
`122`	`122`	`}`
`123`	`123`	`for (size_t i = 0; i < n_groups; ++i) {`