UIImage+Retina4.h

//
//  UIImage+Retina4.h
//  StunOMatic
//
//  Created by Benjamin Stahlhood on 9/12/12.
//  Copyright (c) 2012 DS Media Labs. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UIImage (Retina4)

@end

UIImage+Retina4.m

//
//  UIImage+Retina4.m
//  StunOMatic
//
//  Created by Benjamin Stahlhood on 9/12/12.
//  Copyright (c) 2012 DS Media Labs. All rights reserved.
//

#import "UIImage+Retina4.h"
#import <objc/objc-runtime.h>

static Method origImageNamedMethod = nil;

@implementation UIImage (Retina4)

+ (void)initialize {
    origImageNamedMethod = class_getClassMethod(self, @selector(imageNamed:));
    method_exchangeImplementations(origImageNamedMethod,
                                   class_getClassMethod(self, @selector(retina4ImageNamed:)));
}
                                                   
+ (UIImage *)retina4ImageNamed:(NSString *)imageName {
    
    NSLog(@"Loading image named => %@", imageName);

	CGFloat screenScale = UIScreen.mainScreen.scale;
	CGRect screenBounds = UIScreen.mainScreen.bounds;
	
    NSMutableString *imageNameMutable = [[imageName mutableCopy] autorelease];
    NSRange retinaAtSymbol = [imageName rangeOfString:@"@"];
    if (retinaAtSymbol.location != NSNotFound) {
        [imageNameMutable insertString:@"-568h" atIndex:retinaAtSymbol.location];
    } else {
        if (screenScale == 2.f && screenBounds.size.height == 568.0f) {
            NSRange dot = [imageName rangeOfString:@"."];
            if (dot.location != NSNotFound) {
                [imageNameMutable insertString:@"-568h@2x" atIndex:dot.location];
            } else {
                [imageNameMutable appendString:@"-568h@2x"];
            }
        }
    }
	
	UIImage* testImage = [UIImage retina4ImageNamed:imageNameMutable];
	if (testImage) {
		if (screenScale == 1.f)
			return testImage;
		return [UIImage imageWithCGImage:testImage.CGImage scale:screenScale orientation:testImage.imageOrientation];
	}
	
	return [UIImage retina4ImageNamed:imageName];
}

@end

UIImageRetina4.h

//
//  UIImage+Retina4.h
//  StunOMatic
//
//  Created by Benjamin Stahlhood on 9/12/12.
//  Copyright (c) 2012 DS Media Labs. All rights reserved.
//

#import <UIKit/UIKit.h>

@interface UIImage (Retina4)

@end

UIImageRetina4.m

//
//  UIImage+Retina4.m
//  StunOMatic
//
//  Created by Benjamin Stahlhood on 9/12/12.
//  Copyright (c) 2012 DS Media Labs. All rights reserved.
//

#import "UIImage+Retina4.h"
#import <objc/runtime.h>
#import <objc/message.h>

static Method origImageNamedMethod = nil;

@implementation UIImage (Retina4)

+ (void)initialize {
    origImageNamedMethod = class_getClassMethod(self, @selector(imageNamed:));
    method_exchangeImplementations(origImageNamedMethod,
                                   class_getClassMethod(self, @selector(retina4ImageNamed:)));
}

+ (UIImage *)retina4ImageNamed:(NSString *)imageName {
    
    NSLog(@"Loading image named => %@", imageName);

    CGFloat screenScale = UIScreen.mainScreen.scale;
	CGRect screenBounds = UIScreen.mainScreen.bounds;
	
    NSMutableString *imageNameMutable = [[imageName mutableCopy] autorelease];
    NSRange retinaAtSymbol = [imageName rangeOfString:@"@"];
    if (retinaAtSymbol.location != NSNotFound) {
        [imageNameMutable insertString:@"-568h" atIndex:retinaAtSymbol.location];
    } else {
        if (screenScale == 2.f && screenBounds.size.height == 568.0f) {
            NSRange dot = [imageName rangeOfString:@"."];
            if (dot.location != NSNotFound) {
                [imageNameMutable insertString:@"-568h@2x" atIndex:dot.location];
            } else {
                [imageNameMutable appendString:@"-568h@2x"];
            }
        }
    }
	
	UIImage* testImage = [UIImage retina4ImageNamed:imageNameMutable];
	if (testImage) {
		if (screenScale == 1.f)
			return testImage;
		return [UIImage imageWithCGImage:testImage.CGImage scale:screenScale orientation:testImage.imageOrientation];
	}
	
	return [UIImage retina4ImageNamed:imageName];
}

@end