//

//  CollectionUtils.m

//  MYUtilities

//

//  Created by Jens Alfke on 1/5/08.

//  Copyright 2008 Jens Alfke. All rights reserved.

//

#import "CollectionUtils.h"

#import "Test.h"

NSDictionary* _dictof(const struct _dictpair* pairs, size_t count)

{

    CAssert(count<10000);

    id objects[count], keys[count];

    size_t n = 0;

    for( size_t i=0; i<count; i++,pairs++ ) {

        if( pairs->value ) {

            objects[n] = pairs->value;

            keys[n] = pairs->key;

            n++;

        }

    }

    return [NSDictionary dictionaryWithObjects: objects forKeys: keys count: n];

}

NSMutableDictionary* _mdictof(const struct _dictpair* pairs, size_t count)

{

    CAssert(count<10000);

    id objects[count], keys[count];

    size_t n = 0;

    for( size_t i=0; i<count; i++,pairs++ ) {

        if( pairs->value ) {

            objects[n] = pairs->value;

            keys[n] = pairs->key;

            n++;

        }

    }

    return [NSMutableDictionary dictionaryWithObjects: objects forKeys: keys count: n];

}

NSArray* $apply( NSArray *src, SEL selector, id defaultValue )

{

    NSMutableArray *dst = [NSMutableArray arrayWithCapacity: src.count];

    for( id obj in src ) {

        id result = [obj performSelector: selector] ?: defaultValue;

        [dst addObject: result];

    }

    return dst;

}

NSArray* $applyKeyPath( NSArray *src, NSString *keyPath, id defaultValue )

{

    NSMutableArray *dst = [NSMutableArray arrayWithCapacity: src.count];

    for( id obj in src ) {

        id result = [obj valueForKeyPath: keyPath] ?: defaultValue;

        [dst addObject: result];

    }

    return dst;

}

BOOL $equal(id obj1, id obj2)      // Like -isEqual: but works even if either/both are nil

{

    if( obj1 )

        return obj2 && [obj1 isEqual: obj2];

    else

        return obj2==nil;

}

NSValue* _box(const void *value, const char *encoding)

{

    // file:///Developer/Documentation/DocSets/com.apple.ADC_Reference_Library.DeveloperTools.docset/Contents/Resources/Documents/documentation/DeveloperTools/gcc-4.0.1/gcc/Type-encoding.html

    char e = encoding[0];

    if( e=='r' )                // ignore 'const' modifier

        e = encoding[1];

    switch( e ) {

        case 'c':   return [NSNumber numberWithChar: *(char*)value];

        case 'C':   return [NSNumber numberWithUnsignedChar: *(char*)value];

        case 's':   return [NSNumber numberWithShort: *(short*)value];

        case 'S':   return [NSNumber numberWithUnsignedShort: *(unsigned short*)value];

        case 'i':   return [NSNumber numberWithInt: *(int*)value];

        case 'I':   return [NSNumber numberWithUnsignedInt: *(unsigned int*)value];

        case 'l':   return [NSNumber numberWithLong: *(long*)value];

        case 'L':   return [NSNumber numberWithUnsignedLong: *(unsigned long*)value];

        case 'q':   return [NSNumber numberWithLongLong: *(long long*)value];

        case 'Q':   return [NSNumber numberWithUnsignedLongLong: *(unsigned long long*)value];

        case 'f':   return [NSNumber numberWithFloat: *(float*)value];

        case 'd':   return [NSNumber numberWithDouble: *(double*)value];

        case '*':   return [NSString stringWithUTF8String: *(char**)value];

        case '@':   return *(id*)value;

        default:    return [NSValue value: value withObjCType: encoding];

    }

}

id _cast( Class requiredClass, id object )

{

    if( object && ! [object isKindOfClass: requiredClass] )

        [NSException raise: NSInvalidArgumentException format: @"%@ required, but got %@ %p",

         requiredClass,[object class],object];

    return object;

}

id _castNotNil( Class requiredClass, id object )

{

    if( ! [object isKindOfClass: requiredClass] )

        [NSException raise: NSInvalidArgumentException format: @"%@ required, but got %@ %p",

         requiredClass,[object class],object];

    return object;

}

id _castIf( Class requiredClass, id object )

{

    if( object && ! [object isKindOfClass: requiredClass] )

        object = nil;

    return object;

}

NSArray* _castArrayOf(Class itemClass, NSArray *a)

{

    id item;

    foreach( item, $cast(NSArray,a) )

        _cast(itemClass,item);

    return a;

}

void setObj( id *var, id value )

{

    if( value != *var ) {

        [*var release];

        *var = [value retain];

    }

}

BOOL ifSetObj( id *var, id value )

{

    if( value != *var && ![value isEqual: *var] ) {

        [*var release];

        *var = [value retain];

        return YES;

    } else {

        return NO;

    }

}

void setString( NSString **var, NSString *value )

{

    if( value != *var ) {

        [*var release];

        *var = [value copy];

    }

}

BOOL ifSetString( NSString **var, NSString *value )

{

    if( value != *var && ![value isEqualToString: *var] ) {

        [*var release];

        *var = [value copy];

        return YES;

    } else {

        return NO;

    }

}

NSString* $string( const char *utf8Str )

{

    if( utf8Str )

        return [NSString stringWithCString: utf8Str encoding: NSUTF8StringEncoding];

    else

        return nil;

}

BOOL kvSetSet( id owner, NSString *property, NSMutableSet *set, NSSet *newSet ) {

    CAssert(set);

    if (!newSet)

        newSet = [NSSet set];

    if (![set isEqualToSet: newSet]) {

        [owner willChangeValueForKey: property

                     withSetMutation:NSKeyValueSetSetMutation 

                        usingObjects:newSet]; 

        [set setSet: newSet];

        [owner didChangeValueForKey: property 

                    withSetMutation:NSKeyValueSetSetMutation 

                       usingObjects:newSet]; 

        return YES;

    } else

        return NO;

}

BOOL kvAddToSet( id owner, NSString *property, NSMutableSet *set, id objToAdd ) {

    CAssert(set);

    if (![set containsObject: objToAdd]) {

        NSSet *changedObjects = [[NSSet alloc] initWithObjects: &objToAdd count: 1];

        [owner willChangeValueForKey: property

                     withSetMutation: NSKeyValueUnionSetMutation 

                        usingObjects: changedObjects]; 

        [set addObject: objToAdd];

        [owner didChangeValueForKey: property 

                    withSetMutation: NSKeyValueUnionSetMutation 

                       usingObjects: changedObjects]; 

        [changedObjects release];

        return YES;

    } else

        return NO;

}

BOOL kvRemoveFromSet( id owner, NSString *property, NSMutableSet *set, id objToRemove ) {

    if ([set containsObject: objToRemove]) {

        NSSet *changedObjects = [[NSSet alloc] initWithObjects: &objToRemove count: 1];

        [owner willChangeValueForKey: property

                     withSetMutation: NSKeyValueMinusSetMutation 

                        usingObjects: changedObjects]; 

        [set removeObject: objToRemove];

        [owner didChangeValueForKey: property 

                    withSetMutation: NSKeyValueMinusSetMutation 

                       usingObjects: changedObjects]; 

        [changedObjects release];

        return YES;

    } else

        return NO;

}

@implementation NSArray (MYUtils)

- (BOOL) my_containsObjectIdenticalTo: (id)object

{

    return [self indexOfObjectIdenticalTo: object] != NSNotFound;

}

@end

@implementation NSSet (MYUtils)

+ (NSSet*) my_unionOfSet: (NSSet*)set1 andSet: (NSSet*)set2

{

    if( set1 == set2 || set2.count==0 )

        return set1;

    else if( set1.count==0 )

        return set2;

    else {

        NSMutableSet *result = [set1 mutableCopy];

        [result unionSet: set2];

        return [result autorelease];

    }

}

+ (NSSet*) my_intersectionOfSet: (NSSet*)set1 andSet: (NSSet*)set2

{

    if( set1 == set2 || set1.count==0 )

        return set1;

    else if( set2.count==0 )

        return set2;

    else {

        NSMutableSet *result = [set1 mutableCopy];

        [result intersectSet: set2];

        return [result autorelease];

    }

}

+ (NSSet*) my_differenceOfSet: (NSSet*)set1 andSet: (NSSet*)set2

{

    if( set1.count==0 || set2.count==0 )

        return set1;

    else if( set1==set2 )

        return [NSSet set];

    else {

        NSMutableSet *result = [set1 mutableCopy];

        [result minusSet: set2];

        return [result autorelease];

    }

}

@end

#import "Test.h"

TestCase(CollectionUtils) {

    NSArray *a = $array(@"foo",@"bar",@"baz");

    //Log(@"a = %@",a);

    NSArray *aa = [NSArray arrayWithObjects: @"foo",@"bar",@"baz",nil];

    CAssertEqual(a,aa);

    const char *cstr = "a C string";

    id o = $object(cstr);

    //Log(@"o = %@",o);

    CAssertEqual(o,@"a C string");

    NSDictionary *d = $dict({@"int",    $object(1)},

                            {@"double", $object(-1.1)},

                            {@"char",   $object('x')},

                            {@"ulong",  $object(1234567UL)},

                            {@"longlong",$object(987654321LL)},

                            {@"cstr",   $object(cstr)});

    //Log(@"d = %@",d);

    NSDictionary *dd = [NSDictionary dictionaryWithObjectsAndKeys:

                        [NSNumber numberWithInt: 1],                    @"int",

                        [NSNumber numberWithDouble: -1.1],              @"double",

                        [NSNumber numberWithChar: 'x'],                 @"char",

                        [NSNumber numberWithUnsignedLong: 1234567UL],   @"ulong",

                        [NSNumber numberWithDouble: 987654321LL],       @"longlong",

                        @"a C string",                                  @"cstr",

                        nil];

    CAssertEqual(d,dd);

}

/*

 Copyright (c) 2008, Jens Alfke <jens@mooseyard.com>. All rights reserved.

 Redistribution and use in source and binary forms, with or without modification, are permitted

 provided that the following conditions are met:

 * Redistributions of source code must retain the above copyright notice, this list of conditions

 and the following disclaimer.

 * Redistributions in binary form must reproduce the above copyright notice, this list of conditions

 and the following disclaimer in the documentation and/or other materials provided with the

 distribution.

 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR

 IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 

 FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRI-

 BUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

 (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 

  PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 

 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF 

 THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */