sortCandidates.swift

extension Array where Element == OCRCandidate {

    func sortCandidates() -> [OCRCandidate] {
        self.sorted { (currentCandidate, nextCandidate) -> Bool in
            let currentCandidateBottomY = currentCandidate.boundingPoints.avgBottomY
            let nextCandidateBottomY = nextCandidate.boundingPoints.avgBottomY
            
            if currentCandidateBottomY < nextCandidateBottomY {
                // Rounding item for discrepancies with Y positions.
                let difference = (currentCandidateBottomY - nextCandidateBottomY).roundTo2dp

                // Allowing 0.01 due to image distortion or any other anomalies with the image.
                if abs(difference) <= 0.01 {
                    return currentCandidate.boundingPoints.bottomLeft.x < nextCandidate.boundingPoints.bottomLeft.x
                } else {
                    return true
                }
            }
            return false
        }
    }
}