/* $Id: Controller.mm,v 1.79 2005/11/04 19:41:32 titer Exp $ This file is part of the HandBrake source code. Homepage: . It may be used under the terms of the GNU General Public License. */ #import "HBController.h" #import "HBFocusRingView.h" #import "HBToolbarBadgedItem.h" #import "HBQueueController.h" #import "HBTitleSelectionController.h" #import "HBPresetsManager.h" #import "HBPreset.h" #import "HBMutablePreset.h" #import "HBPictureViewController.h" #import "HBVideoController.h" #import "HBAudioController.h" #import "HBSubtitlesController.h" #import "HBAdvancedController.h" #import "HBChapterTitlesController.h" #import "HBPreviewController.h" #import "HBPreviewGenerator.h" #import "HBPresetsViewController.h" #import "HBAddPresetController.h" @import HandBrakeKit; @interface HBController () { IBOutlet NSTabView *fMainTabView; // Picture controller HBPictureViewController * fPictureViewController; IBOutlet NSTabViewItem * fPictureTab; // Video view controller HBVideoController * fVideoController; IBOutlet NSTabViewItem * fVideoTab; // Subtitles view controller HBSubtitlesController * fSubtitlesViewController; IBOutlet NSTabViewItem * fSubtitlesTab; // Audio view controller HBAudioController * fAudioController; IBOutlet NSTabViewItem * fAudioTab; // Chapters view controller HBChapterTitlesController * fChapterTitlesController; IBOutlet NSTabViewItem * fChaptersTitlesTab; // Advanced options tab HBAdvancedController * fAdvancedOptions; IBOutlet NSTabViewItem * fAdvancedTab; // Picture Preview HBPreviewController * fPreviewController; // Queue panel HBQueueController * fQueueController; // Source box IBOutlet NSProgressIndicator * fScanIndicator; IBOutlet NSBox * fScanHorizontalLine; IBOutlet NSTextField * fSrcDVD2Field; IBOutlet NSPopUpButton * fSrcTitlePopUp; // pts based start / stop IBOutlet NSTextField * fSrcTimeStartEncodingField; IBOutlet NSTextField * fSrcTimeEndEncodingField; // frame based based start / stop IBOutlet NSTextField * fSrcFrameStartEncodingField; IBOutlet NSTextField * fSrcFrameEndEncodingField; IBOutlet NSPopUpButton * fSrcChapterStartPopUp; IBOutlet NSPopUpButton * fSrcChapterEndPopUp; // Bottom IBOutlet NSTextField * fStatusField; IBOutlet NSProgressIndicator * fRipIndicator; BOOL fRipIndicatorShown; // User Preset HBPresetsManager * presetManager; HBPresetsViewController * fPresetsView; IBOutlet NSDrawer * fPresetDrawer; } @property (nonatomic, weak) IBOutlet HBToolbarBadgedItem *showQueueToolbarItem; @property (unsafe_unretained) IBOutlet NSView *openTitleView; @property (nonatomic, readwrite) BOOL scanSpecificTitle; @property (nonatomic, readwrite) NSInteger scanSpecificTitleIdx; @property (nonatomic, readwrite, strong) HBTitleSelectionController *titlesSelectionController; /// The current job. @property (nonatomic, strong, nullable) HBJob *job; /// The current selected preset. @property (nonatomic, strong) HBPreset *currentPreset; /// The current destination. @property (nonatomic, strong) NSURL *currentDestination; /// Whether the job has been edited after a preset was applied. @property (nonatomic) BOOL edited; /// The HBCore used for scanning. @property (nonatomic, strong) HBCore *core; @property (nonatomic, readwrite) NSColor *labelColor; @end @implementation HBController - (instancetype)initWithQueue:(HBQueueController *)queueController presetsManager:(HBPresetsManager *)manager; { self = [super initWithWindowNibName:@"MainWindow"]; if (self) { // Init libhb int loggingLevel = [[[NSUserDefaults standardUserDefaults] objectForKey:@"LoggingLevel"] intValue]; _core = [[HBCore alloc] initWithLogLevel:loggingLevel name:@"ScanCore"]; // Inits the controllers fPreviewController = [[HBPreviewController alloc] init]; fPreviewController.documentController = self; fQueueController = queueController; fQueueController.controller = self; presetManager = manager; if (manager.defaultPreset.isBuiltIn) { _currentPreset = [self presetByAddingDefaultLanguages:manager.defaultPreset]; } else { _currentPreset = manager.defaultPreset; } _scanSpecificTitleIdx = 1; // Check to see if the last destination has been set, use if so, if not, use Movies #ifdef __SANDBOX_ENABLED__ NSData *bookmark = [[NSUserDefaults standardUserDefaults] objectForKey:@"HBLastDestinationDirectoryBookmark"]; if (bookmark) { _currentDestination = [HBUtilities URLFromBookmark:bookmark]; } #else _currentDestination = [[NSUserDefaults standardUserDefaults] URLForKey:@"HBLastDestinationDirectoryURL"]; #endif if (!_currentDestination) { _currentDestination = [NSURL fileURLWithPath:[NSSearchPathForDirectoriesInDomains(NSMoviesDirectory, NSUserDomainMask, YES) firstObject] isDirectory:YES]; } #ifdef __SANDBOX_ENABLED__ [_currentDestination startAccessingSecurityScopedResource]; #endif } return self; } - (void)dealloc { [[NSNotificationCenter defaultCenter] removeObserver:self]; } - (void)windowDidLoad { [self enableUI:NO]; /* For 64 bit builds, the threaded animation in the progress * indicators conflicts with the animation in the advanced tab * for reasons not completely clear. jbrjake found a note in the * 10.5 dev notes regarding this possiblility. It was also noted * that unless specified, setUsesThreadedAnimation defaults to true. * So, at least for now we set the indicator animation to NO for * both the scan and regular progress indicators for both 32 and 64 bit * as it test out fine on both and there is no reason our progress indicators * should require their own thread. */ [fScanIndicator setUsesThreadedAnimation:NO]; [fRipIndicator setUsesThreadedAnimation:NO]; NSSize drawerSize = NSSizeFromString([[NSUserDefaults standardUserDefaults] stringForKey:@"HBDrawerSize"]); if (drawerSize.width > 0) { [fPresetDrawer setContentSize: drawerSize]; } // Show/Hide the Presets drawer upon launch based // on user preference DefaultPresetsDrawerShow if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBDefaultPresetsDrawerShow"]) { [fPresetDrawer open:self]; } // Align the start / stop widgets with the chapter popups NSPoint startPoint = [fSrcChapterStartPopUp frame].origin; startPoint.y += 2; NSPoint endPoint = [fSrcChapterEndPopUp frame].origin; endPoint.y += 2; [fSrcTimeStartEncodingField setFrameOrigin:startPoint]; [fSrcTimeEndEncodingField setFrameOrigin:endPoint]; [fSrcFrameStartEncodingField setFrameOrigin:startPoint]; [fSrcFrameEndEncodingField setFrameOrigin:endPoint]; // Bottom [fStatusField setStringValue:@""]; // Register HBController's Window as a receiver for files/folders drag & drop operations [self.window registerForDraggedTypes:@[NSFilenamesPboardType]]; [fMainTabView registerForDraggedTypes:@[NSFilenamesPboardType]]; // Set up the preset drawer fPresetsView = [[HBPresetsViewController alloc] initWithPresetManager:presetManager]; [fPresetDrawer setContentView:[fPresetsView view]]; fPresetsView.delegate = self; [[fPresetDrawer contentView] setAutoresizingMask:( NSViewWidthSizable | NSViewHeightSizable )]; // Set up the chapters title view fChapterTitlesController = [[HBChapterTitlesController alloc] init]; [fChaptersTitlesTab setView:[fChapterTitlesController view]]; // setup the subtitles view fSubtitlesViewController = [[HBSubtitlesController alloc] init]; [fSubtitlesTab setView:[fSubtitlesViewController view]]; // setup the audio controller fAudioController = [[HBAudioController alloc] init]; [fAudioTab setView:[fAudioController view]]; // setup the advanced view controller fAdvancedOptions = [[HBAdvancedController alloc] init]; [fAdvancedTab setView:[fAdvancedOptions view]]; // setup the video view controller fVideoController = [[HBVideoController alloc] initWithAdvancedController:fAdvancedOptions]; [fVideoTab setView:[fVideoController view]]; // setup the picture view controller fPictureViewController = [[HBPictureViewController alloc] init]; [fPictureTab setView:[fPictureViewController view]]; [[NSUserDefaultsController sharedUserDefaultsController] addObserver:self forKeyPath:@"values.HBShowAdvancedTab" options:NSKeyValueObservingOptionNew | NSKeyValueObservingOptionInitial context:NULL]; [self.window recalculateKeyViewLoop]; } #pragma mark - #pragma mark Drag & drop handling - (nullable NSArray *)fileURLsFromPasteboard:(NSPasteboard *)pasteboard { NSDictionary *options = @{NSPasteboardURLReadingFileURLsOnlyKey: @YES}; return [pasteboard readObjectsForClasses:@[[NSURL class]] options:options]; } - (NSDragOperation)draggingEntered:(id )sender { NSArray *fileURLs = [self fileURLsFromPasteboard:[sender draggingPasteboard]]; [self.window.contentView setShowFocusRing:YES]; return fileURLs.count ? NSDragOperationGeneric : NSDragOperationNone; } - (BOOL)prepareForDragOperation:(id)sender { return YES; } - (BOOL)performDragOperation:(id )sender { NSArray *fileURLs = [self fileURLsFromPasteboard:[sender draggingPasteboard]]; if (fileURLs.count) { [self openURL:fileURLs.firstObject]; } [self.window.contentView setShowFocusRing:NO]; return YES; } - (void)draggingExited:(nullable id )sender { [self.window.contentView setShowFocusRing:NO]; } #pragma mark - KVO - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context { if ([keyPath isEqualToString:@"values.HBShowAdvancedTab"]) { if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBShowAdvancedTab"]) { if (![[fMainTabView tabViewItems] containsObject:fAdvancedTab]) { [fMainTabView insertTabViewItem:fAdvancedTab atIndex:5]; } } else { [fMainTabView removeTabViewItem:fAdvancedTab]; } } else { [super observeValueForKeyPath:keyPath ofObject:object change:change context:context]; } } - (void)enableUI:(BOOL)enabled { if (enabled) { self.labelColor = [NSColor controlTextColor]; } else { self.labelColor = [NSColor disabledControlTextColor]; } fPresetsView.enabled = enabled; } - (NSSize)drawerWillResizeContents:(NSDrawer *) drawer toSize:(NSSize)contentSize { [[NSUserDefaults standardUserDefaults] setObject:NSStringFromSize(contentSize) forKey:@"HBDrawerSize"]; return contentSize; } - (void)setNilValueForKey:(NSString *)key { if ([key isEqualToString:@"scanSpecificTitleIdx"]) { [self setValue:@0 forKey:key]; } } #pragma mark - UI Validation - (BOOL)validateToolbarItem:(NSToolbarItem *)toolbarItem { SEL action = toolbarItem.action; if (self.core.state == HBStateScanning) { if (action == @selector(browseSources:)) { [toolbarItem setImage: [NSImage imageNamed: @"stopencode"]]; [toolbarItem setLabel: NSLocalizedString(@"Cancel Scan", nil)]; [toolbarItem setPaletteLabel: NSLocalizedString(@"Cancel Scanning", nil)]; [toolbarItem setToolTip: NSLocalizedString(@"Cancel Scanning Source", nil)]; return YES; } if (action == @selector(rip:) || action == @selector(addToQueue:)) return NO; } else { if (action == @selector(browseSources:)) { [toolbarItem setImage:[NSImage imageNamed:@"source"]]; [toolbarItem setLabel:NSLocalizedString(@"Open Source", nil)]; [toolbarItem setPaletteLabel:NSLocalizedString(@"Open Source", nil)]; [toolbarItem setToolTip:NSLocalizedString(@"Open Source", nil)]; return YES; } } HBState queueState = fQueueController.core.state; if (queueState == HBStateScanning || queueState == HBStateWorking || queueState == HBStateSearching || queueState == HBStateMuxing) { if (action == @selector(rip:)) { [toolbarItem setImage: [NSImage imageNamed: @"stopencode"]]; [toolbarItem setLabel: NSLocalizedString(@"Stop", nil)]; [toolbarItem setPaletteLabel: NSLocalizedString(@"Stop", nil)]; [toolbarItem setToolTip: NSLocalizedString(@"Stop Encoding", nil)]; return YES; } if (action == @selector(pause:)) { [toolbarItem setImage: [NSImage imageNamed: @"pauseencode"]]; [toolbarItem setLabel: NSLocalizedString(@"Pause", nil)]; [toolbarItem setPaletteLabel: NSLocalizedString(@"Pause Encoding", nil)]; [toolbarItem setToolTip: NSLocalizedString(@"Pause Encoding", nil)]; return YES; } } else if (queueState == HBStatePaused) { if (action == @selector(pause:)) { [toolbarItem setImage: [NSImage imageNamed: @"encode"]]; [toolbarItem setLabel: NSLocalizedString(@"Resume", nil)]; [toolbarItem setPaletteLabel: NSLocalizedString(@"Resume Encoding", nil)]; [toolbarItem setToolTip: NSLocalizedString(@"Resume Encoding", nil)]; return YES; } if (action == @selector(rip:)) return YES; } else { if (action == @selector(rip:)) { [toolbarItem setImage: [NSImage imageNamed: @"encode"]]; if (fQueueController.pendingItemsCount > 0) [toolbarItem setLabel: NSLocalizedString(@"Start Queue", nil)]; else [toolbarItem setLabel: NSLocalizedString(@"Start", nil)]; [toolbarItem setPaletteLabel: NSLocalizedString(@"Start Encoding", nil)]; [toolbarItem setToolTip: NSLocalizedString(@"Start Encoding", nil)]; } if (action == @selector(rip:)) { return (self.job != nil || fQueueController.pendingItemsCount > 0); } if (action == @selector(pause:)) { return NO; } } if (action == @selector(addToQueue:)) { return (self.job != nil); } return YES; } - (BOOL)validateMenuItem:(NSMenuItem *)menuItem { SEL action = menuItem.action; if (action == @selector(addToQueue:) || action == @selector(addAllTitlesToQueue:) || action == @selector(addTitlesToQueue:) || action == @selector(showAddPresetPanel:)) { return self.job && self.window.attachedSheet == nil; } if (action == @selector(selectDefaultPreset:)) { return self.window.attachedSheet == nil; } if (action == @selector(pause:)) { return [fQueueController validateMenuItem:menuItem]; } if (action == @selector(rip:)) { BOOL result = [fQueueController validateMenuItem:menuItem]; if ([menuItem.title isEqualToString:NSLocalizedString(@"Start Encoding", nil)]) { if (!result && self.job) { return YES; } } return result; } if (action == @selector(browseSources:)) { if (self.core.state == HBStateScanning) { return NO; } else { return self.window.attachedSheet == nil; } } if (action == @selector(selectPresetFromMenu:)) { if ([menuItem.representedObject isEqualTo:self.currentPreset] && self.edited == NO) { menuItem.state = NSOnState; } else { menuItem.state = NSOffState; } return (self.job != nil); } if (action == @selector(exportPreset:)) { return [fPresetsView validateUserInterfaceItem:menuItem]; } return YES; } #pragma mark - Get New Source - (void)launchAction { if (self.core.state != HBStateScanning && !self.job) { if ([[NSUserDefaults standardUserDefaults] boolForKey:@"HBShowOpenPanelAtLaunch"]) { [self browseSources:nil]; } } } - (NSModalResponse)runCopyProtectionAlert { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"Copy-Protected sources are not supported.", nil)]; [alert setInformativeText:NSLocalizedString(@"Please note that HandBrake does not support the removal of copy-protection from DVD Discs. You can if you wish use any other 3rd party software for this function.", nil)]; [alert addButtonWithTitle:NSLocalizedString(@"Attempt Scan Anyway", nil)]; [alert addButtonWithTitle:NSLocalizedString(@"Cancel", nil)]; [NSApp requestUserAttention:NSCriticalRequest]; return [alert runModal]; } /** * Here we actually tell hb_scan to perform the source scan, using the path to source and title number */ - (void)scanURL:(NSURL *)fileURL titleIndex:(NSUInteger)index completionHandler:(void(^)(NSArray *titles))completionHandler { // Save the current settings if (self.job) { self.currentPreset = [self createPresetFromCurrentSettings]; } self.job = nil; [fSrcTitlePopUp removeAllItems]; self.window.representedURL = nil; self.window.title = NSLocalizedString(@"HandBrake", nil); NSURL *mediaURL = [HBUtilities mediaURLFromURL:fileURL]; NSString *displayName = [HBUtilities displayNameForURL:fileURL]; NSError *outError = NULL; BOOL suppressWarning = [[NSUserDefaults standardUserDefaults] boolForKey:@"suppressCopyProtectionAlert"]; // Check if we can scan the source and if there is any warning. BOOL canScan = [self.core canScan:mediaURL error:&outError]; // Notify the user that we don't support removal of copy proteciton. if (canScan && [outError code] == 101 && !suppressWarning) { // Only show the user this warning once. They may be using a solution we don't know about. Notifying them each time is annoying. [[NSUserDefaults standardUserDefaults] setBool:YES forKey:@"suppressCopyProtectionAlert"]; if ([self runCopyProtectionAlert] == NSAlertFirstButtonReturn) { // User chose to override our warning and scan the physical dvd anyway, at their own peril. on an encrypted dvd this produces massive log files and fails [HBUtilities writeToActivityLog:"User overrode copy-protection warning - trying to open physical dvd without decryption"]; } else { // User chose to cancel the scan [HBUtilities writeToActivityLog:"Cannot open physical dvd, scan cancelled"]; canScan = NO; } } if (canScan) { int hb_num_previews = [[[NSUserDefaults standardUserDefaults] objectForKey:@"PreviewsNumber"] intValue]; int min_title_duration_seconds = [[[NSUserDefaults standardUserDefaults] objectForKey:@"MinTitleScanSeconds"] intValue]; [self.core scanURL:mediaURL titleIndex:index previews:hb_num_previews minDuration:min_title_duration_seconds progressHandler:^(HBState state, HBProgress progress, NSString *info) { fSrcDVD2Field.stringValue = info; fScanIndicator.hidden = NO; fScanHorizontalLine.hidden = YES; fScanIndicator.doubleValue = progress.percent; } completionHandler:^(HBCoreResult result) { fScanHorizontalLine.hidden = NO; fScanIndicator.hidden = YES; fScanIndicator.indeterminate = NO; fScanIndicator.doubleValue = 0.0; if (result == HBCoreResultDone) { for (HBTitle *title in self.core.titles) { [fSrcTitlePopUp addItemWithTitle:title.description]; } // Set Source Name at top of window with the browsedSourceDisplayName grokked right before -performScan fSrcDVD2Field.stringValue = displayName; self.window.representedURL = mediaURL; self.window.title = mediaURL.lastPathComponent; completionHandler(self.core.titles); } else { // We display a message if a valid source was not chosen fSrcDVD2Field.stringValue = NSLocalizedString(@"No Valid Source Found", @""); } [self.window.toolbar validateVisibleItems]; }]; } } - (void)openURL:(NSURL *)fileURL titleIndex:(NSUInteger)index { [self scanURL:fileURL titleIndex:index completionHandler:^(NSArray *titles) { [[NSDocumentController sharedDocumentController] noteNewRecentDocumentURL:fileURL]; HBTitle *featuredTitle = titles.firstObject; for (HBTitle *title in titles) { if (title.isFeatured) { featuredTitle = title; } } HBJob *job = [self jobFromTitle:featuredTitle]; self.job = job; }]; } - (BOOL)openURL:(NSURL *)fileURL { if (self.core.state != HBStateScanning) { [self openURL:fileURL titleIndex:0]; return YES; } return NO; } /** * Rescans the a job back into the main window */ - (BOOL)openJob:(HBJob *)job { if (self.core.state != HBStateScanning) { [self scanURL:job.fileURL titleIndex:job.titleIdx completionHandler:^(NSArray *titles) { // If the scan was cached, reselect // the original title for (HBTitle *title in titles) { if (title.index == job.titleIdx) { job.title = title; break; } } // Else just one title or a title specific rescan // select the first title if (!job.title) { job.title = titles.firstObject; } self.job = job; }]; return YES; } return NO; } - (HBJob *)jobFromTitle:(HBTitle *)title { // If there is already a title load, save the current settings to a preset if (self.job) { self.currentPreset = [self createPresetFromCurrentSettings]; } HBJob *job = [[HBJob alloc] initWithTitle:title andPreset:self.currentPreset]; job.outputURL = self.currentDestination; job.outputFileName = [HBUtilities defaultNameForJob:job]; return job; } - (void)removeJobObservers { if (self.job) { [[NSNotificationCenter defaultCenter] removeObserver:self name:HBContainerChangedNotification object:_job]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBPictureChangedNotification object:_job.picture]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBFiltersChangedNotification object:_job.filters]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBVideoChangedNotification object:_job.video]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBAudioChangedNotification object:_job.audio]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBChaptersChangedNotification object:_job]; [[NSNotificationCenter defaultCenter] removeObserver:self name:HBRangeChangedNotification object:_job.range]; } } /** * Observe the job settings changes. * This is used to update the file name and extention * and the custom preset string. */ - (void)addJobObservers { if (self.job) { [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(formatChanged:) name:HBContainerChangedNotification object:_job]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(customSettingUsed) name:HBPictureChangedNotification object:_job.picture]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(customSettingUsed) name:HBFiltersChangedNotification object:_job.filters]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(customSettingUsed) name:HBVideoChangedNotification object:_job.video]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(updateFileExtension:) name:HBAudioChangedNotification object:_job.audio]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(updateFileExtension:) name:HBChaptersChangedNotification object:_job]; [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(chapterPopUpChanged:) name:HBRangeChangedNotification object:_job.range]; } } - (void)setJob:(HBJob *)job { [self removeJobObservers]; // Clear the undo manager [_job.undo removeAllActions]; _job.undo = nil; // Retain the new job _job = job; job.undo = self.window.undoManager; // Set the jobs info to the view controllers fPictureViewController.picture = job.picture; fPictureViewController.filters = job.filters; fVideoController.video = job.video; fAudioController.audio = job.audio; fSubtitlesViewController.subtitles = job.subtitles; fChapterTitlesController.job = job; if (job) { fPreviewController.generator = [[HBPreviewGenerator alloc] initWithCore:self.core job:job]; HBTitle *title = job.title; // Update the title selection popup. [fSrcTitlePopUp selectItemWithTitle:title.description]; // If we are a stream type and a batch scan, grok the output file name from title->name upon title change if (title.isStream && self.core.titles.count > 1) { // Change the source to read out the parent folder also fSrcDVD2Field.stringValue = [NSString stringWithFormat:@"%@/%@", title.url.URLByDeletingLastPathComponent.lastPathComponent, title.name]; } } else { fPreviewController.generator = nil; } fPreviewController.picture = job.picture; [self enableUI:(job != nil)]; [self addJobObservers]; } /** * Opens the source browse window, called from Open Source widgets */ - (IBAction)browseSources:(id)sender { if (self.core.state == HBStateScanning) { [self.core cancelScan]; return; } NSOpenPanel *panel = [NSOpenPanel openPanel]; [panel setAllowsMultipleSelection:NO]; [panel setCanChooseFiles:YES]; [panel setCanChooseDirectories:YES]; NSURL *sourceDirectory; if ([[NSUserDefaults standardUserDefaults] URLForKey:@"HBLastSourceDirectoryURL"]) { sourceDirectory = [[NSUserDefaults standardUserDefaults] URLForKey:@"HBLastSourceDirectoryURL"]; } else { sourceDirectory = [NSURL fileURLWithPath:[NSSearchPathForDirectoriesInDomains(NSDesktopDirectory, NSUserDomainMask, YES) firstObject] isDirectory:YES]; } [panel setDirectoryURL:sourceDirectory]; [panel setAccessoryView:self.openTitleView]; if ([panel respondsToSelector:@selector(isAccessoryViewDisclosed)]) { panel.accessoryViewDisclosed = YES; } [panel beginSheetModalForWindow:self.window completionHandler: ^(NSInteger result) { if (result == NSFileHandlingPanelOKButton) { // Set the last searched source directory in the prefs here [[NSUserDefaults standardUserDefaults] setURL:panel.URL.URLByDeletingLastPathComponent forKey:@"HBLastSourceDirectoryURL"]; NSInteger titleIdx = self.scanSpecificTitle ? self.scanSpecificTitleIdx : 0; [self openURL:panel.URL titleIndex:titleIdx]; } }]; } #pragma mark - GUI Controls Changed Methods - (IBAction)browseDestination:(id)sender { // Open a panel to let the user choose and update the text field NSOpenPanel *panel = [NSOpenPanel openPanel]; panel.canChooseFiles = NO; panel.canChooseDirectories = YES; panel.prompt = NSLocalizedString(@"Choose", nil); if (self.job.outputURL) { panel.directoryURL = self.job.outputURL; } [panel beginSheetModalForWindow:self.window completionHandler:^(NSInteger result) { if (result == NSFileHandlingPanelOKButton) { self.job.outputURL = panel.URL; self.currentDestination = panel.URL; // Save this path to the prefs so that on next browse destination window it opens there [[NSUserDefaults standardUserDefaults] setObject:[HBUtilities bookmarkFromURL:panel.URL] forKey:@"HBLastDestinationDirectoryBookmark"]; [[NSUserDefaults standardUserDefaults] setURL:panel.URL forKey:@"HBLastDestinationDirectoryURL"]; } }]; } - (IBAction)titlePopUpChanged:(NSPopUpButton *)sender { HBTitle *title = self.core.titles[sender.indexOfSelectedItem]; HBJob *job = [self jobFromTitle:title]; self.job = job; } - (void)chapterPopUpChanged:(NSNotification *)notification { // We're changing the chapter range - we may need to flip the m4v/mp4 extension if (self.job.container & 0x030000 /*HB_MUX_MASK_MP4*/) { [self updateFileExtension:notification]; } // If Auto Naming is on it might need to be update if it includes the chapters range [self updateFileName]; } - (void)formatChanged:(NSNotification *)notification { [self updateFileExtension:notification]; [self customSettingUsed]; } - (void)updateFileName { [self updateFileExtension:nil]; if ([[NSUserDefaults standardUserDefaults] boolForKey:@"DefaultAutoNaming"] && self.job) { // Generate a new file name NSString *fileName = [HBUtilities automaticNameForJob:self.job]; // Swap the old one with the new one self.job.outputFileName = [NSString stringWithFormat:@"%@.%@", fileName, self.job.outputFileName.pathExtension]; } } - (void)updateFileExtension:(NSNotification *)notification { if (self.job) { NSString *extension = [HBUtilities automaticExtForJob:self.job]; if (![extension isEqualTo:self.job.outputFileName.pathExtension]) { self.job.outputFileName = [[self.job.outputFileName stringByDeletingPathExtension] stringByAppendingPathExtension:extension]; } } } /** * Method to determine if we should change the UI * To reflect whether or not a Preset is being used or if * the user is using "Custom" settings by determining the sender */ - (void)customSettingUsed { // Deselect the currently selected Preset if there is one [fPresetsView deselect]; // Update the preset and file name only if we are not // undoing or redoing, because if so it's already stored // in the undo manager. NSUndoManager *undo = self.window.undoManager; if (!(undo.isUndoing || undo.isRedoing)) { // Change UI to show "Custom" settings are being used self.job.presetName = NSLocalizedString(@"Custom", @""); self.edited = YES; [self updateFileName]; } } #pragma mark - Queue progress - (void)setQueueState:(NSUInteger)count { self.showQueueToolbarItem.badgeValue = count ? @(count).stringValue : nil; } #define WINDOW_HEIGHT 591 #define WINDOW_HEIGHT_OFFSET 36 - (void)setQueueInfo:(NSString *)info progress:(double)progress hidden:(BOOL)hidden { fStatusField.stringValue = info; fRipIndicator.doubleValue = progress; if (hidden) { if (fRipIndicatorShown) { NSRect frame = self.window.frame; if (frame.size.width <= WINDOW_HEIGHT) frame.size.width = WINDOW_HEIGHT; frame.size.height += -WINDOW_HEIGHT_OFFSET; frame.origin.y -= -WINDOW_HEIGHT_OFFSET; [self.window setFrame:frame display:YES animate:YES]; fRipIndicatorShown = NO; // Refresh the toolbar buttons [self.window.toolbar validateVisibleItems]; } } else { // If progress bar hasn't been revealed at the bottom of the window, do // that now. if (!fRipIndicatorShown) { NSRect frame = self.window.frame; if (frame.size.width <= WINDOW_HEIGHT) frame.size.width = WINDOW_HEIGHT; frame.size.height += WINDOW_HEIGHT_OFFSET; frame.origin.y -= WINDOW_HEIGHT_OFFSET; [self.window setFrame:frame display:YES animate:YES]; fRipIndicatorShown = YES; // Refresh the toolbar buttons [self.window.toolbar validateVisibleItems]; } } } #pragma mark - Job Handling /** Check if the job destination if a valid one, if so, call the didEndSelector Note: rework this to use a block in the future @param job the job @param didEndSelector the selector to call if the check is successful */ - (void)runDestinationAlerts:(HBJob *)job didEndSelector:(SEL)didEndSelector { if ([[NSFileManager defaultManager] fileExistsAtPath:job.outputURL.path] == 0) { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"Warning!", @"")]; [alert setInformativeText:NSLocalizedString(@"This is not a valid destination directory!", @"")]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:didEndSelector contextInfo:NULL]; } else if ([job.fileURL isEqual:job.completeOutputURL]) { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"A file already exists at the selected destination.", @"")]; [alert setInformativeText:NSLocalizedString(@"The destination is the same as the source, you can not overwrite your source file!", @"")]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:didEndSelector contextInfo:NULL]; } else if ([[NSFileManager defaultManager] fileExistsAtPath:job.completeOutputURL.path]) { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"A file already exists at the selected destination.", @"")]; [alert setInformativeText:[NSString stringWithFormat:NSLocalizedString(@"Do you want to overwrite %@?", @""), job.completeOutputURL.path]]; [alert addButtonWithTitle:NSLocalizedString(@"Cancel", @"")]; [alert addButtonWithTitle:NSLocalizedString(@"Overwrite", @"")]; [alert setAlertStyle:NSCriticalAlertStyle]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:didEndSelector contextInfo:NULL]; } else if ([fQueueController jobExistAtURL:job.completeOutputURL]) { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"There is already a queue item for this destination.", @"")]; [alert setInformativeText:[NSString stringWithFormat:NSLocalizedString(@"Do you want to overwrite %@?", @""), job.completeOutputURL.path]]; [alert addButtonWithTitle:NSLocalizedString(@"Cancel", @"")]; [alert addButtonWithTitle:NSLocalizedString(@"Overwrite", @"")]; [alert setAlertStyle:NSCriticalAlertStyle]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:didEndSelector contextInfo:NULL]; } else { NSInteger returnCode = NSAlertSecondButtonReturn; NSMethodSignature *methodSignature = [self methodSignatureForSelector:didEndSelector]; NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature]; [invocation setTarget:self]; [invocation setSelector:didEndSelector]; [invocation setArgument:&returnCode atIndex:3]; [invocation invoke]; } } /** * Actually adds a job to the queue */ - (void)doAddToQueue { [fQueueController addJob:[self.job copy]]; } /** * Puts up an alert before ultimately calling doAddToQueue */ - (IBAction)addToQueue:(id)sender { [self runDestinationAlerts:self.job didEndSelector:@selector(overwriteAddToQueueAlertDone:returnCode:contextInfo:)]; } /** * Called from the alert posted by addToQueue * that asks the user if they want to overwrite an exiting movie file. */ - (void)overwriteAddToQueueAlertDone:(NSAlert *)alert returnCode:(NSInteger)returnCode contextInfo:(void *)contextInfo { if (returnCode == NSAlertSecondButtonReturn) { [self doAddToQueue]; } } - (void)doRip { // if there are no jobs in the queue, then add this one to the queue and rip // otherwise, just rip the queue if (fQueueController.pendingItemsCount == 0) { [self doAddToQueue]; } [fQueueController rip:self]; } /** * Puts up an alert before ultimately calling doRip */ - (IBAction)rip:(id)sender { // Rip or Cancel ? if (fQueueController.core.state == HBStateWorking || fQueueController.core.state == HBStatePaused || fQueueController.core.state == HBStateSearching) { // Displays an alert asking user if the want to cancel encoding of current job. [fQueueController cancelRip:self]; } // If there are pending jobs in the queue, then this is a rip the queue else if (fQueueController.pendingItemsCount > 0) { [fQueueController rip:self]; } else { [self runDestinationAlerts:self.job didEndSelector:@selector(overWriteAlertDone:returnCode:contextInfo:)]; } } /** * overWriteAlertDone: called from the alert posted by Rip: that asks the user if they * want to overwrite an exiting movie file. */ - (void)overWriteAlertDone:(NSAlert *)alert returnCode:(NSInteger)returnCode contextInfo:(void *)contextInfo { if (returnCode == NSAlertSecondButtonReturn) { [self doRip]; } } - (IBAction)pause:(id)sender { [fQueueController togglePauseResume:sender]; } #pragma mark - #pragma mark Batch Queue Titles Methods - (IBAction)addTitlesToQueue:(id)sender { self.titlesSelectionController = [[HBTitleSelectionController alloc] initWithTitles:self.core.titles presetName:self.job.presetName delegate:self]; [NSApp beginSheet:self.titlesSelectionController.window modalForWindow:self.window modalDelegate:nil didEndSelector:NULL contextInfo:NULL]; } - (void)didSelectIndexes:(NSIndexSet *)indexes { [self.titlesSelectionController.window orderOut:nil]; [NSApp endSheet:self.titlesSelectionController.window]; [self doAddTitlesAtIndexesToQueue:indexes]; } - (void)doAddTitlesAtIndexesToQueue:(NSIndexSet *)indexes; { NSMutableArray *jobs = [[NSMutableArray alloc] init]; BOOL fileExists = NO; BOOL fileOverwritesSource = NO; // Get the preset from the loaded job. HBPreset *preset = [self createPresetFromCurrentSettings]; for (HBTitle *title in self.core.titles) { if ([indexes containsIndex:title.index]) { HBJob *job = [[HBJob alloc] initWithTitle:title andPreset:preset]; job.outputURL = self.currentDestination; job.outputFileName = [HBUtilities defaultNameForJob:job]; job.title = nil; [jobs addObject:job]; } } NSMutableSet *destinations = [[NSMutableSet alloc] init]; for (HBJob *job in jobs) { if ([destinations containsObject:job.completeOutputURL]) { fileExists = YES; break; } else { [destinations addObject:job.completeOutputURL]; } if ([[NSFileManager defaultManager] fileExistsAtPath:job.completeOutputURL.path] || [fQueueController jobExistAtURL:job.completeOutputURL]) { fileExists = YES; break; } } for (HBJob *job in jobs) { if ([job.fileURL isEqual:job.completeOutputURL]) { fileOverwritesSource = YES; break; } } if (fileOverwritesSource) { NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"A file already exists at the selected destination.", @"")]; [alert setInformativeText:NSLocalizedString(@"The destination is the same as the source, you can not overwrite your source file!", @"")]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:@selector(overwriteAddTitlesToQueueAlertDone:returnCode:contextInfo:) contextInfo:NULL]; } else if (fileExists) { // File exist, warn user NSAlert *alert = [[NSAlert alloc] init]; [alert setMessageText:NSLocalizedString(@"File already exists.", nil)]; [alert setInformativeText:NSLocalizedString(@"One or more file already exists. Do you want to overwrite?", nil)]; [alert addButtonWithTitle:NSLocalizedString(@"Cancel", nil)]; [alert addButtonWithTitle:NSLocalizedString(@"Overwrite", nil)]; [alert setAlertStyle:NSCriticalAlertStyle]; [alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:@selector(overwriteAddTitlesToQueueAlertDone:returnCode:contextInfo:) contextInfo:(void *)CFBridgingRetain(jobs)]; } else { [fQueueController addJobsFromArray:jobs]; } } - (void)overwriteAddTitlesToQueueAlertDone:(NSAlert *)alert returnCode:(NSInteger)returnCode contextInfo:(void *)contextInfo { if (returnCode == NSAlertSecondButtonReturn) { NSArray *jobs = CFBridgingRelease(contextInfo); [fQueueController addJobsFromArray:jobs]; } } - (IBAction)addAllTitlesToQueue:(id)sender { NSMutableIndexSet *indexes = [NSMutableIndexSet indexSet]; for (HBTitle *title in self.core.titles) { [indexes addIndex:title.index]; } [self doAddTitlesAtIndexesToQueue:indexes]; } #pragma mark - Picture - (IBAction)showPreviewWindow:(id)sender { [fPreviewController showWindow:sender]; } - (IBAction)showTabView:(id)sender { NSInteger tag = [sender tag]; [fMainTabView selectTabViewItemAtIndex:tag]; } #pragma mark - Presets View Controller Delegate - (void)selectionDidChange { [self applyPreset:fPresetsView.selectedPreset]; } #pragma mark - Presets - (IBAction)toggleDrawer:(id)sender { if (fPresetDrawer.state == NSDrawerClosedState) { [[NSUserDefaults standardUserDefaults] setBool:YES forKey:@"HBDefaultPresetsDrawerShow"]; } else { [[NSUserDefaults standardUserDefaults] setBool:NO forKey:@"HBDefaultPresetsDrawerShow"]; } [fPresetDrawer toggle:self]; } - (void)setCurrentPreset:(HBPreset *)currentPreset { NSParameterAssert(currentPreset); if (currentPreset != _currentPreset) { NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] setCurrentPreset:_currentPreset]; _currentPreset = currentPreset; } if (!(self.undoManager.isUndoing || self.undoManager.isRedoing)) { // If the preset is one of the built in, set some additional options if (_currentPreset.isBuiltIn) { _currentPreset = [self presetByAddingDefaultLanguages:_currentPreset]; } } } - (HBPreset *)presetByAddingDefaultLanguages:(HBPreset *)preset { HBMutablePreset *mutablePreset = [preset mutableCopy]; NSMutableArray *languages = [NSMutableArray array]; if ([[NSUserDefaults standardUserDefaults] stringForKey:@"AlternateLanguage"]) { NSString *lang = [HBUtilities isoCodeForNativeLang:[[NSUserDefaults standardUserDefaults] stringForKey:@"AlternateLanguage"]]; if (lang) { [languages insertObject:lang atIndex:0]; } } if ([[NSUserDefaults standardUserDefaults] stringForKey:@"DefaultLanguage"]) { NSString *lang = [HBUtilities isoCodeForNativeLang:[[NSUserDefaults standardUserDefaults] stringForKey:@"DefaultLanguage"]]; if (lang) { [languages insertObject:lang atIndex:0]; } } mutablePreset[@"AudioLanguageList"] = languages; return mutablePreset; } - (void)setEdited:(BOOL)edited { if (edited != _edited) { NSUndoManager *undo = self.window.undoManager; [[undo prepareWithInvocationTarget:self] setEdited:_edited]; _edited = edited; } } - (void)applyPreset:(HBPreset *)preset { NSParameterAssert(preset); if (self.job) { self.currentPreset = preset; self.edited = NO; // Remove the job observer so we don't update the file name // too many times while the preset is being applied [self removeJobObservers]; // Apply the preset to the current job [self.job applyPreset:self.currentPreset]; // If Auto Naming is on, update the destination [self updateFileName]; [self addJobObservers]; } } - (IBAction)showAddPresetPanel:(id)sender { BOOL defaultToCustom = ((self.job.picture.width + self.job.picture.cropRight + self.job.picture.cropLeft) < self.job.picture.sourceWidth) || ((self.job.picture.height + self.job.picture.cropTop + self.job.picture.cropBottom) < self.job.picture.sourceHeight); // Show the add panel HBAddPresetController *addPresetController = [[HBAddPresetController alloc] initWithPreset:[self createPresetFromCurrentSettings] customWidth:self.job.picture.width customHeight:self.job.picture.height defaultToCustom:defaultToCustom]; [NSApp beginSheet:addPresetController.window modalForWindow:self.window modalDelegate:self didEndSelector:@selector(sheetDidEnd:returnCode:contextInfo:) contextInfo:(void *)CFBridgingRetain(addPresetController)]; } - (void)sheetDidEnd:(NSWindow *)sheet returnCode:(NSInteger)returnCode contextInfo:(void *)contextInfo { HBAddPresetController *addPresetController = (HBAddPresetController *)CFBridgingRelease(contextInfo); if (returnCode == NSModalResponseContinue) { [presetManager addPreset:addPresetController.preset]; } } - (HBPreset *)createPresetFromCurrentSettings { HBMutablePreset *preset = [self.currentPreset mutableCopy]; // Set whether or not this is a user preset or factory 0 is factory, 1 is user preset[@"Type"] = @1; preset[@"Default"] = @NO; [self.job writeToPreset:preset]; return [preset copy]; } #pragma mark - #pragma mark Import Export Preset(s) - (IBAction)exportPreset:(id)sender { [fPresetsView exportPreset:sender]; } - (IBAction)importPreset:(id)sender { [fPresetsView importPreset:sender]; } #pragma mark - #pragma mark Preset Menu - (IBAction)selectDefaultPreset:(id)sender { [self applyPreset:presetManager.defaultPreset]; [fPresetsView setSelection:_currentPreset]; } - (IBAction)insertFolder:(id)sender { [fPresetsView insertFolder:sender]; } - (IBAction)selectPresetFromMenu:(id)sender { // Retrieve the preset stored in the NSMenuItem HBPreset *preset = [sender representedObject]; [self applyPreset:preset]; [fPresetsView setSelection:preset]; } @end