libs-base/Source/NSGCountedSet.m

/* Concrete implementation of NSCountedSet based on GNU Bag class
   Copyright (C) 1995 Free Software Foundation, Inc.
   
   Written by:  R. Andrew McCallum <mccallum@gnu.ai.mit.edu>
   Created: Sep 1995
   
   This file is part of the GNU Objective C Class 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 <Foundation/NSGSet.h>
#include <objects/NSSet.h>
#include <objects/behavior.h>
#include <objects/Set.h>
#include <objects/eltfuncs.h>
#include <Foundation/NSUtilities.h>
#include <Foundation/NSString.h>

@interface NSGCountedSetEnumerator : NSEnumerator
{
  NSCountedSet *bag;
  void *enum_state;
}
@end

@implementation NSGCountedSetEnumerator

- initWithCountedSet: (NSCountedSet*)d
{
  [super init];
  bag = d;
  [bag retain];
  enum_state = 0;
  return self;
}

- nextObject
{
  elt e;
  if ([bag getNextElement:&e withEnumState:&enum_state])
    return e.id_u;
  else
    return nil;
}

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

@end


@implementation NSGCountedSet

+ (void) initialize
{
  static int done = 0;

  /* xxx This class not yet ready for action. */
  [self notImplemented:_cmd];

  if (!done)
    {
      done = 1;
      class_add_behavior([NSGCountedSet class], [Bag class]);
    }
}

- initWithCapacity: (unsigned)numItems
{
  return [self initWithType:@encode(id)
	       capacity:numItems];
}

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

- (void) addObject: anObject
{
  [self addElement:anObject];
}

- (void) removeObject: anObject
{
  [self removeElement:anObject];
}

- (unsigned int) countForObject: anObject
{
  return [self occurrencesOfElement:anObject];
}

@end