[解決済み] Swift - 方向の変化を検出する方法

質問

1つの画像ビューに2つの画像を追加したいのですが（横向きの画像と縦向きの画像）、swift言語を使用して方向の変化を検出する方法がわかりません。

私はこの答えを試してみましたが、それは1つの画像を取るだけです。

override func viewWillTransitionToSize(size: CGSize, withTransitionCoordinator coordinator: UIViewControllerTransitionCoordinator) {
    if UIDevice.currentDevice().orientation.isLandscape.boolValue {
        print("Landscape")
    } else {
        print("Portrait")
    }
}

私はiOS開発の初心者です。何かアドバイスがあれば、とても感謝します。

どのように解決するのですか？

let const = "Background" //image name
let const2 = "GreyBackground" // image name
    @IBOutlet weak var imageView: UIImageView!
    override func viewDidLoad() {
        super.viewDidLoad()

        imageView.image = UIImage(named: const)
        // Do any additional setup after loading the view.
    }

    override func viewWillTransition(to size: CGSize, with coordinator: UIViewControllerTransitionCoordinator) {
        super.viewWillTransition(to: size, with: coordinator)
        if UIDevice.current.orientation.isLandscape {
            print("Landscape")
            imageView.image = UIImage(named: const2)
        } else {
            print("Portrait")
            imageView.image = UIImage(named: const)
        }
    }