/* Concrete implementation of NSSet based on GNU Set class
   Copyright (C) 1998 Free Software Foundation, Inc.
   
   Written by:  Richard frith-Macdonald <richard@brainstorm.co.Ik>
   Created: October 1998
   
   This file is part of the GNUstep Base Library.
   
   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Library General Public
   License as published by the Free Software Foundation; either
   version 2 of the License, or (at your option) any later version.
   
   This library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Library General Public License for more details.
   
   You should have received a copy of the GNU Library General Public
   License along with this library; if not, write to the Free
   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
   */

#include <config.h>
#include <Foundation/NSSet.h>
#include <base/behavior.h>
#include <Foundation/NSAutoreleasePool.h>
#include <Foundation/NSException.h>
#include <Foundation/NSUtilities.h>
#include <Foundation/NSString.h>
#include <Foundation/NSPortCoder.h>


#define	FAST_MAP_RETAIN_VAL(X)	X
#define	FAST_MAP_RELEASE_VAL(X)	

#include "FastMap.x"

@class	NSSetNonCore;
@class	NSMutableSetNonCore;

@interface NSGCountedSet : NSCountedSet
{
@public
  FastMapTable_t	map;
}
@end

@interface NSGCountedSetEnumerator : NSEnumerator
{
  NSGCountedSet	*set;
  FastMapNode	node;
}
@end

@implementation NSGCountedSetEnumerator

- initWithSet: (NSSet*)d
{
  self = [super init];
  if (self)
    {
      set = [(NSGCountedSet*)d retain];
      node = set->map.firstNode;
    }
  return self;
}

- nextObject
{
  FastMapNode old = node;

  if (node == 0)
    {
      return nil;
    }
  node = node->nextInMap;
  return old->key.o;
}

- (void) dealloc
{
  [set release];
  [super dealloc];
}

@end


@implementation NSGCountedSet

+ (void) initialize
{
  if (self == [NSGCountedSet class])
    {
      class_add_behavior(self, [NSSetNonCore class]);
      class_add_behavior(self, [NSMutableSetNonCore class]);
    }
}

- (void) dealloc
{
  FastMapEmptyMap(&map);
  [super dealloc];
}

- (void) encodeWithCoder: (NSCoder*)aCoder
{
  unsigned	count = map.nodeCount;
  FastMapNode	node = map.firstNode;
  SEL		sel1 = @selector(encodeObject:);
  IMP		imp1 = [aCoder methodForSelector: sel1];
  SEL		sel2 = @selector(encodeValueOfObjCType:at:);
  IMP		imp2 = [aCoder methodForSelector: sel2];
  const char	*type = @encode(unsigned);

  (*imp2)(aCoder, sel2, type, &count);

  while (node != 0)
    {
      (*imp1)(aCoder, sel1, node->key.o);
      (*imp2)(aCoder, sel2, type, &node->value.I);
      node = node->nextInMap;
    }
}

- (id) initWithCoder: (NSCoder*)aCoder
{
  unsigned	count;
  id		value;
  unsigned	valcnt;
  SEL		sel = @selector(decodeValueOfObjCType:at:);
  IMP		imp = [aCoder methodForSelector: sel];
  const char	*utype = @encode(unsigned);
  const char	*otype = @encode(id);

  (*imp)(aCoder, sel, utype, &count);

  FastMapInitWithZoneAndCapacity(&map, [self zone], count);
  while (count-- > 0)
    {
      (*imp)(aCoder, sel, otype, &value);
      (*imp)(aCoder, sel, utype, &valcnt);
      FastMapAddPairNoRetain(&map, (FastMapItem)value, (FastMapItem)valcnt);
    }

  return self;
}

/* Designated initialiser */
- (id) initWithCapacity: (unsigned)cap
{
  FastMapInitWithZoneAndCapacity(&map, [self zone], cap);
  return self;
}

- (id) initWithObjects: (id*)objs count: (unsigned)c
{
  int i;

  if ([self initWithCapacity: c] == nil)
    {
      return nil;
    }
  for (i = 0; i < c; i++)
    {
      FastMapNode     node;

      if (objs[i] == nil)
	{
	  [self autorelease];
	  [NSException raise: NSInvalidArgumentException
		      format: @"Tried to init counted set with nil value"];
	}
      node = FastMapNodeForKey(&map, (FastMapItem)objs[i]);
      if (node == 0)
	{
	  FastMapAddPair(&map,(FastMapItem)objs[i],(FastMapItem)(unsigned)1);
        }
      else
	{
	  node->value.I++;
	}
    }
  return self;
}

- (void) addObject: (NSObject*)anObject
{
  FastMapNode node;

  if (anObject == nil)
    {
      [NSException raise: NSInvalidArgumentException
		  format: @"Tried to nil value to counted set"];
    }

  node = FastMapNodeForKey(&map, (FastMapItem)anObject);
  if (node == 0)
    {
      FastMapAddPair(&map,(FastMapItem)anObject,(FastMapItem)(unsigned)1);
    }
  else
    {
      node->value.I++;
    }
}

- (unsigned) count
{
  return map.nodeCount;
}

- (unsigned) countForObject: (id)anObject
{
  if (anObject)
    {
      FastMapNode node = FastMapNodeForKey(&map, (FastMapItem)anObject);

      if (node)
	{
	  return node->value.I;
	}
    }
  return 0;
}

- (unsigned) hash
{
  return map.nodeCount;
}

- (id) member: (id)anObject
{
  if (anObject)
    {
      FastMapNode node = FastMapNodeForKey(&map, (FastMapItem)anObject);

      if (node)
	{
	  return node->key.o;
	}
    }
  return nil;
}

- (NSEnumerator*) objectEnumerator
{
  return [[[NSGCountedSetEnumerator alloc] initWithSet: self] autorelease];
}

- (void) removeObject: (NSObject*)anObject
{
  if (anObject)
    {
      FastMapBucket       bucket;

      bucket = FastMapBucketForKey(&map, (FastMapItem)anObject);
      if (bucket)
	{
	  FastMapNode     node;

	  node = FastMapNodeForKeyInBucket(bucket, (FastMapItem)anObject);
	  if (node)
	    {
	      if (--node->value.I == 0)
		{
		  FastMapRemoveNodeFromMap(&map, bucket, node);
		  FastMapFreeNode(&map, node);
		}
	    }
	}
    }
}

- (void) removeAllObjects
{
  FastMapCleanMap(&map);
}

@end