1
0
mirror of https://github.com/danog/Telegram.git synced 2024-12-11 17:09:46 +01:00
Telegram/Telegraph/TGMemoryImageCache.m
2014-07-10 18:11:09 +04:00

152 lines
4.1 KiB
Objective-C

/*
* This is the source code of Telegram for iOS v. 1.1
* It is licensed under GNU GPL v. 2 or later.
* You should have received a copy of the license in this archive (see LICENSE).
*
* Copyright Peter Iakovlev, 2013.
*/
#import "TGMemoryImageCache.h"
#import <MTProtoKit/MTQueue.h>
#import <MTProtoKit/MTTime.h>
@interface TGMemoryImageCacheItem : NSObject
@property (nonatomic) id object;
@property (nonatomic, strong) NSDictionary *attributes;
@property (nonatomic) NSUInteger size;
@property (nonatomic) MTAbsoluteTime timestamp;
@end
@implementation TGMemoryImageCacheItem
- (instancetype)initWithObject:(id)object attributes:(NSDictionary *)attributes size:(NSUInteger)size timestamp:(MTAbsoluteTime)timestamp
{
self = [super init];
if (self != nil)
{
_object = object;
_attributes = attributes;
_size = size;
_timestamp = timestamp;
}
return self;
}
@end
@interface TGMemoryImageCache ()
{
MTQueue *_queue;
NSUInteger _softMemoryLimit;
NSUInteger _hardMemoryLimit;
NSMutableDictionary *_cache;
NSUInteger _cacheSize;
}
@end
@implementation TGMemoryImageCache
- (instancetype)initWithSoftMemoryLimit:(NSUInteger)softMemoryLimit hardMemoryLimit:(NSUInteger)hardMemoryLimit
{
self = [super init];
if (self != nil)
{
_queue = [MTQueue mainQueue];
_softMemoryLimit = softMemoryLimit;
_hardMemoryLimit = MAX(hardMemoryLimit, softMemoryLimit);
_cache = [[NSMutableDictionary alloc] init];
}
return self;
}
- (void)_addSize:(NSUInteger)size
{
if (_cacheSize + size > _hardMemoryLimit)
{
__block NSInteger requiredMemory = _cacheSize + size - _softMemoryLimit;
[[_cache keysSortedByValueUsingComparator:^NSComparisonResult(TGMemoryImageCacheItem *item1, TGMemoryImageCacheItem *item2)
{
return item1.timestamp < item2.timestamp ? NSOrderedAscending : NSOrderedDescending;
}] enumerateObjectsUsingBlock:^(id key, __unused NSUInteger idx, BOOL *stop)
{
TGMemoryImageCacheItem *item = _cache[key];
requiredMemory -= item.size;
[_cache removeObjectForKey:key];
if (requiredMemory <= 0 && stop != NULL)
*stop = true;
}];
__block NSUInteger currentCacheSize = 0;
[_cache enumerateKeysAndObjectsUsingBlock:^(__unused id key, TGMemoryImageCacheItem *item, __unused BOOL *stop)
{
currentCacheSize += item.size;
}];
_cacheSize = currentCacheSize;
}
_cacheSize += size;
}
- (UIImage *)imageForKey:(NSString *)key attributes:(__autoreleasing NSDictionary **)attributes
{
if (key == nil)
return nil;
__block id result = nil;
[_queue dispatchOnQueue:^
{
TGMemoryImageCacheItem *item = _cache[key];
if (item != nil)
{
result = item.object;
item.timestamp = MTAbsoluteSystemTime();
if (attributes != NULL)
*attributes = item.attributes;
}
} synchronous:true];
return result;
}
- (void)setImage:(UIImage *)image forKey:(NSString *)key attributes:(NSDictionary *)attributes
{
if (key != nil)
{
[_queue dispatchOnQueue:^
{
TGMemoryImageCacheItem *item = _cache[key];
if (item != nil)
{
if (item.size > _cacheSize)
_cacheSize = 0;
else
_cacheSize -= item.size;
}
if (image != nil)
{
if ([image isKindOfClass:[UIImage class]])
{
CGSize dimensions = image.size;
CGFloat scale = image.scale;
NSUInteger size = (NSUInteger)((dimensions.width * scale * dimensions.height * scale) * 4);
_cache[key] = [[TGMemoryImageCacheItem alloc] initWithObject:image attributes:attributes size:size timestamp:MTAbsoluteSystemTime()];
[self _addSize:size];
}
}
}];
}
}
@end