add SIAlertViewBackgroundStyle support

kyoji2 · kyoji2 · commit caabf571202a · 2013-05-03T11:15:19.000+08:00
diff --git a/SIAlertView/SIAlertView.h b/SIAlertView/SIAlertView.h
@@ -14,6 +14,11 @@ typedef NS_ENUM(NSInteger, SIAlertViewButtonType) {
     SIAlertViewButtonTypeCancel
 };
 
+typedef NS_ENUM(NSInteger, SIAlertViewBackgroundStyle) {
+    SIAlertViewBackgroundStyleGradient = 0,
+    SIAlertViewBackgroundStyleSolid,
+};
+
 typedef NS_ENUM(NSInteger, SIAlertViewTransitionStyle) {
     SIAlertViewTransitionStyleSlideFromBottom = 0,
     SIAlertViewTransitionStyleSlideFromTop,
@@ -31,6 +36,7 @@ typedef void(^SIAlertViewHandler)(SIAlertView *alertView);
 @property (nonatomic, copy) NSString *message;
 
 @property (nonatomic, assign) SIAlertViewTransitionStyle transitionStyle; // default is SIAlertViewTransitionStyleSlideFromBottom
+@property (nonatomic, assign) SIAlertViewBackgroundStyle backgroundStyle; // default is SIAlertViewButtonTypeGradient
 
 @property (nonatomic, copy) SIAlertViewHandler willShowHandler;
 @property (nonatomic, copy) SIAlertViewHandler didShowHandler;
diff --git a/SIAlertView/SIAlertView.m b/SIAlertView/SIAlertView.m
@@ -63,21 +63,17 @@ - (void)hideAnimated:(BOOL)animated;
 
 @interface SIAlertBackgroundWindow ()
 
-@property (nonatomic, strong) UIView *backgroundView;
+@property (nonatomic, assign) SIAlertViewBackgroundStyle style;
 
 @end
 
 @implementation SIAlertBackgroundWindow
 
-- (id)initWithFrame:(CGRect)frame
+- (id)initWithFrame:(CGRect)frame andStyle:(SIAlertViewBackgroundStyle)style
 {
     self = [super initWithFrame:frame];
     if (self) {
-        self.backgroundView = [[UIView alloc] initWithFrame:CGRectMake(0, 0, frame.size.width, frame.size.height)];
-        self.backgroundView.backgroundColor = [UIColor colorWithWhite:0 alpha:0.5];
-        self.backgroundView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
-        [self addSubview:self.backgroundView];
-        
+        self.style = style;
         self.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
         self.opaque = NO;
         self.windowLevel = UIWindowLevelAlert;
@@ -89,7 +85,7 @@ - (void)show
 {
     [UIView animateWithDuration:0.3
                      animations:^{
-                         self.backgroundView.alpha = 1;
+                         self.alpha = 1;
                      }];
 }
 
@@ -101,13 +97,41 @@ - (void)hideAnimated:(BOOL)animated
     }
     [UIView animateWithDuration:0.3
                      animations:^{
-                         self.backgroundView.alpha = 0;
+                         self.alpha = 0;
                      }
                      completion:^(BOOL finished) {
                          [SIAlertView removeSharedBackground];
                      }];
 }
 
+- (void)drawRect:(CGRect)rect
+{
+    CGContextRef context = UIGraphicsGetCurrentContext();
+    switch (self.style) {
+        case SIAlertViewBackgroundStyleGradient:
+        {
+            size_t locationsCount = 2;
+            CGFloat locations[2] = {0.0f, 1.0f};
+            CGFloat colors[8] = {0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.75f};
+            CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
+            CGGradientRef gradient = CGGradientCreateWithColorComponents(colorSpace, colors, locations, locationsCount);
+            CGColorSpaceRelease(colorSpace);
+            
+            CGPoint center = CGPointMake(self.bounds.size.width / 2, self.bounds.size.height / 2);
+            CGFloat radius = MIN(self.bounds.size.width, self.bounds.size.height) ;
+            CGContextDrawRadialGradient (context, gradient, center, 0, center, radius, kCGGradientDrawsAfterEndLocation);
+            CGGradientRelease(gradient);
+            break;
+        }
+        case SIAlertViewBackgroundStyleSolid:
+        {
+            [[UIColor colorWithWhite:0 alpha:0.5] set];
+            CGContextFillRect(context, self.bounds);
+            break;
+        }
+    }
+}
+
 @end
 
 #pragma mark - SIAlertItem
@@ -643,7 +667,7 @@ - (id)initWithTitle:(NSString *)title andMessage:(NSString *)message
 + (SIAlertBackgroundWindow *)sharedBackground
 {
     if (!__si_alert_background_window) {
-        __si_alert_background_window = [[SIAlertBackgroundWindow alloc] initWithFrame:[UIScreen mainScreen].bounds];
+        __si_alert_background_window = [[SIAlertBackgroundWindow alloc] initWithFrame:[UIScreen mainScreen].bounds andStyle:[SIAlertView currentAlertView].backgroundStyle];
         [__si_alert_background_window makeKeyAndVisible];
     }
     return __si_alert_background_window;
@@ -732,6 +756,11 @@ - (void)show
         self.willShowHandler(self);
     }
     
+    self.visible = YES;
+    
+    [SIAlertView setAnimating:YES];
+    [SIAlertView setCurrentAlertView:self];
+    
     // transition for background if needed
     if ([SIAlertView sharedQueue].count == 1) {
         [[SIAlertView sharedBackground] show];
@@ -751,11 +780,6 @@ - (void)show
     }
     [self.alertWindow makeKeyAndVisible];
     
-    self.visible = YES;
-    
-    [SIAlertView setAnimating:YES];
-    [SIAlertView setCurrentAlertView:self];
-    
     [self.viewController transitionInCompletion:^{
         if (self.didShowHandler) {
             self.didShowHandler(self);
diff --git a/SIAlertViewExample/SIAlertViewExample/ViewController.m b/SIAlertViewExample/SIAlertViewExample/ViewController.m
@@ -130,6 +130,7 @@ - (IBAction)alert3:(id)sender
                               NSLog(@"OK Clicked");
                           }];
     alertView.transitionStyle = SIAlertViewTransitionStyleDropDown;
+    alertView.backgroundStyle = SIAlertViewBackgroundStyleSolid;
     
     alertView.willShowHandler = ^(SIAlertView *alertView) {
         NSLog(@"%@, willShowHandler3", alertView);
