session-ios/Signal/src/view controllers/TSAttachmentAdapter.m

//
//  TSAttachementAdapter.m
//  Signal
//
//  Created by Frederic Jacobs on 17/12/14.
//  Copyright (c) 2014 Open Whisper Systems. All rights reserved.
//

#import "TSAttachmentAdapter.h"

#import "UIDevice+TSHardwareVersion.h"
#import "JSQMessagesMediaViewBubbleImageMasker.h"

@interface TSAttachmentAdapter ()

@property UIImage *image;

@property (strong, nonatomic) UIImageView *cachedImageView;
@property (assign, nonatomic, readonly) BOOL isImageAttachment;

@end

@implementation TSAttachmentAdapter

- (instancetype)initWithAttachment:(TSAttachmentStream*)attachment{
    self = [super init];
    
    if (self) {
        _image = [UIImage imageWithCGImage:attachment.image.CGImage];
        _cachedImageView = nil;
        _isImageAttachment = YES;
    }
    return self;
}

- (void)dealloc
{
    _image = nil;
    _cachedImageView = nil;
}

- (void)setAppliesMediaViewMaskAsOutgoing:(BOOL)appliesMediaViewMaskAsOutgoing
{
    [super setAppliesMediaViewMaskAsOutgoing:appliesMediaViewMaskAsOutgoing];
    _cachedImageView = nil;
}

#pragma mark - JSQMessageMediaData protocol

- (UIView *)mediaView
{
    if (self.image == nil) {
        return nil;
    }
    
    if (self.cachedImageView == nil) {
        CGSize size = [self mediaViewDisplaySize];
        UIImageView *imageView = [[UIImageView alloc] initWithImage:self.image];
        imageView.frame = CGRectMake(0.0f, 0.0f, size.width, size.height);
        imageView.contentMode = UIViewContentModeScaleAspectFill;
        imageView.clipsToBounds = YES;
        [JSQMessagesMediaViewBubbleImageMasker applyBubbleImageMaskToMediaView:imageView isOutgoing:self.appliesMediaViewMaskAsOutgoing];
        self.cachedImageView = imageView;
    }
    
    return self.cachedImageView;
}

- (CGSize)mediaViewDisplaySize
{
    return [self getBubbleSizeForImage:_image];
}

-(BOOL)isImage
{
    return _isImageAttachment;
}

#pragma mark - Utility

-(CGSize)getBubbleSizeForImage:(UIImage*)image
{
    CGFloat aspectRatio = image.size.height / image.size.width ;
    
    if ([[UIDevice currentDevice] isiPhoneVersionSixOrMore])
    {
        return [self getLargeSizeForAspectRatio:aspectRatio];
    } else {
        return [self getSmallSizeForAspectRatio:aspectRatio];
    }
}

-(CGSize)getLargeSizeForAspectRatio:(CGFloat)ratio
{
    return ratio > 1.0f ? [self largePortraitSize] : [self largeLandscapeSize];
}

-(CGSize)getSmallSizeForAspectRatio:(CGFloat)ratio
{
    return ratio > 1.0f ? [self smallPortraitSize] : [self smallLandscapeSize];
}


- (CGSize)largePortraitSize
{
    return CGSizeMake(220.0f, 310.0f);
}

- (CGSize)smallPortraitSize
{
    return CGSizeMake(150.0f, 210.0f);
}

- (CGSize)largeLandscapeSize
{
    return CGSizeMake(310.0f, 220.0f);
}

- (CGSize)smallLandscapeSize
{
    return CGSizeMake(210.0f, 150.0f);
}

@end