react-native/ReactKit/Base/RCTSparseArray.m

// Copyright 2004-present Facebook. All Rights Reserved.

#import "RCTSparseArray.h"

@implementation RCTSparseArray
{
  NSMutableDictionary *_storage;
}

- (instancetype)init
{
  return [self initWithCapacity:0];
}

- (instancetype)initWithCapacity:(NSUInteger)capacity
{
  if (self = [super init]) {
    _storage = [NSMutableDictionary dictionaryWithCapacity:capacity];
  }

  return self;
}

- (instancetype)initWithSparseArray:(RCTSparseArray *)sparseArray
{
  if (self = [super init]) {
    _storage = [sparseArray->_storage copy];
  }

  return self;
}

- (instancetype)initWithStorage:(NSDictionary *)storage
{
  if ((self = [super init])) {
    _storage = [storage copy];
  }
  return self;
}

+ (instancetype)sparseArray
{
  return [[self alloc] init];
}

+ (instancetype)sparseArrayWithCapacity:(NSUInteger)capacity
{
  return [[self alloc] initWithCapacity:capacity];
}

+ (instancetype)sparseArrayWithSparseArray:(RCTSparseArray *)sparseArray
{
  return [[self alloc] initWithSparseArray:sparseArray];
}

- (id)objectAtIndexedSubscript:(NSUInteger)idx
{
  return _storage[@(idx)];
}

- (void)setObject:(id)obj atIndexedSubscript:(NSUInteger)idx
{
  self[@(idx)] = obj;
}

- (id)objectForKeyedSubscript:(NSNumber *)key
{
  return _storage[key];
}

- (void)setObject:(id)obj forKeyedSubscript:(NSNumber *)key
{
  if (obj) {
    _storage[key] = obj;
  } else {
    [_storage removeObjectForKey:key];
  }
}

- (NSUInteger)count
{
  return _storage.count;
}

- (NSArray *)allIndexes
{
  return _storage.allKeys;
}

- (NSArray *)allObjects
{
  return _storage.allValues;
}

- (void)enumerateObjectsUsingBlock:(void (^)(id obj, NSNumber *idx, BOOL *stop))block
{
  NSParameterAssert(block != nil);
  [_storage enumerateKeysAndObjectsUsingBlock:^(NSNumber *key, id obj, BOOL *stop) {
    block(obj, key, stop);
  }];
}

- (void)enumerateObjectsWithOptions:(NSEnumerationOptions)opts usingBlock:(void (^)(id obj, NSNumber *idx, BOOL *stop))block
{
  NSParameterAssert(block != nil);
  [_storage enumerateKeysAndObjectsWithOptions:opts usingBlock:^(NSNumber *key, id obj, BOOL *stop) {
    block(obj, key, stop);
  }];
}

- (id)copyWithZone:(NSZone *)zone
{
  return [[[self class] allocWithZone:zone] initWithSparseArray:self];
}

@end