Add ability to set custom offset in to the stopEachSection scroll mode

Leta0n · Mar 5, 2018 · 607a4a2 · 607a4a2
1 parent e5f7945
commit 607a4a2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/Example/JTAppleCalendar iOS Example/Example Calendars/ViewController.swift b/Example/JTAppleCalendar iOS Example/Example Calendars/ViewController.swift
@@ -44,7 +44,7 @@ class ViewController: UIViewController {
     }
 
     @IBAction func scrollFixed(_ sender: Any) {
-        calendarView.scrollingMode = .stopAtEachSection
+        calendarView.scrollingMode = .stopAtEachSection(customOffset: 0)
     }
     @IBAction func showPrepost(_ sender: UIButton) {
         prePostVisibility = {state, cell in

diff --git a/Sources/CalendarEnums.swift b/Sources/CalendarEnums.swift
@@ -62,8 +62,8 @@ public enum ReadingOrientation {
 public enum ScrollingMode: Equatable {
     /// stopAtEachCalendarFrame - non-continuous scrolling that will stop at each frame
     case stopAtEachCalendarFrame
-    /// stopAtEachSection - non-continuous scrolling that will stop at each section
-    case stopAtEachSection
+    /// stopAtEachSection - non-continuous scrolling that will stop at each section with custom offset
+    case stopAtEachSection(customOffset: CGFloat)
     /// stopAtEach - non-continuous scrolling that will stop at each custom interval
     case stopAtEach(customInterval: CGFloat)
     /// nonStopToSection - continuous scrolling that will stop at a section

diff --git a/Sources/UIScrollViewDelegates.swift b/Sources/UIScrollViewDelegates.swift
@@ -125,7 +125,7 @@ extension JTAppleCalendarView: UIScrollViewDelegate {
                 setTargetContentOffset(scrollDirection == .horizontal ? targetContentOffset.pointee.x : targetContentOffset.pointee.y)
             #endif
             break
-        case .stopAtEachSection:
+        case .stopAtEachSection(let offset):
             var calculatedOffSet: CGFloat = 0
             if scrollDirection == .horizontal ||
                 (scrollDirection == .vertical && !calendarViewLayout.thereAreHeaders && cachedConfiguration.generateOutDates == .tillEndOfGrid) {
@@ -148,7 +148,7 @@ extension JTAppleCalendarView: UIScrollViewDelegate {
                     }
                 }
             }
-            setTargetContentOffset(calculatedOffSet)
+            setTargetContentOffset(calculatedOffSet + offset)
         case .nonStopToSection, .nonStopToCell, .nonStopTo:
             let diff = abs(theTargetContentOffset - contentOffset)
             let targetSection = calendarLayout.sectionFromOffset(theTargetContentOffset)