Swiftgram/Source/Private/ASRectMap.mm
Adlai Holler 2b12654810
Use NS_RETURNS_RETAINED macro to save time (#843)
* Use NS_RETURNS_RETAINED macro to save time

* Update changelog and do the thing with the license headers
2018-03-21 08:13:09 -07:00

79 lines
1.7 KiB
Plaintext

//
// ASRectMap.mm
// Texture
//
// Copyright (c) 2017-present, Pinterest, Inc. All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
#import "ASRectMap.h"
#import "ASObjectDescriptionHelpers.h"
#import <UIKit/UIGeometry.h>
#import <unordered_map>
@implementation ASRectMap {
std::unordered_map<void *, CGRect> _map;
}
+ (ASRectMap *)rectMapForWeakObjectPointers NS_RETURNS_RETAINED
{
return [[self alloc] init];
}
- (CGRect)rectForKey:(id)key
{
auto result = _map.find((__bridge void *)key);
if (result != _map.end()) {
// result->first is the key; result->second is the value, a CGRect.
return result->second;
} else {
return CGRectNull;
}
}
- (void)setRect:(CGRect)rect forKey:(id)key
{
if (key) {
_map[(__bridge void *)key] = rect;
}
}
- (void)removeRectForKey:(id)key
{
if (key) {
_map.erase((__bridge void *)key);
}
}
- (id)copyWithZone:(NSZone *)zone
{
ASRectMap *copy = [ASRectMap rectMapForWeakObjectPointers];
copy->_map = _map;
return copy;
}
- (NSMutableArray<NSDictionary *> *)propertiesForDescription
{
NSMutableArray *result = [NSMutableArray array];
// { ptr1->rect1 ptr2->rect2 ptr3->rect3 }
NSMutableString *str = [NSMutableString string];
for (auto it = _map.begin(); it != _map.end(); it++) {
[str appendFormat:@" %@->%@", it->first, NSStringFromCGRect(it->second)];
}
[result addObject:@{ @"ASRectMap": str }];
return result;
}
- (NSString *)description
{
return ASObjectDescriptionMakeWithoutObject([self propertiesForDescription]);
}
@end