Fixed #1: Header only resized after did appear

roadfire · Aug 18, 2015 · 098883f · 098883f
1 parent 2388405
commit 098883f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/AutoLayoutTableHeaderView/CustomHeaderView.swift b/AutoLayoutTableHeaderView/CustomHeaderView.swift
@@ -12,8 +12,7 @@ class CustomHeaderView: UIView {
 
     @IBOutlet var label: UILabel!
 
-    override func layoutSubviews()
-    {
+    override func layoutSubviews() {
         super.layoutSubviews()
         label.preferredMaxLayoutWidth = label.bounds.width
     }

diff --git a/AutoLayoutTableHeaderView/ViewController.swift b/AutoLayoutTableHeaderView/ViewController.swift
@@ -12,14 +12,12 @@ class ViewController: UIViewController {
 
     @IBOutlet var tableView: UITableView!
 
-    override func viewDidAppear(animated: Bool)
-    {
-        super.viewDidAppear(animated)
+    override func viewDidLayoutSubviews() {
+        super.viewDidLayoutSubviews()
         sizeHeaderToFit()
     }
 
-    func sizeHeaderToFit()
-    {
+    func sizeHeaderToFit() {
         let headerView = tableView.tableHeaderView!
 
         headerView.setNeedsLayout()