mirror of
https://github.com/danog/Telegram.git
synced 2024-12-11 17:09:46 +01:00
152 lines
4.1 KiB
Objective-C
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
|