画面サイズを取得する方法です。
import UIKit

let width = UIScreen.main.bounds.size.width
let height = UIScreen.main.bounds.size.height