filters.py

#!/usr/bin/env python

# Nicolas Seriot
# 2015-04-11

import argparse
import os
import time

from Foundation import NSBundle, NSClassFromString, NSDictionary, NSURL

success = NSBundle.bundleWithPath_("/System/Library/PrivateFrameworks/PhotoLibraryPrivate.framework/Versions/A/Frameworks/PAImaging.framework").load()
assert success == True

CIImage = NSClassFromString("CIImage")
assert CIImage

CIFilter = NSClassFromString("CIFilter")
assert CIFilter

NSBitmapImageRep = NSClassFromString("NSBitmapImageRep")
assert NSBitmapImageRep

IPAPhotoAdjustmentStackSerializer_v10 = NSClassFromString("IPAPhotoAdjustmentStackSerializer_v10")
assert IPAPhotoAdjustmentStackSerializer_v10

ipaPASS = IPAPhotoAdjustmentStackSerializer_v10.alloc().init()

def apply_cifilter_with_name(filter_name, orientation, in_path, out_path, dry_run=False):
    
    print "-- in: ", in_path
    print "-- out:", out_path
    
    assert in_path
    assert out_path
    
    assert filter_name in ["CIPhotoEffectTonal", "CIPhotoEffectMono", "CIPhotoEffectInstant", "CIPhotoEffectTransfer",
                           "CIPhotoEffectProcess", "CIPhotoEffectChrome", "CIPhotoEffectNoir", "CIPhotoEffectFade"]
                     
    url = NSURL.alloc().initFileURLWithPath_(in_path)
    ci_image = CIImage.imageWithContentsOfURL_(url)
    assert ci_image
    
    in_creation_timestamp = os.path.getmtime(in_path)
    print time.ctime(in_creation_timestamp)
    
    if orientation != None and orientation != 1:
        print "-- orientation:", orientation
        ci_image = ci_image.imageByApplyingOrientation_(orientation)
        
    ci_filter = CIFilter.filterWithName_(filter_name)
    assert ci_filter
    
    ci_filter.setValue_forKey_(ci_image, "inputImage")
    ci_filter.setDefaults()
    
    ci_image_result = ci_filter.outputImage()
    assert ci_image_result
    
    bitmap_rep = NSBitmapImageRep.alloc().initWithCIImage_(ci_image_result)
    assert bitmap_rep
    
    properties = { "NSImageCompressionFactor" : 0.9 }
    data = bitmap_rep.representationUsingType_properties_(3, properties) # 3 for JPEG
    
    if dry_run:
        print "-- dryrun, don't write", out_path
        return
    
    assert data.writeToFile_atomically_(out_path, True)
    
    os.utime(out_path, (time.time(), in_creation_timestamp))

def read_aae_file(path):

    plist = NSDictionary.dictionaryWithContentsOfFile_(path)
    
    if plist["adjustmentFormatIdentifier"] != "com.apple.photo":
        print "-- bad format identifier:", plist["adjustmentFormatIdentifier"]
        return None, None
    
    data = plist["adjustmentData"]
    
    d = ipaPASS.archiveFromData_error_(data, None)
    
    adjustments = d["adjustments"]
    orientation = d["metadata"]["orientation"]
    
    effect_names = [ d_["settings"]["effectName"] for d_ in d["adjustments"] if d_["identifier"] == "Effect"]
    
    if len(effect_names) == 0:
        print "-- no effect name"
        return None, None
    
    filter_name = "CIPhotoEffect" + effect_names[0]
    print "-- filter:", filter_name

    return filter_name, orientation

def main():
    
    parser = argparse.ArgumentParser(description='Restore filters on photos imported from iOS 8 with Image Capture.')
    parser.add_argument("-o", "--overwrite", action='store_true', default=False, help="overwrite original photos with filtered photos, remove AAE files")
    parser.add_argument("-d", "--dryrun", action='store_true', default=False, help="don't write anything on disk")
    parser.add_argument("path", help="path to folder with JPG and AAC files")
    args = parser.parse_args()
    
    aae_files = [ os.path.join(args.path, f) for f in os.listdir(args.path) if f.endswith('.AAE') ]
    
    for aae in aae_files:

        print "-- reading", aae
		
        filter_name, orientation = read_aae_file(aae)
        if not filter_name:
            continue
        
        name, ext = os.path.splitext(aae)
        jpg_in = name + ".JPG"
        
        if args.overwrite and not args.dryrun:
    	    print "-- removing", aae
            os.remove(aae)
        
        if not os.path.exists(jpg_in):
            print "-- missing file:", jpg_in
            continue
                
        jpg_out = jpg_in if args.overwrite else (name + "_" + filter_name + ".JPG")
        
        apply_cifilter_with_name(filter_name, orientation, jpg_in, jpg_out, args.dryrun)

        del(jpg_in)
        del(jpg_out)

if __name__=='__main__':
    main()