//
//  NSImage+ScreenShot.m
//  YunPan for Mac
//
//  Created by 61 on 13-11-25.
//
//

#import "NSImage+ScreenShot.h"

@implementation NSImage (ScreenShot)
+ (NSImage *)mainScreenShot
{
    NSRect screenRect = [[NSScreen mainScreen] frame];
    CGImageRef cgImage = CGWindowListCreateImage(screenRect, kCGWindowListOptionOnScreenOnly, kCGNullWindowID, kCGWindowImageDefault);
    NSBitmapImageRep *rep = [[NSBitmapImageRep alloc] initWithCGImage:cgImage];
    CGImageRelease(cgImage);
    NSImage *image = [[NSImage alloc] init];
    [image addRepresentation:rep];
    return image;
}
@end