import 'photo_classifier_platform_interface.dart';
import 'models.dart';

/// 照片分类插件主类
class PhotoClassifier {
  
  /// 开始照片分类过程，返回一个提供实时更新的流
  /// 
  /// [types] 指定要执行的分类类型，可以是以下值的组合:
  /// - 'similar': 查找相似图片组
  /// - 'people': 查找包含人物的图片
  /// - 'screenshot': 查找屏幕截图
  /// - 'blurry': 查找模糊图片
  /// 
  /// 返回一个流，发出ClassificationEvent事件，包含进度和当前结果
  Stream<ClassificationEvent?> startClassificationStream({
    List<PhotoImageClassifyType> types = PhotoImageClassifyType.values,
  }) {
    return PhotoClassifierPlatform.instance.startClassificationStream(types: types);
  }
  
  
  /// 配置照片分类器
  /// 
  /// [batchSize] 每批处理的图片数量
  /// [maxConcurrentProcessing] 最大并发处理数量
  /// [similarityThreshold] 相似度判定阈值（0.0-1.0），值越大表示要求越相似
  Future configureClassifier({
    int? batchSize, 
    int? maxConcurrentProcessing, 
    double? similarityThreshold,
  }) {
    return PhotoClassifierPlatform.instance.configureClassifier(
      batchSize: batchSize,
      maxConcurrentProcessing: maxConcurrentProcessing,
      similarityThreshold: similarityThreshold,
    );
  }
  
  /// 重置照片分类器
  Future resetClassifier() {
    return PhotoClassifierPlatform.instance.resetClassifier();
  }
  
  /// 检查是否已加载完所有资源
  Future checkAssetsLoaded() {
    return PhotoClassifierPlatform.instance.checkAssetsLoaded();
  }
}