Gawker/ScreenImageSource.m at master · wnka/Gawker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
//
//  ScreenImageSource.m
//  Gawker
//
//  Created by phil piwonka on 3/7/06.
//  Copyright 2006 __MyCompanyName__. All rights reserved.
//

#import "ScreenImageSource.h"
#import "ImageText.h"

static inline void swapcopy32(void * src, void * dst, int bytecount);
static void swizzleBitmap(NSBitmapImageRep * bitmap);

@interface ScreenImageSource (PrivateMethods)
- (BOOL)setupGLForScreenNumber:(int)screenNum;
- (NSImage *)captureFrame;
- (void)setRecentImage:(NSImage *)newImage;
@end

@implementation ScreenImageSource
- (id)init
{
    if (self = [super init]) {
        int initialScreen = 0;
        isEnabled = NO;
        timeZone = [[NSTimeZone localTimeZone] retain];
        recentImage = nil;

        [self setSourceDescription:NSFullUserName()];
        [self setSourceSubDescription:@"My Desktop"];

        if (![self setupGLForScreenNumber:initialScreen]) {
            NSLog(@"Error setting up screen grabbing");
            [self autorelease];
            return nil;
        }
    }

    return self;
}

- (void)dealloc
{
    [self stop];

    [timeZone release];

    [recentImage release];
    [recentTime release];

    [sourceDescription release];
    [sourceSubDescription release];

    [super dealloc];
}

- (BOOL)start
{
    if (isEnabled) {
        // already enabled
        return YES;
    }

    isEnabled = YES;
    [self setRecentImage:[self captureFrame]];
    NSNotificationCenter *nc = [NSNotificationCenter defaultCenter];
    [nc postNotificationName:@"SourceConnect"
        object:self
        userInfo:nil];
    return YES;
}

- (BOOL)stop;
{
    isEnabled = NO;
    // stop a timer here
    return YES;
}

- (BOOL)isEnabled
{
    return isEnabled;
}

- (BOOL)setEnabled:(BOOL)state
{
    BOOL success;
    if (state) {
        success = [self start];
    }
    else {
        success = [self stop];
    }
    return success;
}

- (NSString *)sourceDescription
{
    return sourceDescription;
}

- (void)setSourceDescription:(NSString *)newDesc
{
    [newDesc retain];
    [sourceDescription release];
    sourceDescription = newDesc;
}

- (NSString *)sourceSubDescription
{
    return sourceSubDescription;
}

- (void)setSourceSubDescription:(NSString *)newDesc
{
    [newDesc retain];
    [sourceSubDescription release];
    sourceSubDescription = newDesc;
}

- (NSImage *)recentImage
{
    NSImage *image = nil;
    if (isEnabled) {
        [self setRecentImage:[self captureFrame]];
        image = recentImage;
    }
    return image;
}

- (NSString *)recentTime
{
    return recentTime;
}

- (void)setScreenToGrab:(int)screenNum
{
    if (screenNum != [[NSScreen screens] indexOfObject:screen]) {
        NSLog(@"Need to update which screen to grab from.");
        [self setupGLForScreenNumber:screenNum];
    }
}

@end

@implementation ScreenImageSource (PrivateMethods)

- (BOOL)setupGLForScreenNumber:(int)screenNum
{
    if (screen) {
        [screen release];
    }
    screen = [[[NSScreen screens] objectAtIndex:screenNum] retain];

    return YES;
}

- (NSImage *)captureFrame
{
    NSRect scaledRect = NSMakeRect(0, 0, 640, 480);

    NSDictionary* screenDictionary = [screen deviceDescription];
    NSNumber* screenID = [screenDictionary objectForKey:@"NSScreenNumber"];
    CGDirectDisplayID aID = [screenID unsignedIntValue];
    // make a snapshot of the current display

    CGImageRef image = CGDisplayCreateImage(aID);
    NSImage* newImage = [[[NSImage alloc] initWithCGImage:image size:scaledRect.size] autorelease];
    CGImageRelease(image);

    return newImage;
}

- (void)setRecentImage:(NSImage *)newImage
{
    [newImage retain];
    [recentImage release];
    recentImage = newImage;
    [recentTime release];
    recentTime = [[ImageText timeString:timeZone] retain];
}

@end