/* $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 "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 NSTextField * fQueueStatus;
IBOutlet NSProgressIndicator * fRipIndicator;
BOOL fRipIndicatorShown;
// User Preset
HBPresetsManager * presetManager;
HBPresetsViewController * fPresetsView;
IBOutlet NSDrawer * fPresetDrawer;
}
@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;
/// 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;
_currentPreset = manager.defaultPreset;
_scanSpecificTitleIdx = 1;
}
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]];
// 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)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;
}
#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 and scan the selected title", 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];
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)
{
// Set Source Name at top of window with the browsedSourceDisplayName grokked right before -performScan
fSrcDVD2Field.stringValue = displayName;
[fSrcTitlePopUp addItemWithTitle:title.description];
// See if this is the main feature according
if (title.isFeatured)
{
[fSrcTitlePopUp selectItemWithTitle:title.description];
}
}
// Select the first item is nothing is selected
if (!fSrcTitlePopUp.selectedItem)
{
[fSrcTitlePopUp selectItemAtIndex:0];
}
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];
HBJob *job = [self jobFromTitle:titles.firstObject];
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)
{
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.destURL = [HBUtilities destURLForJob: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;
// 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:NSHomeDirectory()] URLByAppendingPathComponent:@"Desktop"];
}
[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)
{
// Check if we selected a folder or not
id outValue = nil;
[panel.URL getResourceValue:&outValue forKey:NSURLIsDirectoryKey error:NULL];
// we set the last searched source directory in the prefs here
if ([outValue boolValue])
{
[[NSUserDefaults standardUserDefaults] setURL:panel.URL forKey:@"HBLastSourceDirectoryURL"];
}
else
{
[[NSUserDefaults standardUserDefaults] setURL:panel.URL.URLByDeletingLastPathComponent forKey:@"HBLastSourceDirectoryURL"];
}
NSInteger titleIdx = self.scanSpecificTitle ? self.scanSpecificTitle : 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
NSSavePanel *panel = [NSSavePanel savePanel];
if (self.job.destURL)
{
panel.directoryURL = self.job.destURL.URLByDeletingLastPathComponent;
panel.nameFieldStringValue = self.job.destURL.lastPathComponent;
}
[panel beginSheetModalForWindow:self.window completionHandler:^(NSInteger result)
{
if (result == NSFileHandlingPanelOKButton)
{
self.job.destURL = panel.URL;
// Save this path to the prefs so that on next browse destination window it opens there
[[NSUserDefaults standardUserDefaults] setURL:panel.URL.URLByDeletingLastPathComponent
forKey:@"HBLastDestinationDirectory"];
}
}];
}
- (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.destURL = [[self.job.destURL URLByDeletingLastPathComponent] URLByAppendingPathComponent:
[NSString stringWithFormat:@"%@.%@", fileName, self.job.destURL.pathExtension]];
}
}
- (void)updateFileExtension:(NSNotification *)notification
{
if (self.job)
{
NSString *extension = [HBUtilities automaticExtForJob:self.job];
if (![extension isEqualTo:self.job.destURL.pathExtension])
{
self.job.destURL = [[self.job.destURL URLByDeletingPathExtension] URLByAppendingPathExtension: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:(NSString *)info
{
fQueueStatus.stringValue = info;
}
#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
/**
* 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
{
// We get the destination directory from the destination field here
NSString *destinationDirectory = self.job.destURL.path.stringByDeletingLastPathComponent;
// We check for a valid destination here
if ([[NSFileManager defaultManager] fileExistsAtPath:destinationDirectory] == 0)
{
NSAlert *alert = [[NSAlert alloc] init];
[alert setMessageText:NSLocalizedString(@"Warning!", @"")];
[alert setInformativeText:NSLocalizedString(@"This is not a valid destination directory!", @"")];
[alert runModal];
return;
}
if ([[NSFileManager defaultManager] fileExistsAtPath:self.job.destURL.path])
{
// File exist, warn user
NSAlert *alert = [[NSAlert alloc] init];
[alert setMessageText:NSLocalizedString(@"File already exists.", @"")];
[alert setInformativeText:[NSString stringWithFormat:NSLocalizedString(@"Do you want to overwrite %@?", @""), self.job.destURL.path]];
[alert addButtonWithTitle:NSLocalizedString(@"Cancel", @"")];
[alert addButtonWithTitle:NSLocalizedString(@"Overwrite", @"")];
[alert setAlertStyle:NSCriticalAlertStyle];
[alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:@selector(overwriteAddToQueueAlertDone:returnCode:contextInfo:) contextInfo:NULL];
}
else if ([fQueueController jobExistAtURL:self.job.destURL])
{
// File exist in queue, warn user
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 %@?", @""), self.job.destURL.path]];
[alert addButtonWithTitle:NSLocalizedString(@"Cancel", @"")];
[alert addButtonWithTitle:NSLocalizedString(@"Overwrite", @"")];
[alert setAlertStyle:NSCriticalAlertStyle];
[alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:@selector(overwriteAddToQueueAlertDone:returnCode:contextInfo:) contextInfo:NULL];
}
else
{
[self doAddToQueue];
}
}
/**
* 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)
{
// Displays an alert asking user if the want to cancel encoding of current job.
[fQueueController cancelRip:self];
return;
}
// If there are pending jobs in the queue, then this is a rip the queue
if (fQueueController.pendingItemsCount > 0)
{
[fQueueController rip:self];
return;
}
// Before adding jobs to the queue, check for a valid destination.
NSString *destinationDirectory = self.job.destURL.path.stringByDeletingLastPathComponent;
if ([[NSFileManager defaultManager] fileExistsAtPath:destinationDirectory] == 0)
{
NSAlert *alert = [[NSAlert alloc] init];
[alert setMessageText:NSLocalizedString(@"Invalid destination.", @"")];
[alert setInformativeText:NSLocalizedString(@"The current destination folder is not a valid.", @"")];
[alert runModal];
return;
}
// We check for duplicate name here
if ([[NSFileManager defaultManager] fileExistsAtPath:self.job.destURL.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 %@?", @""), self.job.destURL.path]];
[alert addButtonWithTitle:NSLocalizedString(@"Cancel", @"")];
[alert addButtonWithTitle:NSLocalizedString(@"Overwrite", @"")];
[alert setAlertStyle:NSCriticalAlertStyle];
[alert beginSheetModalForWindow:self.window modalDelegate:self didEndSelector:@selector(overWriteAlertDone:returnCode:contextInfo:) contextInfo:NULL];
// overWriteAlertDone: will be called when the alert is dismissed. It will call doRip.
}
else
{
[self doRip];
}
}
/**
* 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;
// 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.destURL = [HBUtilities destURLForJob:job];
job.title = nil;
[jobs addObject:job];
}
}
NSMutableSet *destinations = [[NSMutableSet alloc] init];
for (HBJob *job in jobs)
{
if ([destinations containsObject:job.destURL])
{
fileExists = YES;
break;
}
else
{
[destinations addObject:job.destURL];
}
if ([[NSFileManager defaultManager] fileExistsAtPath:job.destURL.path] || [fQueueController jobExistAtURL:job.destURL])
{
fileExists = YES;
break;
}
}
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;
}
}
- (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:preset];
// 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