BFRecordScreenKit/BFRecordScreenKit/Classes/BFRecordItemModel.swift

//
//  BFRecordItemModel.swift
//  BFRecordScreenKit
//
//  Created by 胡志强 on 2021/12/6.
//

import Foundation
import Photos
import BFFramework

struct SplitRecordRange {
    var isRecord:Bool = false
    var range:CMTimeRange
    var index:Int
}

public class BFRecordItemModel: NSObject {
//    var baseMaterial : AVURLAsset?
    var localPath : String?
    var materialDuraion: Double = 0.0
    var fetchCoverImg : ((UIImage)->Void)?
    var fetchAVUrlAsset : ((AVURLAsset)->Void)?
    var fetchPlayItem : ((AVPlayerItem)->Void)?
    var dealedDurationRanges = [SplitRecordRange]()                     // 录音切割的时间区间，合成导出时计算
    public var voiceStickers = [PQVoiceModel]()                         //
    public var videoStickers = [PQEditVisionTrackMaterialsModel]()      // 合成导出时计算
    public var imageStickers = [PQEditVisionTrackMaterialsModel]()      //
    public var titleStickers = [PQEditSubTitleModel]()                  // 字幕贴纸
    public var coverImg : UIImage?
    public var index = 0
    public var width = 0
    public var height = 0
    
    func initOriginData(phasset:PHAsset){
        width = phasset.pixelWidth
        height = phasset.pixelHeight
        
        fetchCoverImage(phasset)
        fetchPlayItem(phasset)
        fetchAVUrlAsset(phasset)
        
    }
    
    func fetchCoverImage(_ phasset:PHAsset) {
        let option = PHImageRequestOptions()
        option.isNetworkAccessAllowed = true //允许下载iCloud的图片
        option.resizeMode = .fast
        option.deliveryMode = .highQualityFormat
        PHImageManager.default().requestImage(for: phasset,
                                       targetSize: CGSize(width: width, height: height),
                                      contentMode: .aspectFit,
                                          options: option)
        { [weak self] (image, nil) in
             // 设置首帧/封面
            if image != nil {
                self?.fetchCoverImg?(image!)
                self?.coverImg = image
            }
        }
    }

    func fetchPlayItem(_ phasset:PHAsset) {
        let options = PHVideoRequestOptions()
        options.isNetworkAccessAllowed = true
        options.deliveryMode = .automatic
        if phasset.mediaType == .image {
            
        }else if phasset.mediaType == .video{
            PHImageManager.default().requestPlayerItem(forVideo:phasset, options: options, resultHandler: { [weak self] playerItem, info in
                
                guard let item = playerItem else {
                    cShowHUB(superView: nil, msg: "视频获取失败:\(self?.index ?? 0)")
                    return
                }
                self?.fetchPlayItem?(item)
            })
        }
    }
    
    func fetchAVUrlAsset(_ phasset:PHAsset){
        let options = PHVideoRequestOptions()
        options.isNetworkAccessAllowed = true
        options.deliveryMode = .automatic

        PHCachingImageManager().requestAVAsset(forVideo: phasset, options: options, resultHandler: {[weak self] (asset: AVAsset?, audioMix: AVAudioMix?, info) in
            if let urlAsset = asset as? AVURLAsset {
                self?.materialDuraion = urlAsset.duration.seconds
                self?.localPath = ((urlAsset.url.absoluteString).removingPercentEncoding)?.replacingOccurrences(of: "file://", with: "")
                self?.fetchAVUrlAsset?(urlAsset)
            }
        })
    }
    
    func generationTimeRanges(needSort:Bool = false) {
        
        dealedDurationRanges.removeAll()
        
        var start : Double = 0
        
        var list: [PQVoiceModel]
        list = voiceStickers.sorted { model1, model2 in
            model1.startTime < model2.startTime
        }
        
        for model in list {
            
            if model.startTime > start{
                //
                let range = CMTimeRange(start: CMTime(seconds: start, preferredTimescale: 100), duration: CMTime(seconds: model.startTime - start, preferredTimescale: 100))
                dealedDurationRanges.append(SplitRecordRange(isRecord: false, range: range, index: -1))
            }

            let ind = voiceStickers.firstIndex(of: model)
            let range = CMTimeRange(start: CMTime(seconds: model.startTime, preferredTimescale: 100), end: CMTime(seconds: model.endTime, preferredTimescale: 100))
            dealedDurationRanges.append(SplitRecordRange(isRecord: true, range: range, index: ind ?? -1))
            start = model.endTime
        }
        if start < materialDuraion {
            let range = CMTimeRange(start: CMTime(seconds: start, preferredTimescale: 100), end: CMTime(seconds: materialDuraion, preferredTimescale: 100))
            dealedDurationRanges.append(SplitRecordRange(isRecord: false, range: range, index: -1))
        }
        
    }
}