OSDN Git Service

514f7b32f08d214f4f01c6d939dd7283884e1540
[handbrake-jp/handbrake-jp-git.git] / macosx / HBAudioController.m
1 //
2 //  HBAudioController.m
3 //  HandBrake
4 //
5 //  Created on 2010-08-24.
6 //
7
8 #import "HBAudioController.h"
9 #import "Controller.h"
10 #import "HBAudio.h"
11 #import "hb.h"
12
13 NSString *keyAudioTrackIndex = @"keyAudioTrackIndex";
14 NSString *keyAudioTrackName = @"keyAudioTrackName";
15 NSString *keyAudioInputBitrate = @"keyAudioInputBitrate";
16 NSString *keyAudioInputSampleRate = @"keyAudioInputSampleRate";
17 NSString *keyAudioInputCodec = @"keyAudioInputCodec";
18 NSString *keyAudioInputChannelLayout = @"keyAudioInputChannelLayout";
19 NSString *HBMixdownChangedNotification = @"HBMixdownChangedNotification";
20
21 @implementation HBAudioController
22
23 #pragma mark -
24 #pragma mark Accessors
25
26 @synthesize masterTrackArray;
27 @synthesize noneTrack;
28 @synthesize videoContainerTag;
29
30 - (id) init
31
32 {
33         if (self = [super init]) {
34                 [self setVideoContainerTag: [NSNumber numberWithInt: HB_MUX_MP4]];
35                 audioArray = [[NSMutableArray alloc] init];
36         }
37         return self;
38 }
39
40 - (void) dealloc
41
42 {
43         [[NSNotificationCenter defaultCenter] removeObserver: self];
44         [masterTrackArray release];
45         [noneTrack release];
46         [audioArray release];
47         [self setVideoContainerTag: nil];
48         [super dealloc];
49         return;
50 }
51
52 - (void) setHBController: (id) aController
53
54 {
55         NSNotificationCenter *center = [NSNotificationCenter defaultCenter];
56         myController = aController;
57
58         /* register that we are interested in changes made to the video container */
59         [center addObserver: self selector: @selector(containerChanged:) name: HBContainerChangedNotification object: aController];
60         [center addObserver: self selector: @selector(titleChanged:) name: HBTitleChangedNotification object: aController];
61         return;
62 }
63
64 - (void) _clearAudioArray
65
66 {
67         while (0 < [self countOfAudioArray]) {
68                 [self removeObjectFromAudioArrayAtIndex: 0];
69         }
70         return;
71 }
72
73 #pragma mark -
74 #pragma mark HBController Support
75
76 - (void) prepareAudioForQueueFileJob: (NSMutableDictionary *) aDict
77
78 {
79         unsigned int audioArrayCount = [self countOfAudioArray];
80         for (unsigned int counter = 0; counter < audioArrayCount; counter++) {
81                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: counter];
82                 if (YES == [anAudio enabled]) {
83                         NSString *prefix = [NSString stringWithFormat: @"Audio%d", counter + 1];
84                         NSNumber *sampleRateToUse = (0 == [[[anAudio sampleRate] objectForKey: keyAudioSamplerate] intValue]) ?
85                                                                 [[anAudio track] objectForKey: keyAudioInputSampleRate] :
86                                                                 [[anAudio sampleRate] objectForKey: keyAudioSamplerate];
87                 
88                         [aDict setObject: [[anAudio track] objectForKey: keyAudioTrackIndex] forKey: [prefix stringByAppendingString: @"Track"]];
89                         [aDict setObject: [[anAudio track] objectForKey: keyAudioTrackName] forKey: [prefix stringByAppendingString: @"TrackDescription"]];
90                         [aDict setObject: [[anAudio codec] objectForKey: keyAudioCodecName] forKey: [prefix stringByAppendingString: @"Encoder"]];
91                         [aDict setObject: [[anAudio mixdown] objectForKey: keyAudioMixdownName] forKey: [prefix stringByAppendingString: @"Mixdown"]];
92                         [aDict setObject: [[anAudio sampleRate] objectForKey: keyAudioSampleRateName] forKey: [prefix stringByAppendingString: @"Samplerate"]];
93                         [aDict setObject: [[anAudio bitRate] objectForKey: keyAudioBitrateName] forKey: [prefix stringByAppendingString: @"Bitrate"]];
94                         [aDict setObject: [anAudio drc] forKey: [prefix stringByAppendingString: @"TrackDRCSlider"]];
95                 
96                         prefix = [NSString stringWithFormat: @"JobAudio%d", counter + 1];
97                         [aDict setObject: [[anAudio codec] objectForKey: keyAudioCodec] forKey: [prefix stringByAppendingString: @"Encoder"]];
98                         [aDict setObject: [[anAudio mixdown] objectForKey: keyAudioMixdown] forKey: [prefix stringByAppendingString: @"Mixdown"]];
99                         [aDict setObject: sampleRateToUse forKey: [prefix stringByAppendingString: @"Samplerate"]];
100                         [aDict setObject: [[anAudio bitRate] objectForKey: keyAudioBitrate] forKey: [prefix stringByAppendingString: @"Bitrate"]];
101                 }
102         }
103         return;
104 }
105
106 - (void) prepareAudioForJob: (hb_job_t *) aJob
107
108 {
109         unsigned int i;
110         
111         //      First clear out any audio tracks in the job currently
112     int audiotrack_count = hb_list_count(aJob->list_audio);
113     for(i = 0; i < audiotrack_count; i++)
114     {
115         hb_audio_t *temp_audio = (hb_audio_t *) hb_list_item(aJob->list_audio, 0);
116         hb_list_rem(aJob->list_audio, temp_audio);
117     }
118
119         //      Now add audio tracks based on the current settings
120         unsigned int audioArrayCount = [self countOfAudioArray];
121         for (i = 0; i < audioArrayCount; i++) {
122                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: i];
123                 if (YES == [anAudio enabled]) {
124                         NSNumber *sampleRateToUse = (0 == [[[anAudio sampleRate] objectForKey: keyAudioSamplerate] intValue]) ?
125                                                                                 [[anAudio track] objectForKey: keyAudioInputSampleRate] :
126                                                                                 [[anAudio sampleRate] objectForKey: keyAudioSamplerate];
127                         
128                         hb_audio_config_t *audio = (hb_audio_config_t *) calloc(1, sizeof(*audio));
129                         hb_audio_config_init(audio);
130                         audio->in.track = [[[anAudio track] objectForKey: keyAudioTrackIndex] intValue] - 1;
131                         /* We go ahead and assign values to our audio->out.<properties> */
132                         audio->out.track = audio->in.track;
133                         audio->out.codec = [[[anAudio codec] objectForKey: keyAudioCodec] intValue];
134                         audio->out.mixdown = [[[anAudio mixdown] objectForKey: keyAudioMixdown] intValue];
135                         audio->out.bitrate = [[[anAudio bitRate] objectForKey: keyAudioBitrate] intValue];
136                         audio->out.samplerate = [sampleRateToUse intValue];
137                         audio->out.dynamic_range_compression = [[anAudio drc] floatValue];
138         
139                         hb_audio_add(aJob, audio);
140                         free(audio);
141                 }
142         }
143         return;
144 }
145
146 - (void) prepareAudioForPreset: (NSMutableArray *) anArray
147
148 {
149         unsigned int audioArrayCount = [self countOfAudioArray];
150         unsigned int i;
151
152         for (i = 0; i < audioArrayCount; i++) {
153                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: i];
154                 if (YES == [anAudio enabled]) {
155                         NSMutableDictionary *dict = [[NSMutableDictionary alloc] initWithCapacity: 7];
156                         [dict setObject: [[anAudio track] objectForKey: keyAudioTrackIndex] forKey: @"AudioTrack"];
157                         [dict setObject: [[anAudio track] objectForKey: keyAudioTrackName] forKey: @"AudioTrackDescription"];
158                         [dict setObject: [[anAudio codec] objectForKey: keyAudioCodecName] forKey: @"AudioEncoder"];
159                         [dict setObject: [[anAudio mixdown] objectForKey: keyAudioMixdownName] forKey: @"AudioMixdown"];
160                         [dict setObject: [[anAudio sampleRate] objectForKey: keyAudioSampleRateName] forKey: @"AudioSamplerate"];
161                         [dict setObject: [[anAudio bitRate] objectForKey: keyAudioBitrateName] forKey: @"AudioBitrate"];
162                         [dict setObject: [anAudio drc] forKey: @"AudioTrackDRCSlider"];
163                         [anArray addObject: dict];
164                         [dict release];
165                 }
166         }
167         return;
168 }
169
170 - (void) addTracksFromQueue: (NSMutableDictionary *) aQueue
171
172 {
173         NSString *base;
174         int value;
175         int maximumNumberOfAllowedAudioTracks = [HBController maximumNumberOfAllowedAudioTracks];
176
177         //      Reinitialize the configured list of audio tracks
178         [self _clearAudioArray];
179         
180         //      The following is the pattern to follow, but with Audio%dTrack being the key to seek...
181         //      Can we assume that there will be no skip in the data?
182         for (unsigned int i = 1; i <= maximumNumberOfAllowedAudioTracks; i++) {
183                 base = [NSString stringWithFormat: @"Audio%d", i];
184                 value = [[aQueue objectForKey: [base stringByAppendingString: @"Track"]] intValue];
185                 if (0 < value) {
186                         HBAudio *newAudio = [[HBAudio alloc] init];
187                         [newAudio setController: self];
188                         [self insertObject: newAudio inAudioArrayAtIndex: [self countOfAudioArray]];
189                         [newAudio setVideoContainerTag: [self videoContainerTag]];
190                         [newAudio setTrackFromIndex: value];
191                         [newAudio setCodecFromName: [aQueue objectForKey: [base stringByAppendingString: @"Encoder"]]];
192                         [newAudio setMixdownFromName: [aQueue objectForKey: [base stringByAppendingString: @"Mixdown"]]];
193                         [newAudio setSampleRateFromName: [aQueue objectForKey: [base stringByAppendingString: @"Samplerate"]]];
194                         [newAudio setBitRateFromName: [aQueue objectForKey: [base stringByAppendingString: @"Bitrate"]]];
195                         [newAudio setDrc: [aQueue objectForKey: [base stringByAppendingString: @"TrackDRCSlider"]]];
196                         [newAudio release];
197                 }
198         }
199
200         [self switchingTrackFromNone: nil];     // see if we need to add one to the list
201         
202         return;
203 }
204
205 //      This routine takes the preset and will return the value for the key AudioList
206 //      if it exists, otherwise it creates an array from the data in the present.
207 - (NSArray *) _presetAudioArrayFromPreset: (NSMutableDictionary *) aPreset
208
209 {
210         NSArray *retval = [aPreset objectForKey: @"AudioList"];
211
212         if (nil == retval) {
213                 int maximumNumberOfAllowedAudioTracks = [HBController maximumNumberOfAllowedAudioTracks];
214                 NSString *base;
215                 NSMutableArray *whatToUse = [NSMutableArray array];
216                 for (unsigned int i = 1; i <= maximumNumberOfAllowedAudioTracks; i++) {
217                         base = [NSString stringWithFormat: @"Audio%d", i];
218                         if (nil != [aPreset objectForKey: [base stringByAppendingString: @"Track"]]) {
219                                 [whatToUse addObject: [NSDictionary dictionaryWithObjectsAndKeys:
220                                                                            [aPreset objectForKey: [base stringByAppendingString: @"Encoder"]], @"AudioEncoder",
221                                                                            [aPreset objectForKey: [base stringByAppendingString: @"Mixdown"]], @"AudioMixdown",
222                                                                            [aPreset objectForKey: [base stringByAppendingString: @"Samplerate"]], @"AudioSamplerate",
223                                                                            [aPreset objectForKey: [base stringByAppendingString: @"Bitrate"]], @"AudioBitrate",
224                                                                            [aPreset objectForKey: [base stringByAppendingString: @"TrackDRCSlider"]], @"AudioTrackDRCSlider",
225                                                                            nil]];
226                         }
227                 }
228                 retval = whatToUse;
229         }
230         return retval;
231 }
232
233 //      This uses the templateAudioArray from the preset to create the audios for the specified trackIndex
234 - (void) _processPresetAudioArray: (NSArray *) templateAudioArray forTrack: (unsigned int) trackIndex andType: (int) aType
235
236 {
237         NSEnumerator *enumerator = [templateAudioArray objectEnumerator];
238         NSDictionary *dict;
239         NSString *key;
240         int maximumNumberOfAllowedAudioTracks = [HBController maximumNumberOfAllowedAudioTracks];
241         
242         while (nil != (dict = [enumerator nextObject])) {
243                 if ([self countOfAudioArray] < maximumNumberOfAllowedAudioTracks) {
244                         HBAudio *newAudio = [[HBAudio alloc] init];
245                         [newAudio setController: self];
246                         [self insertObject: newAudio inAudioArrayAtIndex: [self countOfAudioArray]];
247                         [newAudio setVideoContainerTag: [self videoContainerTag]];
248                         [newAudio setTrackFromIndex: trackIndex];
249                         key = [dict objectForKey: @"AudioEncoder"];
250                         if (0 == aType &&
251                                 YES == [[NSUserDefaults standardUserDefaults] boolForKey: @"UseCoreAudio"] &&
252                                 YES == [key isEqualToString: @"AAC (faac)"]
253                                 ) {
254                                 key = @"AAC (CoreAudio)";
255                         }
256                         if (YES == [[NSUserDefaults standardUserDefaults] boolForKey: @"AC3PassthruDefaultsToAC3"] &&
257                                 YES == [key isEqualToString: @"AC3 Passthru"]) {
258                                 if (NO == [newAudio setCodecFromName: key]) {
259                                         key = @"AC3";
260                                 }
261                         }
262                         //      If our preset wants us to support a codec that the track does not support, instead
263                         //      of changing the codec we remove the audio instead.
264                         if (YES == [newAudio setCodecFromName: key]) {
265                                 [newAudio setMixdownFromName: [dict objectForKey: @"AudioMixdown"]];
266                                 [newAudio setSampleRateFromName: [dict objectForKey: @"AudioSamplerate"]];
267                                 [newAudio setBitRateFromName: [dict objectForKey: @"AudioBitrate"]];
268                                 [newAudio setDrc: [dict objectForKey: @"AudioTrackDRCSlider"]];
269                         }
270                         else {
271                                 [self removeObjectFromAudioArrayAtIndex: [self countOfAudioArray] - 1];
272                         }
273                         [newAudio release];
274                 }
275         }
276         return;
277 }
278
279 //      This matches the FIRST track with the specified prefix, otherwise it uses the defaultIfNotFound value
280 - (unsigned int) _trackWithTitlePrefix: (NSString *) prefix defaultIfNotFound: (unsigned int) defaultIfNotFound
281
282 {
283         unsigned int retval = defaultIfNotFound;
284         int count = [masterTrackArray count];
285         NSString *languageTitle;
286         BOOL found = NO;
287         
288         //      We search for the prefix noting that our titles have the format %d: %s where the %s is the prefix
289         for (unsigned int i = 1; i < count && NO == found; i++) {       //      Note that we skip the "None" track
290                 languageTitle = [[masterTrackArray objectAtIndex: i] objectForKey: keyAudioTrackName];
291                 if (YES == [[languageTitle substringFromIndex: [languageTitle rangeOfString: @" "].location + 1] hasPrefix: prefix]) {
292                         retval = i;
293                         found = YES;
294                 }
295         }
296         return retval;
297 }
298
299 //      When we add a track and we do not have a preset to use for the track we use
300 //      this bogus preset to do the dirty work.
301 - (NSMutableDictionary *) _defaultPreset
302
303 {
304         static NSMutableDictionary *retval = nil;
305         
306         if (nil == retval) {
307                 retval = [[NSMutableDictionary dictionaryWithObjectsAndKeys:
308                 [NSArray arrayWithObject:
309                 [NSDictionary dictionaryWithObjectsAndKeys:
310                 [NSNumber numberWithInt: 1],     @"AudioTrack",
311                 @"AAC (faac)",                   @"AudioEncoder",
312                 @"Dolby Pro Logic II",           @"AudioMixdown",
313                 @"Auto",                         @"AudioSamplerate",
314                 @"160",                          @"AudioBitrate",
315                 [NSNumber numberWithFloat: 0.0], @"AudioTrackDRCSlider",
316                 nil]], @"AudioList", nil] retain];
317         }
318         return retval;
319 }
320
321 - (void) addTracksFromPreset: (NSMutableDictionary *) aPreset allTracks: (BOOL) allTracks
322
323 {
324         id whatToUse = [self _presetAudioArrayFromPreset: aPreset];
325         NSString *preferredLanguageName = [[NSUserDefaults standardUserDefaults] stringForKey: @"DefaultLanguage"];
326         int preferredLanguage = [self _trackWithTitlePrefix: preferredLanguageName defaultIfNotFound: 1];
327
328         //      Reinitialize the configured list of audio tracks
329         [self _clearAudioArray];
330         
331         [self _processPresetAudioArray: whatToUse forTrack: preferredLanguage andType: [[aPreset objectForKey: @"Type"] intValue]];
332         if (YES == allTracks) {
333                 unsigned int count = [masterTrackArray count];
334                 for (unsigned int i = 1; i < count; i++) {
335                         if (i != preferredLanguage) {
336                                 [self _processPresetAudioArray: whatToUse forTrack: i andType: [[aPreset objectForKey: @"Type"] intValue]];
337                         }
338                 }
339         }
340
341         return;
342 }
343
344 - (void) _ensureAtLeastOneNonEmptyTrackExists
345
346 {
347         int count = [self countOfAudioArray];
348         if (0 == count || NO == [[self objectInAudioArrayAtIndex: 0] enabled]) {
349                 [self addTracksFromPreset: [self _defaultPreset] allTracks: NO];
350                 }
351         [self switchingTrackFromNone: nil];     //      this ensures there is a None track at the end of the list
352         return;
353 }
354
355 - (void) addTracksFromPreset: (NSMutableDictionary *) aPreset
356
357 {
358         [self addTracksFromPreset: aPreset allTracks: NO];
359         [self _ensureAtLeastOneNonEmptyTrackExists];
360         return;
361 }
362
363 - (void) addAllTracksFromPreset: (NSMutableDictionary *) aPreset
364
365 {
366         [self addTracksFromPreset: aPreset allTracks: YES];
367         [self _ensureAtLeastOneNonEmptyTrackExists];
368         return;
369 }
370
371 - (BOOL) anyCodecMatches: (int) aCodecValue
372
373 {
374         BOOL retval = NO;
375         unsigned int audioArrayCount = [self countOfAudioArray];
376         for (unsigned int i = 0; i < audioArrayCount && NO == retval; i++) {
377                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: i];
378         if (YES == [anAudio enabled] && aCodecValue == [[[anAudio codec] objectForKey: keyAudioCodec] intValue]) {
379                         retval = YES;
380                 }
381         }
382         return retval;
383 }
384
385 - (void) addNewAudioTrack
386
387 {
388         HBAudio *newAudio = [[HBAudio alloc] init];
389         [newAudio setController: self];
390         [self insertObject: newAudio inAudioArrayAtIndex: [self countOfAudioArray]];
391         [newAudio setVideoContainerTag: [self videoContainerTag]];
392         [newAudio setTrack: noneTrack];
393         [newAudio setDrc: [NSNumber numberWithFloat: 0.0]];
394         [newAudio release];     
395         return;
396 }
397
398 #pragma mark -
399 #pragma mark Notification Handling
400
401 - (void) settingTrackToNone: (HBAudio *) newNoneTrack
402
403 {
404         //      If this is not the last track in the array we need to remove it.  We then need to see if a new
405         //      one needs to be added (in the case when we were at maximum count and this switching makes it
406         //      so we are no longer at maximum.
407         unsigned int index = [audioArray indexOfObject: newNoneTrack];
408
409         if (NSNotFound != index && index < [self countOfAudioArray] - 1) {
410                 [self removeObjectFromAudioArrayAtIndex: index];
411         }
412         [self switchingTrackFromNone: nil];     // see if we need to add one to the list
413         return;
414 }
415
416 - (void) switchingTrackFromNone: (HBAudio *) noLongerNoneTrack
417
418 {
419         int count = [self countOfAudioArray];
420         BOOL needToAdd = NO;
421         int maximumNumberOfAllowedAudioTracks = [HBController maximumNumberOfAllowedAudioTracks];
422
423         //      If there is no last track that is None and we are less than our maximum number of permitted tracks, we add one.
424         if (count < maximumNumberOfAllowedAudioTracks) {
425                 if (0 < count) {
426                         HBAudio *lastAudio = [self objectInAudioArrayAtIndex: count - 1];
427                         if (YES == [lastAudio enabled]) {
428                                 needToAdd = YES;
429                         }
430                 }
431                 else {
432                         needToAdd = YES;
433                 }
434         }
435
436         if (YES == needToAdd) {
437                 [self addNewAudioTrack];
438         }
439         return;
440 }
441
442 //      This gets called whenever the video container changes.
443 - (void) containerChanged: (NSNotification *) aNotification
444
445 {
446         NSDictionary *notDict = [aNotification userInfo];
447
448         [self setVideoContainerTag: [notDict objectForKey: keyContainerTag]];
449
450         //      Update each of the instances because this value influences possible settings.
451         NSEnumerator *enumerator = [audioArray objectEnumerator];
452         HBAudio *audioObject;
453
454         while (nil != (audioObject = [enumerator nextObject])) {
455                 [audioObject setVideoContainerTag: [self videoContainerTag]];
456         }
457         return;
458 }
459
460 - (void) titleChanged: (NSNotification *) aNotification
461
462 {
463         NSDictionary *notDict = [aNotification userInfo];
464         NSData *theData = [notDict objectForKey: keyTitleTag];
465         hb_title_t *title = NULL;
466
467         [theData getBytes: &title length: sizeof(title)];
468         if (title) {
469                 hb_audio_config_t *audio;
470                 hb_list_t *list = title->list_audio;
471                 int i, count = hb_list_count(list);
472
473                 //      Reinitialize the master list of available audio tracks from this title
474                 [masterTrackArray release];
475                 masterTrackArray = [[NSMutableArray alloc] init];
476                 [noneTrack release];
477                 noneTrack = [[NSDictionary dictionaryWithObjectsAndKeys:
478                                          [NSNumber numberWithInt: 0], keyAudioTrackIndex,
479                                          NSLocalizedString(@"None", @"None"), keyAudioTrackName,
480                                          [NSNumber numberWithInt: 0], keyAudioInputCodec,
481                                                          nil] retain];
482                 [masterTrackArray addObject: noneTrack];
483                 for (i = 0; i < count; i++) {
484                         audio = (hb_audio_config_t *) hb_list_audio_config_item(list, i);
485                         [masterTrackArray addObject: [NSDictionary dictionaryWithObjectsAndKeys:
486                                                                                   [NSNumber numberWithInt: i + 1], keyAudioTrackIndex,
487                                                                                   [NSString stringWithFormat: @"%d: %s", i, audio->lang.description], keyAudioTrackName,
488                                                                                   [NSNumber numberWithInt: audio->in.bitrate / 1000], keyAudioInputBitrate,
489                                                                                   [NSNumber numberWithInt: audio->in.samplerate], keyAudioInputSampleRate,
490                                                                                   [NSNumber numberWithInt: audio->in.codec], keyAudioInputCodec,
491                                                                                   [NSNumber numberWithInt: audio->in.channel_layout], keyAudioInputChannelLayout,
492                                                                                   nil]];
493                 }
494         }
495
496         //      Reinitialize the configured list of audio tracks
497         [self _clearAudioArray];
498
499         if (NO == [myController hasValidPresetSelected]) {
500                 [self _ensureAtLeastOneNonEmptyTrackExists];
501         }
502         return;
503 }
504
505 #pragma mark -
506 #pragma mark KVC
507
508 - (unsigned int) countOfAudioArray
509
510 {
511         return [audioArray count];
512 }
513
514 - (HBAudio *) objectInAudioArrayAtIndex: (unsigned int) index
515
516 {
517         return [audioArray objectAtIndex: index];
518 }
519
520 - (void) insertObject: (HBAudio *) audioObject inAudioArrayAtIndex: (unsigned int) index;
521
522 {
523         [audioArray insertObject: audioObject atIndex: index];
524         return;
525 }
526
527 - (void) removeObjectFromAudioArrayAtIndex: (unsigned int) index
528
529 {
530         [audioArray removeObjectAtIndex: index];
531         return;
532 }
533
534 @end