//
//  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
{
  let 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 (let &e : _map) {
    [str appendFormat:@" %@->%@", e.first, NSStringFromCGRect(e.second)];
  }
  [result addObject:@{ @"ASRectMap": str }];

  return result;
}

- (NSString *)description
{
  return ASObjectDescriptionMakeWithoutObject([self propertiesForDescription]);
}

@end