-
Notifications
You must be signed in to change notification settings - Fork 3
/
listing2.html
executable file
·474 lines (395 loc) · 20.6 KB
/
listing2.html
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
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN"
"http://www.w3.org/TR/1999/REC-html401-19991224/loose.dtd">
<html>
<head>
<!-- BEGIN META TAG INFO -->
<meta http-equiv="content-type" content="text/html; charset=UTF-8">
<link rel="home" href="http://developer.apple.com/">
<link rel="find" href="http://developer.apple.com/search/">
<link rel="stylesheet" type="text/css" href="../../documentation/css/adcstyle.css" title="fonts">
<script language="JavaScript" src="../../documentation/js/adc.js" type="text/javascript"></script>
<!-- END META TAG INFO -->
<!-- BEGIN TITLE -->
<title>CIColorTracking - /AppController.m</title>
<!-- END TITLE -->
<script language="JavaScript">
function JumpToNewPage() {
window.location=document.scpopupmenu.gotop.value;
return true;
}
</script>
</head>
<!-- BEGIN BODY OPEN -->
<body>
<!--END BODY OPEN -->
<!-- START CENTER OPEN -->
<center>
<!-- END CENTER OPEN -->
<!-- BEGIN LOGO AND SEARCH -->
<!--#include virtual="/includes/adcnavbar"-->
<!-- END LOGO AND SEARCH -->
<!-- START BREADCRUMB -->
<div id="breadcrumb">
<table width="680" border="0" cellpadding="0" cellspacing="0">
<tr>
<td scope="row"><img width="340" height="10" src="images/1dot.gif" alt=""></td>
<td><img width="340" height="10" src="images/1dot.gif" alt=""></td>
</tr>
<tr valign="middle">
<td align="left" colspan="2">
<a href="http://developer.apple.com/">ADC Home</a> > <a href="../../referencelibrary/index.html">Reference Library</a> > <a href="../../samplecode/index.html">Sample Code</a> > <a href="../../samplecode/GraphicsImaging/index.html">Graphics & Imaging</a> > <a href="../../samplecode/GraphicsImaging/idxQuartz-date.html">Quartz</a> > <A HREF="javascript:location.replace('index.html');">CIColorTracking</A> >
</td>
</tr>
<tr>
<td colspan="2" scope="row"><img width="680" height="35" src="images/1dot.gif" alt=""></td>
</tr>
</table>
</div>
<!-- END BREADCRUMB -->
<!-- START MAIN CONTENT -->
<!-- START TITLE GRAPHIC AND INTRO-->
<table width="680" border="0" cellpadding="0" cellspacing="0">
<tr align="left" valign="top">
<td><h1><div id="pagehead">CIColorTracking</div></h1></td>
</tr>
</table>
<!-- END TITLE GRAPHIC AND INTRO -->
<!-- START WIDE COLUMN -->
<table width="680" border="0" cellpadding="0" cellspacing="0">
<tr align="left" valign="top">
<td id="scdetails">
<h2>/AppController.m</h2>
<form name="scpopupmenu" onSubmit="return false;" method=post>
<p><strong>View Source Code:</strong>
<select name="gotop" onChange="JumpToNewPage();" style="width:340px"><option selected value="ingnore">Select File</option>
<option value="listing1.html">/AppController.h</option>
<option value="listing2.html">/AppController.m</option>
<option value="listing3.html">/main.m</option>
<option value="listing4.html">/ReadMe.txt</option>
<option value="listing5.html">/TrackingImageUnit/TrackingFilters.h</option>
<option value="listing6.html">/TrackingImageUnit/TrackingFilters.m</option>
<option value="listing7.html">/VideoCIView.h</option>
<option value="listing8.html">/VideoCIView.m</option>
<option value="listing9.html">/VideoHandler.h</option>
<option value="listing10.html">/VideoHandler.m</option></select>
</p>
</form>
<p><strong><a href="CIColorTracking.zip">Download Sample</a></strong> (“CIColorTracking.zip”, 2.70M)<BR>
<strong><a href="CIColorTracking.dmg">Download Sample</a></strong> (“CIColorTracking.dmg”, 3.02M)</p>
<!--
<p><strong><a href="#">Download Sample</a></strong> (“filename.sit”, 500K)</p>
-->
</td>
</tr>
<tr>
<td scope="row"><img width="680" height="10" src="images/1dot.gif" alt=""><br>
<img height="1" width="680" src="images/1dot_919699.gif" alt=""><br>
<img width="680" height="20" src="images/1dot.gif" alt=""></td>
</tr>
<tr>
<td scope="row">
<!--googleon: index -->
<pre class="sourcecodebox">/*
Project: CIColorTracking
File: AppController.m
Abstract:
This is the implementation file for AppController, a class that handles the interaction between the
GUI, the video, and the tracking filters, allowing you to select a source movie, change
tracking options, view the tracking video, see a preview of the mask used for the video,
and overlay the mask image onto the tracking video.
Version 1.0
Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc.
("Apple") in consideration of your agreement to the following terms, and your
use, installation, modification or redistribution of this Apple software
constitutes acceptance of these terms. If you do not agree with these terms,
please do not use, install, modify or redistribute this Apple software.
In consideration of your agreement to abide by the following terms, and subject
to these terms, Apple grants you a personal, non-exclusive license, under
Apple's copyrights in this original Apple software (the "Apple Software"), to
use, reproduce, modify and redistribute the Apple Software, with or without
modifications, in source and/or binary forms; provided that if you redistribute
the Apple Software in its entirety and without modifications, you must retain
this notice and the following text and disclaimers in all such redistributions
of the Apple Software.
Neither the name, trademarks, service marks or logos of Apple Inc. may be used
to endorse or promote products derived from the Apple Software without specific
prior written permission from Apple. Except as expressly stated in this notice,
no other rights or licenses, express or implied, are granted by Apple herein,
including but not limited to any patent rights that may be infringed by your
derivative works or by other works in which the Apple Software may be
incorporated.
The Apple Software is provided by Apple on an "AS IS" basis. APPLE MAKES NO
WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
WARRANTIES OF NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION ALONE OR IN
COMBINATION WITH YOUR PRODUCTS.
IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION AND/OR
DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER UNDER THEORY OF
CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR OTHERWISE, EVEN IF
APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Copyright (C) 2008 Apple Inc. All Rights Reserved.
*/
#import "AppController.h"
#import "VideoCIView.h"
#import "VideoHandler.h"
@interface AppController (private)
- (void)_loadMovie:(NSString *)path;
- (void)_renderFilterChain:(BOOL)rechain;
@end
@implementation AppController
//--------------------------------------------------------------------------------------------------
- (void)awakeFromNib
{
lock = [[NSRecursiveLock alloc] init];
// Load all Image Units that are installed in /Library/Graphics/Image Units
[CIPlugIn loadAllPlugIns];
// Get a list of all filters.
NSArray *filterList = [CIFilter filterNamesInCategories:nil];
// If the filter need for color tracking is not in the list, then
// load the filter from the PlugIns directory of the application.
if(![filterList containsObject:@"GPUGemsAreaMean"])
{
NSString *path = [[[NSBundle mainBundle] builtInPlugInsPath] stringByAppendingPathComponent:@"TrackingImageUnit.plugin"];
NSURL *pluginURL = [NSURL fileURLWithPath:path];
[CIPlugIn loadPlugIn:pluginURL allowNonExecutable:NO];
}
// Update the filters list.
filterList = [CIFilter filterNamesInCategories:nil];
// Check to make sure that the filter needed for color tracking is on the list.
if(![filterList containsObject:@"GPUGemsAreaMean"])
{
NSRunCriticalAlertPanel(@"Required filters are missing", @"The Tracking Image Unit is not installed", @"Quit", nil, nil, nil);
[NSApp terminate:self];
}
// Load the filters that you need for color tracking. Some of these are built-in to Core Image,
// Others are provided by the Tracking Image Unit.
cropFilter = [[CIFilter filterWithName:@"CICrop"] retain];
stage1_Filter_maskFromColor = [[CIFilter filterWithName:@"MaskFromColor"] retain];
stage2_Filter_multiplyCoordinates = [[CIFilter filterWithName:@"CoordinateMaskWithColor"] retain];
stage2_Filter_areaMean = [[CIFilter filterWithName:@"GPUGemsAreaMean"] retain];
stage2_Filter_coordinateNormalize = [[CIFilter filterWithName:@"Centroid"] retain];
stage3_Filter_compositeResult = [[CIFilter filterWithName:@"CompositeFilter"] retain];
postProcessing_Filter_colorMatrix = [[CIFilter filterWithName:@"CIColorMatrix"] retain];
postProcessing_Filter_composite = [[CIFilter filterWithName:@"CISourceOverCompositing"] retain];
// Set initial values for filter.
[cropFilter setValue:[CIVector vectorWithX:70.0 Y:0.0 Z:640.0 W:480.0] forKey:@"inputRectangle"];
[stage1_Filter_maskFromColor setDefaults];
[stage2_Filter_areaMean setDefaults];
[stage3_Filter_compositeResult setDefaults];
[postProcessing_Filter_colorMatrix setDefaults];
CIVector *alphaVector = [CIVector vectorWithX:0.0 Y:0.0 Z:0.0 W:[maskOpacitySlider floatValue]];
[postProcessing_Filter_colorMatrix setValue:alphaVector forKey:@"inputAVector"];
// Read the overlay image from the resources.
NSString *imagePath = [[NSBundle bundleForClass:[self class]] pathForResource:@"Duck" ofType:@"tiff"];
[stage3_Filter_compositeResult setValue:[CIImage imageWithContentsOfURL:[NSURL fileURLWithPath:imagePath]] forKey:@"inputOverlayImage"];
}
//--------------------------------------------------------------------------------------------------
- (void)applicationWillFinishLaunching:(NSNotification *)note
{
NSString *moviePath;
CIColor *cicolor;
// Check whether the the MoviePath key is set to a pathname.
moviePath = [[NSUserDefaults standardUserDefaults] stringForKey:@"MoviePath"];
// If so, load the movie located at that path.
if(moviePath)
{
[self _loadMovie:moviePath];
}
// Get the default tracking color.
NSData *theData = [[[NSUserDefaults standardUserDefaults] stringForKey:@"TrackingColor"] retain];
if (theData != nil)
trackingColor = (NSColor *)[NSUnarchiver unarchiveObjectWithData:theData];
// Set the default tracking color as the value of the input color parameters for the mask from color filter.
if(!trackingColor)
trackingColor = [[NSColor colorWithCIColor: [stage1_Filter_maskFromColor valueForKey:@"inputColor"]] retain];
[colorTrackingBox setColor:trackingColor];
// Set the defatult tracking tolerance value.
NSNumber *trackingToleranceValue = (NSNumber *)[[NSUserDefaults standardUserDefaults] stringForKey:@"TrackingTolerance"];
if(!trackingToleranceValue)
trackingToleranceValue = [stage1_Filter_maskFromColor valueForKey:@"inputThreshold"];
trackingTolerance = [trackingToleranceValue floatValue];
[toleranceSlider setFloatValue:trackingTolerance];
[toleranceField setFloatValue:trackingTolerance];
// Allocate a CIColor object and initialize it with the tracking color.
cicolor = [[CIColor alloc] initWithColor: trackingColor];
// Set the color input value and threshold value for the mask from color filter.
[stage1_Filter_maskFromColor setValue:[cicolor autorelease] forKey:@"inputColor"];
[stage1_Filter_maskFromColor setValue:[NSNumber numberWithFloat:trackingTolerance] forKey:@"inputThreshold"];
[self _renderFilterChain:YES];
}
//--------------------------------------------------------------------------------------------------
- (void)dealloc
{
[trackingColor release];
[videoHandler release];
[lock release];
[stage1_Filter_maskFromColor release];
[stage2_Filter_multiplyCoordinates release];
[stage2_Filter_areaMean release];
[stage2_Filter_coordinateNormalize release];
[stage3_Filter_compositeResult release];
[postProcessing_Filter_colorMatrix release];
[postProcessing_Filter_composite release];
[super dealloc];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)movieScrubAction:(id)sender
{
[videoHandler scrubMovie:[sender doubleValue]];
[playButton setImage:[videoHandler isPlaying] ? [NSImage imageNamed:@"FS_Pause_Normal.tif"] : [NSImage imageNamed:@"FS_Play_Normal.tif"]];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)selectMovie:(id)sender
{
NSOpenPanel *openPanel;
int rv;
openPanel = [NSOpenPanel openPanel];
[openPanel setCanChooseDirectories:NO];
[openPanel setAllowsMultipleSelection:NO];
[openPanel setResolvesAliases:YES];
[openPanel setCanChooseFiles:YES];
rv = [openPanel runModalForTypes:nil];
if(rv == NSFileHandlingPanelOKButton)
[self _loadMovie:[openPanel filename]];
}
//--------------------------------------------------------------------------------------------------
- (void)_loadMovie:(NSString *)path
{
if(!videoHandler)
videoHandler = [[VideoHandler alloc] initWithView:videoView delegate:self];
if([videoHandler loadMovieFromPath:path])
{
[[NSUserDefaults standardUserDefaults] setObject:path forKey:@"MoviePath"];
[[NSUserDefaults standardUserDefaults] synchronize];
[maskOpacitySlider setEnabled:YES];
[showCompositeCheckBox setEnabled:YES];
[playButton setEnabled:YES];
[videoScrubber setEnabled:YES];
}
}
//--------------------------------------------------------------------------------------------------
- (IBAction)setCompositeVisible:(id)sender
{
[lock lock];
[self _renderFilterChain:NO];
[lock unlock];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)setMaskOverlayOpacity:(id)sender
{
[lock lock];
CIVector *alphaVector = [CIVector vectorWithX:0.0 Y:0.0 Z:0.0 W:[sender floatValue]];
[postProcessing_Filter_colorMatrix setValue:alphaVector forKey:@"inputAVector"];
[self _renderFilterChain:YES];
[lock unlock];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)setTrackingColor:(id)sender
{
[lock lock];
[trackingColor release];
trackingColor = [[colorTrackingBox color] retain];
[[NSUserDefaults standardUserDefaults] setObject:[NSArchiver archivedDataWithRootObject:trackingColor] forKey:@"TrackingColor"];
[[NSUserDefaults standardUserDefaults] synchronize];
[stage1_Filter_maskFromColor setValue:[CIColor colorWithRed:[trackingColor redComponent] green:[trackingColor greenComponent] blue:[trackingColor blueComponent]] forKey:@"inputColor"];
[self _renderFilterChain:YES];
[lock unlock];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)setTrackingTolerance:(id)sender
{
[lock lock];
trackingTolerance = [sender floatValue];
[[NSUserDefaults standardUserDefaults] setObject:[NSNumber numberWithFloat:trackingTolerance] forKey:@"TrackingTolerance"];
[[NSUserDefaults standardUserDefaults] synchronize];
[toleranceSlider setFloatValue:trackingTolerance];
[toleranceField setFloatValue:trackingTolerance];
[stage1_Filter_maskFromColor setValue:[NSNumber numberWithFloat:trackingTolerance] forKey:@"inputThreshold"];
[self _renderFilterChain:YES];
[lock unlock];
}
//--------------------------------------------------------------------------------------------------
- (IBAction)togglePlayback:(id)sender
{
[videoHandler togglePlayback];
[playButton setImage:[videoHandler isPlaying] ? [NSImage imageNamed:@"FS_Pause_Normal.tif"] : [NSImage imageNamed:@"FS_Play_Normal.tif"]];
}
//--------------------------------------------------------------------------------------------------
- (void)renderFrame:(CVImageBufferRef)inFrame progress:(double)inProgress
{
CIImage *image;
[lock lock];
[cropFilter setValue:[CIImage imageWithCVImageBuffer:inFrame] forKey:@"inputImage"];
image = [cropFilter valueForKey:@"outputImage"];
[stage1_Filter_maskFromColor setValue: image forKey:@"inputImage"];
[stage3_Filter_compositeResult setValue: image forKey:@"inputImage"];
[self _renderFilterChain:YES];
[videoScrubber setDoubleValue:inProgress];
[lock unlock];
}
//--------------------------------------------------------------------------------------------------
- (void)_renderFilterChain:(BOOL)rechain
{
if([stage1_Filter_maskFromColor valueForKey:@"inputImage"])
{
if(rechain)
{
CGRect extent;
[stage2_Filter_multiplyCoordinates setValue:[stage1_Filter_maskFromColor valueForKey:@"outputImage"] forKey:@"inputImage"];
extent = [[stage2_Filter_multiplyCoordinates valueForKey:@"outputImage"] extent];
[stage2_Filter_areaMean setValue: [CIVector vectorWithX: CGRectGetMinX(extent) Y: CGRectGetMinY(extent)
Z: CGRectGetWidth(extent) W: CGRectGetHeight(extent)] forKey:@"inputExtent"];
[stage2_Filter_areaMean setValue:[stage2_Filter_multiplyCoordinates valueForKey:@"outputImage"] forKey:@"inputImage"];
[stage2_Filter_coordinateNormalize setValue:[stage2_Filter_areaMean valueForKey:@"outputImage"] forKey:@"inputImage"];
[stage3_Filter_compositeResult setValue:[stage2_Filter_coordinateNormalize valueForKey:@"outputImage"] forKey:@"inputLocation"];
[postProcessing_Filter_colorMatrix setValue:[stage1_Filter_maskFromColor valueForKey:@"outputImage"] forKey:@"inputImage"];
if([showCompositeCheckBox state] == NSOnState)
[postProcessing_Filter_composite setValue:[stage3_Filter_compositeResult valueForKey:@"outputImage"] forKey:@"inputBackgroundImage"];
else
[postProcessing_Filter_composite setValue:[stage1_Filter_maskFromColor valueForKey:@"inputImage"] forKey:@"inputBackgroundImage"];
[postProcessing_Filter_composite setValue:[postProcessing_Filter_colorMatrix valueForKey:@"outputImage"] forKey:@"inputImage"];
}
[maskView setImage:[stage1_Filter_maskFromColor valueForKey:@"outputImage"]];
[videoView setImage:[postProcessing_Filter_composite valueForKey:@"outputImage"]];
}
}
@end
</pre>
<!--googleoff: index -->
</td>
</tr>
</table>
<!-- END WIDE COLUMN -->
<!-- END MAIN CONTENT -->
<table width="680" border="0" cellpadding="0" cellspacing="0">
<tr>
<td><div style="width: 100%; height: 1px; background-color: #919699; margin-top: 5px; margin-bottom: 15px"></div></td>
</tr>
<tr>
<td align="center"><br/>
<table border="0" cellpadding="0" cellspacing="0" class="graybox">
<tr>
<th>Did this document help you?</th>
</tr>
<tr>
<td>
<div style="margin-bottom: 8px"><a href="http://developer.apple.com/feedback/?v=1&url=/samplecode/CIColorTracking/listing2.html%3Fid%3DDTS10004327-2.0&media=dvd" target=_new>Yes</a>: Tell us what works for you.</div>
<div style="margin-bottom: 8px"><a href="http://developer.apple.com/feedback/?v=2&url=/samplecode/CIColorTracking/listing2.html%3Fid%3DDTS10004327-2.0&media=dvd" target=_new>It’s good, but:</a> Report typos, inaccuracies, and so forth.</div>
<div><a href="http://developer.apple.com/feedback/?v=3&url=/samplecode/CIColorTracking/listing2.html%3Fid%3DDTS10004327-2.0&media=dvd" target=_new>It wasn’t helpful</a>: Tell us what would have helped.</div>
</td>
</tr>
</table>
</td>
</tr>
</table>
<!-- START BOTTOM APPLE NAVIGATION -->
<!--#include virtual="/includes/footer"-->
<!-- END BOTTOM APPLE NAVIGATION -->
<!-- START CENTER CLOSE -->
</center>
<!-- END CENTER CLOSE -->
</body>
</html>