EFQRCodeRecognizer.swift 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. //
  2. // EFQRCodeRecognizer.swift
  3. // EFQRCode
  4. //
  5. // Created by EyreFree on 2017/3/28.
  6. //
  7. // Copyright (c) 2017 EyreFree <eyrefree@eyrefree.org>
  8. //
  9. // Permission is hereby granted, free of charge, to any person obtaining a copy
  10. // of this software and associated documentation files (the "Software"), to deal
  11. // in the Software without restriction, including without limitation the rights
  12. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13. // copies of the Software, and to permit persons to whom the Software is
  14. // furnished to do so, subject to the following conditions:
  15. //
  16. // The above copyright notice and this permission notice shall be included in
  17. // all copies or substantial portions of the Software.
  18. //
  19. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25. // THE SOFTWARE.
  26. #if os(iOS) || os(macOS) || os(tvOS)
  27. import CoreImage
  28. @objcMembers
  29. public class EFQRCodeRecognizer: NSObject {
  30. private var image: CGImage? {
  31. didSet {
  32. contentArray = nil
  33. }
  34. }
  35. public func setImage(image: CGImage?) {
  36. self.image = image
  37. }
  38. private var contentArray: [String]?
  39. public init(image: CGImage) {
  40. self.image = image
  41. }
  42. public func recognize() -> [String]? {
  43. if nil == contentArray {
  44. contentArray = getQRString()
  45. }
  46. return contentArray
  47. }
  48. // Get QRCodes from image
  49. private func getQRString() -> [String]? {
  50. guard let finalImage = image else {
  51. return nil
  52. }
  53. let result = finalImage.toCIImage().recognizeQRCode(options: [CIDetectorAccuracy: CIDetectorAccuracyHigh])
  54. if result.isEmpty {
  55. return finalImage.grayscale()?.toCIImage().recognizeQRCode(
  56. options: [CIDetectorAccuracy: CIDetectorAccuracyLow]
  57. )
  58. }
  59. return result
  60. }
  61. }
  62. #endif