diff --git a/public/gears/src/main/scala/nl/lumc/sasc/biopet/pipelines/gears/GearsKraken.scala b/public/gears/src/main/scala/nl/lumc/sasc/biopet/pipelines/gears/GearsKraken.scala index d9d2fd06346724bcbc99daabd1b17b83efda3d58..3c7b0d0f5f7aa8cb1c381eee1deab693b110f4a4 100644 --- a/public/gears/src/main/scala/nl/lumc/sasc/biopet/pipelines/gears/GearsKraken.scala +++ b/public/gears/src/main/scala/nl/lumc/sasc/biopet/pipelines/gears/GearsKraken.scala @@ -124,9 +124,16 @@ object GearsKraken { case (k, v) => val node = <node name={ k }></node> val sizes = samples.map { sample => - <val> - { getValue(sample, (path ::: k :: Nil).tail, "size").getOrElse(0) } - </val> + if (k == "root") { + val unclassified = summaries(sample)("unclassified").asInstanceOf[Map[String, Any]]("size").asInstanceOf[Long] + <val> + {getValue(sample, (path ::: k :: Nil).tail, "size").getOrElse(0).toString.toLong + unclassified} + </val> + } else { + <val> + {getValue(sample, (path ::: k :: Nil).tail, "size").getOrElse(0)} + </val> + } } val size = <size>{ sizes }</size> node.copy(child = size ++ createNodes(v.asInstanceOf[mutable.Map[String, Any]], path ::: k :: Nil))