]> git.wincent.com - wikitext.git/blob - ext/ary.h
Add NO_ITEM macro to 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, long idx)
45 {
46     return ary->count > idx ? ary->entries[idx] : INT_MAX;
47 }
48
49 inline int ary_delete_at(ary_t *ary, long idx)
50 {
51     // dirty optimization: we know we'll only ever be called to delete the last element of the array
52     if (ary->count > 0)
53     {
54         ary->count--;
55         return 1;
56     }
57     return 0;
58 }
59
60 inline void ary_push(ary_t *ary, int val)
61 {
62     if (ary->count == ary->max)
63     {
64         ary->max += DEFAULT_ENTRY_COUNT;
65         REALLOC_N(ary->entries, int, ary->max);
66     }
67     ary->entries[ary->count] = val;
68     ary->count++;
69 }
70
71 inline int ary_includes(ary_t *ary, int val)
72 {
73     for (int i = 0, max = ary->count; i < max; i++)
74     {
75         if (ary->entries[i] == val)
76             return 1;
77     }
78     return 0;
79 }
80
81 // returns a count indicating the number of times the value appears in the collection
82 // refactored from _Wikitext_count()
83 inline int ary_count(ary_t *ary, int item)
84 {
85     int count = 0;
86     for (int i = 0, max = ary->count; i < max; i++)
87     {
88         if (ary->entries[i] == item)
89             count++;
90     }
91     return count;
92 }
93