IPad screenshot

IPad screenshot

//Create an image about the screen of the iOS device
+ (UIImage *) screenshot{

    UIWindow *mainWindow = [[[UIApplication sharedApplication] windows] firstObject];
    
    if(UIGraphicsBeginImageContextWithOptions != NULL)
    {
        UIGraphicsBeginImageContextWithOptions(mainWindow.frame.size, NO, 0.0f);
    } else {
        UIGraphicsBeginImageContext(mainWindow.frame.size);
    }
    //Get all the views with the same size!
    for (UIWindow *window in [[UIApplication sharedApplication] windows] ) {
        if (CGSizeEqualToSize(window.frame.size, mainWindow.frame.size)) {
            [window drawViewHierarchyInRect:mainWindow.frame afterScreenUpdates:YES];
        }
    }
    
    UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
    UIGraphicsEndImageContext();
    return image;
}