如何在iPad屏幕旋转后获得新的窗口大小？

import UIKit

class ViewController: UIViewController {
    override func willTransition(to newCollection: UITraitCollection, with coordinator: UIViewControllerTransitionCoordinator) {
        super.willTransition(to: newCollection, with: coordinator)
        
        coordinator.animate(alongsideTransition: { (context) in
            // 在屏幕旋转动画过程中执行的代码
            
            // 获取新的窗口大小
            let screenSize = UIScreen.main.bounds.size
            let screenWidth = screenSize.width
            let screenHeight = screenSize.height
            
            // 根据新的窗口大小进行布局调整
            // ...
            
        }) { (context) in
            // 在屏幕旋转动画完成后执行的代码
        }
    }
}