Clean/plugins/classify_photo/ios/Classes/ClassifyPhotoPlugin.swift

import Flutter
import Photos
import UIKit

public class ClassifyPhotoPlugin: NSObject, FlutterPlugin {

  var photoClassifier = ClassifyPhoto()

  public static func register(with registrar: FlutterPluginRegistrar) {
    let channel = FlutterMethodChannel(name: "classify_photo", binaryMessenger: registrar.messenger())
    let instance = ClassifyPhotoPlugin()
    registrar.addMethodCallDelegate(instance, channel: channel)
  }

  public func handle(_ call: FlutterMethodCall, result: @escaping FlutterResult) {
    print("iOS: Received method call: \(call.method)")
    switch call.method {
    case "getPhoto":
        self.getPhoto(flutterResult: result)
    case "getPlatformVersion":
      result("iOS " + UIDevice.current.systemVersion)
    default:
      result(FlutterMethodNotImplemented)
    }
  }

  private func getPhoto(flutterResult: @escaping FlutterResult) {

      DispatchQueue.global(qos: .userInitiated).async { [weak self] in
           guard let self = self else { return }

           let fetchOptions = PHFetchOptions()
           let allPhotos = PHAsset.fetchAssets(with: .image, options: fetchOptions)

           photoClassifier.classifyPhotos(
               assets: allPhotos,
               progressHandler: { (stage, progress) in
                   print("Progress: \(stage) - \(progress)")
               },
               completion: { result in
                   var resultData: [[String: Any]] = []
                   let mainGroup = DispatchGroup()
                   
                   // 处理截图
                   mainGroup.enter()
                   self.processPhotoGroup(assets: result.screenshots, groupName: "screenshots") { groupData in
                       if !groupData.isEmpty {
                           resultData.append(["group": groupData, "type": "screenshots"])
                       }
                       mainGroup.leave()
                   }
                   
                   // 处理相似照片组
                   for photoGroup in result.similarPhotos {
                       mainGroup.enter()
                       self.processPhotoGroup(assets: photoGroup, groupName: "similar") { groupData in
                           if !groupData.isEmpty {
                               resultData.append(["group": groupData, "type": "similar"])
                           }
                           mainGroup.leave()
                       }
                   }
                   
                   // 处理地点分组
                   for (location, assets) in result.locations {
                       mainGroup.enter()
                       self.processPhotoGroup(assets: assets, groupName: location) { groupData in
                           if !groupData.isEmpty {
                               resultData.append(["group": groupData, "type": "location", "name": location])
                           }
                           mainGroup.leave()
                       }
                   }
                   
                   // 处理人物分组
                   for (person, assets) in result.people {
                       mainGroup.enter()
                       self.processPhotoGroup(assets: assets, groupName: person) { groupData in
                           if !groupData.isEmpty {
                               resultData.append(["group": groupData, "type": "people"])
                           }
                           mainGroup.leave()
                       }
                   }
                   
                   mainGroup.notify(queue: .main) {
                       print("Final result count: \(resultData.count)")
                       flutterResult(resultData)
                   }
               }
           )
       }

  // 在后台队列处理照片
//       DispatchQueue.global(qos: .userInitiated).async {
//           let fetchOptions = PHFetchOptions()
//           let allPhotos = PHAsset.fetchAssets(with: .image, options: fetchOptions)
//           var resultData: [[String: Any]] = []
//
//           // 创建一个组来存储所有照片
//           var allPhotosGroup: [[String: Any]] = []
//           let semaphore = DispatchSemaphore(value: 0)
//
//           allPhotos.enumerateObjects { (asset, index, stop) in
//               // 获取图片的文件URL
//               let options = PHContentEditingInputRequestOptions()
//               options.isNetworkAccessAllowed = true
//
//               asset.requestContentEditingInput(with: options) { (input, info) in
//                   if let input = input, let url = input.fullSizeImageURL {
//                       let photoInfo: [String: Any] = [
//                           "path": url.path,
//                           "id": asset.localIdentifier,
//                           "width": asset.pixelWidth,
//                           "height": asset.pixelHeight,
//                           "creationDate": asset.creationDate?.timeIntervalSince1970 ?? 0
//                       ]
//                       allPhotosGroup.append(photoInfo)
//                   }
//                   semaphore.signal()
//               }
//               semaphore.wait()
//           }
//
//           if !allPhotosGroup.isEmpty {
//               resultData.append(["group": allPhotosGroup])
//           }
//
//           // 在主线程返回结果
//           DispatchQueue.main.async {
//               flutterResult(resultData)
//           }
//       }
//      // 在后台队列处理照片
//      DispatchQueue.global(qos: .userInitiated).async { [weak self] in
//          guard let self = self else { return }
//
//          let fetchOptions = PHFetchOptions()
//          let allPhotos = PHAsset.fetchAssets(with: .image, options: fetchOptions)
//
//          photoClassifier.classifyPhotos(
//              assets: allPhotos,
//              progressHandler: { (stage, progress) in
//                  // 进度更新
//              },
//              completion: { result in
//                  var similarPhotos = result.similarPhotos
//                  var resultData: [[String: Any]] = []
//
//                  for photoGroup in similarPhotos {
//                      var groupData: [[String: Any]] = []
//                      let semaphore = DispatchSemaphore(value: 0)
//
//                      for asset in photoGroup {
//                          let options = PHImageRequestOptions()
//                          options.deliveryMode = .highQualityFormat
//                          options.isSynchronous = false
//
//                          PHImageManager.default().requestImage(
//                              for: asset,
//                              targetSize: CGSize(width: 800, height: 800),
//                              contentMode: .aspectFit,
//                              options: options
//                          ) { image, info in
//                              if let image = image,
//                                 let imageData = image.jpegData(compressionQuality: 0.8) {
//                                  let photoInfo: [String: Any] = [
//                                      "data": FlutterStandardTypedData(bytes: imageData),
//                                      "id": asset.localIdentifier,
//                                      "width": image.size.width,
//                                      "height": image.size.height,
//                                      "creationDate": asset.creationDate?.timeIntervalSince1970 ?? 0
//                                  ]
//                                  groupData.append(photoInfo)
//                              }
//                              semaphore.signal()
//                          }
//                          semaphore.wait()
//                      }
//
//                      if !groupData.isEmpty {
//                          resultData.append(["group": groupData])
//                      }
//                  }
//
//                  // 在主线程返回结果
//                  DispatchQueue.main.async {
//                      flutterResult(resultData)
//                  }
//              }
//          )
//      }
  }
    
    // 辅助方法：处理照片组
      private func processPhotoGroup(assets: [PHAsset], groupName: String, completion: @escaping ([[String: Any]]) -> Void) {
          let photoProcessGroup = DispatchGroup()
          var groupData: [[String: Any]] = []
          
          for asset in assets {
              photoProcessGroup.enter()
              
              let options = PHContentEditingInputRequestOptions()
              options.isNetworkAccessAllowed = true
              
              asset.requestContentEditingInput(with: options) { (input, info) in
                  defer { photoProcessGroup.leave() }
                  
                  if let input = input, let url = input.fullSizeImageURL {
                      let photoInfo: [String: Any] = [
                          "path": url.path,
                          "id": asset.localIdentifier,
                          "width": asset.pixelWidth,
                          "height": asset.pixelHeight,
                          "creationDate": asset.creationDate?.timeIntervalSince1970 ?? 0
                      ]
                      groupData.append(photoInfo)
                  }
              }
          }
          
          photoProcessGroup.notify(queue: .main) {
              completion(groupData)
          }
      }
}