Fixes #18

DemoExpandingCollection/DemoExpandingCollection/ViewControllers/DemoViewController/DemoViewController.swift

@@ -22,6 +22,7 @@ class DemoViewController: ExpandingViewController {
 extension DemoViewController {
   
   override func viewDidLoad() {
+    itemSize = CGSize(width: 256, height: 335)
     super.viewDidLoad()
     
     registerCell()
@@ -104,7 +105,8 @@ extension DemoViewController {
 
 extension DemoViewController {
   
-  func collectionView(collectionView: UICollectionView, willDisplayCell cell: UICollectionViewCell, forItemAtIndexPath indexPath: NSIndexPath) {
+  override func collectionView(collectionView: UICollectionView, willDisplayCell cell: UICollectionViewCell, forItemAtIndexPath indexPath: NSIndexPath) {
+    super.collectionView(collectionView, willDisplayCell: cell, forItemAtIndexPath: indexPath)
     guard let cell = cell as? DemoCollectionViewCell else { return }
 
     let index = indexPath.row % items.count

Source/ExpandingViewController/CollectionView/Cells/BasePageCollectionCell.swift

@@ -188,7 +188,19 @@ extension BasePageCollectionCell {
     frame.origin.x -= yOffset / 2
     frame.size.width += yOffset
   }
-
+  
+  
+  func configureCellViewConstraintsWithSize(size: CGSize) {
+    guard isOpened == false && frontContainerView.getConstraint(.Width)?.constant != size.width else { return }
+    
+    [frontContainerView, backContainerView].forEach {
+      let constraintWidth = $0.getConstraint(.Width)
+      constraintWidth?.constant = size.width
+      
+      let constraintHeight = $0.getConstraint(.Height)
+      constraintHeight?.constant = size.height
+    }
+  }
 }
 
 // MARK: NSCoding

Source/ExpandingViewController/ExpandingViewController.swift

@@ -107,6 +107,13 @@ extension ExpandingViewController {
 
 extension ExpandingViewController: UICollectionViewDataSource, UICollectionViewDelegate {
   
+  public func collectionView(collectionView: UICollectionView, willDisplayCell cell: UICollectionViewCell, forItemAtIndexPath indexPath: NSIndexPath) {
+    guard case let cell as BasePageCollectionCell = cell else {
+      return
+    }
+    
+    cell.configureCellViewConstraintsWithSize(itemSize)
+  }
   public func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
     fatalError("need emplementation in subclass")
   }

expanding-collection.podspec

@@ -1,6 +1,6 @@
 Pod::Spec.new do |s|
   s.name         = 'expanding-collection'
-  s.version      = '0.2.0'
+  s.version      = '0.2.1'
   s.summary      = 'Transition animtion from CollectionView to TableView'
   s.license      = 'MIT'
   s.homepage     = 'https://github.com/Ramotion/expanding-collection'