Added delegate method for the "did not select" completion case so the…

… delegate can remove the calendar view properly.
erburrell · Sep 9, 2012 · b15b822 · b15b822
1 parent 30ac4db
commit b15b822
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/Demo/OCCalendar/OCViewController.m b/Demo/OCCalendar/OCViewController.m
@@ -100,6 +100,13 @@ - (void)completedWithStartDate:(NSDate *)startDate endDate:(NSDate *)endDate {
     calVC = nil;
 }
 
+-(void) completedWithNoSelection{
+    [calVC.view removeFromSuperview];
+    calVC.delegate = nil;
+    [calVC release];
+    calVC = nil;
+}
+
 
 #pragma mark - 
 #pragma mark Prettifying Methods...

diff --git a/OCCalendarView/OCCalendarViewController.h b/OCCalendarView/OCCalendarViewController.h
@@ -15,6 +15,8 @@
 
 -(void)completedWithStartDate:(NSDate *)startDate endDate:(NSDate *)endDate;
 
+-(void)completedWithNoSelection;
+
 @end
 
 @interface OCCalendarViewController : UIViewController <UIGestureRecognizerDelegate> {

diff --git a/OCCalendarView/OCCalendarViewController.m b/OCCalendarView/OCCalendarViewController.m
@@ -108,7 +108,7 @@ - (void)removeCalView {
         else
             [self.delegate completedWithStartDate:endDate endDate:startDate];
     } else {
-        //Do something else if it's not selected.
+        [self.delegate completedWithNoSelection];
     }
 
     [calView removeFromSuperview];
-Original file line number
+Diff line change
@@ Expand Up / @@ -15,6 +15,8 @@ @@
     -(void)completedWithStartDate:(NSDate *)startDate endDate:(NSDate *)endDate;
+    -(void)completedWithNoSelection;
     @end
     @interface OCCalendarViewController : UIViewController <UIGestureRecognizerDelegate> {
@@ Expand Down @@