1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
|
/*
Copyright (c) 1998, 1999, 2000, 2001, 2003, 2004 Benhur Stein
This file is part of Paj.
Paj is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the
Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
Paj 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 Lesser General Public License
for more details.
You should have received a copy of the GNU Lesser General Public License
along with Paj; if not, write to the Free Software Foundation, Inc.,
59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
*/
// BusyEnumerator.m
// 19980227 BS creation
#include "BusyEnumerator.h"
#include "ReduceEntity.h"
#include "BusyState.h"
@implementation BusyEnumerator
- (id)initWithEnumerator:(NSEnumerator *)original
container:(PajeContainer *)cont
fromTime:(NSDate *)start
toTime:(NSDate *)end
concurrentTypes:(NSSet *)busySet
{
self = [super init];
enumerator = [original retain];
busyEntities = [busySet retain];
array = [[BusyArray alloc] initWithEntityType:[BusyState entityType] container:cont startTime:start endTime:end];
firstFase = YES;
return self;
}
- (void)dealloc
{
[enumerator release];
[busyEntities release];
[array release];
[super dealloc];
}
- (id)nextObject
{
id obj;
if (firstFase) {
do {
obj = [enumerator nextObject];
if ((nil != obj) && [busyEntities containsObject:[obj name]])
[array addObject:obj];
} while (obj);
if (/*sendOriginal && */(nil != obj)) {
return obj;
} else {
firstFase = NO;
[enumerator release];
enumerator = [[array objectEnumeratorOfClass:[CountReduceEntity class]] retain];
}
}
// second fase -- take objects from the array enumerator
return [enumerator nextObject];
}
@end
|