macosx/HBFilters.m

   1 /*  HBFilters.m $
   2
   3  This file is part of the HandBrake source code.
   4  Homepage: <http://handbrake.fr/>.
   5  It may be used under the terms of the GNU General Public License. */
   6
   7 #import "HBFilters.h"
   8 #import "HBCodingUtilities.h"
   9 #include "hb.h"
  10
  11 NSString * const HBFiltersChangedNotification = @"HBFiltersChangedNotification";
  12
  13 @interface HBFilters ()
  14
  15 @property (nonatomic, readwrite, getter=areNotificationsEnabled) BOOL notificationsEnabled;
  16
  17 @end
  18
  19 @implementation HBFilters
  20
  21 - (instancetype)init
  22 {
  23     self = [super init];
  24     if (self)
  25     {
  26         _detelecine = @"off";
  27         _detelecineCustomString = @"";
  28         _deinterlace = @"off";
  29         _deinterlaceCustomString = @"";
  30         _decomb = @"off";
  31         _decombCustomString = @"";
  32         _denoise = @"off";
  33         _denoiseCustomString = @"";
  34         _denoisePreset = @"medium";
  35         _denoiseTune = @"none";
  36
  37         _notificationsEnabled = YES;
  38     }
  39     return self;
  40 }
  41
  42 - (void)postChangedNotification
  43 {
  44     if (self.areNotificationsEnabled)
  45     {
  46         [[NSNotificationCenter defaultCenter] postNotification: [NSNotification notificationWithName:HBFiltersChangedNotification
  47                                                                                               object:self
  48                                                                                             userInfo:nil]];
  49     }
  50 }
  51
  52 #pragma mark - Setters
  53
  54 - (void)setDetelecine:(NSString *)detelecine
  55 {
  56     if (detelecine)
  57     {
  58         _detelecine = [detelecine copy];
  59     }
  60     else
  61     {
  62         _detelecine = @"off";
  63     }
  64     [self postChangedNotification];
  65 }
  66
  67 // Override setter to avoid nil values.
  68 - (void)setDetelecineCustomString:(NSString *)detelecineCustomString
  69 {
  70     if (detelecineCustomString)
  71     {
  72         _detelecineCustomString = [detelecineCustomString copy];
  73     }
  74     else
  75     {
  76         _detelecineCustomString = @"";
  77     }
  78
  79     [self postChangedNotification];
  80 }
  81
  82 - (void)setDeinterlace:(NSString *)deinterlace
  83 {
  84     if (deinterlace)
  85     {
  86         _deinterlace = [deinterlace copy];
  87     }
  88     else
  89     {
  90         _deinterlace = @"off";
  91     }
  92     [self postChangedNotification];
  93 }
  94
  95 - (void)setDeinterlaceCustomString:(NSString *)deinterlaceCustomString
  96 {
  97     if (deinterlaceCustomString)
  98     {
  99         _deinterlaceCustomString = [deinterlaceCustomString copy];
 100     }
 101     else
 102     {
 103         _deinterlaceCustomString = @"";
 104     }
 105
 106     [self postChangedNotification];
 107 }
 108
 109 - (void)setDecomb:(NSString *)decomb
 110 {
 111     if (decomb)
 112     {
 113         _decomb = [decomb copy];
 114     }
 115     else
 116     {
 117         _decomb = @"off";
 118     }
 119     [self postChangedNotification];
 120 }
 121
 122 - (void)setDecombCustomString:(NSString *)decombCustomString
 123 {
 124     if (decombCustomString)
 125     {
 126         _decombCustomString = [decombCustomString copy];
 127     }
 128     else
 129     {
 130         _decombCustomString = @"";
 131     }
 132
 133     [self postChangedNotification];
 134 }
 135
 136 - (void)setDenoise:(NSString *)denoise
 137 {
 138     if (denoise)
 139     {
 140         _denoise = [denoise copy];
 141     }
 142     else
 143     {
 144         _denoise = @"";
 145     }
 146
 147     [self postChangedNotification];
 148 }
 149
 150 - (void)setDenoisePreset:(NSString *)denoisePreset
 151 {
 152     if (denoisePreset)
 153     {
 154         _denoisePreset = [denoisePreset copy];
 155     }
 156     else
 157     {
 158         _denoisePreset = @"medium";
 159     }
 160
 161     [self postChangedNotification];
 162 }
 163
 164 - (void)setDenoiseTune:(NSString *)denoiseTune
 165 {
 166     if (denoiseTune)
 167     {
 168         _denoiseTune = [denoiseTune copy];
 169     }
 170     else
 171     {
 172         _denoiseTune = @"none";
 173     }
 174
 175     [self postChangedNotification];
 176 }
 177
 178 - (void)setDenoiseCustomString:(NSString *)denoiseCustomString
 179 {
 180     if (denoiseCustomString)
 181     {
 182         _denoiseCustomString = [denoiseCustomString copy];
 183     }
 184     else
 185     {
 186         _denoiseCustomString = @"";
 187     }
 188
 189     [self postChangedNotification];
 190 }
 191
 192 - (void)setDeblock:(int)deblock
 193 {
 194     _deblock = deblock;
 195     [self postChangedNotification];
 196 }
 197
 198 - (void)setGrayscale:(BOOL)grayscale
 199 {
 200     _grayscale = grayscale;
 201     [self postChangedNotification];
 202 }
 203
 204 - (void)setUseDecomb:(BOOL)useDecomb
 205 {
 206     _useDecomb = useDecomb;
 207     [self postChangedNotification];
 208 }
 209
 210 + (NSSet *)keyPathsForValuesAffectingValueForKey:(NSString *)key
 211 {
 212     NSSet *retval = nil;
 213
 214     if ([key isEqualToString:@"summary"])
 215     {
 216         retval = [NSSet setWithObjects:@"detelecine", @"detelecineCustomString", @"useDecomb", @"deinterlace", @"deinterlaceCustomString", @"decomb", @"decombCustomString", @"denoise", @"denoisePreset", @"denoiseTune", @"denoiseCustomString", @"deblock", @"grayscale", nil];
 217     }
 218     if ([key isEqualToString:@"customDetelecineSelected"] ||
 219         [key isEqualToString:@"customDecombSelected"] ||
 220         [key isEqualToString:@"customDeinterlaceSelected"])
 221     {
 222         retval = [NSSet setWithObjects:@"detelecine", @"decomb", @"deinterlace", @"useDecomb", nil];
 223     }
 224     if ([key isEqualToString:@"denoiseTunesAvailable"] ||
 225         [key isEqualToString:@"customDenoiseSelected"])
 226     {
 227         retval = [NSSet setWithObjects:@"denoise", @"denoisePreset", nil];
 228     }
 229     if ([key isEqualToString:@"denoiseEnabled"])
 230     {
 231         retval = [NSSet setWithObject:@"denoise"];
 232     }
 233     if ([key isEqualToString:@"deblockSummary"])
 234     {
 235         retval = [NSSet setWithObject:@"deblock"];
 236     }
 237     return retval;
 238 }
 239
 240
 241 #pragma mark - NSCopying
 242
 243 - (instancetype)copyWithZone:(NSZone *)zone
 244 {
 245     HBFilters *copy = [[[self class] alloc] init];
 246
 247     if (copy)
 248     {
 249         copy->_detelecine = [_detelecine copy];
 250         copy->_detelecineCustomString = [_detelecineCustomString copy];
 251
 252         copy->_deinterlace = [_deinterlace copy];
 253         copy->_deinterlaceCustomString = [_deinterlaceCustomString copy];
 254
 255         copy->_decomb = [_decomb copy];
 256         copy->_decombCustomString = [_decombCustomString copy];
 257
 258         copy->_denoise = [_denoise copy];
 259         copy->_denoisePreset = [_denoisePreset copy];
 260         copy->_denoiseTune = [_denoiseTune copy];
 261         copy->_denoiseCustomString = [_denoiseCustomString copy];
 262
 263         copy->_deblock = _deblock;
 264         copy->_grayscale = _grayscale;
 265
 266         copy->_useDecomb = _useDecomb;
 267     }
 268
 269     return copy;
 270 }
 271
 272 #pragma mark - NSCoding
 273
 274 + (BOOL)supportsSecureCoding
 275 {
 276     return YES;
 277 }
 278
 279 - (void)encodeWithCoder:(NSCoder *)coder
 280 {
 281     [coder encodeInt:1 forKey:@"HBFiltersVersion"];
 282
 283     encodeObject(_detelecine);
 284     encodeObject(_detelecineCustomString);
 285
 286     encodeObject(_deinterlace);
 287     encodeObject(_deinterlaceCustomString);
 288
 289     encodeObject(_decomb);
 290     encodeObject(_decombCustomString);
 291
 292     encodeObject(_denoise);
 293     encodeObject(_denoisePreset);
 294     encodeObject(_denoiseTune);
 295     encodeObject(_denoiseCustomString);
 296
 297     encodeInt(_deblock);
 298     encodeBool(_grayscale);
 299
 300     encodeBool(_useDecomb);
 301 }
 302
 303 - (instancetype)initWithCoder:(NSCoder *)decoder
 304 {
 305     self = [super init];
 306
 307     decodeObject(_detelecine, NSString);
 308     decodeObject(_detelecineCustomString, NSString);
 309
 310     decodeObject(_deinterlace, NSString);
 311     decodeObject(_deinterlaceCustomString, NSString);
 312
 313     decodeObject(_decomb, NSString);
 314     decodeObject(_decombCustomString, NSString);
 315
 316     decodeObject(_denoise, NSString);
 317     decodeObject(_denoisePreset, NSString);
 318     decodeObject(_denoiseTune, NSString);
 319     decodeObject(_denoiseCustomString, NSString);
 320
 321     decodeInt(_deblock);
 322     decodeBool(_grayscale);
 323
 324     decodeBool(_useDecomb);
 325
 326     _notificationsEnabled = YES;
 327
 328     return self;
 329 }
 330
 331 #pragma mark - Presets and queue
 332
 333 - (void)writeToPreset:(NSMutableDictionary *)preset
 334 {
 335     preset[@"PictureDecombDeinterlace"] = @(self.useDecomb);
 336
 337     preset[@"PictureDeinterlace"] = self.deinterlace;
 338     preset[@"PictureDeinterlaceCustom"] = self.deinterlaceCustomString;
 339
 340     preset[@"PictureDecomb"] = self.decomb;
 341     preset[@"PictureDecombCustom"] = self.decombCustomString;
 342
 343     preset[@"PictureDetelecine"] = self.detelecine;
 344     preset[@"PictureDetelecineCustom"] = self.detelecineCustomString;
 345
 346     preset[@"PictureDenoiseFilter"] = self.denoise;
 347     preset[@"PictureDenoisePreset"] = self.denoisePreset;
 348     preset[@"PictureDenoiseTune"] = self.denoiseTune;
 349     preset[@"PictureDenoiseCustom"] = self.denoiseCustomString;
 350
 351     preset[@"PictureDeblock"] = @(self.deblock);
 352     preset[@"VideoGrayScale"] = @(self.grayscale);
 353 }
 354
 355 - (void)applyPreset:(NSDictionary *)preset
 356 {
 357     self.notificationsEnabled = NO;
 358
 359     // If the preset has an objectForKey:@"UsesPictureFilters", and handle the filters here
 360     if ([preset[@"UsesPictureFilters"] boolValue])
 361     {
 362         // We only allow *either* Decomb or Deinterlace. So check for the PictureDecombDeinterlace key.
 363         self.useDecomb = [preset[@"PictureDecombDeinterlace"] boolValue];
 364
 365         self.decomb = preset[@"PictureDecomb"];
 366         self.decombCustomString = preset[@"PictureDecombCustom"];
 367
 368         self.deinterlace = preset[@"PictureDeinterlace"];
 369         self.deinterlaceCustomString = preset[@"PictureDeinterlaceCustom"];
 370
 371         //Detelecine
 372         self.detelecine = preset[@"PictureDetelecine"];
 373         self.detelecineCustomString = preset[@"PictureDetelecineCustom"];
 374
 375         // Denoise
 376         self.denoise = preset[@"PictureDenoiseFilter"];
 377         self.denoisePreset = preset[@"PictureDenoisePreset"];
 378         self.denoiseTune = preset[@"PictureDenoiseTune"];
 379
 380         self.denoiseCustomString = preset[@"PictureDenoiseCustom"];
 381
 382         // Deblock
 383         if ([preset[@"PictureDeblock"] intValue] == 1)
 384         {
 385             // if its a one, then its the old on/off deblock, set on to 5
 386             self.deblock = 5;
 387         }
 388         else
 389         {
 390             // use the settings intValue
 391             self.deblock = [preset[@"PictureDeblock"] intValue];
 392         }
 393
 394         self.grayscale = [preset[@"VideoGrayScale"] boolValue];
 395     }
 396
 397     self.notificationsEnabled = YES;
 398 }
 399
 400 @end