isl_arg_parse: allow options after arguments
[isl.git] / isl_arg.c
blobfb7eab77c5035612eb464154495585e773c5ac5a
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
8 */
10 #include <stdio.h>
11 #include <stdlib.h>
12 #include <string.h>
14 #include <isl/arg.h>
15 #include <isl/ctx.h>
17 static void set_default_choice(struct isl_arg *arg, void *opt)
19 *(unsigned *)(((char *)opt) + arg->offset) = arg->u.choice.default_value;
22 static void set_default_flags(struct isl_arg *arg, void *opt)
24 *(unsigned *)(((char *)opt) + arg->offset) = arg->u.flags.default_value;
27 static void set_default_bool(struct isl_arg *arg, void *opt)
29 *(unsigned *)(((char *)opt) + arg->offset) = arg->u.b.default_value;
32 static void set_default_child(struct isl_arg *arg, void *opt)
34 void *child = calloc(1, arg->u.child.size);
36 if (child)
37 isl_arg_set_defaults(arg->u.child.child, child);
39 *(void **)(((char *)opt) + arg->offset) = child;
42 static void set_default_user(struct isl_arg *arg, void *opt)
44 arg->u.user.init(((char *)opt) + arg->offset);
47 static void set_default_long(struct isl_arg *arg, void *opt)
49 *(long *)(((char *)opt) + arg->offset) = arg->u.l.default_value;
52 static void set_default_ulong(struct isl_arg *arg, void *opt)
54 *(unsigned long *)(((char *)opt) + arg->offset) = arg->u.ul.default_value;
57 static void set_default_str(struct isl_arg *arg, void *opt)
59 const char *str = NULL;
60 if (arg->u.str.default_value)
61 str = strdup(arg->u.str.default_value);
62 *(const char **)(((char *)opt) + arg->offset) = str;
65 void isl_arg_set_defaults(struct isl_arg *arg, void *opt)
67 int i;
69 for (i = 0; arg[i].type != isl_arg_end; ++i) {
70 switch (arg[i].type) {
71 case isl_arg_choice:
72 set_default_choice(&arg[i], opt);
73 break;
74 case isl_arg_flags:
75 set_default_flags(&arg[i], opt);
76 break;
77 case isl_arg_bool:
78 set_default_bool(&arg[i], opt);
79 break;
80 case isl_arg_child:
81 set_default_child(&arg[i], opt);
82 break;
83 case isl_arg_user:
84 set_default_user(&arg[i], opt);
85 break;
86 case isl_arg_long:
87 set_default_long(&arg[i], opt);
88 break;
89 case isl_arg_ulong:
90 set_default_ulong(&arg[i], opt);
91 break;
92 case isl_arg_arg:
93 case isl_arg_str:
94 set_default_str(&arg[i], opt);
95 break;
96 case isl_arg_version:
97 case isl_arg_end:
98 break;
103 void isl_arg_free(struct isl_arg *arg, void *opt)
105 int i;
107 if (!opt)
108 return;
110 for (i = 0; arg[i].type != isl_arg_end; ++i) {
111 switch (arg[i].type) {
112 case isl_arg_child:
113 isl_arg_free(arg[i].u.child.child,
114 *(void **)(((char *)opt) + arg[i].offset));
115 break;
116 case isl_arg_arg:
117 case isl_arg_str:
118 free(*(char **)(((char *)opt) + arg[i].offset));
119 break;
120 case isl_arg_user:
121 if (arg[i].u.user.clear)
122 arg[i].u.user.clear(((char *)opt) + arg[i].offset);
123 break;
124 case isl_arg_bool:
125 case isl_arg_choice:
126 case isl_arg_flags:
127 case isl_arg_long:
128 case isl_arg_ulong:
129 case isl_arg_version:
130 case isl_arg_end:
131 break;
135 free(opt);
138 static int print_arg_help(struct isl_arg *decl, const char *prefix, int no)
140 int len = 0;
142 if (!decl->long_name) {
143 printf(" -%c", decl->short_name);
144 return 4;
147 if (decl->short_name)
148 printf(" -%c, --", decl->short_name);
149 else
150 printf(" --");
151 len += 8;
153 if (prefix) {
154 printf("%s-", prefix);
155 len += strlen(prefix) + 1;
157 if (no) {
158 printf("no-");
159 len += 3;
161 printf("%s", decl->long_name);
162 len += strlen(decl->long_name);
164 return len;
167 const void *isl_memrchr(const void *s, int c, size_t n)
169 const char *p = s;
170 while (n-- > 0)
171 if (p[n] == c)
172 return p + n;
173 return NULL;
176 static int print_help_msg(struct isl_arg *decl, int pos)
178 int len;
179 const char *s;
181 if (!decl->help_msg)
182 return pos;
184 if (pos >= 29)
185 printf("\n%30s", "");
186 else
187 printf("%*s", 30 - pos, "");
189 s = decl->help_msg;
190 len = strlen(s);
191 while (len > 45) {
192 const char *space = isl_memrchr(s, ' ', 45);
193 int l;
195 if (!space)
196 space = strchr(s + 45, ' ');
197 if (!space)
198 break;
199 l = space - s;
200 printf("%.*s", l, s);
201 s = space + 1;
202 len -= l + 1;
203 printf("\n%30s", "");
206 printf("%s", s);
207 return len;
210 static void print_default(struct isl_arg *decl, const char *def, int pos)
212 int i;
213 const char *default_prefix = "[default: ";
214 const char *default_suffix = "]";
215 int len;
217 len = strlen(default_prefix) + strlen(def) + strlen(default_suffix);
219 if (!decl->help_msg) {
220 if (pos >= 29)
221 printf("\n%30s", "");
222 else
223 printf("%*s", 30 - pos, "");
224 pos = 0;
225 } else {
226 if (pos + len >= 48)
227 printf("\n%30s", "");
228 else
229 printf(" ");
231 printf("%s%s%s", default_prefix, def, default_suffix);
234 static void print_default_choice(struct isl_arg *decl, int pos)
236 int i;
237 const char *s = "none";
239 for (i = 0; decl->u.choice.choice[i].name; ++i)
240 if (decl->u.choice.choice[i].value == decl->u.choice.default_value) {
241 s = decl->u.choice.choice[i].name;
242 break;
245 print_default(decl, s, pos);
248 static void print_choice_help(struct isl_arg *decl, const char *prefix)
250 int i;
251 int pos;
253 pos = print_arg_help(decl, prefix, 0);
254 printf("=");
255 pos++;
257 for (i = 0; decl->u.choice.choice[i].name; ++i) {
258 if (i) {
259 printf("|");
260 pos++;
262 printf("%s", decl->u.choice.choice[i].name);
263 pos += strlen(decl->u.choice.choice[i].name);
266 pos = print_help_msg(decl, pos);
267 print_default_choice(decl, pos);
269 printf("\n");
272 static void print_default_flags(struct isl_arg *decl, int pos)
274 int i, first;
275 const char *default_prefix = "[default: ";
276 const char *default_suffix = "]";
277 int len = strlen(default_prefix) + strlen(default_suffix);
279 for (i = 0; decl->u.flags.flags[i].name; ++i)
280 if ((decl->u.flags.default_value & decl->u.flags.flags[i].mask) ==
281 decl->u.flags.flags[i].value)
282 len += strlen(decl->u.flags.flags[i].name);
284 if (!decl->help_msg) {
285 if (pos >= 29)
286 printf("\n%30s", "");
287 else
288 printf("%*s", 30 - pos, "");
289 pos = 0;
290 } else {
291 if (pos + len >= 48)
292 printf("\n%30s", "");
293 else
294 printf(" ");
296 printf("%s", default_prefix);
298 for (first = 1, i = 0; decl->u.flags.flags[i].name; ++i)
299 if ((decl->u.flags.default_value & decl->u.flags.flags[i].mask) ==
300 decl->u.flags.flags[i].value) {
301 if (!first)
302 printf(",");
303 printf("%s", decl->u.flags.flags[i].name);
304 first = 0;
307 printf("%s", default_suffix);
310 static void print_flags_help(struct isl_arg *decl, const char *prefix)
312 int i, j;
313 int pos;
315 pos = print_arg_help(decl, prefix, 0);
316 printf("=");
317 pos++;
319 for (i = 0; decl->u.flags.flags[i].name; ++i) {
320 if (i) {
321 printf(",");
322 pos++;
324 for (j = i;
325 decl->u.flags.flags[j].mask == decl->u.flags.flags[i].mask;
326 ++j) {
327 if (j != i) {
328 printf("|");
329 pos++;
331 printf("%s", decl->u.flags.flags[j].name);
332 pos += strlen(decl->u.flags.flags[j].name);
334 i = j - 1;
337 pos = print_help_msg(decl, pos);
338 print_default_flags(decl, pos);
340 printf("\n");
343 static void print_bool_help(struct isl_arg *decl, const char *prefix)
345 int pos;
346 int no = decl->u.b.default_value == 1;
347 pos = print_arg_help(decl, prefix, no);
348 pos = print_help_msg(decl, pos);
349 print_default(decl, no ? "yes" : "no", pos);
350 printf("\n");
353 static int print_argument_name(struct isl_arg *decl, const char *name, int pos)
355 printf("%c<%s>", decl->long_name ? '=' : ' ', name);
356 return pos + 3 + strlen(name);
359 static void print_long_help(struct isl_arg *decl, const char *prefix)
361 int pos;
362 pos = print_arg_help(decl, prefix, 0);
363 if (decl->u.l.default_value != decl->u.l.default_selected) {
364 printf("[");
365 pos++;
367 printf("=long");
368 pos += 5;
369 if (decl->u.l.default_value != decl->u.l.default_selected) {
370 printf("]");
371 pos++;
373 print_help_msg(decl, pos);
374 printf("\n");
377 static void print_ulong_help(struct isl_arg *decl, const char *prefix)
379 int pos;
380 pos = print_arg_help(decl, prefix, 0);
381 printf("=ulong");
382 pos += 6;
383 print_help_msg(decl, pos);
384 printf("\n");
387 static void print_str_help(struct isl_arg *decl, const char *prefix)
389 int pos;
390 const char *a = decl->argument_name ? decl->argument_name : "string";
391 pos = print_arg_help(decl, prefix, 0);
392 pos = print_argument_name(decl, a, pos);
393 pos = print_help_msg(decl, pos);
394 if (decl->u.str.default_value)
395 print_default(decl, decl->u.str.default_value, pos);
396 printf("\n");
399 static void print_help(struct isl_arg *arg, const char *prefix)
401 int i;
403 for (i = 0; arg[i].type != isl_arg_end; ++i) {
404 switch (arg[i].type) {
405 case isl_arg_flags:
406 print_flags_help(&arg[i], prefix);
407 break;
408 case isl_arg_choice:
409 print_choice_help(&arg[i], prefix);
410 break;
411 case isl_arg_bool:
412 print_bool_help(&arg[i], prefix);
413 break;
414 case isl_arg_long:
415 print_long_help(&arg[i], prefix);
416 break;
417 case isl_arg_ulong:
418 print_ulong_help(&arg[i], prefix);
419 break;
420 case isl_arg_str:
421 print_str_help(&arg[i], prefix);
422 break;
423 case isl_arg_version:
424 case isl_arg_arg:
425 case isl_arg_child:
426 case isl_arg_user:
427 case isl_arg_end:
428 break;
432 for (i = 0; arg[i].type != isl_arg_end; ++i) {
433 if (arg[i].type != isl_arg_child)
434 continue;
436 printf("\n");
437 if (arg[i].help_msg)
438 printf(" %s\n", arg[i].help_msg);
439 print_help(arg[i].u.child.child, arg[i].long_name);
443 static const char *prog_name(const char *prog)
445 const char *slash;
447 slash = strrchr(prog, '/');
448 if (slash)
449 prog = slash + 1;
450 if (strncmp(prog, "lt-", 3) == 0)
451 prog += 3;
453 return prog;
456 static int any_version(struct isl_arg *decl)
458 int i;
460 for (i = 0; decl[i].type != isl_arg_end; ++i) {
461 switch (decl[i].type) {
462 case isl_arg_version:
463 return 1;
464 case isl_arg_child:
465 if (any_version(decl[i].u.child.child))
466 return 1;
467 break;
468 default:
469 break;
473 return 0;
476 static void print_help_and_exit(struct isl_arg *arg, const char *prog)
478 int i;
480 printf("Usage: %s [OPTION...]", prog_name(prog));
482 for (i = 0; arg[i].type != isl_arg_end; ++i)
483 if (arg[i].type == isl_arg_arg)
484 printf(" %s", arg[i].argument_name);
486 printf("\n\n");
488 print_help(arg, NULL);
489 if (any_version(arg)) {
490 printf("\n");
491 printf(" -V, --version\n");
494 exit(0);
497 static const char *skip_name(struct isl_arg *decl, const char *arg,
498 const char *prefix, int need_argument, int *has_argument)
500 const char *equal;
501 const char *name;
502 const char *end;
504 if (arg[0] == '-' && arg[1] && arg[1] == decl->short_name) {
505 if (need_argument && !arg[2])
506 return NULL;
507 if (has_argument)
508 *has_argument = arg[2] != '\0';
509 return arg + 2;
511 if (!decl->long_name)
512 return NULL;
514 if (strncmp(arg, "--", 2))
515 return NULL;
517 name = arg + 2;
518 equal = strchr(name, '=');
519 if (need_argument && !equal)
520 return NULL;
522 if (has_argument)
523 *has_argument = !!equal;
524 end = equal ? equal : name + strlen(name);
526 if (prefix) {
527 size_t prefix_len = strlen(prefix);
528 if (strncmp(name, prefix, prefix_len) == 0 &&
529 name[prefix_len] == '-')
530 name += prefix_len + 1;
533 if (end - name != strlen(decl->long_name) ||
534 strncmp(name, decl->long_name, end - name))
535 return NULL;
537 return equal ? equal + 1 : end;
540 static int parse_choice_option(struct isl_arg *decl, char **arg,
541 const char *prefix, void *opt)
543 int i;
544 int has_argument;
545 const char *choice;
547 choice = skip_name(decl, arg[0], prefix, 0, &has_argument);
548 if (!choice)
549 return 0;
551 if (!has_argument && (!arg[1] || arg[1][0] == '-')) {
552 unsigned u = decl->u.choice.default_selected;
553 if (decl->u.choice.set)
554 decl->u.choice.set(opt, u);
555 else
556 *(unsigned *)(((char *)opt) + decl->offset) = u;
558 return 1;
561 if (!has_argument)
562 choice = arg[1];
564 for (i = 0; decl->u.choice.choice[i].name; ++i) {
565 unsigned u;
567 if (strcmp(choice, decl->u.choice.choice[i].name))
568 continue;
570 u = decl->u.choice.choice[i].value;
571 if (decl->u.choice.set)
572 decl->u.choice.set(opt, u);
573 else
574 *(unsigned *)(((char *)opt) + decl->offset) = u;
576 return has_argument ? 1 : 2;
579 return 0;
582 static int set_flag(struct isl_arg *decl, unsigned *val, const char *flag,
583 size_t len)
585 int i;
587 for (i = 0; decl->u.flags.flags[i].name; ++i) {
588 if (strncmp(flag, decl->u.flags.flags[i].name, len))
589 continue;
591 *val &= ~decl->u.flags.flags[i].mask;
592 *val |= decl->u.flags.flags[i].value;
594 return 1;
597 return 0;
600 static int parse_flags_option(struct isl_arg *decl, char **arg,
601 const char *prefix, void *opt)
603 int has_argument;
604 const char *flags;
605 const char *comma;
606 unsigned val;
608 flags = skip_name(decl, arg[0], prefix, 0, &has_argument);
609 if (!flags)
610 return 0;
612 if (!has_argument && !arg[1])
613 return 0;
615 if (!has_argument)
616 flags = arg[1];
618 val = *(unsigned *)(((char *)opt) + decl->offset);
620 while ((comma = strchr(flags, ',')) != NULL) {
621 if (!set_flag(decl, &val, flags, comma - flags))
622 return 0;
623 flags = comma + 1;
625 if (!set_flag(decl, &val, flags, strlen(flags)))
626 return 0;
628 *(unsigned *)(((char *)opt) + decl->offset) = val;
630 return has_argument ? 1 : 2;
633 static int parse_bool_option(struct isl_arg *decl, const char *arg,
634 const char *prefix, void *opt)
636 if (skip_name(decl, arg, prefix, 0, NULL)) {
637 *(unsigned *)(((char *)opt) + decl->offset) = 1;
639 return 1;
642 if (!decl->long_name)
643 return 0;
645 if (strncmp(arg, "--", 2))
646 return 0;
647 arg += 2;
649 if (prefix) {
650 size_t prefix_len = strlen(prefix);
651 if (strncmp(arg, prefix, prefix_len) == 0 &&
652 arg[prefix_len] == '-') {
653 arg += prefix_len + 1;
654 prefix = NULL;
658 if (strncmp(arg, "no-", 3))
659 return 0;
660 arg += 3;
662 if (prefix) {
663 size_t prefix_len = strlen(prefix);
664 if (strncmp(arg, prefix, prefix_len) == 0 &&
665 arg[prefix_len] == '-')
666 arg += prefix_len + 1;
669 if (!strcmp(arg, decl->long_name)) {
670 *(unsigned *)(((char *)opt) + decl->offset) = 0;
672 return 1;
675 return 0;
678 static int parse_str_option(struct isl_arg *decl, char **arg,
679 const char *prefix, void *opt)
681 int has_argument;
682 const char *s;
683 char **p = (char **)(((char *)opt) + decl->offset);
685 s = skip_name(decl, arg[0], prefix, 0, &has_argument);
686 if (!s)
687 return 0;
689 if (has_argument) {
690 *p = strdup(s);
691 return 1;
694 if (arg[1]) {
695 *p = strdup(arg[1]);
696 return 2;
699 return 0;
702 static int parse_long_option(struct isl_arg *decl, char **arg,
703 const char *prefix, void *opt)
705 int has_argument;
706 const char *val;
707 char *endptr;
708 long *p = (long *)(((char *)opt) + decl->offset);
710 val = skip_name(decl, arg[0], prefix, 0, &has_argument);
711 if (!val)
712 return 0;
714 if (has_argument) {
715 long l = strtol(val, NULL, 0);
716 if (decl->u.l.set)
717 decl->u.l.set(opt, l);
718 else
719 *p = l;
720 return 1;
723 if (arg[1]) {
724 long l = strtol(arg[1], &endptr, 0);
725 if (*endptr == '\0') {
726 if (decl->u.l.set)
727 decl->u.l.set(opt, l);
728 else
729 *p = l;
730 return 2;
734 if (decl->u.l.default_value != decl->u.l.default_selected) {
735 if (decl->u.l.set)
736 decl->u.l.set(opt, decl->u.l.default_selected);
737 else
738 *p = decl->u.l.default_selected;
739 return 1;
742 return 0;
745 static int parse_ulong_option(struct isl_arg *decl, char **arg,
746 const char *prefix, void *opt)
748 int has_argument;
749 const char *val;
750 char *endptr;
751 unsigned long *p = (unsigned long *)(((char *)opt) + decl->offset);
753 val = skip_name(decl, arg[0], prefix, 0, &has_argument);
754 if (!val)
755 return 0;
757 if (has_argument) {
758 *p = strtoul(val, NULL, 0);
759 return 1;
762 if (arg[1]) {
763 unsigned long ul = strtoul(arg[1], &endptr, 0);
764 if (*endptr == '\0') {
765 *p = ul;
766 return 2;
770 return 0;
773 static int parse_option(struct isl_arg *decl, char **arg,
774 const char *prefix, void *opt);
776 static int parse_child_option(struct isl_arg *decl, char **arg, void *opt)
778 return parse_option(decl->u.child.child, arg, decl->long_name,
779 *(void **)(((char *)opt) + decl->offset));
782 static int parse_option(struct isl_arg *decl, char **arg,
783 const char *prefix, void *opt)
785 int i;
787 for (i = 0; decl[i].type != isl_arg_end; ++i) {
788 int parsed = 0;
789 switch (decl[i].type) {
790 case isl_arg_choice:
791 parsed = parse_choice_option(&decl[i], arg, prefix, opt);
792 break;
793 case isl_arg_flags:
794 parsed = parse_flags_option(&decl[i], arg, prefix, opt);
795 break;
796 case isl_arg_long:
797 parsed = parse_long_option(&decl[i], arg, prefix, opt);
798 break;
799 case isl_arg_ulong:
800 parsed = parse_ulong_option(&decl[i], arg, prefix, opt);
801 break;
802 case isl_arg_bool:
803 parsed = parse_bool_option(&decl[i], *arg, prefix, opt);
804 break;
805 case isl_arg_str:
806 parsed = parse_str_option(&decl[i], arg, prefix, opt);
807 break;
808 case isl_arg_child:
809 parsed = parse_child_option(&decl[i], arg, opt);
810 break;
811 case isl_arg_arg:
812 case isl_arg_user:
813 case isl_arg_version:
814 case isl_arg_end:
815 break;
817 if (parsed)
818 return parsed;
821 return 0;
824 static void print_version(struct isl_arg *decl)
826 int i;
828 for (i = 0; decl[i].type != isl_arg_end; ++i) {
829 switch (decl[i].type) {
830 case isl_arg_version:
831 decl[i].u.version.print_version();
832 break;
833 case isl_arg_child:
834 print_version(decl[i].u.child.child);
835 break;
836 default:
837 break;
842 static void print_version_and_exit(struct isl_arg *decl)
844 print_version(decl);
846 exit(0);
849 static int drop_argument(int argc, char **argv, int drop, int n)
851 for (; drop < argc; ++drop)
852 argv[drop] = argv[drop + n];
854 return argc - n;
857 static int n_arg(struct isl_arg *arg)
859 int i;
860 int n_arg = 0;
862 for (i = 0; arg[i].type != isl_arg_end; ++i)
863 if (arg[i].type == isl_arg_arg)
864 n_arg++;
866 return n_arg;
869 static int next_arg(struct isl_arg *arg, int a)
871 for (++a; arg[a].type != isl_arg_end; ++a)
872 if (arg[a].type == isl_arg_arg)
873 return a;
875 return -1;
878 int isl_arg_parse(struct isl_arg *arg, int argc, char **argv, void *opt,
879 unsigned flags)
881 int a = -1;
882 int skip = 0;
883 int i;
884 int n;
886 n = n_arg(arg);
888 for (i = 1; i < argc; ++i) {
889 if (strcmp(argv[i], "--help") == 0)
890 print_help_and_exit(arg, argv[0]);
893 for (i = 1; i < argc; ++i) {
894 if ((strcmp(argv[i], "--version") == 0 ||
895 strcmp(argv[i], "-V") == 0) && any_version(arg))
896 print_version_and_exit(arg);
899 while (argc > 1 + skip) {
900 int parsed;
901 if (argv[1 + skip][0] != '-') {
902 a = next_arg(arg, a);
903 if (a >= 0) {
904 const char **p;
905 p = (const char **)(((char *)opt)+arg[a].offset);
906 *p = strdup(argv[1 + skip]);
907 argc = drop_argument(argc, argv, 1 + skip, 1);
908 --n;
909 } else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
910 fprintf(stderr, "%s: extra argument: %s\n",
911 prog_name(argv[0]), argv[1 + skip]);
912 exit(-1);
913 } else
914 ++skip;
915 continue;
917 parsed = parse_option(arg, &argv[1 + skip], NULL, opt);
918 if (parsed)
919 argc = drop_argument(argc, argv, 1 + skip, parsed);
920 else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
921 fprintf(stderr, "%s: unrecognized option: %s\n",
922 prog_name(argv[0]), argv[1 + skip]);
923 exit(-1);
924 } else
925 ++skip;
928 if (n > 0) {
929 fprintf(stderr, "%s: expecting %d more argument(s)\n",
930 prog_name(argv[0]), n);
931 exit(-1);
934 return argc;