diff --git a/basisu_enc.h b/basisu_enc.h index 7d42121..05656da 100644 --- a/basisu_enc.h +++ b/basisu_enc.h @@ -1070,7 +1070,7 @@ namespace basisu assert(node.is_leaf()); var_heap.delete_top(); - + if (node.m_training_vecs.size() > 1) { if (split_node(node_index, var_heap, l_children, r_children)) @@ -1197,11 +1197,11 @@ namespace basisu } if ((l_child.m_var > 0.0f) && (l_child.m_training_vecs.size() > 1)) - var_heap.add_heap(l_child_index, l_var); - + var_heap.add_heap(l_child_index, l_child.m_var); + if ((r_child.m_var > 0.0f) && (r_child.m_training_vecs.size() > 1)) - var_heap.add_heap(r_child_index, r_var); - + var_heap.add_heap(r_child_index, r_child.m_var); + return true; }