add contentHeight method

harry6 · Jul 26, 2014 · 3578139 · 3578139
1 parent 378e9c9
commit 3578139
Show file tree

Hide file tree

Showing 5 changed files with 62 additions and 6 deletions.
diff --git a/Example/RACollectionViewReorderableTripletLayout-Demo/Main.storyboard b/Example/RACollectionViewReorderableTripletLayout-Demo/Main.storyboard
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8" standalone="no"?>
-<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="5056" systemVersion="13D65" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" initialViewController="1Nm-aK-gXn">
+<document type="com.apple.InterfaceBuilder3.CocoaTouch.Storyboard.XIB" version="3.0" toolsVersion="5056" systemVersion="13E28" targetRuntime="iOS.CocoaTouch" propertyAccessControl="none" useAutolayout="YES" initialViewController="1Nm-aK-gXn">
     <dependencies>
         <plugIn identifier="com.apple.InterfaceBuilder.IBCocoaTouchPlugin" version="3733"/>
     </dependencies>

diff --git a/RACollectionViewReorderableTripletLayout.podspec b/RACollectionViewReorderableTripletLayout.podspec
@@ -1,12 +1,12 @@
 Pod::Spec.new do |s|
   s.name         = "RACollectionViewReorderableTripletLayout"
-  s.version      = "0.0.8"
+  s.version      = "0.0.9"
   s.summary      = "The custom collectionView layout that can perform reordering of cells by dragging it."
   s.license      = { :type => 'MIT', :file => 'LICENSE' }
   s.homepage     = "https://github.com/ra1028"
   s.author       = { "ra1028" => "r.fe51028.r@gmail.com" }
   s.platform     = :ios, '7.0'
-  s.source       = { :git => "https://github.com/ra1028/RACollectionViewReorderableTripletLayout.git", :tag => "0.0.8" }
+  s.source       = { :git => "https://github.com/ra1028/RACollectionViewReorderableTripletLayout.git", :tag => "0.0.9" }
   s.requires_arc = true
   s.source_files =  'RACollectionViewReorderableTripletLayout/*.{h,m}'
 end
diff --git a/RACollectionViewReorderableTripletLayout/RACollectionViewReorderableTripletLayout.h b/RACollectionViewReorderableTripletLayout/RACollectionViewReorderableTripletLayout.h
@@ -8,7 +8,7 @@
 
 #import "RACollectionViewTripletLayout.h"
 
-@protocol RACollectionViewReorderableTripletLayoutDataSource <UICollectionViewDataSource>
+@protocol RACollectionViewReorderableTripletLayoutDataSource <RACollectionViewTripletLayoutDatasource>
 
 @optional
 

diff --git a/RACollectionViewReorderableTripletLayout/RACollectionViewTripletLayout.h b/RACollectionViewReorderableTripletLayout/RACollectionViewTripletLayout.h
@@ -22,15 +22,18 @@
 
 @end
 
+@protocol RACollectionViewTripletLayoutDatasource <UICollectionViewDataSource>
 
+@end
 
 @interface RACollectionViewTripletLayout : UICollectionViewFlowLayout
 
 @property (nonatomic, assign) id<RACollectionViewDelegateTripletLayout> delegate;
+@property (nonatomic, assign) id<RACollectionViewTripletLayoutDatasource> datasource;
 @property (nonatomic, assign, readonly) CGSize largeCellSize;
 @property (nonatomic, assign, readonly) CGSize smallCellSize;
 
-//needs override
-- (BOOL)shouldUpdateAttributesArray;
+- (BOOL)shouldUpdateAttributesArray; //needs override
+- (CGFloat)contentHeight;
 
 @end
diff --git a/RACollectionViewReorderableTripletLayout/RACollectionViewTripletLayout.m b/RACollectionViewReorderableTripletLayout/RACollectionViewTripletLayout.m
@@ -55,6 +55,59 @@ - (void)prepareLayout
     }
 }
 
+- (CGFloat)contentHeight
+{
+    CGFloat contentHeight = 0;
+    NSInteger numberOfSections = self.collectionView.numberOfSections;
+    CGSize collectionViewSize = self.collectionView.bounds.size;
+
+    UIEdgeInsets insets = UIEdgeInsetsZero;
+    if ([self.delegate respondsToSelector:@selector(insetsForCollectionView:)]) {
+        insets = [self.delegate insetsForCollectionView:self.collectionView];
+    }
+    CGFloat sectionSpacing = 0;
+    if ([self.delegate respondsToSelector:@selector(sectionSpacingForCollectionView:)]) {
+        sectionSpacing = [self.delegate sectionSpacingForCollectionView:self.collectionView];
+    }
+    CGFloat itemSpacing = 0;
+    if ([self.delegate respondsToSelector:@selector(minimumInteritemSpacingForCollectionView:)]) {
+        itemSpacing = [self.delegate minimumInteritemSpacingForCollectionView:self.collectionView];
+    }
+    CGFloat lineSpacing = 0;
+    if ([self.delegate respondsToSelector:@selector(minimumLineSpacingForCollectionView:)]) {
+       lineSpacing = [self.delegate minimumLineSpacingForCollectionView:self.collectionView];
+    }
+
+    contentHeight += insets.top + insets.bottom + sectionSpacing * (numberOfSections - 1);
+
+    CGFloat lastSmallCellHeight = 0;
+    for (NSInteger i = 0; i < numberOfSections; i++) {
+        NSInteger numberOfLines = ceil((CGFloat)[self.collectionView numberOfItemsInSection:i] / 3.f);
+
+        CGFloat largeCellSideLength = (2.f * (collectionViewSize.width - insets.left - insets.right) - itemSpacing) / 3.f;
+        CGFloat smallCellSideLength = (largeCellSideLength - itemSpacing) / 2.f;
+        CGSize largeCellSize = CGSizeMake(largeCellSideLength, largeCellSideLength);
+        CGSize smallCellSize = CGSizeMake(smallCellSideLength, smallCellSideLength);
+        if ([self.delegate respondsToSelector:@selector(collectionView:sizeForLargeItemsInSection:)]) {
+            if (!CGSizeEqualToSize([self.delegate collectionView:self.collectionView sizeForLargeItemsInSection:i], RACollectionViewTripletLayoutStyleSquare)) {
+                largeCellSize = [self.delegate collectionView:self.collectionView sizeForLargeItemsInSection:i];
+                smallCellSize = CGSizeMake(collectionViewSize.width - largeCellSize.width - itemSpacing - insets.left - insets.right, (largeCellSize.height / 2.f) - (itemSpacing / 2.f));
+            }
+        }
+        lastSmallCellHeight = smallCellSize.height;
+        CGFloat largeCellHeight = largeCellSize.height;
+        CGFloat lineHeight = numberOfLines * (largeCellHeight + lineSpacing) - lineSpacing;
+        contentHeight += lineHeight;
+    }
+
+    NSInteger numberOfItemsInLastSection = [self.collectionView numberOfItemsInSection:numberOfSections -1];
+    if ((numberOfItemsInLastSection - 1) % 3 == 0 && (numberOfItemsInLastSection - 1) % 6 != 0) {
+        contentHeight -= lastSmallCellHeight + itemSpacing;
+    }
+
+    return contentHeight;
+}
+
 - (id<RACollectionViewDelegateTripletLayout>)delegate
 {
     return (id<RACollectionViewDelegateTripletLayout>)self.collectionView.delegate;