/
AppController.m
376 lines (296 loc) · 12.2 KB
/
AppController.m
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
//
// AppController.m
// spgw4ffmpeg
//
// Created by Patrick on 04.05.09.
// Copyright 2009 Patrick Mosby. All rights reserved.
//
#import "AppController.h"
@implementation AppController
@synthesize inputFilePath, outputFilePath, allowedFileTypes;
- (id)init
{
self = [super init];
if (self) {
conversionSuccessful = NO;
self.inputFilePath = nil;
self.outputFilePath = nil;
[NSApp setDelegate:self];
self.allowedFileTypes = [NSArray arrayWithObjects:@"flv", @"avi", @"mp4", @"mov", @"wmv", @"divx", @"h264", @"mkv", @"m4v", @"3gp", @"mpg", @"mpeg", nil];
}
return self;
}
- (void)awakeFromNib
{
}
-(IBAction)startTranscode:(id)sender
{
// Validate all user inputs
// Show progress sheet and start the progress indicator
[NSApp beginSheet:progressSheet modalForWindow:mainWindow modalDelegate:self didEndSelector:NULL contextInfo:nil];
[progressIndicator startAnimation:self];
// NSLog(@"startTimeCode: %@", [startTimeCodeField stringValue]);
// NSLog(@"durationTimeCode: %@", [durationTimeCodeField stringValue]);
// NSLog(@"quality: %@", [[qualityRadioGroup selectedCell] title]);
NSString *startTimeCode = [startTimeCodeField stringValue];
NSString *durationTimeCode = [durationTimeCodeField stringValue];
NSBundle *bundle = [NSBundle mainBundle];
NSString *ffmpegPath = [bundle pathForAuxiliaryExecutable:@"ffmpeg"];
// NSLog(@"ffmpegPath: %@", ffmpegPath);
// Start ffmpeg
task = [[NSTask alloc] init];
[task setLaunchPath:ffmpegPath];
NSMutableArray *tempArgs = [NSMutableArray arrayWithObjects: @"-i",
[inputFileField stringValue],
@"-acodec",
@"libfaac",
@"-ab",
@"128k",
@"-vcodec",
@"libx264",
@"-threads",
@"2",
@"-y",
[outputFileField stringValue], nil];
NSArray *qualityArray;
// NSLog(@"qualityRadioGroup: %@", [[qualityRadioGroup selectedCell] title]);
// NSLog(@"TEMPARGS %@", tempArgs);
if ([[[qualityRadioGroup selectedCell] title] isEqualToString:@"Lossless"]) {
qualityArray = [NSArray arrayWithObjects:@"-crf", @"18", @"-me_method", @"umh", @"-subq", @"6", nil];
[tempArgs addObjectsFromArray:qualityArray];
// NSLog(@"tempargs: %@", tempArgs);
} else if ([[[qualityRadioGroup selectedCell] title] isEqualToString:@"Recompress"]) {
qualityArray = [NSArray arrayWithObjects:@"-b", @"1000k", @"-subq", @"4", nil];
[tempArgs addObjectsFromArray:qualityArray];
// NSLog(@"tempargs: %@", tempArgs);
}
NSArray *timecodeArray;
// -- convert from startTimeCode to end of file
// else if startTimeCode is not equal to "00:00:00" and durationTimeCode is equal to "00:00:00" then
if (![startTimeCode isEqualToString:@"00:00:00"] && [durationTimeCode isEqualToString:@"00:00:00"]) {
timecodeArray = [NSArray arrayWithObjects:@"-ss", startTimeCode, nil];
[tempArgs addObjectsFromArray:timecodeArray];
} else if (![startTimeCode isEqualToString:@"00:00:00"] && ![durationTimeCode isEqualToString:@"00:00:00"]) {
timecodeArray = [NSArray arrayWithObjects:@"-ss", startTimeCode, @"-t", durationTimeCode, nil];
[tempArgs addObjectsFromArray:timecodeArray];
} else if ([startTimeCode isEqualToString:@"00:00:00"] && ![durationTimeCode isEqualToString:@"00:00:00"]) {
timecodeArray = [NSArray arrayWithObjects:@"-t", durationTimeCode, nil];
[tempArgs addObjectsFromArray:timecodeArray];
}
//-- convert from startTimeCode to durationTimeCode
//else if startTimeCode is not equal to "00:00:00" and durationTimeCode is not equal to "00:00:00" then
//-- convert from start of file to durationTimeCode
NSArray *args = [NSArray arrayWithArray:tempArgs];
// NSLog(@"args: %@", args);
[task setArguments:args];
// NSLog(@"arguments: %@", [task arguments]);
// Release the old pipe
[pipe release];
// Create a new pipe
pipe = [[NSPipe alloc] init];
[task setStandardOutput:pipe];
NSFileHandle *fh = [pipe fileHandleForReading];
NSNotificationCenter *nc;
nc = [NSNotificationCenter defaultCenter];
[nc removeObserver:self];
[nc addObserver:self selector:@selector(dataReady:) name:NSFileHandleReadCompletionNotification object:fh];
[nc addObserver:self selector:@selector(taskTerminated:) name:NSTaskDidTerminateNotification object:task];
[logView setString:@"FUCKER!"];
[fh readInBackgroundAndNotify];
[task launch];
}
-(void)appendData:(NSData *)d
{
// NSLog(@"Trying to append");
NSString *s = [[NSString alloc] initWithData:d encoding:NSUTF8StringEncoding];
NSTextStorage *ts = [logView textStorage];
[ts replaceCharactersInRange:NSMakeRange([ts length], 0) withString:s];
[s release];
}
-(void)dataReady:(NSNotification *)n
{
NSData *d;
d = [[n userInfo] valueForKey:NSFileHandleNotificationDataItem];
// NSLog(@"The notification is: %@", n);
// NSLog(@"The data is: %@", d);
[self appendData:d];
// If the task is running start reading again
if ([task isRunning]) {
// NSLog(@"Reading again");
[[pipe fileHandleForReading] readInBackgroundAndNotify];
}
}
-(IBAction)showInputChooserPanel:(id)sender
{
if ([inputChooseButton title] == @"Clear") {
self.inputFilePath = nil;
[inputFileField setStringValue:@""];
[inputChooseButton setTitle:@"Choose"];
} else {
NSString *moviesDirectory = [NSHomeDirectory() stringByAppendingPathComponent:@"Movies"];
NSOpenPanel *panel = [NSOpenPanel openPanel];
// Run the open panel
[panel beginSheetForDirectory:moviesDirectory
file:nil
types:allowedFileTypes
modalForWindow:mainWindow
modalDelegate:self
didEndSelector:@selector(inputChooserPanelDidEnd:returnCode:contextInfo:)
contextInfo:NULL];
}
}
- (void)inputChooserPanelDidEnd:(NSOpenPanel *)openPanel
returnCode:(int)returnCode
contextInfo:(void *)x
{
// Did they choose "Open"?
if (returnCode == NSOKButton) {
self.inputFilePath = [openPanel filename];
[inputFileField setStringValue:inputFilePath];
[inputChooseButton setTitle:@"Clear"];
// NSLog(@"inputFile: %@", path);
}
}
-(IBAction)showOutputChooserPanel:(id)sender
{
if ([outputChooseButton title] == @"Clear") {
self.outputFilePath = nil;
[outputFileField setStringValue:@""];
[outputChooseButton setTitle:@"Choose"];
} else {
NSString *moviesDirectory = [NSHomeDirectory() stringByAppendingPathComponent:@"Movies"];
NSSavePanel *panel = [NSSavePanel savePanel];
[panel beginSheetForDirectory:moviesDirectory
file:@"Extracted Video.mp4"
modalForWindow:mainWindow
modalDelegate:self
didEndSelector:@selector(savePanelDidEnd:returnCode:contextInfo:)
contextInfo:NULL];
}
}
- (void)savePanelDidEnd:(NSSavePanel *)savePanel
returnCode:(int)returnCode
contextInfo:(void *)x
{
if (returnCode == NSOKButton) {
self.outputFilePath = [savePanel filename];
[outputFileField setStringValue:outputFilePath];
[outputChooseButton setTitle:@"Clear"];
// NSLog(@"outputFile: %@", path);
}
}
-(IBAction)cancelTranscode:(id)sender
{
// Is the task running?
if (task) {
// Close progress sheet
[progressSheet orderOut:nil];
[NSApp endSheet:progressSheet];
// Ask the user if he's sure
NSAlert *alert = [[[NSAlert alloc] init] autorelease];
[alert addButtonWithTitle:@"Yes"];
[alert addButtonWithTitle:@"No"];
[alert setMessageText:@"Are you sure?"];
[alert setInformativeText:@"Temporary movie file will be deleted."];
[alert setAlertStyle:NSWarningAlertStyle];
[alert beginSheetModalForWindow:mainWindow modalDelegate:self didEndSelector:@selector(alertDidEnd:returnCode:contextInfo:) contextInfo:nil];
}
// NSLog(@"Something went wrong…");
}
- (void)alertDidEnd:(NSAlert *)alert returnCode:(int)returnCode contextInfo:(void *)contextInfo {
// NSLog(@"alertDidEnd called");
if (returnCode == NSAlertFirstButtonReturn) {
// Kill ffmpeg
[task terminate];
// Delete the temporary outputfile
NSFileManager *fm = [[NSFileManager alloc] init];
[fm removeFileAtPath:[outputFileField stringValue] handler:nil];
[progressIndicator stopAnimation:self];
// [startTranscodeButton setNextState];
} else {
// Close the confirmation alert sheet
[[alert window] orderOut:self];
// Reopen the progress sheet
[NSApp beginSheet:progressSheet modalForWindow:mainWindow modalDelegate:self didEndSelector:NULL contextInfo:nil];
}
}
-(void)taskTerminated:(NSNotification *)note
{
// NSLog(@"taskTerminated:");
int termStatus = [task terminationStatus];
// NSLog(@"terminationStatus is: %d", termStatus);
[progressIndicator stopAnimation:self];
[progressSheet orderOut:nil];
[NSApp endSheet:progressSheet];
NSAlert *alert = [[[NSAlert alloc] init] autorelease];
[alert addButtonWithTitle:@"OK"];
if (termStatus == 255) {
[alert setMessageText:@"Conversion canceled"];
[alert setAlertStyle:NSWarningAlertStyle];
} else if (termStatus == 1) {
[alert setMessageText:@"Sorry but it seems that I can't handle this movie file"];
[alert setAlertStyle:NSWarningAlertStyle];
} else {
[alert setMessageText:@"Conversion finished"];
[alert setAlertStyle:NSInformationalAlertStyle];
conversionSuccessful = YES;
}
[self playSound:conversionSuccessful];
[alert beginSheetModalForWindow:mainWindow modalDelegate:self didEndSelector:nil contextInfo:nil];
[task release];
task = nil;
conversionSuccessful = NO;
[startTranscodeButton setState:0];
[startTranscodeButton setEnabled: NO];
[inputFileField setStringValue:@""];
[inputChooseButton setTitle:@"Choose"];
[outputFileField setStringValue:@""];
[outputChooseButton setTitle:@"Choose"];
}
-(IBAction)showLogWindow:(id)sender
{
// NSLog(@"Log Window Button clicked");
[logWindow makeKeyAndOrderFront:self];
}
- (BOOL)application:(NSApplication *)theApplication openFile:(NSString *)filename
{
NSString *fileextension = [filename pathExtension];
if (![allowedFileTypes containsObject:fileextension]) {
NSAlert *alert = [[[NSAlert alloc] init] autorelease];
[alert addButtonWithTitle:@"OK"];
[alert setMessageText:@"Sorry but it seems that I can't handle this kind of file"];
[alert setAlertStyle:NSWarningAlertStyle];
[alert beginSheetModalForWindow:mainWindow modalDelegate:self didEndSelector:nil contextInfo:nil];
} else {
self.inputFilePath = filename;
[inputFileField setStringValue:inputFilePath];
[inputChooseButton setTitle:@"Clear"];
}
return YES;
}
# pragma mark Open PDF for Help
-(IBAction)openHelp:(id)sender
{
// get path of bundle
NSBundle *bundle = [NSBundle mainBundle];
NSString *bundlePath = [bundle bundlePath];
// location of iCutMyPorn Help file
NSString *helpFile = @"/Contents/Resources/iCutMyPorn Help.pdf";
// concatenate bundlePath and helpFile
NSString *help = [bundlePath stringByAppendingString:helpFile];
// open it with your favourite PDF file viewer
[[NSWorkspace sharedWorkspace] openFile:help];
}
# pragma mark successful/failed sounds
-(void)playSound:(BOOL)success
{
NSSound *successSound;
// system sounds in /Library/Sounds and ~/Library/Sounds will be played automatically when NSSound is used
if (success == YES) {
successSound = [NSSound soundNamed:@"complete"];
} else if (success == NO) {
successSound = [NSSound soundNamed:@"Basso"];
}
[successSound play];
}
@end