func recognize()

in ViT4MNIST/ViT4MNIST/HandwrittenDigitRecogizer.swift [25:50]

• 21 lines of code
• 8 McCabe index (conditional complexity)

    func recognize(allPoints: [[CGPoint]], width: CGFloat, height: CGFloat) -> String? {
        if (allPoints.count == 0) {
            return nil
        }
        let BLANK = -MNIST_STD / MNIST_MEAN
        let NON_BLANK = (1.0 - MNIST_STD) / MNIST_MEAN

        var inputs = Array(repeating: BLANK, count: MNIST_IMAGE_SIZE * MNIST_IMAGE_SIZE)
        
        // loop through each stroke
        for consecutivePoints in allPoints {
            // loop through each point in the stroke
            for point in consecutivePoints {
                if point.x > width || point.y > height ||
                    point.x < 0 || point.y < 0 {
                    continue;
                }
                let x = MNIST_IMAGE_SIZE * Int(point.x) / Int(width)
                let y = MNIST_IMAGE_SIZE * Int(point.y) / Int(height)
                let loc = y * MNIST_IMAGE_SIZE + x
                inputs[loc] = NON_BLANK
            }
        }
        
        return module.recognize(points: inputs)
    }