こんにちは、Ikoba(<a href="http://twitter.com/ikoba0911">@ikoba0911</a>)です。 
今回はUIButtonに設定したimageのサイズを後から変更する方法について、備忘録的に書いていきたいと思います。 
そもそも、imageの画像を最初からデザイン準拠で作成する方が良いと思うのでなかなか使わない知識かもしれません。
今回はNGパターンに関しても記載していきます。
<h2 id="目次">目次</h2>
<ul>
<li><a href="#ng%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B31--viewcontroller%E5%81%B4%E3%81%AE%E3%83%A9%E3%82%A4%E3%83%95%E3%82%B5%E3%82%A4%E3%82%AF%E3%83%AB%E3%81%A7%E5%91%BC%E3%81%B6">NGパターン1 : ViewController側のライフサイクルで呼ぶ</a></li>
<li><a href="#ng%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B32--%E3%82%AB%E3%82%B9%E3%82%BF%E3%83%A0%E3%83%93%E3%83%A5%E3%83%BC%E3%81%AB%E3%81%97%E3%81%A6awakefromnib%E3%81%A7%E5%91%BC%E3%81%B6">NGパターン2 : カスタムビューにしてawakeFromNibで呼ぶ</a></li>
<li><a href="#ng%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B33--%E3%82%AB%E3%82%B9%E3%82%BF%E3%83%A0%E3%83%93%E3%83%A5%E3%83%BC%E3%81%A7%E5%AE%9A%E7%BE%A9%E3%81%97%E3%81%9F%E3%82%B5%E3%82%A4%E3%82%BA%E5%A4%89%E6%9B%B4%E3%81%AE%E9%96%A2%E6%95%B0%E3%82%92viewcontroller%E3%81%A7%E5%91%BC%E3%81%B3%E3%81%9D%E3%81%AE%E5%BE%8C%E3%82%AB%E3%82%B9%E3%82%BF%E3%83%A0%E3%83%93%E3%83%A5%E3%83%BC%E3%81%AE%E5%86%8D%E6%8F%8F%E7%94%BB%E3%81%AE%E5%87%A6%E7%90%86%E3%82%92%E5%91%BC%E3%81%B3%E5%87%BA%E3%81%99">NGパターン3 : カスタムビューで定義したサイズ変更の関数をViewControllerで呼び、その後カスタムビューの再描画の処理を呼び出す</a></li>
<li><a href="#%E6%88%90%E5%8A%9F%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B3--%E3%82%AB%E3%82%B9%E3%82%BF%E3%83%A0%E3%83%93%E3%83%A5%E3%83%BC%E3%81%A7%E5%AE%9A%E7%BE%A9%E3%81%97%E3%81%9F%E3%82%B5%E3%82%A4%E3%82%BA%E5%A4%89%E6%9B%B4%E3%81%AE%E9%96%A2%E6%95%B0%E3%82%92%E3%82%AB%E3%82%B9%E3%82%BF%E3%83%A0%E3%83%93%E3%83%A5%E3%83%BC%E5%86%85%E3%81%AElayoutsubviews%E3%81%A7%E5%91%BC%E3%81%B3%E5%87%BA%E3%81%99">成功パターン : カスタムビューで定義したサイズ変更の関数を、カスタムビュー内のlayoutSubviews()で呼び出す</a></li>
<li><a href="#%E3%81%BE%E3%81%A8%E3%82%81">まとめ</a></li>
</ul>
<h2 id="ngパターン1--viewcontroller側のライフサイクルで呼ぶ">NGパターン1 : ViewController側のライフサイクルで呼ぶ</h2>
一番最初に試すとしたら下記のような形でしょう。 
この形で書いても残念ながら変更されません。 
参照の仕方は合っているのですが、ViewControllerではUIButtonのsubViewsに存在するimageViewを変更するタイミングを検知することができません。 
同様にviewWillAppear()やviewDidLayoutSubviews()などで呼び出しても変更されません。
<div class="remark-highlight"><pre class="language-swift"><code class="language-swift">import UIKit

class ViewController: UIViewController {

@IBOutlet weak var testButton: UIButton!

 override func viewDidLoad() {
 super.viewDidLoad()
 setUpImage()
 }

 func setUpImage(imageSize: CGSize = CGSize(width: 10, height: 10)) {

 if let buttonImage = testButton.imageView {
 buttonImage.frame.size = imageSize
 }
 }
}
</code></pre></div>
<h2 id="ngパターン2--カスタムビューにしてawakefromnibで呼ぶ">NGパターン2 : カスタムビューにしてawakeFromNibで呼ぶ</h2>
次に考えたのはカスタムビューでawakeFromNib()内でサイズを変更しようとしました。 
これもまた失敗で、画像のサイズは元のサイズに戻ってしまいます。
UIButtonのカスタムクラス
<div class="remark-highlight"><pre class="language-swift"><code class="language-swift">import UIKit

internal final class TestButton: UIButton {

 override func awakeFromNib() {
 super.awakeFromNib()
 setUpImage()
 }

 func setUpImage(imageSize: CGSize = CGSize(width: 10, height: 10)) {
 if let image = imageView {
 image.frame.size = imageSize
 }
 }
}
</code></pre></div>
<h2 id="ngパターン3--カスタムビューで定義したサイズ変更の関数をviewcontrollerで呼びその後カスタムビューの再描画の処理を呼び出す">NGパターン3 : カスタムビューで定義したサイズ変更の関数をViewControllerで呼び、その後カスタムビューの再描画の処理を呼び出す</h2>
それならばと思い、カスタムビューからサイズ変更用の関数を呼び出して直後に再描画関係の関数を呼べば良いかという検討をしました。 
再描画関係の関数とは「setNeedsDisplay」、「setNeedsLayout」、「layoutIfNeeded」、「layoutSubviews」を指します。
<div class="remark-highlight"><pre class="language-swift"><code class="language-swift">import UIKit

class ViewController: UIViewController {

@IBOutlet weak var testButton: TestButton!

 override func viewDidLoad() {
 super.viewDidLoad() 
 testButton.setUpImage() // NGパターン2で定義したイメージの変更用の関数
 testButton.setNeedsDisplay()
 testButton.setNeedsLayout()
 testButton.layoutIfNeeded()
 testButton.layoutSubviews()
 }
}
</code></pre></div>
<h2 id="成功パターン--カスタムビューで定義したサイズ変更の関数をカスタムビュー内のlayoutsubviewsで呼び出す">成功パターン : カスタムビューで定義したサイズ変更の関数を、カスタムビュー内のlayoutSubviews()で呼び出す</h2>
そして最終的に行き着いたのが、調べたら良く出てくるカスタムビュー内のlayoutSubviews()内でサイズ変更を呼び出す形です。 
layoutSubviews()は複数回呼び出されるのでその点を留意する必要がありますが、この形が成功するパターンでした。
<div class="remark-highlight"><pre class="language-swift"><code class="language-swift">import UIKit

internal final class TestButton: UIButton {
 override func layoutSubviews() {
 super.layoutSubviews()
 setUpImage()
 }

 func setUpImage(imageSize: CGSize = CGSize(width: 10, height: 10)) {
 if let image = imageView {
 image.frame.size = imageSize
 }
 }
}
</code></pre></div>
<h2 id="まとめ">まとめ</h2>
割といろいろ検討したのですが、なかなかうまくいかないものだなと改めて思いました。 
今回は成功パターンよりもNGパターンの方が参考になるかもしれません。
備忘録できな感じですがどなたかの参考になったら幸いです！

Storyboard上でのUIButtonのimageのサイズ変更について記載しています。

【Swift】Storyboard上で設定したUIButtonのimageのサイズを変更する方法【UIKit】