1v1/MSYOUPAI/NewCode/Module/DynamicDetail/ViewModel/YMDynamicDetailInfoAlbumCellViewModel.m

//
//  YMDynamicDetailInfoAlbumCellViewModel.m
//  MSYOUPAI
//
//  Created by YoMi on 2024/3/21.
//  Copyright © 2024 MS. All rights reserved.
//

#import "YMDynamicDetailInfoAlbumCellViewModel.h"
#import "YMDynamicListModel.h"
@interface YMDynamicDetailInfoAlbumCellViewModel ()
/// 相册Url
@property (nonatomic, copy, readwrite) NSString *albumType;
/// 相册Url
@property (nonatomic, copy, readwrite) NSString *albumUrl;
/// 相册尺寸
@property (nonatomic, assign, readwrite) CGSize albumSize;
@end

@implementation YMDynamicDetailInfoAlbumCellViewModel
- (void)ym_initialize{
    [super ym_initialize];
    if ([self.params[ParamsModel] isKindOfClass:[YMDynamicImageModel class]]) {
        YMDynamicImageModel *model = self.params[ParamsModel];
        self.albumType = model.exts;
        self.albumUrl = model.url;
        
        if ([model.exts  isEqualToString:@"mp4"]) {
            AVURLAsset *asset = [AVURLAsset assetWithURL:[LCTools getImageUrlWithAddress:model.url]];
            NSArray *array = asset.tracks;
            CGSize videoSize = CGSizeZero;
            for (AVAssetTrack *track in array) {
                if ([track.mediaType isEqualToString:AVMediaTypeVideo]){
                    videoSize = track.naturalSize;
                }
            }
            self.albumSize = CGSizeMake(videoSize.width*0.8, videoSize.height*0.8);
        }else{
            self.albumSize = CGSizeMake([model.width floatValue]*0.8, [model.height floatValue]*0.8);
        }
        
    }
    
}

@end