err.no Git - sope/blob - sope-ical/NGiCal/iCalRepeatableEntityObject.m

   1 /*
   2   Copyright (C) 2004-2005 SKYRIX Software AG
   3
   4   This file is part of SOPE.
   5
   6   SOPE is free software; you can redistribute it and/or modify it under
   7   the terms of the GNU Lesser General Public License as published by the
   8   Free Software Foundation; either version 2, or (at your option) any
   9   later version.
  10
  11   SOPE is distributed in the hope that it will be useful, but WITHOUT ANY
  12   WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13   FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  14   License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with SOPE; see the file COPYING.  If not, write to the
  18   Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  19   02111-1307, USA.
  20 */
  21
  22 #include "iCalRepeatableEntityObject.h"
  23 #include <NGExtensions/NGCalendarDateRange.h>
  24 #include "iCalRecurrenceRule.h"
  25 #include "iCalRecurrenceCalculator.h"
  26 #include "common.h"
  27
  28 @implementation iCalRepeatableEntityObject
  29
  30 - (void)dealloc {
  31   [self->rRules  release];
  32   [self->exRules release];
  33   [self->exDates release];
  34   [super dealloc];
  35 }
  36
  37 /* Accessors */
  38
  39 - (void)removeAllRecurrenceRules {
  40   [self->rRules removeAllObjects];
  41 }
  42 - (void)addToRecurrenceRules:(id)_rrule {
  43   if (_rrule == nil) return;
  44   if (self->rRules == nil)
  45     self->rRules = [[NSMutableArray alloc] initWithCapacity:1];
  46   [self->rRules addObject:_rrule];
  47 }
  48 - (void)setRecurrenceRules:(NSArray *)_rrules {
  49   if (_rrules == self->rRules)
  50     return;
  51   [self->rRules release];
  52   self->rRules = [_rrules mutableCopy];
  53 }
  54 - (BOOL)hasRecurrenceRules {
  55   return [self->rRules count] > 0 ? YES : NO;
  56 }
  57 - (NSArray *)recurrenceRules {
  58   return self->rRules;
  59 }
  60
  61 - (void)removeAllExceptionRules {
  62   [self->exRules removeAllObjects];
  63 }
  64 - (void)addToExceptionRules:(id)_rrule {
  65   if (_rrule == nil) return;
  66   if (self->exRules == nil)
  67     self->exRules = [[NSMutableArray alloc] initWithCapacity:1];
  68   [self->exRules addObject:_rrule];
  69 }
  70 - (void)setExceptionRules:(NSArray *)_rrules {
  71   if (_rrules == self->exRules)
  72     return;
  73   [self->exRules release];
  74   self->exRules = [_rrules mutableCopy];
  75 }
  76 - (BOOL)hasExceptionRules {
  77   return [self->exRules count] > 0 ? YES : NO;
  78 }
  79 - (NSArray *)exceptionRules {
  80   return self->exRules;
  81 }
  82
  83 - (void)removeAllExceptionDates {
  84   [self->exDates removeAllObjects];
  85 }
  86 - (void)setExceptionDates:(NSArray *)_exDates {
  87   if (_exDates == self->exDates)
  88     return;
  89   [self->exDates release];
  90   self->exDates = [_exDates mutableCopy];
  91 }
  92 - (void)addToExceptionDates:(id)_date {
  93   if (_date == nil) return;
  94   if (self->exDates == nil)
  95     self->exDates = [[NSMutableArray alloc] initWithCapacity:4];
  96   [self->exDates addObject:_date];
  97 }
  98 - (NSArray *)exceptionDates {
  99   return self->exDates;
 100 }
 101
 102 /* Convenience */
 103
 104 - (BOOL)isRecurrent {
 105   return [self hasRecurrenceRules] ? YES : NO;
 106 }
 107
 108 /* Matching */
 109
 110 - (BOOL)isWithinCalendarDateRange:(NGCalendarDateRange *)_range
 111   firstInstanceCalendarDateRange:(NGCalendarDateRange *)_fir
 112 {
 113   NSArray *ranges;
 114
 115   ranges = [self recurrenceRangesWithinCalendarDateRange:_range
 116                  firstInstanceCalendarDateRange:_fir];
 117   return [ranges count] > 0;
 118 }
 119
 120 - (NSArray *)recurrenceRangesWithinCalendarDateRange:(NGCalendarDateRange *)_r
 121   firstInstanceCalendarDateRange:(NGCalendarDateRange *)_fir
 122 {
 123   iCalRecurrenceRule       *rule;
 124   iCalRecurrenceCalculator *calc;
 125   NSMutableArray           *ranges;
 126   unsigned                 i, count, rCount;
 127
 128   ranges = [NSMutableArray array];
 129   count  = [self->rRules count];
 130   for (i = 0; i < count; i++) {
 131     NSArray *rs;
 132
 133     rule = [self->rRules objectAtIndex:i];
 134     calc = [iCalRecurrenceCalculator recurrenceCalculatorForRecurrenceRule:rule
 135                                      withFirstInstanceCalendarDateRange:_fir];
 136     rs   = [calc recurrenceRangesWithinCalendarDateRange:_r];
 137     [ranges addObjectsFromArray:rs];
 138   }
 139
 140   if (![ranges count])
 141     return nil;
 142
 143   /* test if any exceptions do match */
 144   count = [self->exRules count];
 145   for (i = 0; i < count; i++) {
 146     NSArray *rs;
 147
 148     rule = [self->exRules objectAtIndex:i];
 149     calc = [iCalRecurrenceCalculator recurrenceCalculatorForRecurrenceRule:rule
 150                                      withFirstInstanceCalendarDateRange:_fir];
 151     rs   = [calc recurrenceRangesWithinCalendarDateRange:_r];
 152     [ranges removeObjectsInArray:rs];
 153   }
 154
 155   if (![ranges count])
 156     return nil;
 157
 158   /* exception dates are also possible */
 159   rCount = [ranges count];
 160   count  = [self->exDates count];
 161   for (i = 0; i < count; i++) {
 162     NSCalendarDate      *exDate;
 163     NGCalendarDateRange *r;
 164     unsigned            k;
 165
 166     exDate = [self->exDates objectAtIndex:i];
 167     for (k = 0; k < rCount; k++) {
 168       r = [ranges objectAtIndex:(rCount - k) - 1];
 169       if ([r containsDate:exDate]) {
 170         [ranges removeObjectAtIndex:k];
 171       }
 172     }
 173   }
 174   return ranges;
 175 }
 176
 177
 178 /* this is the outmost bound possible, not necessarily the real last date */
 179 - (NSCalendarDate *)lastPossibleRecurrenceStartDateUsingFirstInstanceCalendarDateRange:(NGCalendarDateRange *)_r
 180 {
 181   NSCalendarDate *date;
 182   unsigned       i, count;
 183
 184   count = [self->rRules count];
 185   if (!count)
 186     return nil;
 187
 188   date  = nil;
 189   for (i = 0; i < count; i++) {
 190     iCalRecurrenceRule       *rule;
 191     iCalRecurrenceCalculator *calc;
 192     NSCalendarDate           *rdate;
 193
 194     rule = [self->rRules objectAtIndex:i];
 195     if ([rule isInfinite])
 196       return nil; /* rule is not bound, hence no limit */
 197     calc  = [iCalRecurrenceCalculator recurrenceCalculatorForRecurrenceRule:rule
 198                                       withFirstInstanceCalendarDateRange:_r];
 199     rdate = [[calc lastInstanceCalendarDateRange] startDate];
 200     if (date == nil || [date compare:rdate] == NSOrderedAscending)
 201       date = rdate;
 202   }
 203   return date;
 204 }
 205
 206 @end