NSArray+YYAdd.m 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. //
  2. // NSArray+YYAdd.m
  3. // YYKit <https://github.com/ibireme/YYKit>
  4. //
  5. // Created by ibireme on 13/4/4.
  6. // Copyright (c) 2015 ibireme.
  7. //
  8. // This source code is licensed under the MIT-style license found in the
  9. // LICENSE file in the root directory of this source tree.
  10. //
  11. #import "NSArray+YYAdd.h"
  12. #import "YYKitMacro.h"
  13. #import "NSData+YYAdd.h"
  14. YYSYNTH_DUMMY_CLASS(NSArray_YYAdd)
  15. @implementation NSArray (YYAdd)
  16. + (NSArray *)arrayWithPlistData:(NSData *)plist {
  17. if (!plist) return nil;
  18. NSArray *array = [NSPropertyListSerialization propertyListWithData:plist options:NSPropertyListImmutable format:NULL error:NULL];
  19. if ([array isKindOfClass:[NSArray class]]) return array;
  20. return nil;
  21. }
  22. + (NSArray *)arrayWithPlistString:(NSString *)plist {
  23. if (!plist) return nil;
  24. NSData *data = [plist dataUsingEncoding:NSUTF8StringEncoding];
  25. return [self arrayWithPlistData:data];
  26. }
  27. - (NSData *)plistData {
  28. return [NSPropertyListSerialization dataWithPropertyList:self format:NSPropertyListBinaryFormat_v1_0 options:kNilOptions error:NULL];
  29. }
  30. - (NSString *)plistString {
  31. NSData *xmlData = [NSPropertyListSerialization dataWithPropertyList:self format:NSPropertyListXMLFormat_v1_0 options:kNilOptions error:NULL];
  32. if (xmlData) return xmlData.utf8String;
  33. return nil;
  34. }
  35. - (id)randomObject {
  36. if (self.count) {
  37. return self[arc4random_uniform((u_int32_t)self.count)];
  38. }
  39. return nil;
  40. }
  41. - (id)objectOrNilAtIndex:(NSUInteger)index {
  42. return index < self.count ? self[index] : nil;
  43. }
  44. - (NSString *)jsonStringEncoded {
  45. if ([NSJSONSerialization isValidJSONObject:self]) {
  46. NSError *error = nil;
  47. NSData *jsonData = [NSJSONSerialization dataWithJSONObject:self options:0 error:&error];
  48. NSString *json = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
  49. if (!error) return json;
  50. }
  51. return nil;
  52. }
  53. - (NSString *)jsonPrettyStringEncoded {
  54. if ([NSJSONSerialization isValidJSONObject:self]) {
  55. NSError *error = nil;
  56. NSData *jsonData = [NSJSONSerialization dataWithJSONObject:self options:NSJSONWritingPrettyPrinted error:&error];
  57. NSString *json = [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
  58. if (!error) return json;
  59. }
  60. return nil;
  61. }
  62. @end
  63. @implementation NSMutableArray (YYAdd)
  64. + (NSMutableArray *)arrayWithPlistData:(NSData *)plist {
  65. if (!plist) return nil;
  66. NSMutableArray *array = [NSPropertyListSerialization propertyListWithData:plist options:NSPropertyListMutableContainersAndLeaves format:NULL error:NULL];
  67. if ([array isKindOfClass:[NSMutableArray class]]) return array;
  68. return nil;
  69. }
  70. + (NSMutableArray *)arrayWithPlistString:(NSString *)plist {
  71. if (!plist) return nil;
  72. NSData *data = [plist dataUsingEncoding:NSUTF8StringEncoding];
  73. return [self arrayWithPlistData:data];
  74. }
  75. - (void)removeFirstObject {
  76. if (self.count) {
  77. [self removeObjectAtIndex:0];
  78. }
  79. }
  80. #pragma clang diagnostic push
  81. #pragma clang diagnostic ignored "-Wobjc-protocol-method-implementation"
  82. - (void)removeLastObject {
  83. if (self.count) {
  84. [self removeObjectAtIndex:self.count - 1];
  85. }
  86. }
  87. #pragma clang diagnostic pop
  88. - (id)popFirstObject {
  89. id obj = nil;
  90. if (self.count) {
  91. obj = self.firstObject;
  92. [self removeFirstObject];
  93. }
  94. return obj;
  95. }
  96. - (id)popLastObject {
  97. id obj = nil;
  98. if (self.count) {
  99. obj = self.lastObject;
  100. [self removeLastObject];
  101. }
  102. return obj;
  103. }
  104. - (void)appendObject:(id)anObject {
  105. [self addObject:anObject];
  106. }
  107. - (void)prependObject:(id)anObject {
  108. [self insertObject:anObject atIndex:0];
  109. }
  110. - (void)appendObjects:(NSArray *)objects {
  111. if (!objects) return;
  112. [self addObjectsFromArray:objects];
  113. }
  114. - (void)prependObjects:(NSArray *)objects {
  115. if (!objects) return;
  116. NSUInteger i = 0;
  117. for (id obj in objects) {
  118. [self insertObject:obj atIndex:i++];
  119. }
  120. }
  121. - (void)insertObjects:(NSArray *)objects atIndex:(NSUInteger)index {
  122. NSUInteger i = index;
  123. for (id obj in objects) {
  124. [self insertObject:obj atIndex:i++];
  125. }
  126. }
  127. - (void)reverse {
  128. NSUInteger count = self.count;
  129. int mid = floor(count / 2.0);
  130. for (NSUInteger i = 0; i < mid; i++) {
  131. [self exchangeObjectAtIndex:i withObjectAtIndex:(count - (i + 1))];
  132. }
  133. }
  134. - (void)shuffle {
  135. for (NSUInteger i = self.count; i > 1; i--) {
  136. [self exchangeObjectAtIndex:(i - 1)
  137. withObjectAtIndex:arc4random_uniform((u_int32_t)i)];
  138. }
  139. }
  140. @end