`count` and `i` are used interchangeably, why not use a single variable?

The patch attached does that, renaming both to segment_number.

Additionally, at the end of the loop, usages of segment_number are changed to
n_segments for better readability (segment_number is a loop index and should
not be used outside of the loop in presence of better alternatives).