bbf005e79ac6ab058ab4c099d68da06fd7d9bb0a
[blender.git] / source / blender / blenlib / intern / BLI_args.c
1 /*
2  * A general argument parsing module
3  *
4  *
5  * ***** BEGIN GPL LICENSE BLOCK *****
6  *
7  * This program is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License
9  * as published by the Free Software Foundation; either version 2
10  * of the License, or (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20  *
21  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
22  * All rights reserved.
23  *
24  * The Original Code is: all of this file.
25  *
26  * Contributor(s): none yet.
27  *
28  * ***** END GPL LICENSE BLOCK *****
29  */
30
31 /** \file blender/blenlib/intern/BLI_args.c
32  *  \ingroup bli
33  */
34
35
36 #include <ctype.h> /* for tolower */
37 #include <string.h>
38
39 #include "MEM_guardedalloc.h"
40
41 #include "BLI_utildefines.h"
42 #include "BLI_listbase.h"
43 #include "BLI_string.h"
44 #include "BLI_args.h"
45 #include "BLI_ghash.h"
46
47 static char NO_DOCS[] = "NO DOCUMENTATION SPECIFIED";
48
49 struct bArgDoc;
50 typedef struct bArgDoc {
51         struct bArgDoc *next, *prev;
52         const char *short_arg;
53         const char *long_arg;
54         const char *documentation;
55         int  done;
56 } bArgDoc;
57
58 typedef struct bAKey {
59         const char *arg;
60         uintptr_t pass; /* cast easier */
61         int case_str; /* case specific or not */
62 } bAKey;
63
64 typedef struct bArgument {
65         bAKey *key;
66         BA_ArgCallback func;
67         void *data;
68         bArgDoc *doc;
69 } bArgument;
70
71 struct bArgs {
72         ListBase docs;
73         GHash  *items;
74         int     argc;
75         const char  **argv;
76         int       *passes;
77 };
78
79 static unsigned int case_strhash(const void *ptr)
80 {
81         const char *s= ptr;
82         unsigned int i= 0;
83         unsigned char c;
84
85         while ( (c= tolower(*s++)) )
86                 i= i*37 + c;
87
88         return i;
89 }
90
91 static unsigned int     keyhash(const void *ptr)
92 {
93         const bAKey *k = ptr;
94         return case_strhash(k->arg); // ^ BLI_ghashutil_inthash((void*)k->pass);
95 }
96
97 static int keycmp(const void *a, const void *b)
98 {
99         const bAKey *ka = a;
100         const bAKey *kb = b;
101         if (ka->pass == kb->pass || ka->pass == -1 || kb->pass == -1) { /* -1 is wildcard for pass */
102                 if (ka->case_str == 1 || kb->case_str == 1)
103                         return BLI_strcasecmp(ka->arg, kb->arg);
104                 else
105                         return strcmp(ka->arg, kb->arg);
106         } else {
107                 return BLI_ghashutil_intcmp((const void*)ka->pass, (const void*)kb->pass);
108         }
109 }
110
111 static bArgument *lookUp(struct bArgs *ba, const char *arg, int pass, int case_str)
112 {
113         bAKey key;
114
115         key.case_str = case_str;
116         key.pass = pass;
117         key.arg = arg;
118
119         return BLI_ghash_lookup(ba->items, &key);
120 }
121
122 bArgs *BLI_argsInit(int argc, const char **argv)
123 {
124         bArgs *ba = MEM_callocN(sizeof(bArgs), "bArgs");
125         ba->passes = MEM_callocN(sizeof(int) * argc, "bArgs passes");
126         ba->items = BLI_ghash_new(keyhash, keycmp, "bArgs passes gh");
127         ba->docs.first = ba->docs.last = NULL;
128         ba->argc = argc;
129         ba->argv = argv;
130
131         return ba;
132 }
133
134 static void freeItem(void *val)
135 {
136         MEM_freeN(val);
137 }
138
139 void BLI_argsFree(struct bArgs *ba)
140 {
141         BLI_ghash_free(ba->items, freeItem, freeItem);
142         MEM_freeN(ba->passes);
143         BLI_freelistN(&ba->docs);
144         MEM_freeN(ba);
145 }
146
147 void BLI_argsPrint(struct bArgs *ba)
148 {
149         int i;
150         for (i = 0; i < ba->argc; i++) {
151                 printf("argv[%d] = %s\n", i, ba->argv[i]);
152         }
153 }
154
155 const char **BLI_argsArgv(struct bArgs *ba)
156 {
157         return ba->argv;
158 }
159
160 static bArgDoc *internalDocs(struct bArgs *ba, const char *short_arg, const char *long_arg, const char *doc)
161 {
162         bArgDoc *d;
163
164         d = MEM_callocN(sizeof(bArgDoc), "bArgDoc");
165
166         if (doc == NULL)
167                 doc = NO_DOCS;
168
169         d->short_arg = short_arg;
170         d->long_arg = long_arg;
171         d->documentation = doc;
172
173         BLI_addtail(&ba->docs, d);
174
175         return d;
176 }
177
178 static void internalAdd(struct bArgs *ba, const char *arg, int pass, int case_str, BA_ArgCallback cb, void *data, bArgDoc *d)
179 {
180         bArgument *a;
181         bAKey *key;
182
183         a = lookUp(ba, arg, pass, case_str);
184
185         if (a) {
186                 printf("WARNING: conflicting argument\n");
187                 printf("\ttrying to add '%s' on pass %i, %scase sensitive\n", arg, pass, case_str == 1? "not ": "");
188                 printf("\tconflict with '%s' on pass %i, %scase sensitive\n\n", a->key->arg, (int)a->key->pass, a->key->case_str == 1? "not ": "");
189         }
190
191         a = MEM_callocN(sizeof(bArgument), "bArgument");
192         key = MEM_callocN(sizeof(bAKey), "bAKey");
193
194         key->arg = arg;
195         key->pass = pass;
196         key->case_str = case_str;
197
198         a->key = key;
199         a->func = cb;
200         a->data = data;
201         a->doc = d;
202
203         BLI_ghash_insert(ba->items, key, a);
204 }
205
206 void BLI_argsAddCase(struct bArgs *ba, int pass, const char *short_arg, int short_case, const char *long_arg, int long_case, const char *doc, BA_ArgCallback cb, void *data)
207 {
208         bArgDoc *d = internalDocs(ba, short_arg, long_arg, doc);
209
210         if (short_arg)
211                 internalAdd(ba, short_arg, pass, short_case, cb, data, d);
212
213         if (long_arg)
214                 internalAdd(ba, long_arg, pass, long_case, cb, data, d);
215
216
217 }
218
219 void BLI_argsAdd(struct bArgs *ba, int pass, const char *short_arg, const char *long_arg, const char *doc, BA_ArgCallback cb, void *data)
220 {
221         BLI_argsAddCase(ba, pass, short_arg, 0, long_arg, 0, doc, cb, data);
222 }
223
224 static void internalDocPrint(bArgDoc *d)
225 {
226         if (d->short_arg && d->long_arg)
227                 printf("%s or %s", d->short_arg, d->long_arg);
228         else if (d->short_arg)
229                 printf("%s", d->short_arg);
230         else if (d->long_arg)
231                 printf("%s", d->long_arg);
232
233         printf(" %s\n\n", d->documentation);
234 }
235
236 void BLI_argsPrintArgDoc(struct bArgs *ba, const char *arg)
237 {
238         bArgument *a = lookUp(ba, arg, -1, -1);
239
240         if (a)
241         {
242                 bArgDoc *d = a->doc;
243
244                 internalDocPrint(d);
245
246                 d->done = 1;
247         }
248 }
249
250 void BLI_argsPrintOtherDoc(struct bArgs *ba)
251 {
252         bArgDoc *d;
253
254         for( d = ba->docs.first; d; d = d->next)
255         {
256                 if (d->done == 0)
257                 {
258                         internalDocPrint(d);
259                 }
260         }
261 }
262
263 void BLI_argsParse(struct bArgs *ba, int pass, BA_ArgCallback default_cb, void *default_data)
264 {
265         int i = 0;
266
267         for( i = 1; i < ba->argc; i++) { /* skip argv[0] */
268                 if (ba->passes[i] == 0) {
269                          /* -1 signal what side of the comparison it is */
270                         bArgument *a = lookUp(ba, ba->argv[i], pass, -1);
271                         BA_ArgCallback func = NULL;
272                         void *data = NULL;
273
274                         if (a) {
275                                 func = a->func;
276                                 data = a->data;
277                         } else {
278                                 func = default_cb;
279                                 data = default_data;
280                         }
281
282                         if (func) {
283                                 int retval = func(ba->argc - i, ba->argv + i, data);
284
285                                 if (retval >= 0) {
286                                         int j;
287
288                                         /* use extra arguments */
289                                         for (j = 0; j <= retval; j++) {
290                                                 ba->passes[i + j] = pass;
291                                         }
292                                         i += retval;
293                                 }
294                                 else if (retval == -1) {
295                                         if (a) {
296                                                 if (a->key->pass != -1)
297                                                         ba->passes[i] = pass;
298                                         }
299                                         break;
300                                 }
301                         }
302                 }
303         }
304 }