@@ -35,7 +35,7 @@ class ViewController: UIViewController, UIPickerViewDelegate, UIPickerViewDataSo
3535 let images = [
3636 ImageDisplay ( file: " do-not-feed-birds " , name: " Image 1 " ) ,
3737 ImageDisplay ( file: " walk-on-grass " , name: " Image 2 " ) ,
38- ]
38+ ]
3939
4040 override func viewDidLoad( ) {
4141 super. viewDidLoad ( )
@@ -57,8 +57,6 @@ class ViewController: UIViewController, UIPickerViewDelegate, UIPickerViewDataSo
5757 runCloudTextRecognition ( with: imageView. image!)
5858 }
5959
60-
61-
6260 // MARK: Text Recognition
6361
6462 func runTextRecognition( with image: UIImage ) {
@@ -71,57 +69,60 @@ class ViewController: UIViewController, UIPickerViewDelegate, UIPickerViewDataSo
7169 func runCloudTextRecognition( with image: UIImage ) {
7270 let visionImage = VisionImage ( image: image)
7371 cloudTextDetector. detect ( in: visionImage) { features, error in
72+ if let error = error {
73+ print ( " Received error: \( error) " )
74+ return
75+ }
76+
7477 self . processCloudResult ( from: features, error: error)
7578 }
7679 }
80+
7781
7882 // MARK: Image Drawing
7983
8084 func processResult( from text: [ VisionText ] ? , error: Error ? ) {
8185 removeFrames ( )
82- if let features = text, let image = imageView. image {
83- for text in features {
84- if let block = text as? VisionTextBlock {
85- for line in block. lines {
86- for element in line. elements {
87- self . addFrameView (
88- featureFrame: element. frame,
89- imageSize: image. size,
90- viewFrame: self . imageView. frame,
91- text: element. text
92- )
93- }
86+ guard let features = text, let image = imageView. image else {
87+ return
88+ }
89+ for text in features {
90+ if let block = text as? VisionTextBlock {
91+ for line in block. lines {
92+ for element in line. elements {
93+ self . addFrameView (
94+ featureFrame: element. frame,
95+ imageSize: image. size,
96+ viewFrame: self . imageView. frame,
97+ text: element. text
98+ )
9499 }
95100 }
96101 }
97102 }
98103 }
104+
99105
100106 func processCloudResult( from text: VisionCloudText ? , error: Error ? ) {
101107 removeFrames ( )
102- if let features = text, let image = imageView. image, let pages = features. pages {
103- for page in pages {
104- if let blocks = page. blocks {
105- for block in blocks {
106- if let paragraphs = block. paragraphs {
107- for paragraph in paragraphs {
108- if let words = paragraph. words {
109- for word in words {
110- self . addFrameView (
111- featureFrame: word. frame,
112- imageSize: image. size,
113- viewFrame: self . imageView. frame
114- )
115- }
116- }
117- }
118- }
108+ guard let features = text, let image = imageView. image, let pages = features. pages else {
109+ return
110+ }
111+ for page in pages {
112+ for block in page. blocks ?? [ ] {
113+ for paragraph in block. paragraphs ?? [ ] {
114+ for word in paragraph. words ?? [ ] {
115+ self . addFrameView (
116+ featureFrame: word. frame,
117+ imageSize: image. size,
118+ viewFrame: self . imageView. frame
119+ )
119120 }
120121 }
121-
122122 }
123123 }
124124 }
125+
125126
126127 /// Converts a feature frame to a frame UIView that is displayed over the image.
127128 ///
@@ -184,10 +185,10 @@ class ViewController: UIViewController, UIPickerViewDelegate, UIPickerViewDataSo
184185 textLayer. string = text
185186 textLayer. fontSize = 12.0
186187 textLayer. foregroundColor = Constants . lineColor
187- let center = CGPoint ( x: rect. midX, y: rect. midX )
188+ let center = CGPoint ( x: rect. midX, y: rect. midY )
188189 textLayer. position = center
189- textLayer. isHidden = false
190- textLayer. alignmentMode = kCAAlignmentLeft
190+ textLayer. frame = rect
191+ textLayer. alignmentMode = kCAAlignmentCenter
191192 textLayer. contentsScale = UIScreen . main. scale
192193 frameSublayer. addSublayer ( textLayer)
193194 }
0 commit comments