UIApplicationUserDidTakeScreenshotNotification检查截屏并获取截屏后的图片

- (void)registerTakeScreenShotNotice

{

    __weak typeof(self) weakSelf = self;

    NSOperationQueue *mainQueue = [NSOperationQueue mainQueue];

    [[NSNotificationCenter defaultCenter] addObserverForName:UIApplicationUserDidTakeScreenshotNotification

                                                      object:nil

                                                       queue:mainQueue

                                                  usingBlock:^(NSNotification *note) {

                                                      // executes after screenshot

                                                      NSLog(@"截屏咯");

                                                      [weakSelf userDidTakeScreenshot];

                                                  }];

}

//截屏响应

- (void)userDidTakeScreenshot

{

    NSLog(@"检测到截屏");

    

    //人为截屏, 模拟用户截屏行为, 获取所截图片

    UIImage *image_ = [self imageWithScreenshot];

}

/**

 *  返回截取到的图片

 *

 *  @return UIImage *

 */

+ (UIImage *)imageWithScreenshot

{

    NSData *imageData = [self dataWithScreenshotInPNGFormat];

    return [UIImage imageWithData:imageData];

}

/**

 *  截取当前屏幕

 *

 *  @return NSData *

 */

- (NSData *)dataWithScreenshotInPNGFormat

{

    CGSize imageSize = CGSizeZero;

    UIInterfaceOrientation orientation = [UIApplication sharedApplication].statusBarOrientation;

    if (UIInterfaceOrientationIsPortrait(orientation))

        imageSize = [UIScreen mainScreen].bounds.size;

    else

        imageSize = CGSizeMake([UIScreen mainScreen].bounds.size.height, [UIScreen mainScreen].bounds.size.width);

    

    UIGraphicsBeginImageContextWithOptions(imageSize, NO, 0);

    CGContextRef context = UIGraphicsGetCurrentContext();

    for (UIWindow *window in [[UIApplication sharedApplication] windows])

    {

        CGContextSaveGState(context);

        CGContextTranslateCTM(context, window.center.x, window.center.y);

        CGContextConcatCTM(context, window.transform);

        CGContextTranslateCTM(context, -window.bounds.size.width * window.layer.anchorPoint.x, -window.bounds.size.height * window.layer.anchorPoint.y);

        if (orientation == UIInterfaceOrientationLandscapeLeft)

        {

            CGContextRotateCTM(context, M_PI_2);

            CGContextTranslateCTM(context, 0, -imageSize.width);

        }

        else if (orientation == UIInterfaceOrientationLandscapeRight)

        {

            CGContextRotateCTM(context, -M_PI_2);

            CGContextTranslateCTM(context, -imageSize.height, 0);

        } else if (orientation == UIInterfaceOrientationPortraitUpsideDown) {

            CGContextRotateCTM(context, M_PI);

            CGContextTranslateCTM(context, -imageSize.width, -imageSize.height);

        }

        if ([window respondsToSelector:@selector(drawViewHierarchyInRect:afterScreenUpdates:)])

        {

            [window drawViewHierarchyInRect:window.bounds afterScreenUpdates:YES];

        }

        else

        {

            [window.layer renderInContext:context];

        }

        CGContextRestoreGState(context);

    }

    

    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();

    UIGraphicsEndImageContext();

    

    return UIImagePNGRepresentation(image);

}