截取整个屏幕的内容(截图代码)

下面这段代码功能是截取整个屏幕的图片内容,我已经给大家抽取成方法,大家在使用的时候可以直接将代码拿到项目中便可使用了,代码如下:

/**
  直接截取屏幕内容,返回一个image(这个方法的功能就是截图)
 */
- (UIImage *)screenImage{
    
    UIWindow *window = [UIApplication sharedApplication].keyWindow;
    //开启图形上下文
    UIGraphicsBeginImageContext(window.size);
    //获取到开启的图形上下文
    CGContextRef context = UIGraphicsGetCurrentContext();
    //拿到window的layer,将其渲染到上下文
    [window.layer renderInContext:context];
    //获取图片
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    //结束图形上下文
    UIGraphicsEndImageContext();
    
    return image;
}