Refactors Data Cell and Data CellViewModel to incorporate DataCellTyp…

…eValue
pavankataria · Mar 13, 2017 · 724a6f0 · 724a6f0
1 parent b454c08
commit 724a6f0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/SwiftDataTables/Classes/DataCell/DataCell.swift b/SwiftDataTables/Classes/DataCell/DataCell.swift
@@ -15,11 +15,6 @@ class DataCell: UICollectionViewCell {
 
     //MARK: - Lifecycle
     func setup(_ viewModel: DataCellViewModel){
-        self.dataLabel.text = viewModel.data
-
-//        self.highlight(viewModel.highlighted)
+        self.dataLabel.text = viewModel.data.stringRepresentation
     }
-//    var representedElementKind = {
-//        
-//    }
 }
diff --git a/SwiftDataTables/Classes/DataCell/DataCellViewModel.swift b/SwiftDataTables/Classes/DataCell/DataCellViewModel.swift
@@ -14,10 +14,11 @@ open class DataCellViewModel: VirtualPositionTrackable, CollectionViewCellRepres
     var xPositionRunningTotal: CGFloat?  = nil
     var yPositionRunningTotal: CGFloat?  = nil
     var virtualHeight: CGFloat = 0
-    let data: String
+    let data: DataTableValueType
+
     var highlighted: Bool = false
     //MARK: - Lifecycle
-    init(data: String){
+    init(data: DataTableValueType){
         self.data = data
     }
     static func registerCell(collectionView: UICollectionView) {