如何在swift中根据单元格中的图像调整CollectionView单元格高度？

func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
    // 获取当前单元格的数据
    let imageData = yourDataSource[indexPath.item].imageData
    
    // 根据图像数据计算单元格的高度
    let image = UIImage(data: imageData)
    let aspectRatio = image?.size.width / image?.size.height
    let cellWidth = collectionView.frame.width - sectionInset.left - sectionInset.right
    let cellHeight = cellWidth / aspectRatio
    
    return CGSize(width: cellWidth, height: cellHeight)
}

class YourViewController: UIViewController, UICollectionViewDelegateFlowLayout {
    // ...
    
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 设置CollectionView的delegate
        yourCollectionView.delegate = self
    }
    
    // 实现UICollectionViewDelegateFlowLayout协议中的方法
    // ...
}