With this return value, developers can add any decoration view for JT…

…CalendarMenuView
jonathantribouharet · Oct 19, 2016 · a143d1b · a143d1b
1 parent 173452d
commit a143d1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/JTCalendar/Views/JTCalendarMenuView.h b/JTCalendar/Views/JTCalendarMenuView.h
@@ -22,5 +22,5 @@
  */
 - (void)commonInit;
 
-- (UIView*)centerView;
+- (CGSize)centerTextSize;
 @end
diff --git a/JTCalendar/Views/JTCalendarMenuView.m b/JTCalendar/Views/JTCalendarMenuView.m
@@ -73,9 +73,14 @@ - (void)commonInit
     }
 }
 
-- (UIView*)centerView
+- (CGSize)centerTextSize
 {
-	return _centerView;
+	if (_centerView && [_centerView isKindOfClass:[UILabel class]])
+	{
+       UILabel* label = (UILabel*)_centerView;
+	   return [label.text sizeWithAttributes:@{NSFontAttributeName:[label font]}];
+	}
+	return CGSizeZero;
 }
 
 - (void)layoutSubviews