CollectionUtils.h
author Jens Alfke <jens@mooseyard.com>
Wed May 07 16:47:44 2008 -0700 (2008-05-07)
changeset 8 5588347dfcbd
parent 7 59addced5e2a
child 9 823e7e74088e
permissions -rw-r--r--
* Added $apply and some other collection utils.
* Moved logging code to a separate code segment.
* Fixed uninitialized variable in Target.
     1 //
     2 //  CollectionUtils.h
     3 //  MYUtilities
     4 //
     5 //  Created by Jens Alfke on 1/5/08.
     6 //  Copyright 2008 Jens Alfke. All rights reserved.
     7 //
     8 
     9 #import <Cocoa/Cocoa.h>
    10 
    11 // Collection creation conveniences:
    12 
    13 #define $array(OBJS...)     ({id objs[]={OBJS}; \
    14                               [NSArray arrayWithObjects: objs count: sizeof(objs)/sizeof(id)];})
    15 #define $marray(OBJS...)    ({id objs[]={OBJS}; \
    16                               [NSMutableArray arrayWithObjects: objs count: sizeof(objs)/sizeof(id)];})
    17 
    18 #define $dict(PAIRS...)     ({struct _dictpair pairs[]={PAIRS}; \
    19                               _dictof(pairs,sizeof(pairs)/sizeof(struct _dictpair));})
    20 #define $mdict(PAIRS...)    ({struct _dictpair pairs[]={PAIRS}; \
    21                               _mdictof(pairs,sizeof(pairs)/sizeof(struct _dictpair));})
    22 
    23 #define $object(VAL)        ({__typeof(VAL) v=(VAL); _box(&v,@encode(__typeof(v)));})
    24 
    25 
    26 // Apply a selector to each array element, returning an array of the results:
    27 NSArray* $apply( NSArray *src, SEL selector, id defaultValue );
    28 NSArray* $applyKeyPath( NSArray *src, NSString *keyPath, id defaultValue );
    29 
    30 
    31 // Object conveniences:
    32 
    33 BOOL $equal(id obj1, id obj2);      // Like -isEqual: but works even if either/both are nil
    34 
    35 #define $sprintf(FORMAT, ARGS... )  [NSString stringWithFormat: (FORMAT), ARGS]
    36 
    37 #define $cast(CLASSNAME,OBJ)        ((CLASSNAME*)(_cast([CLASSNAME class],(OBJ))))
    38 #define $castNotNil(CLASSNAME,OBJ)  ((CLASSNAME*)(_castNotNil([CLASSNAME class],(OBJ))))
    39 #define $castIf(CLASSNAME,OBJ)      ((CLASSNAME*)(_castIf([CLASSNAME class],(OBJ))))
    40 #define $castArrayOf(ITEMCLASSNAME,OBJ) _castArrayOf([ITEMCLASSNAME class],(OBJ)))
    41 
    42 void setObj( id *var, id value );
    43 BOOL ifSetObj( id *var, id value );
    44 void setString( NSString **var, NSString *value );
    45 BOOL ifSetString( NSString **var, NSString *value );
    46 
    47 
    48 #define $true   ((NSNumber*)kCFBooleanTrue)
    49 #define $false  ((NSNumber*)kCFBooleanFalse)
    50 
    51 
    52 @interface NSArray (MYUtils)
    53 - (BOOL) my_containsObjectIdenticalTo: (id)object;
    54 @end
    55 
    56 
    57 @interface NSSet (MYUtils)
    58 + (NSSet*) my_unionOfSet: (NSSet*)set1 andSet: (NSSet*)set2;
    59 + (NSSet*) my_intersectionOfSet: (NSSet*)set1 andSet: (NSSet*)set2;
    60 + (NSSet*) my_differenceOfSet: (NSSet*)set1 andSet: (NSSet*)set2;
    61 @end
    62 
    63 
    64 #pragma mark -
    65 #pragma mark FOREACH:
    66     
    67 #if MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_5
    68 #define foreach(VAR,ARR) for(VAR in ARR)
    69 
    70 #else
    71 struct foreachstate {NSArray *array; unsigned n, i;};
    72 static inline struct foreachstate _initforeach( NSArray *arr ) {
    73     struct foreachstate s;
    74     s.array = arr;
    75     s.n = [arr count];
    76     s.i = 0;
    77     return s;
    78 }
    79 #define foreach(VAR,ARR) for( struct foreachstate _s = _initforeach((ARR)); \
    80                                    _s.i<_s.n && ((VAR)=[_s.array objectAtIndex: _s.i], YES); \
    81                                    _s.i++ )
    82 #endif
    83 
    84 
    85 // Internals (don't use directly)
    86 struct _dictpair { id key; id value; };
    87 NSDictionary* _dictof(const struct _dictpair*, size_t count);
    88 NSMutableDictionary* _mdictof(const struct _dictpair*, size_t count);
    89 NSValue* _box(const void *value, const char *encoding);
    90 id _cast(Class,id);
    91 id _castNotNil(Class,id);
    92 id _castIf(Class,id);
    93 NSArray* _castArrayOf(Class,NSArray*);