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 == [key isEqualToString: @"AC3 Passthru"]) {
 257                                 if (NO == [newAudio setCodecFromName: key]) {
 258                                         key = @"AC3";
 259                                 }
 260                         }
 261                         //      If our preset wants us to support a codec that the track does not support, instead
 262                         //      of changing the codec we remove the audio instead.
 263                         if (YES == [newAudio setCodecFromName: key]) {
 264                                 [newAudio setMixdownFromName: [dict objectForKey: @"AudioMixdown"]];
 265                                 [newAudio setSampleRateFromName: [dict objectForKey: @"AudioSamplerate"]];
 266                                 [newAudio setBitRateFromName: [dict objectForKey: @"AudioBitrate"]];
 267                                 [newAudio setDrc: [dict objectForKey: @"AudioTrackDRCSlider"]];
 268                         }
 269                         else {
 270                                 [self removeObjectFromAudioArrayAtIndex: [self countOfAudioArray] - 1];
 271                         }
 272                         [newAudio release];
 273                 }
 274         }
 275         return;
 276 }
 277
 278 //      This matches the FIRST track with the specified prefix, otherwise it uses the defaultIfNotFound value
 279 - (unsigned int) _trackWithTitlePrefix: (NSString *) prefix defaultIfNotFound: (unsigned int) defaultIfNotFound
 280
 281 {
 282         unsigned int retval = defaultIfNotFound;
 283         int count = [masterTrackArray count];
 284         NSString *languageTitle;
 285         BOOL found = NO;
 286
 287         //      We search for the prefix noting that our titles have the format %d: %s where the %s is the prefix
 288         for (unsigned int i = 1; i < count && NO == found; i++) {       //      Note that we skip the "None" track
 289                 languageTitle = [[masterTrackArray objectAtIndex: i] objectForKey: keyAudioTrackName];
 290                 if (YES == [[languageTitle substringFromIndex: [languageTitle rangeOfString: @" "].location + 1] hasPrefix: prefix]) {
 291                         retval = i;
 292                         found = YES;
 293                 }
 294         }
 295         return retval;
 296 }
 297
 298 - (void) addTracksFromPreset: (NSMutableDictionary *) aPreset allTracks: (BOOL) allTracks
 299
 300 {
 301         id whatToUse = [self _presetAudioArrayFromPreset: aPreset];
 302         NSString *preferredLanguageName = [[NSUserDefaults standardUserDefaults] stringForKey: @"DefaultLanguage"];
 303         int preferredLanguage = [self _trackWithTitlePrefix: preferredLanguageName defaultIfNotFound: 1];
 304
 305         //      Reinitialize the configured list of audio tracks
 306         [self _clearAudioArray];
 307
 308         [self _processPresetAudioArray: whatToUse forTrack: preferredLanguage andType: [[aPreset objectForKey: @"Type"] intValue]];
 309         if (YES == allTracks) {
 310                 unsigned int count = [masterTrackArray count];
 311                 for (unsigned int i = 1; i < count; i++) {
 312                         if (i != preferredLanguage) {
 313                                 [self _processPresetAudioArray: whatToUse forTrack: i andType: [[aPreset objectForKey: @"Type"] intValue]];
 314                         }
 315                 }
 316         }
 317
 318         [self switchingTrackFromNone: nil];     // see if we need to add one to the list
 319
 320         return;
 321 }
 322
 323 - (void) addTracksFromPreset: (NSMutableDictionary *) aPreset
 324
 325 {
 326         [self addTracksFromPreset: aPreset allTracks: NO];
 327         return;
 328 }
 329
 330 - (void) addAllTracksFromPreset: (NSMutableDictionary *) aPreset
 331
 332 {
 333         [self addTracksFromPreset: aPreset allTracks: YES];
 334         return;
 335 }
 336
 337 - (BOOL) anyCodecMatches: (int) aCodecValue
 338
 339 {
 340         BOOL retval = NO;
 341         unsigned int audioArrayCount = [self countOfAudioArray];
 342         for (unsigned int i = 0; i < audioArrayCount && NO == retval; i++) {
 343                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: i];
 344         if (YES == [anAudio enabled] && aCodecValue == [[[anAudio codec] objectForKey: keyAudioCodec] intValue]) {
 345                         retval = YES;
 346                 }
 347         }
 348         return retval;
 349 }
 350
 351 - (void) addNewAudioTrack
 352
 353 {
 354         HBAudio *newAudio = [[HBAudio alloc] init];
 355         [newAudio setController: self];
 356         [self insertObject: newAudio inAudioArrayAtIndex: [self countOfAudioArray]];
 357         [newAudio setVideoContainerTag: [self videoContainerTag]];
 358         [newAudio setTrack: noneTrack];
 359         [newAudio setDrc: [NSNumber numberWithFloat: 0.0]];
 360         [newAudio release];
 361         return;
 362 }
 363
 364 #pragma mark -
 365 #pragma mark Notification Handling
 366
 367 - (void) settingTrackToNone: (HBAudio *) newNoneTrack
 368
 369 {
 370         //      If this is not the last track in the array we need to remove it.  We then need to see if a new
 371         //      one needs to be added (in the case when we were at maximum count and this switching makes it
 372         //      so we are no longer at maximum.
 373         unsigned int index = [audioArray indexOfObject: newNoneTrack];
 374
 375         if (NSNotFound != index && index < [self countOfAudioArray] - 1) {
 376                 [self removeObjectFromAudioArrayAtIndex: index];
 377         }
 378         [self switchingTrackFromNone: nil];     // see if we need to add one to the list
 379         return;
 380 }
 381
 382 - (void) switchingTrackFromNone: (HBAudio *) noLongerNoneTrack
 383
 384 {
 385         int count = [self countOfAudioArray];
 386         BOOL needToAdd = NO;
 387         int maximumNumberOfAllowedAudioTracks = [HBController maximumNumberOfAllowedAudioTracks];
 388
 389         //      If there is no last track that is None and we are less than our maximum number of permitted tracks, we add one.
 390         if (count < maximumNumberOfAllowedAudioTracks) {
 391                 if (0 < count) {
 392                         HBAudio *lastAudio = [self objectInAudioArrayAtIndex: count - 1];
 393                         if (YES == [lastAudio enabled]) {
 394                                 needToAdd = YES;
 395                         }
 396                 }
 397                 else {
 398                         needToAdd = YES;
 399                 }
 400         }
 401
 402         if (YES == needToAdd) {
 403                 [self addNewAudioTrack];
 404         }
 405         return;
 406 }
 407
 408 //      This gets called whenever the video container changes.
 409 - (void) containerChanged: (NSNotification *) aNotification
 410
 411 {
 412         NSDictionary *notDict = [aNotification userInfo];
 413
 414         [self setVideoContainerTag: [notDict objectForKey: keyContainerTag]];
 415
 416         //      Update each of the instances because this value influences possible settings.
 417         NSEnumerator *enumerator = [audioArray objectEnumerator];
 418         HBAudio *audioObject;
 419
 420         while (nil != (audioObject = [enumerator nextObject])) {
 421                 [audioObject setVideoContainerTag: [self videoContainerTag]];
 422         }
 423         return;
 424 }
 425
 426 - (void) titleChanged: (NSNotification *) aNotification
 427
 428 {
 429         NSDictionary *notDict = [aNotification userInfo];
 430         NSData *theData = [notDict objectForKey: keyTitleTag];
 431         hb_title_t *title = NULL;
 432
 433         [theData getBytes: &title length: sizeof(title)];
 434         if (title) {
 435                 hb_audio_config_t *audio;
 436                 hb_list_t *list = title->list_audio;
 437                 int i, count = hb_list_count(list);
 438
 439                 //      Reinitialize the master list of available audio tracks from this title
 440                 [masterTrackArray release];
 441                 masterTrackArray = [[NSMutableArray alloc] init];
 442                 [noneTrack release];
 443                 noneTrack = [[NSDictionary dictionaryWithObjectsAndKeys:
 444                                          [NSNumber numberWithInt: 0], keyAudioTrackIndex,
 445                                          NSLocalizedString(@"None", @"None"), keyAudioTrackName,
 446                                          [NSNumber numberWithInt: 0], keyAudioInputCodec,
 447                                                          nil] retain];
 448                 [masterTrackArray addObject: noneTrack];
 449                 for (i = 0; i < count; i++) {
 450                         audio = (hb_audio_config_t *) hb_list_audio_config_item(list, i);
 451                         [masterTrackArray addObject: [NSDictionary dictionaryWithObjectsAndKeys:
 452                                                                                   [NSNumber numberWithInt: i + 1], keyAudioTrackIndex,
 453                                                                                   [NSString stringWithFormat: @"%d: %s", i, audio->lang.description], keyAudioTrackName,
 454                                                                                   [NSNumber numberWithInt: audio->in.bitrate / 1000], keyAudioInputBitrate,
 455                                                                                   [NSNumber numberWithInt: audio->in.samplerate], keyAudioInputSampleRate,
 456                                                                                   [NSNumber numberWithInt: audio->in.codec], keyAudioInputCodec,
 457                                                                                   [NSNumber numberWithInt: audio->in.channel_layout], keyAudioInputChannelLayout,
 458                                                                                   nil]];
 459                 }
 460         }
 461
 462         //      Reinitialize the configured list of audio tracks
 463         [self _clearAudioArray];
 464
 465         if (NO == [myController hasValidPresetSelected]) {
 466                 NSString *preferredLanguageName = [[NSUserDefaults standardUserDefaults] stringForKey: @"DefaultLanguage"];
 467                 int preferredLanguage = [self _trackWithTitlePrefix: preferredLanguageName defaultIfNotFound: 1];
 468                 [self addNewAudioTrack];
 469                 HBAudio *anAudio = [self objectInAudioArrayAtIndex: 0];
 470                 [anAudio setTrackFromIndex: preferredLanguage];
 471         }
 472         return;
 473 }
 474
 475 #pragma mark -
 476 #pragma mark KVC
 477
 478 - (unsigned int) countOfAudioArray
 479
 480 {
 481         return [audioArray count];
 482 }
 483
 484 - (HBAudio *) objectInAudioArrayAtIndex: (unsigned int) index
 485
 486 {
 487         return [audioArray objectAtIndex: index];
 488 }
 489
 490 - (void) insertObject: (HBAudio *) audioObject inAudioArrayAtIndex: (unsigned int) index;
 491
 492 {
 493         [audioArray insertObject: audioObject atIndex: index];
 494         return;
 495 }
 496
 497 - (void) removeObjectFromAudioArrayAtIndex: (unsigned int) index
 498
 499 {
 500         [audioArray removeObjectAtIndex: index];
 501         return;
 502 }
 503
 504 @end