isl_options_parse: print default on boolean options
[isl.git] / isl_arg.c
blob8f7ac1df7fd57849cd5658277f298741a1506969
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 void print_long_help(struct isl_arg *decl, const char *prefix)
355 int pos;
356 pos = print_arg_help(decl, prefix, 0);
357 if (decl->u.l.default_value != decl->u.l.default_selected) {
358 printf("[");
359 pos++;
361 printf("=long");
362 pos += 5;
363 if (decl->u.l.default_value != decl->u.l.default_selected) {
364 printf("]");
365 pos++;
367 print_help_msg(decl, pos);
368 printf("\n");
371 static void print_ulong_help(struct isl_arg *decl, const char *prefix)
373 int pos;
374 pos = print_arg_help(decl, prefix, 0);
375 printf("=ulong");
376 pos += 6;
377 print_help_msg(decl, pos);
378 printf("\n");
381 static void print_str_help(struct isl_arg *decl, const char *prefix)
383 int pos;
384 const char *a = decl->argument_name ? decl->argument_name : "string";
385 pos = print_arg_help(decl, prefix, 0);
386 printf("=%s", a);
387 pos += 1 + strlen(a);
388 print_help_msg(decl, pos);
389 printf("\n");
392 static void print_help(struct isl_arg *arg, const char *prefix)
394 int i;
396 for (i = 0; arg[i].type != isl_arg_end; ++i) {
397 switch (arg[i].type) {
398 case isl_arg_flags:
399 print_flags_help(&arg[i], prefix);
400 break;
401 case isl_arg_choice:
402 print_choice_help(&arg[i], prefix);
403 break;
404 case isl_arg_bool:
405 print_bool_help(&arg[i], prefix);
406 break;
407 case isl_arg_long:
408 print_long_help(&arg[i], prefix);
409 break;
410 case isl_arg_ulong:
411 print_ulong_help(&arg[i], prefix);
412 break;
413 case isl_arg_str:
414 print_str_help(&arg[i], prefix);
415 break;
416 case isl_arg_version:
417 printf(" -V, --version\n");
418 break;
419 case isl_arg_arg:
420 case isl_arg_child:
421 case isl_arg_user:
422 case isl_arg_end:
423 break;
427 for (i = 0; arg[i].type != isl_arg_end; ++i) {
428 if (arg[i].type != isl_arg_child)
429 continue;
431 printf("\n");
432 if (arg[i].help_msg)
433 printf(" %s\n", arg[i].help_msg);
434 print_help(arg[i].u.child.child, arg[i].long_name);
438 static const char *prog_name(const char *prog)
440 const char *slash;
442 slash = strrchr(prog, '/');
443 if (slash)
444 prog = slash + 1;
445 if (strncmp(prog, "lt-", 3) == 0)
446 prog += 3;
448 return prog;
451 static void print_help_and_exit(struct isl_arg *arg, const char *prog)
453 int i;
455 printf("Usage: %s [OPTION...]", prog_name(prog));
457 for (i = 0; arg[i].type != isl_arg_end; ++i)
458 if (arg[i].type == isl_arg_arg)
459 printf(" %s", arg[i].argument_name);
461 printf("\n\n");
463 print_help(arg, NULL);
465 exit(0);
468 static const char *skip_name(struct isl_arg *decl, const char *arg,
469 const char *prefix, int need_argument, int *has_argument)
471 const char *equal;
472 const char *name;
473 const char *end;
475 if (arg[0] == '-' && arg[1] && arg[1] == decl->short_name) {
476 if (need_argument && !arg[2])
477 return NULL;
478 if (has_argument)
479 *has_argument = arg[2] != '\0';
480 return arg + 2;
483 if (strncmp(arg, "--", 2))
484 return NULL;
486 name = arg + 2;
487 equal = strchr(name, '=');
488 if (need_argument && !equal)
489 return NULL;
491 if (has_argument)
492 *has_argument = !!equal;
493 end = equal ? equal : name + strlen(name);
495 if (prefix) {
496 size_t prefix_len = strlen(prefix);
497 if (strncmp(name, prefix, prefix_len) == 0 &&
498 name[prefix_len] == '-')
499 name += prefix_len + 1;
502 if (end - name != strlen(decl->long_name) ||
503 strncmp(name, decl->long_name, end - name))
504 return NULL;
506 return equal ? equal + 1 : end;
509 static int parse_choice_option(struct isl_arg *decl, char **arg,
510 const char *prefix, void *opt)
512 int i;
513 int has_argument;
514 const char *choice;
516 choice = skip_name(decl, arg[0], prefix, 0, &has_argument);
517 if (!choice)
518 return 0;
520 if (!has_argument && (!arg[1] || arg[1][0] == '-')) {
521 unsigned u = decl->u.choice.default_selected;
522 if (decl->u.choice.set)
523 decl->u.choice.set(opt, u);
524 else
525 *(unsigned *)(((char *)opt) + decl->offset) = u;
527 return 1;
530 if (!has_argument)
531 choice = arg[1];
533 for (i = 0; decl->u.choice.choice[i].name; ++i) {
534 unsigned u;
536 if (strcmp(choice, decl->u.choice.choice[i].name))
537 continue;
539 u = decl->u.choice.choice[i].value;
540 if (decl->u.choice.set)
541 decl->u.choice.set(opt, u);
542 else
543 *(unsigned *)(((char *)opt) + decl->offset) = u;
545 return has_argument ? 1 : 2;
548 return 0;
551 static int set_flag(struct isl_arg *decl, unsigned *val, const char *flag,
552 size_t len)
554 int i;
556 for (i = 0; decl->u.flags.flags[i].name; ++i) {
557 if (strncmp(flag, decl->u.flags.flags[i].name, len))
558 continue;
560 *val &= ~decl->u.flags.flags[i].mask;
561 *val |= decl->u.flags.flags[i].value;
563 return 1;
566 return 0;
569 static int parse_flags_option(struct isl_arg *decl, char **arg,
570 const char *prefix, void *opt)
572 int has_argument;
573 const char *flags;
574 const char *comma;
575 unsigned val;
577 flags = skip_name(decl, arg[0], prefix, 0, &has_argument);
578 if (!flags)
579 return 0;
581 if (!has_argument && !arg[1])
582 return 0;
584 if (!has_argument)
585 flags = arg[1];
587 val = *(unsigned *)(((char *)opt) + decl->offset);
589 while ((comma = strchr(flags, ',')) != NULL) {
590 if (!set_flag(decl, &val, flags, comma - flags))
591 return 0;
592 flags = comma + 1;
594 if (!set_flag(decl, &val, flags, strlen(flags)))
595 return 0;
597 *(unsigned *)(((char *)opt) + decl->offset) = val;
599 return has_argument ? 1 : 2;
602 static int parse_bool_option(struct isl_arg *decl, const char *arg,
603 const char *prefix, void *opt)
605 if (skip_name(decl, arg, prefix, 0, NULL)) {
606 *(unsigned *)(((char *)opt) + decl->offset) = 1;
608 return 1;
611 if (strncmp(arg, "--", 2))
612 return 0;
613 arg += 2;
615 if (prefix) {
616 size_t prefix_len = strlen(prefix);
617 if (strncmp(arg, prefix, prefix_len) == 0 &&
618 arg[prefix_len] == '-') {
619 arg += prefix_len + 1;
620 prefix = NULL;
624 if (strncmp(arg, "no-", 3))
625 return 0;
626 arg += 3;
628 if (prefix) {
629 size_t prefix_len = strlen(prefix);
630 if (strncmp(arg, prefix, prefix_len) == 0 &&
631 arg[prefix_len] == '-')
632 arg += prefix_len + 1;
635 if (!strcmp(arg, decl->long_name)) {
636 *(unsigned *)(((char *)opt) + decl->offset) = 0;
638 return 1;
641 return 0;
644 static int parse_str_option(struct isl_arg *decl, char **arg,
645 const char *prefix, void *opt)
647 int has_argument;
648 const char *s;
649 char **p = (char **)(((char *)opt) + decl->offset);
651 s = skip_name(decl, arg[0], prefix, 0, &has_argument);
652 if (!s)
653 return 0;
655 if (has_argument) {
656 *p = strdup(s);
657 return 1;
660 if (arg[1]) {
661 *p = strdup(arg[1]);
662 return 2;
665 return 0;
668 static int parse_long_option(struct isl_arg *decl, char **arg,
669 const char *prefix, void *opt)
671 int has_argument;
672 const char *val;
673 char *endptr;
674 long *p = (long *)(((char *)opt) + decl->offset);
676 val = skip_name(decl, arg[0], prefix, 0, &has_argument);
677 if (!val)
678 return 0;
680 if (has_argument) {
681 long l = strtol(val, NULL, 0);
682 if (decl->u.l.set)
683 decl->u.l.set(opt, l);
684 else
685 *p = l;
686 return 1;
689 if (arg[1]) {
690 long l = strtol(arg[1], &endptr, 0);
691 if (*endptr == '\0') {
692 if (decl->u.l.set)
693 decl->u.l.set(opt, l);
694 else
695 *p = l;
696 return 2;
700 if (decl->u.l.default_value != decl->u.l.default_selected) {
701 if (decl->u.l.set)
702 decl->u.l.set(opt, decl->u.l.default_selected);
703 else
704 *p = decl->u.l.default_selected;
705 return 1;
708 return 0;
711 static int parse_ulong_option(struct isl_arg *decl, char **arg,
712 const char *prefix, void *opt)
714 int has_argument;
715 const char *val;
716 char *endptr;
717 unsigned long *p = (unsigned long *)(((char *)opt) + decl->offset);
719 val = skip_name(decl, arg[0], prefix, 0, &has_argument);
720 if (!val)
721 return 0;
723 if (has_argument) {
724 *p = strtoul(val, NULL, 0);
725 return 1;
728 if (arg[1]) {
729 unsigned long ul = strtoul(arg[1], &endptr, 0);
730 if (*endptr == '\0') {
731 *p = ul;
732 return 2;
736 return 0;
739 static int parse_option(struct isl_arg *decl, char **arg,
740 const char *prefix, void *opt);
742 static int parse_child_option(struct isl_arg *decl, char **arg, void *opt)
744 return parse_option(decl->u.child.child, arg, decl->long_name,
745 *(void **)(((char *)opt) + decl->offset));
748 static int parse_option(struct isl_arg *decl, char **arg,
749 const char *prefix, void *opt)
751 int i;
753 for (i = 0; decl[i].type != isl_arg_end; ++i) {
754 int parsed = 0;
755 switch (decl[i].type) {
756 case isl_arg_choice:
757 parsed = parse_choice_option(&decl[i], arg, prefix, opt);
758 break;
759 case isl_arg_flags:
760 parsed = parse_flags_option(&decl[i], arg, prefix, opt);
761 break;
762 case isl_arg_long:
763 parsed = parse_long_option(&decl[i], arg, prefix, opt);
764 break;
765 case isl_arg_ulong:
766 parsed = parse_ulong_option(&decl[i], arg, prefix, opt);
767 break;
768 case isl_arg_bool:
769 parsed = parse_bool_option(&decl[i], *arg, prefix, opt);
770 break;
771 case isl_arg_str:
772 parsed = parse_str_option(&decl[i], arg, prefix, opt);
773 break;
774 case isl_arg_child:
775 parsed = parse_child_option(&decl[i], arg, opt);
776 break;
777 case isl_arg_arg:
778 case isl_arg_user:
779 case isl_arg_version:
780 case isl_arg_end:
781 break;
783 if (parsed)
784 return parsed;
787 return 0;
790 static int any_version(struct isl_arg *decl)
792 int i;
794 for (i = 0; decl[i].type != isl_arg_end; ++i) {
795 switch (decl[i].type) {
796 case isl_arg_version:
797 return 1;
798 case isl_arg_child:
799 if (any_version(decl[i].u.child.child))
800 return 1;
801 break;
802 default:
803 break;
807 return 0;
810 static void print_version(struct isl_arg *decl)
812 int i;
814 for (i = 0; decl[i].type != isl_arg_end; ++i) {
815 switch (decl[i].type) {
816 case isl_arg_version:
817 decl[i].u.version.print_version();
818 break;
819 case isl_arg_child:
820 print_version(decl[i].u.child.child);
821 break;
822 default:
823 break;
828 static void print_version_and_exit(struct isl_arg *decl)
830 print_version(decl);
832 exit(0);
835 static int drop_argument(int argc, char **argv, int drop, int n)
837 for (; drop < argc; ++drop)
838 argv[drop] = argv[drop + n];
840 return argc - n;
843 static int n_arg(struct isl_arg *arg)
845 int i;
846 int n_arg = 0;
848 for (i = 0; arg[i].type != isl_arg_end; ++i)
849 if (arg[i].type == isl_arg_arg)
850 n_arg++;
852 return n_arg;
855 int isl_arg_parse(struct isl_arg *arg, int argc, char **argv, void *opt,
856 unsigned flags)
858 int skip = 0;
859 int i;
860 int n;
862 n = n_arg(arg);
864 for (i = 1; i < argc; ++i) {
865 if (strcmp(argv[i], "--help") == 0)
866 print_help_and_exit(arg, argv[0]);
869 for (i = 1; i < argc; ++i) {
870 if ((strcmp(argv[i], "--version") == 0 ||
871 strcmp(argv[i], "-V") == 0) && any_version(arg))
872 print_version_and_exit(arg);
875 while (argc > 1 + skip) {
876 int parsed;
877 if (argv[1 + skip][0] != '-')
878 break;
879 parsed = parse_option(arg, &argv[1 + skip], NULL, opt);
880 if (parsed)
881 argc = drop_argument(argc, argv, 1 + skip, parsed);
882 else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
883 fprintf(stderr, "%s: unrecognized option: %s\n",
884 prog_name(argv[0]), argv[1 + skip]);
885 exit(-1);
886 } else
887 ++skip;
890 if (ISL_FL_ISSET(flags, ISL_ARG_ALL) ? argc != 1 + n
891 : argc < 1 + skip + n) {
892 fprintf(stderr, "%s: expecting %d arguments\n",
893 prog_name(argv[0]), n);
894 exit(-1);
897 for (i = 0; arg[i].type != isl_arg_end; ++i) {
898 const char *str;
899 if (arg[i].type != isl_arg_arg)
900 continue;
901 str = strdup(argv[1 + skip]);
902 *(const char **)(((char *)opt) + arg[i].offset) = str;
903 argc = drop_argument(argc, argv, 1 + skip, 1);
906 return argc;