refactor

Author: muyexi

Source/OriginalRow.swift

@@ -11,9 +11,17 @@ class OriginalRow {
         self.originalIndexPath = originalIndexPath
     }
 
-    var hiddenReal: Bool = false
+    var hidden: Bool = false
 
-    var hiddenPlanned: Bool = false
+    var hiding: Bool = false {
+        willSet {
+            if !hidden && newValue {
+                batchOperation = .delete
+            } else if hidden && !newValue {
+                batchOperation = .insert
+            }
+        }
+    }
 
     var batchOperation: BatchOperation = .none
 
@@ -24,19 +32,9 @@ class OriginalRow {
     var height: CGFloat = CGFloat.greatestFiniteMagnitude
 
     func update() {
-        if !hiddenPlanned && batchOperation == .none {
+        if !hiding && batchOperation == .none {
             batchOperation = .update
         }
     }
 
-    func set(hidden: Bool) {
-        if !hiddenReal && hidden {
-            batchOperation = .delete
-        } else if hiddenReal && !hidden {
-            batchOperation = .insert
-        }
-        
-        hiddenPlanned = hidden
-    }
-    
 }

Source/OriginalSection.swift

@@ -5,7 +5,7 @@ class OriginalSection {
     var rows: [OriginalRow] = []
 
     func numberOfVissibleRows() -> Int {
-        return rows.filter { !$0.hiddenPlanned }.count
+        return rows.filter { !$0.hiding }.count
     }
 
 }

Source/OriginalTable.swift

@@ -60,7 +60,7 @@ class OriginalTable {
 
     func vissibleOriginalRowWithIndexPath(_ indexPath: IndexPath) -> OriginalRow {
         let section = sections[indexPath.section]
-        let visibleRows = section.rows.filter { !$0.hiddenPlanned }
+        let visibleRows = section.rows.filter { !$0.hiding }
 
         return visibleRows[indexPath.row]
     }
@@ -76,7 +76,7 @@ class OriginalTable {
         var indexRow = originalRow.originalIndexPath.row
 
         let section = sections[indexSection]
-        indexRow = section.rows[0..<indexRow].filter { !$0.hiddenPlanned }.count
+        indexRow = section.rows[0..<indexRow].filter { !$0.hiding }.count
 
         return IndexPath(row: indexRow, section: indexSection)
     }
@@ -86,7 +86,7 @@ class OriginalTable {
         var indexRow = originalRow.originalIndexPath.row
 
         let section = sections[indexSection]
-        indexRow = section.rows[0..<indexRow].filter { !$0.hiddenReal }.count
+        indexRow = section.rows[0..<indexRow].filter { !$0.hidden }.count
 
         return IndexPath(row: indexRow, section: indexSection)
     }
@@ -112,7 +112,7 @@ class OriginalTable {
         }
 
         allRows.forEach { (row) in
-            row.hiddenReal = row.hiddenPlanned
+            row.hidden = row.hiding
             row.batchOperation = BatchOperation.none
         }
     }

Source/StaticTableViewController.swift

@@ -26,7 +26,7 @@ open class StaticTableViewController: UITableViewController, OriginalTableConfig
     open func set(cells: UITableViewCell..., hidden: Bool) {
         cells.forEach { cell in
             let row = originalTable!.originalRowWithTableViewCell(cell)
-            row.set(hidden: hidden)
+            row.hiding = hidden
         }
     }
 
@@ -38,7 +38,7 @@ open class StaticTableViewController: UITableViewController, OriginalTableConfig
     }
 
     open func isHidden(cell: UITableViewCell) -> Bool {
-        return originalTable!.originalRowWithTableViewCell(cell).hiddenReal
+        return originalTable!.originalRowWithTableViewCell(cell).hidden
     }
 
     open func reloadData(animated: Bool) {