Chomp/CHOMPEnumeratorCollectTrampoline.m

   1 //
   2 //  CHOMPIterationCollectEnumerator.m
   3 //  Chomp
   4 //
   5 //  Created by Michael Ash on 12/16/04.
   6 //
   7
   8 #import "CHOMPEnumeratorCollectTrampoline.h"
   9
  10
  11 @implementation CHOMPEnumeratorCollectTrampoline
  12
  13 - (void)prepareWithInvocation:(NSInvocation *)invocation
  14 {
  15         if([target isKindOfClass:[NSSet class]])
  16                 returnCollection = [[NSMutableSet alloc] init];
  17         else
  18                 returnCollection = [[NSMutableArray alloc] init];
  19 }
  20
  21 - (void)processReturnValue:retVal
  22 {
  23         if(!retVal)
  24                 retVal = [NSNull null];
  25         [returnCollection addObject:retVal];
  26 }
  27
  28 - getReturnValue
  29 {
  30         return returnCollection;
  31 }
  32
  33 - (void)finalizeWithInvocation:(NSInvocation *)invocation
  34 {
  35         // autorelease because we're returning it and it can't disappear right away
  36         [returnCollection autorelease];
  37         returnCollection = nil;
  38 }
  39
  40 - (void)dealloc
  41 {
  42         [returnCollection release];
  43         [super dealloc];
  44 }
  45
  46 @end