/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */
#import <Foundation/Foundation.h>
#import "MyEnumerator.h"

// init array

void nsarray_empty_array_constant() {
  NSArray* array = [NSArray array];

  for (int i = 0; i < array.count; i++) {
  }
}

void nsarray_init_constant() {
  NSArray* array = [[NSArray alloc] init];

  for (int i = 0; i < array.count; i++) {
  }
}

void nsarray_init_with_array_linear(NSArray* array) {
  NSArray* ref_array = [[NSArray alloc] initWithArray:array];

  for (int i = 0; i < ref_array.count; i++) {
  }
}

void nsarray_init_with_array_constant() {
  NSArray* array = [[NSArray alloc] init];
  nsarray_init_with_array_linear(array);
}

void nsarray_init_with_array_copy_linear(NSArray* array) {
  NSArray* copy_array = [[NSArray alloc] initWithArray:array copyItems:YES];
  for (int i = 0; i < copy_array.count; i++) {
  }
}

NSArray* nsarray_init_with_objects_constant() {
  NSString* strings[3];
  strings[0] = @"First";
  strings[1] = @"Second";
  strings[2] = @"Third";

  NSArray* array = [NSArray arrayWithObjects:strings count:2];
  for (int i = 0; i < array.count; i++) {
  }
}

NSArray* nsarray_array_with_objects_constant() {
  NSDate* aDate = [NSDate distantFuture];
  NSValue* aValue = @(5);
  NSString* aString = @"hello";

  NSArray* array = [NSArray arrayWithObjects:aDate, aValue, aString, nil];
  for (id item in array) {
  }

  return array;
}

// derive new array

NSArray* nsarray_add_object_constant(id obj) {
  NSArray* array = [[NSArray alloc] init];
  return [array arrayByAddingObject:obj];
}

NSArray* nsarray_add_objects_from_array_linear(NSArray* append_array) {
  NSArray* array = [[NSArray alloc] init];
  NSArray* new_array = [array arrayByAddingObjectsFromArray:append_array];

  for (int i = 0; i < new_array.count; i++) {
  }
}

// query element

void nsarray_get_first_constant() {
  NSArray* array = @[ @1, @2 ];
  for (int i = 0; i < [[array firstObject] integerValue]; i++) {
  }
}

void nsarray_access_constant() {
  NSArray* array = @[ @1.0f, @2.0f, @3.0f, @4.0f, @5.0f, @6.0f, @7.0f, @8.0f ];
  for (int i = 0; i < 4; i++) {
    [array objectAtIndex:i];
  }
}

void nsarray_object_at_indexed_constant() {
  NSArray* array = @[ @2, @3 ];

  for (int i = 0; i < [array[0] integerValue]; i++) {
  }
}

void nsarray_access_linear(NSArray* array) {
  id obj;
  for (int i = 0; i < array.count; i++) {
    obj = array[i];
  }
}

void nsarray_find_linear(NSArray* array) {
  for (int i = 0; i < array.count && i != [array[i] integerValue]; i++) {
  }
}

void nsarray_contains_object_linear(NSArray* array) {
  [array containsObject:@1];
}

id nsarray_first_object_constant(NSArray* array) { return array.firstObject; }

id nsarray_last_object_constant(NSArray* array) { return array.lastObject; }

// find element
NSInteger nsarray_binary_search_log_FN(NSArray* sorted_array) {
  NSNumber* target = @5;
  return [sorted_array indexOfObject:target
                       inSortedRange:NSMakeRange(0, sorted_array.count)
                             options:NSBinarySearchingFirstEqual
                     usingComparator:^(id lhs, id rhs) {
                       return [lhs compare:rhs];
                     }];
}

// sort array

NSArray* nsarray_sort_using_descriptors_constant() {
  NSArray* array = @[ @"Grapes", @"Apples", @"Oranges" ];
  NSSortDescriptor* sd = [[NSSortDescriptor alloc] initWithKey:nil
                                                     ascending:YES];
  return [array sortedArrayUsingDescriptors:@[ sd ]];
}

NSArray* nsarray_sort_using_descriptors_nlogn(NSArray* array) {
  NSSortDescriptor* sd = [[NSSortDescriptor alloc] initWithKey:nil
                                                     ascending:YES];
  return [array sortedArrayUsingDescriptors:@[ sd ]];
}

// iterate through array

void nsarray_iterate_linear(NSArray* array) {
  NSInteger sum = 0;
  for (id obj in array) {
    sum += (NSInteger)obj;
  }
}

void nsarray_enumerator_linear(NSArray* array) {
  NSEnumerator* enumerator = [array objectEnumerator];

  id obj;
  NSInteger sum = 0;

  while (obj = [enumerator nextObject]) {
    sum += (NSInteger)obj;
  }
}

void nsarray_enumerator_param_linear(NSEnumerator* enumerator) {
  id obj;
  while (obj = [enumerator nextObject]) {
  }
}

void multiple_nsarray_enumerators_param_linear(bool b,
                                               NSEnumerator* enumerator1,
                                               NSEnumerator* enumerator2) {
  NSEnumerator* enumerator = b ? enumerator1 : enumerator2;
  id obj;
  while (obj = [enumerator nextObject]) {
  }
}

void call_nsarray_enumerator_param_linear(NSArray* array) {
  NSEnumerator* enumerator = [array objectEnumerator];
  nsarray_enumerator_param_linear_FN(enumerator);
}

void nsarray_next_object_linear(NSArray* array) {
  for (id item in array) {
  }
}

void nsarray_next_object_constant() {
  NSArray* array = @[ @1, @2 ];
  for (id item in array) {
  }
}

// compare array
boolean_t nsarray_is_equal_to_array_linear(NSArray* array1, NSArray* array2) {
  return [array1 isEqualToArray:array2];
}

// count

void nsarray_count_bounded_linear(NSArray* array) {
  for (int i = 0; i < array.count; i++) {
  }
}

void nsarray_copy_linear(NSArray* array) {
  NSArray* copy = [array mutableCopy];
  for (int i = 0; i < copy.count; i++) {
  }
}

void call_my_enumerator_next_object_linear(MyEnumerator* enumerator) {
  // NSEnumerator.nextObject should be replaced to MyEnumerator.nextObject
  NSString* s = [enumerator nextObject];
}

// The cost analyzer cannot reason the amortized complexity.
void loop_with_my_enumerator_next_object_linear_FP(MyEnumerator* enumerator) {
  NSString* s;
  while (s = [enumerator nextObject]) {
  }
}

void enumerate_via_block_linear(NSArray* array) {
  [array enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL* stop){
    NSLog(@"index: %@", index);
  }];
}

@interface MyBlock : NSObject
@end

@implementation MyBlock

+ (void)call_enumerate_via_block_param_quadratic:(NSArray*)x:(int)size {
  void (^b)(id, NSUInteger, BOOL*) =
      ^(id object, NSUInteger indexPath, BOOL* stop) {
        for (int i = 0; i < size; i++) {
        }
      };
  // Here, captured arg size is passed as the first argument to the block,
  // not the array x.
  [x enumerateObjectsUsingBlock:b];
}

@end