]> git.wincent.com - wikitext.git/blob - ext/ary.h
Fix off-by-one error in ary.h
[wikitext.git] / ext / ary.h
1 // Copyright 2008 Wincent Colaiuta
2 // This program is free software: you can redistribute it and/or modify
3 // it under the terms of the GNU General Public License as published by
4 // the Free Software Foundation, either version 3 of the License, or
5 // (at your option) any later version.
6 //
7 // This program is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10 // GNU General Public License for more details.
11 //
12 // You should have received a copy of the GNU General Public License
13 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
14
15 #include <ruby/ruby.h>
16
17 typedef struct
18 {
19     int     count;
20     int     max;
21     int     *entries;
22 } ary_t;
23
24 // in the test suite array count goes no higher than 25 or 26
25 #define DEFAULT_ENTRY_COUNT 64
26
27 #define NO_ITEM(item) (item == INT_MAX)
28
29 inline ary_t *ary_new(void)
30 {
31     ary_t *ary      = ALLOC_N(ary_t, 1);
32     ary->count      = 0;
33     ary->max        = DEFAULT_ENTRY_COUNT;
34     ary->entries    = ALLOC_N(int, DEFAULT_ENTRY_COUNT);
35     return ary;
36 }
37
38 inline void ary_free(ary_t *ary)
39 {
40     free(ary->entries);
41     free(ary);
42 }
43
44 inline int ary_entry(ary_t *ary, int idx)
45 {
46     if (idx < 0)
47         idx = ary->count + idx;
48     return (idx >= 0 && ary->count > idx) ? ary->entries[idx] : INT_MAX;
49 }
50
51 inline int ary_delete_at(ary_t *ary, int idx)
52 {
53     // dirty optimization: we know we'll only ever be called to delete the last element of the array
54     if (ary->count > 0)
55     {
56         ary->count--;
57         return 1;
58     }
59     return 0;
60 }
61
62 inline void ary_push(ary_t *ary, int val)
63 {
64     if (ary->count == ary->max)
65     {
66         ary->max += DEFAULT_ENTRY_COUNT;
67         REALLOC_N(ary->entries, int, ary->max);
68     }
69     ary->entries[ary->count] = val;
70     ary->count++;
71 }
72
73 inline int ary_includes(ary_t *ary, int val)
74 {
75     for (int i = 0, max = ary->count; i < max; i++)
76     {
77         if (ary->entries[i] == val)
78             return 1;
79     }
80     return 0;
81 }
82
83 // returns a count indicating the number of times the value appears in the collection
84 // refactored from _Wikitext_count()
85 inline int ary_count(ary_t *ary, int item)
86 {
87     int count = 0;
88     for (int i = 0, max = ary->count; i < max; i++)
89     {
90         if (ary->entries[i] == item)
91             count++;
92     }
93     return count;
94 }
95