Swiftgram/submodules/LegacyComponents/Sources/TGMediaEditingContext.m
Ilya Laktyushin 9c486969ea Various fixes
2025-02-14 21:36:20 +04:00

1386 lines
44 KiB
Objective-C

#import "TGMediaEditingContext.h"
#import "LegacyComponentsInternal.h"
#import "TGStringUtils.h"
#import <LegacyComponents/UIImage+TG.h>
#import "TGPhotoEditorUtils.h"
#import "PGPhotoEditorValues.h"
#import "TGVideoEditAdjustments.h"
#import "TGModernCache.h"
#import "TGMemoryImageCache.h"
#import "TGMediaAsset.h"
#import "TGPaintingData.h"
@interface TGMediaImageUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) id representation;
+ (instancetype)imageUpdateWithItem:(id<TGMediaEditableItem>)item representation:(id)representation;
@end
@interface TGMediaAdjustmentsUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) id<TGMediaEditAdjustments> adjustments;
+ (instancetype)adjustmentsUpdateWithItem:(id<TGMediaEditableItem>)item adjustments:(id<TGMediaEditAdjustments>)adjustments;
@end
@interface TGMediaCaptionUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSAttributedString *caption;
+ (instancetype)captionUpdateWithItem:(id<TGMediaEditableItem>)item caption:(NSAttributedString *)caption;
@end
@interface TGMediaTimerUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSNumber *timer;
+ (instancetype)timerUpdateWithItem:(id<TGMediaEditableItem>)item timer:(NSNumber *)timer;
+ (instancetype)timerUpdate:(NSNumber *)timer;
@end
@interface TGMediaSpoilerUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly) bool spoiler;
+ (instancetype)spoilerUpdateWithItem:(id<TGMediaEditableItem>)item spoiler:(bool)spoiler;
+ (instancetype)spoilerUpdate:(bool)spoiler;
@end
@interface TGMediaPriceUpdate : NSObject
@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSNumber *price;
+ (instancetype)priceUpdateWithItem:(id<TGMediaEditableItem>)item price:(NSNumber *)price;
+ (instancetype)priceUpdate:(NSNumber *)timer;
@end
@interface TGModernCache (Private)
- (void)cleanup;
@end
@interface TGMediaEditingContext ()
{
NSString *_contextId;
NSMutableDictionary *_captions;
NSMutableDictionary *_adjustments;
NSMutableDictionary *_timers;
NSNumber *_timer;
NSMutableDictionary *_spoilers;
NSMutableDictionary *_prices;
SQueue *_queue;
NSMutableDictionary *_temporaryRepCache;
TGMemoryImageCache *_imageCache;
TGMemoryImageCache *_thumbnailImageCache;
TGMemoryImageCache *_paintingImageCache;
TGMemoryImageCache *_stillPaintingImageCache;
TGMemoryImageCache *_originalImageCache;
TGMemoryImageCache *_originalThumbnailImageCache;
TGMemoryImageCache *_coverImageCache;
NSMutableDictionary *_coverPositions;
TGModernCache *_diskCache;
NSURL *_fullSizeResultsUrl;
NSURL *_paintingDatasUrl;
NSURL *_paintingImagesUrl;
NSURL *_stillPaintingImagesUrl;
NSURL *_videoPaintingImagesUrl;
NSMutableArray *_storeVideoPaintingImages;
NSMutableDictionary *_faces;
SPipe *_representationPipe;
SPipe *_thumbnailImagePipe;
SPipe *_coverImagePipe;
SPipe *_adjustmentsPipe;
SPipe *_captionPipe;
SPipe *_timerPipe;
SPipe *_spoilerPipe;
SPipe *_pricePipe;
SPipe *_fullSizePipe;
SPipe *_cropPipe;
SPipe *_captionAbovePipe;
NSAttributedString *_forcedCaption;
bool _captionAbove;
}
@end
@implementation TGMediaEditingContext
- (instancetype)init
{
self = [super init];
if (self != nil)
{
_contextId = [NSString stringWithFormat:@"%ld", lrand48()];
_queue = [[SQueue alloc] init];
_captions = [[NSMutableDictionary alloc] init];
_adjustments = [[NSMutableDictionary alloc] init];
_timers = [[NSMutableDictionary alloc] init];
_spoilers = [[NSMutableDictionary alloc] init];
_prices = [[NSMutableDictionary alloc] init];
_imageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
hardMemoryLimit:[[self class] imageHardMemoryLimit]];
_thumbnailImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] thumbnailImageSoftMemoryLimit]
hardMemoryLimit:[[self class] thumbnailImageHardMemoryLimit]];
_paintingImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
hardMemoryLimit:[[self class] imageHardMemoryLimit]];
_stillPaintingImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
hardMemoryLimit:[[self class] imageHardMemoryLimit]];
_originalImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] originalImageSoftMemoryLimit]
hardMemoryLimit:[[self class] originalImageHardMemoryLimit]];
_originalThumbnailImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] thumbnailImageSoftMemoryLimit]
hardMemoryLimit:[[self class] thumbnailImageHardMemoryLimit]];
_coverImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] thumbnailImageSoftMemoryLimit] * 10
hardMemoryLimit:[[self class] thumbnailImageHardMemoryLimit] * 10];
_coverPositions = [[NSMutableDictionary alloc] init];
NSString *diskCachePath = [[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[[self class] diskCachePath]];
_diskCache = [[TGModernCache alloc] initWithPath:diskCachePath size:[[self class] diskMemoryLimit]];
_fullSizeResultsUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"photoeditorresults/%@", _contextId]]];
[[NSFileManager defaultManager] createDirectoryAtPath:_fullSizeResultsUrl.path withIntermediateDirectories:true attributes:nil error:nil];
_paintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"paintingimages/%@", _contextId]]];
[[NSFileManager defaultManager] createDirectoryAtPath:_paintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
_stillPaintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:@"stillpaintingimages"]];
[[NSFileManager defaultManager] createDirectoryAtPath:_stillPaintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
_videoPaintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:@"videopaintingimages"]];
[[NSFileManager defaultManager] createDirectoryAtPath:_videoPaintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
_paintingDatasUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"paintingdatas/%@", _contextId]]];
[[NSFileManager defaultManager] createDirectoryAtPath:_paintingDatasUrl.path withIntermediateDirectories:true attributes:nil error:nil];
_storeVideoPaintingImages = [[NSMutableArray alloc] init];
_faces = [[NSMutableDictionary alloc] init];
_temporaryRepCache = [[NSMutableDictionary alloc] init];
_representationPipe = [[SPipe alloc] init];
_thumbnailImagePipe = [[SPipe alloc] init];
_adjustmentsPipe = [[SPipe alloc] init];
_captionPipe = [[SPipe alloc] init];
_coverImagePipe = [[SPipe alloc] init];
_timerPipe = [[SPipe alloc] init];
_spoilerPipe = [[SPipe alloc] init];
_pricePipe = [[SPipe alloc] init];
_fullSizePipe = [[SPipe alloc] init];
_cropPipe = [[SPipe alloc] init];
_captionAbovePipe = [[SPipe alloc] init];
}
return self;
}
- (void)dealloc
{
[self cleanup];
}
- (void)cleanup
{
TGModernCache *diskCache = _diskCache;
TGDispatchAfter(10.0, dispatch_get_main_queue(), ^{
[diskCache cleanup];
});
[[NSFileManager defaultManager] removeItemAtPath:_fullSizeResultsUrl.path error:nil];
[[NSFileManager defaultManager] removeItemAtPath:_paintingImagesUrl.path error:nil];
[[NSFileManager defaultManager] removeItemAtPath:_paintingDatasUrl.path error:nil];
}
+ (instancetype)contextForCaptionsOnly
{
TGMediaEditingContext *context = [[TGMediaEditingContext alloc] init];
context->_inhibitEditing = true;
return context;
}
#pragma mark -
- (SSignal *)imageSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
return [self imageSignalForItem:item withUpdates:true];
}
- (SSignal *)imageSignalForItem:(NSObject<TGMediaEditableItem> *)item withUpdates:(bool)withUpdates
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return [SSignal fail:nil];
SSignal *updateSignal = [[_representationPipe.signalProducer() filter:^bool(TGMediaImageUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
}] map:^id(TGMediaImageUpdate *update)
{
return update.representation;
}];
if ([self _adjustmentsForItemId:itemId] == nil)
{
SSignal *signal = [SSignal single:nil];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}
NSString *imageUri = [TGMediaEditingContext _imageUriForItemId:itemId];
SSignal *signal = [[self _imageSignalForItemId:itemId imageCache:_imageCache imageDiskUri:imageUri synchronous:false] catch:^SSignal *(__unused id error)
{
id temporaryRep = [_temporaryRepCache objectForKey:itemId];
SSignal *signal = [SSignal single:temporaryRep];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}
- (SSignal *)thumbnailImageSignalForItem:(id<TGMediaEditableItem>)item
{
return [self thumbnailImageSignalForIdentifier:item.uniqueIdentifier];
}
- (SSignal *)thumbnailImageSignalForItem:(id<TGMediaEditableItem>)item withUpdates:(bool)withUpdates synchronous:(bool)synchronous
{
return [self thumbnailImageSignalForIdentifier:item.uniqueIdentifier withUpdates:withUpdates synchronous: synchronous];
}
- (SSignal *)thumbnailImageSignalForIdentifier:(NSString *)identifier {
return [self thumbnailImageSignalForIdentifier:identifier withUpdates:true synchronous:false];
}
- (SSignal *)thumbnailImageSignalForIdentifier:(NSString *)identifier withUpdates:(bool)withUpdates synchronous:(bool)synchronous {
NSString *itemId = [self _contextualIdForItemId:identifier];
if (itemId == nil)
return [SSignal fail:nil];
SSignal *updateSignal = [[_thumbnailImagePipe.signalProducer() filter:^bool(TGMediaImageUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:identifier];
}] map:^id(TGMediaImageUpdate *update)
{
return update.representation;
}];
if ([self _adjustmentsForItemId:itemId] == nil)
{
SSignal *signal = [SSignal single:nil];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}
NSString *imageUri = [TGMediaEditingContext _thumbnailImageUriForItemId:itemId];
SSignal *signal = [[self _imageSignalForItemId:itemId imageCache:_thumbnailImageCache imageDiskUri:imageUri synchronous:synchronous] catch:^SSignal *(__unused id error)
{
SSignal *signal = [SSignal single:nil];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}];
if (withUpdates)
signal = [signal then:updateSignal];
return signal;
}
- (SSignal *)fastImageSignalForItem:(NSObject<TGMediaEditableItem> *)item withUpdates:(bool)withUpdates
{
return [[self thumbnailImageSignalForItem:item withUpdates:false synchronous:true] then:[self imageSignalForItem:item withUpdates:withUpdates]];
}
- (SSignal *)_imageSignalForItemId:(NSString *)itemId imageCache:(TGMemoryImageCache *)imageCache imageDiskUri:(NSString *)imageDiskUri synchronous:(bool)synchronous
{
if (itemId == nil)
return [SSignal fail:nil];
SSignal *signal = [[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
{
void (^completionBlock)(UIImage *) = ^(UIImage *result) {
if (result == nil)
{
NSData *imageData = [_diskCache getValueForKey:[imageDiskUri dataUsingEncoding:NSUTF8StringEncoding]];
if (imageData.length > 0)
{
result = [UIImage imageWithData:imageData];
[imageCache setImage:result forKey:itemId attributes:NULL];
}
}
if (result != nil)
{
[subscriber putNext:result];
[subscriber putCompletion];
}
else
{
[subscriber putError:nil];
}
};
if (synchronous) {
UIImage *result = [imageCache imageForKey:itemId attributes:NULL];
completionBlock(result);
} else {
[imageCache imageForKey:itemId attributes:NULL completion:^(UIImage *result) {
completionBlock(result);
}];
}
return nil;
}];
return synchronous ? signal : [signal startOn:_queue];
}
- (void)_clearPreviousImageForItemId:(NSString *)itemId
{
[_imageCache setImage:nil forKey:itemId attributes:NULL];
NSString *imageUri = [[self class] _imageUriForItemId:itemId];
[_diskCache setValue:[NSData data] forKey:[imageUri dataUsingEncoding:NSUTF8StringEncoding]];
}
- (UIImage *)paintingImageForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
UIImage *result = [_paintingImageCache imageForKey:itemId attributes:NULL];
if (result == nil)
{
NSURL *imageUrl = [_paintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
UIImage *diskImage = [UIImage imageWithContentsOfFile:imageUrl.path];
if (diskImage != nil)
{
result = diskImage;
[_paintingImageCache setImage:result forKey:itemId attributes:NULL];
}
}
return result;
}
- (UIImage *)stillPaintingImageForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
UIImage *result = [_stillPaintingImageCache imageForKey:itemId attributes:NULL];
if (result == nil)
{
NSURL *imageUrl = [_stillPaintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
UIImage *diskImage = [UIImage imageWithContentsOfFile:imageUrl.path];
if (diskImage != nil)
{
result = diskImage;
[_stillPaintingImageCache setImage:result forKey:itemId attributes:NULL];
}
}
return result;
}
#pragma mark - Caption
- (NSAttributedString *)captionForItem:(id<TGMediaEditableItem>)item
{
if (_forcedCaption != nil)
return _forcedCaption;
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return _captions[itemId];
}
- (void)setCaption:(NSAttributedString *)caption forItem:(id<TGMediaEditableItem>)item
{
if (_forcedCaption != nil)
{
_forcedCaption = caption;
_captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:item caption:caption]);
return;
}
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
if (caption.length > 0)
_captions[itemId] = caption;
else
[_captions removeObjectForKey:itemId];
_captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:item caption:caption]);
}
- (bool)isForcedCaption {
return _forcedCaption.string.length > 0;
}
- (SSignal *)forcedCaption
{
__weak TGMediaEditingContext *weakSelf = self;
SSignal *updateSignal = [_captionPipe.signalProducer() map:^NSAttributedString *(TGMediaCaptionUpdate *update)
{
__strong TGMediaEditingContext *strongSelf = weakSelf;
if (strongSelf.isForcedCaption) {
return strongSelf->_forcedCaption;
} else {
return nil;
}
}];
return [[SSignal single:_forcedCaption] then:updateSignal];
}
- (void)setForcedCaption:(NSAttributedString *)caption
{
[self setForcedCaption:caption skipUpdate:false];
}
- (void)setForcedCaption:(NSAttributedString *)caption skipUpdate:(bool)skipUpdate
{
_forcedCaption = caption;
if (!skipUpdate) {
_captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:nil caption:caption]);
}
}
- (SSignal *)captionSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *uniqueIdentifier = item.uniqueIdentifier;
SSignal *updateSignal = [[_captionPipe.signalProducer() filter:^bool(TGMediaCaptionUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:uniqueIdentifier];
}] map:^NSAttributedString *(TGMediaCaptionUpdate *update)
{
return update.caption;
}];
return [[SSignal single:[self captionForItem:item]] then:updateSignal];
}
#pragma mark -
- (id<TGMediaEditAdjustments>)adjustmentsForItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return [self _adjustmentsForItemId:itemId];
}
- (id<TGMediaEditAdjustments>)_adjustmentsForItemId:(NSString *)itemId
{
if (itemId == nil)
return nil;
return _adjustments[itemId];
}
- (void)setAdjustments:(id<TGMediaEditAdjustments>)adjustments forItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
id<TGMediaEditAdjustments> previousAdjustments = _adjustments[itemId];
if (adjustments != nil)
_adjustments[itemId] = adjustments;
else
[_adjustments removeObjectForKey:itemId];
bool cropChanged = false;
if (![previousAdjustments cropAppliedForAvatar:false] && [adjustments cropAppliedForAvatar:false])
cropChanged = true;
else if ([previousAdjustments cropAppliedForAvatar:false] && ![adjustments cropAppliedForAvatar:false])
cropChanged = true;
else if ([previousAdjustments cropAppliedForAvatar:false] && [adjustments cropAppliedForAvatar:false] && ![previousAdjustments isCropEqualWith:adjustments])
cropChanged = true;
if (cropChanged)
_cropPipe.sink(@true);
_adjustmentsPipe.sink([TGMediaAdjustmentsUpdate adjustmentsUpdateWithItem:item adjustments:adjustments]);
}
- (SSignal *)adjustmentsSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
SSignal *updateSignal = [[_adjustmentsPipe.signalProducer() filter:^bool(TGMediaAdjustmentsUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
}] map:^id<TGMediaEditAdjustments>(TGMediaAdjustmentsUpdate *update)
{
return update.adjustments;
}];
return [[SSignal single:[self adjustmentsForItem:item]] then:updateSignal];
}
- (SSignal *)cropAdjustmentsUpdatedSignal
{
return _cropPipe.signalProducer();
}
- (SSignal *)adjustmentsUpdatedSignal
{
return [_adjustmentsPipe.signalProducer() map:^id(__unused id value)
{
return @true;
}];
}
#pragma mark -
- (NSNumber *)timerForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return [self _timerForItemId:itemId];
}
- (NSNumber *)_timerForItemId:(NSString *)itemId
{
if (itemId == nil)
return nil;
return _timers[itemId];
}
- (void)setTimer:(NSNumber *)timer forItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
if (timer.integerValue != 0)
_timers[itemId] = timer;
else
[_timers removeObjectForKey:itemId];
_timerPipe.sink([TGMediaTimerUpdate timerUpdateWithItem:item timer:timer]);
}
- (SSignal *)timerSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
SSignal *updateSignal = [[_timerPipe.signalProducer() filter:^bool(TGMediaTimerUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
}] map:^NSNumber *(TGMediaTimerUpdate *update)
{
return update.timer;
}];
return [[SSignal single:[self timerForItem:item]] then:updateSignal];
}
- (SSignal *)timersUpdatedSignal
{
return [_timerPipe.signalProducer() map:^id(__unused id value)
{
return @true;
}];
}
#pragma mark -
- (bool)spoilerForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return [self _spoilerForItemId:itemId];
}
- (bool)_spoilerForItemId:(NSString *)itemId
{
if (itemId == nil)
return nil;
return _spoilers[itemId];
}
- (void)setSpoiler:(bool)spoiler forItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
if (spoiler)
_spoilers[itemId] = @true;
else
[_spoilers removeObjectForKey:itemId];
_spoilerPipe.sink([TGMediaSpoilerUpdate spoilerUpdateWithItem:item spoiler:spoiler]);
}
- (SSignal *)spoilerSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
SSignal *updateSignal = [[_spoilerPipe.signalProducer() filter:^bool(TGMediaSpoilerUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
}] map:^NSNumber *(TGMediaSpoilerUpdate *update)
{
return @(update.spoiler);
}];
return [[SSignal single:@([self spoilerForItem:item])] then:updateSignal];
}
- (SSignal *)spoilerSignalForIdentifier:(NSString *)identifier
{
SSignal *updateSignal = [[_spoilerPipe.signalProducer() filter:^bool(TGMediaSpoilerUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:identifier];
}] map:^NSNumber *(TGMediaSpoilerUpdate *update)
{
return @(update.spoiler);
}];
return [[SSignal single:@([self _spoilerForItemId:identifier])] then:updateSignal];
}
- (SSignal *)spoilersUpdatedSignal
{
return [_spoilerPipe.signalProducer() map:^id(__unused id value)
{
return @true;
}];
}
#pragma mark -
- (NSNumber *)priceForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return [self _priceForItemId:itemId];
}
- (NSNumber *)_priceForItemId:(NSString *)itemId
{
if (itemId == nil)
return nil;
return _prices[itemId];
}
- (void)setPrice:(NSNumber *)price forItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
if (price.integerValue != 0)
_prices[itemId] = price;
else
[_prices removeObjectForKey:itemId];
_pricePipe.sink([TGMediaPriceUpdate priceUpdateWithItem:item price:price]);
}
- (SSignal *)priceSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
SSignal *updateSignal = [[_pricePipe.signalProducer() filter:^bool(TGMediaPriceUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
}] map:^NSNumber *(TGMediaPriceUpdate *update)
{
return update.price;
}];
return [[SSignal single:[self priceForItem:item]] then:updateSignal];
}
- (SSignal *)priceSignalForIdentifier:(NSString *)identifier
{
SSignal *updateSignal = [[_pricePipe.signalProducer() filter:^bool(TGMediaPriceUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:identifier];
}] map:^NSNumber *(TGMediaPriceUpdate *update)
{
return update.price;
}];
return [[SSignal single:[self _priceForItemId:identifier]] then:updateSignal];
}
- (SSignal *)pricesUpdatedSignal
{
return [_pricePipe.signalProducer() map:^id(__unused id value)
{
return @true;
}];
}
#pragma mark -
- (void)setImage:(UIImage *)image thumbnailImage:(UIImage *)thumbnailImage forItem:(id<TGMediaEditableItem>)item synchronous:(bool)synchronous
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
void (^block)(void) = ^
{
[_temporaryRepCache removeObjectForKey:itemId];
NSString *imageUri = [[self class] _imageUriForItemId:itemId];
[_imageCache setImage:image forKey:itemId attributes:NULL];
if (image != nil)
{
NSData *imageData = UIImageJPEGRepresentation(image, 0.95f);
[_diskCache setValue:imageData forKey:[imageUri dataUsingEncoding:NSUTF8StringEncoding]];
}
_representationPipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:image]);
NSString *thumbnailImageUri = [[self class] _thumbnailImageUriForItemId:itemId];
[_thumbnailImageCache setImage:thumbnailImage forKey:itemId attributes:NULL];
if (thumbnailImage != nil)
{
NSData *imageData = UIImageJPEGRepresentation(thumbnailImage, 0.87f);
[_diskCache setValue:imageData forKey:[thumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
}
if ([item isKindOfClass:[TGMediaAsset class]] && ((TGMediaAsset *)item).isVideo)
_thumbnailImagePipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:thumbnailImage]);
};
if (synchronous)
[_queue dispatchSync:block];
else
[_queue dispatch:block];
}
- (bool)setPaintingData:(NSData *)data entitiesData:(NSData *)entitiesData image:(UIImage *)image stillImage:(UIImage *)stillImage forItem:(NSObject<TGMediaEditableItem> *)item dataUrl:(NSURL **)dataOutUrl entitiesDataUrl:(NSURL **)entitiesDataOutUrl imageUrl:(NSURL **)imageOutUrl forVideo:(bool)video
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return false;
NSURL *imagesDirectory = video ? _videoPaintingImagesUrl : _paintingImagesUrl;
NSURL *imageUrl = [imagesDirectory URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
NSURL *dataUrl = [_paintingDatasUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.dat", [TGStringUtils md5:itemId]]];
NSURL *entitiesDataUrl = [_paintingDatasUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@_entities.dat", [TGStringUtils md5:itemId]]];
[_paintingImageCache setImage:image forKey:itemId attributes:NULL];
NSData *imageData = UIImagePNGRepresentation(image);
[[NSFileManager defaultManager] removeItemAtURL:imageUrl error:nil];
bool imageSuccess = [imageData writeToURL:imageUrl options:NSDataWritingAtomic error:nil];
[[NSFileManager defaultManager] removeItemAtURL:dataUrl error:nil];
bool dataSuccess = [data writeToURL:dataUrl options:NSDataWritingAtomic error:nil];
bool entitiesDataSuccess = [entitiesData writeToURL:entitiesDataUrl options:NSDataWritingAtomic error:nil];
if (imageSuccess && imageOutUrl != NULL)
*imageOutUrl = imageUrl;
if (dataSuccess && dataOutUrl != NULL)
*dataOutUrl = dataUrl;
if (entitiesDataSuccess && entitiesDataOutUrl != NULL)
*entitiesDataOutUrl = entitiesDataUrl;
if (video)
[_storeVideoPaintingImages addObject:imageUrl];
if (stillImage != nil) {
NSURL *stillImageUrl = [_stillPaintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
[_stillPaintingImageCache setImage:stillImage forKey:itemId attributes:NULL];
NSData *stillImageData = UIImagePNGRepresentation(stillImage);
[[NSFileManager defaultManager] removeItemAtURL:stillImageUrl error:nil];
[stillImageData writeToURL:stillImageUrl options:NSDataWritingAtomic error:nil];
if (video)
[_storeVideoPaintingImages addObject:stillImageUrl];
}
return (image == nil || imageSuccess) && (data == nil || dataSuccess);
}
- (void)clearPaintingData
{
for (NSURL *url in _storeVideoPaintingImages)
{
[[NSFileManager defaultManager] removeItemAtURL:url error:NULL];
}
}
- (bool)isCaptionAbove {
return _captionAbove;
}
- (SSignal *)captionAbove
{
__weak TGMediaEditingContext *weakSelf = self;
SSignal *updateSignal = [_captionAbovePipe.signalProducer() map:^NSNumber *(NSNumber *update)
{
__strong TGMediaEditingContext *strongSelf = weakSelf;
return @(strongSelf->_captionAbove);
}];
return [[SSignal single:@(_captionAbove)] then:updateSignal];
}
- (void)setCaptionAbove:(bool)captionAbove
{
_captionAbove = captionAbove;
_captionAbovePipe.sink(@(captionAbove));
}
- (SSignal *)facesForItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return [SSignal fail:nil];
NSArray *faces = _faces[itemId];
return [[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
{
[subscriber putNext:faces];
[subscriber putCompletion];
return nil;
}];
}
- (void)setFaces:(NSArray *)faces forItem:(NSObject<TGMediaEditableItem> *)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
if (faces.count > 0)
_faces[itemId] = faces;
else
[_faces removeObjectForKey:itemId];
}
- (SSignal *)coverImageSignalForIdentifier:(NSString *)identifier
{
NSString *itemId = [TGMediaEditingContext _coverImageUriForItemId:identifier];
if (itemId == nil)
return [SSignal fail:nil];
SSignal *updateSignal = [[_coverImagePipe.signalProducer() filter:^bool(TGMediaImageUpdate *update)
{
return [update.item.uniqueIdentifier isEqualToString:identifier];
}] map:^id(TGMediaImageUpdate *update)
{
return update.representation;
}];
return [[SSignal single:[_coverImageCache imageForKey:itemId attributes:NULL]]
then:updateSignal];
}
- (SSignal *)coverImageSignalForItem:(NSObject<TGMediaEditableItem> *)item {
return [self coverImageSignalForIdentifier:item.uniqueIdentifier];
}
- (UIImage *)coverImageForItem:(NSObject<TGMediaEditableItem> *)item {
NSString *itemId = [TGMediaEditingContext _coverImageUriForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return [_coverImageCache imageForKey:itemId attributes:NULL];
}
- (NSNumber *)coverPositionForItem:(NSObject<TGMediaEditableItem> *)item {
NSString *itemId = [TGMediaEditingContext _coverImageUriForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
return _coverPositions[itemId];
}
- (void)setCoverImage:(UIImage *)image position:(NSNumber *)position forItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [TGMediaEditingContext _coverImageUriForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
[_coverImageCache setImage:image forKey:itemId attributes:NULL];
_coverImagePipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:image]);
if (position != nil) {
[_coverPositions setObject:position forKey:itemId];
} else {
[_coverPositions removeObjectForKey:itemId];
}
}
- (void)setFullSizeImage:(UIImage *)image forItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
NSData *imageData = UIImageJPEGRepresentation(image, 0.7f);
NSURL *url = [_fullSizeResultsUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.jpg", [TGStringUtils md5:itemId]]];
bool succeed = [imageData writeToURL:url options:NSDataWritingAtomic error:nil];
if (succeed)
_fullSizePipe.sink(itemId);
}
- (NSURL *)_fullSizeImageUrlForItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return nil;
NSURL *url = [_fullSizeResultsUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.jpg", [TGStringUtils md5:itemId]]];
if ([[NSFileManager defaultManager] fileExistsAtPath:url.path])
return url;
return nil;
}
- (SSignal *)fullSizeImageUrlForItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
id<TGMediaEditAdjustments> adjustments = [self adjustmentsForItem:item];
if (![adjustments isKindOfClass:[PGPhotoEditorValues class]])
return [SSignal complete];
PGPhotoEditorValues *editorValues = (PGPhotoEditorValues *)adjustments;
if (![editorValues toolsApplied] && ![editorValues hasPainting])
return [SSignal complete];
if ([editorValues.paintingData hasAnimation])
return [SSignal complete];
NSURL *url = [self _fullSizeImageUrlForItem:item];
if (url != nil)
return [SSignal single:url];
return [[[_fullSizePipe.signalProducer() filter:^bool(NSString *identifier)
{
return [identifier isEqualToString:itemId];
}] mapToSignal:^SSignal *(__unused id next)
{
NSURL *url = [self _fullSizeImageUrlForItem:item];
if (url != nil)
return [SSignal single:url];
else
return [SSignal complete];
}] timeout:5.0 onQueue:_queue orSignal:[SSignal complete]];
}
- (void)setTemporaryRep:(id)rep forItem:(id<TGMediaEditableItem>)item
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
return;
UIImage *thumbnailImage = nil;
if ([rep isKindOfClass:[UIImage class]])
{
UIImage *image = (UIImage *)rep;
image.degraded = true;
image.edited = true;
CGSize fillSize = TGPhotoThumbnailSizeForCurrentScreen();
fillSize.width = CGCeil(fillSize.width);
fillSize.height = CGCeil(fillSize.height);
CGSize size = TGScaleToFillSize(image.size, fillSize);
UIGraphicsBeginImageContextWithOptions(size, true, 0.0f);
CGContextRef context = UIGraphicsGetCurrentContext();
CGContextSetInterpolationQuality(context, kCGInterpolationMedium);
[image drawInRect:CGRectMake(0, 0, size.width, size.height)];
thumbnailImage = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();
}
[_queue dispatchSync:^
{
[self _clearPreviousImageForItemId:itemId];
if (rep != nil)
[_temporaryRepCache setObject:rep forKey:itemId];
else
[_temporaryRepCache removeObjectForKey:itemId];
_representationPipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:rep]);
if (thumbnailImage != nil)
{
[_thumbnailImageCache setImage:thumbnailImage forKey:itemId attributes:NULL];
_thumbnailImagePipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:thumbnailImage]);
}
}];
}
#pragma mark - Original Images
- (void)requestOriginalImageForItem:(id<TGMediaEditableItem>)item completion:(void (^)(UIImage *))completion
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
{
if (completion != nil)
completion(nil);
return;
}
__block UIImage *result = [_originalImageCache imageForKey:itemId attributes:NULL];
if (result != nil)
{
if (completion != nil)
completion(result);
}
else
{
[_queue dispatch:^
{
NSString *originalImageUri = [[self class] _originalImageUriForItemId:itemId];
NSData *imageData = [_diskCache getValueForKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
if (imageData != nil)
{
result = [UIImage imageWithData:imageData];
[_originalImageCache setImage:result forKey:itemId attributes:NULL];
}
if (completion != nil)
completion(result);
}];
}
}
- (void)requestOriginalThumbnailImageForItem:(id<TGMediaEditableItem>)item completion:(void (^)(UIImage *))completion
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil)
{
if (completion != nil)
completion(nil);
return;
}
__block UIImage *result = [_originalThumbnailImageCache imageForKey:itemId attributes:NULL];
if (result != nil)
{
if (completion != nil)
completion(result);
}
else
{
[_queue dispatch:^
{
NSString *originalThumbnailImageUri = [[self class] _originalThumbnailImageUriForItemId:itemId];
NSData *imageData = [_diskCache getValueForKey:[originalThumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
if (imageData != nil)
{
result = [UIImage imageWithData:imageData];
[_originalThumbnailImageCache setImage:result forKey:itemId attributes:NULL];
}
if (completion != nil)
completion(result);
}];
}
}
- (void)setOriginalImage:(UIImage *)image forItem:(id<TGMediaEditableItem>)item synchronous:(bool)synchronous
{
NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
if (itemId == nil || image == nil)
return;
if ([_originalImageCache imageForKey:itemId attributes:NULL] != nil)
return;
void (^block)(void) = ^
{
if (image != nil)
{
NSString *originalImageUri = [[self class] _originalImageUriForItemId:itemId];
NSData *existingImageData = [_diskCache getValueForKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
if (existingImageData.length > 0)
return;
[_originalImageCache setImage:image forKey:itemId attributes:NULL];
NSData *imageData = UIImageJPEGRepresentation(image, 0.95f);
[_diskCache setValue:imageData forKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
CGFloat thumbnailImageSide = TGPhotoThumbnailSizeForCurrentScreen().width;
CGSize targetSize = TGScaleToSize(image.size, CGSizeMake(thumbnailImageSide, thumbnailImageSide));
UIGraphicsBeginImageContextWithOptions(targetSize, true, 0.0f);
[image drawInRect:CGRectMake(0, 0, targetSize.width, targetSize.height)];
UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();
[_originalThumbnailImageCache setImage:image forKey:itemId attributes:NULL];
NSString *originalThumbnailImageUri = [[self class] _originalThumbnailImageUriForItemId:itemId];
NSData *thumbnailImageData = UIImageJPEGRepresentation(image, 0.87f);
[_diskCache setValue:thumbnailImageData forKey:[originalThumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
}
};
if (synchronous)
[_queue dispatchSync:block];
else
[_queue dispatch:block];
}
+ (NSString *)_originalImageUriForItemId:(NSString *)itemId
{
return [NSString stringWithFormat:@"photo-editor-original://%@", itemId];
}
+ (NSString *)_originalThumbnailImageUriForItemId:(NSString *)itemId
{
return [NSString stringWithFormat:@"photo-editor-original-thumb://%@", itemId];
}
#pragma mark - URI
- (NSString *)_contextualIdForItemId:(NSString *)itemId
{
if (itemId == nil)
return nil;
return itemId;
//return [NSString stringWithFormat:@"%@hm%@", _contextId, itemId];
}
+ (NSString *)_imageUriForItemId:(NSString *)itemId
{
return [NSString stringWithFormat:@"%@://%@", [self imageUriScheme], itemId];
}
+ (NSString *)_thumbnailImageUriForItemId:(NSString *)itemId
{
return [NSString stringWithFormat:@"%@://%@", [self thumbnailImageUriScheme], itemId];
}
+ (NSString *)_coverImageUriForItemId:(NSString *)itemId
{
return [NSString stringWithFormat:@"%@://%@", @"photo-editor-cover", itemId];
}
#pragma mark - Constants
+ (NSString *)imageUriScheme
{
return @"photo-editor";
}
+ (NSString *)thumbnailImageUriScheme
{
return @"photo-editor-thumb";
}
+ (NSString *)diskCachePath
{
return @"photoeditorcache_v1";
}
+ (NSUInteger)diskMemoryLimit
{
return 512 * 1024 * 1024;
}
+ (NSUInteger)imageSoftMemoryLimit
{
return 13 * 1024 * 1024;
}
+ (NSUInteger)imageHardMemoryLimit
{
return 15 * 1024 * 1024;
}
+ (NSUInteger)originalImageSoftMemoryLimit
{
return 12 * 1024 * 1024;
}
+ (NSUInteger)originalImageHardMemoryLimit
{
return 14 * 1024 * 1024;
}
+ (NSUInteger)thumbnailImageSoftMemoryLimit
{
return 2 * 1024 * 1024;
}
+ (NSUInteger)thumbnailImageHardMemoryLimit
{
return 3 * 1024 * 1024;
}
@end
@implementation TGMediaImageUpdate
+ (instancetype)imageUpdateWithItem:(id<TGMediaEditableItem>)item representation:(id)representation
{
TGMediaImageUpdate *update = [[TGMediaImageUpdate alloc] init];
update->_item = item;
update->_representation = representation;
return update;
}
@end
@implementation TGMediaAdjustmentsUpdate
+ (instancetype)adjustmentsUpdateWithItem:(id<TGMediaEditableItem>)item adjustments:(id<TGMediaEditAdjustments>)adjustments
{
TGMediaAdjustmentsUpdate *update = [[TGMediaAdjustmentsUpdate alloc] init];
update->_item = item;
update->_adjustments = adjustments;
return update;
}
@end
@implementation TGMediaCaptionUpdate
+ (instancetype)captionUpdateWithItem:(id<TGMediaEditableItem>)item caption:(NSAttributedString *)caption
{
TGMediaCaptionUpdate *update = [[TGMediaCaptionUpdate alloc] init];
update->_item = item;
update->_caption = caption;
return update;
}
@end
@implementation TGMediaTimerUpdate
+ (instancetype)timerUpdateWithItem:(id<TGMediaEditableItem>)item timer:(NSNumber *)timer
{
TGMediaTimerUpdate *update = [[TGMediaTimerUpdate alloc] init];
update->_item = item;
update->_timer = timer;
return update;
}
+ (instancetype)timerUpdate:(NSNumber *)timer
{
TGMediaTimerUpdate *update = [[TGMediaTimerUpdate alloc] init];
update->_timer = timer;
return update;
}
@end
@implementation TGMediaSpoilerUpdate
+ (instancetype)spoilerUpdateWithItem:(id<TGMediaEditableItem>)item spoiler:(bool)spoiler
{
TGMediaSpoilerUpdate *update = [[TGMediaSpoilerUpdate alloc] init];
update->_item = item;
update->_spoiler = spoiler;
return update;
}
+ (instancetype)spoilerUpdate:(bool)spoiler
{
TGMediaSpoilerUpdate *update = [[TGMediaSpoilerUpdate alloc] init];
update->_spoiler = spoiler;
return update;
}
@end
@implementation TGMediaPriceUpdate
+ (instancetype)priceUpdateWithItem:(id<TGMediaEditableItem>)item price:(NSNumber *)price
{
TGMediaPriceUpdate *update = [[TGMediaPriceUpdate alloc] init];
update->_item = item;
update->_price = price;
return update;
}
+ (instancetype)priceUpdate:(NSNumber *)price
{
TGMediaPriceUpdate *update = [[TGMediaPriceUpdate alloc] init];
update->_price = price;
return update;
}
@end