ltt/branches/poly/lttv/option.c

   1
   2 #include <popt.h>
   3 #include <glib.h>
   4 #include <lttv/option.h>
   5
   6 typedef struct _LttvOption {
   7   char *long_name;
   8   char char_name;
   9   char *description;
  10   char *arg_description;
  11   LttvOptionType t;
  12   gpointer p;
  13   LttvOptionHook hook;
  14   gpointer hook_data;
  15 } LttvOption;
  16
  17 GHashTable *options;
  18
  19
  20 static void
  21 list_options(gpointer key, gpointer value, gpointer user_data)
  22 {
  23   g_ptr_array_add((GPtrArray *)user_data, value);
  24 }
  25
  26
  27 static void
  28 free_option(LttvOption *option)
  29 {
  30   g_free(option->long_name);
  31   g_free(option->description);
  32   g_free(option->arg_description);
  33   g_free(option);
  34 }
  35
  36
  37 void lttv_option_init(int argc, char **argv)
  38 {
  39   options = g_hash_table_new(g_str_hash, g_str_equal);
  40 }
  41
  42
  43 void lttv_option_destroy()
  44 {
  45   LttvOption option;
  46
  47   GPtrArray *list = g_ptr_array_new();
  48
  49   int i;
  50
  51   g_hash_table_foreach(options, list_options, list);
  52   g_hash_table_destroy(options);
  53
  54   for(i = 0 ; i < list->len ; i++) {
  55     free_option((LttvOption *)list->pdata[i]);
  56   }
  57   g_ptr_array_free(list, TRUE);
  58 }
  59
  60
  61 void lttv_option_add(const char *long_name, const char char_name,
  62     const char *description, const char *arg_description,
  63     const LttvOptionType t, void *p,
  64     const LttvOptionHook h, void *hook_data)
  65 {
  66   LttvOption *option;
  67
  68   if(g_hash_table_lookup(options, long_name) != NULL) {
  69     g_warning("duplicate option");
  70     return;
  71   }
  72
  73   option = g_new(LttvOption, 1);
  74   option->long_name = g_strdup(long_name);
  75   option->char_name = char_name;
  76   option->description = g_strdup(description);
  77   option->arg_description = g_strdup(arg_description);
  78   option->t = t;
  79   option->p = p;
  80   option->hook = h;
  81   option->hook_data = hook_data;
  82   g_hash_table_insert(options, option->long_name, option);
  83 }
  84
  85
  86 void
  87 lttv_option_remove(const char *long_name)
  88 {
  89   LttvOption *option = g_hash_table_lookup(options, long_name);
  90
  91   if(option == NULL) {
  92     g_warning("trying to remove unknown option %s", long_name);
  93     return;
  94   }
  95   g_hash_table_remove(options, long_name);
  96   free_option(option);
  97 }
  98
  99
 100 static int poptToLTT[] = {
 101   POPT_ARG_NONE, POPT_ARG_STRING, POPT_ARG_INT, POPT_ARG_LONG
 102 };
 103
 104 static struct poptOption endOption = { NULL, '\0', 0, NULL, 0};
 105
 106
 107 static void
 108 build_popts(GPtrArray **plist, struct poptOption **ppopts, poptContext *pc,
 109     int argc, char **argv)
 110 {
 111   LttvOption *option;
 112
 113   GPtrArray *list;
 114
 115   struct poptOption *popts;
 116
 117   poptContext c;
 118
 119   guint i;
 120
 121   list = g_ptr_array_new();
 122
 123   g_hash_table_foreach(options, list_options, list);
 124
 125   /* Build a popt options array from our list */
 126
 127   popts = g_new(struct poptOption, list->len + 1);
 128
 129   for(i = 0 ; i < list->len ; i++) {
 130     option = (LttvOption *)list->pdata[i];
 131     popts[i].longName = option->long_name;
 132     popts[i].shortName = option->char_name;
 133     popts[i].descrip = option->description;
 134     popts[i].argDescrip = option->arg_description;
 135     popts[i].argInfo = poptToLTT[option->t];
 136     popts[i].arg = option->p;
 137     popts[i].val = i + 1;
 138   }
 139
 140   /* Terminate the array for popt and create the context */
 141
 142   popts[list->len] = endOption;
 143   c = poptGetContext(argv[0], argc, (const char**)argv, popts, 0);
 144
 145   *plist = list;
 146   *ppopts = popts;
 147   *pc = c;
 148 }
 149
 150
 151 static void
 152 destroy_popts(GPtrArray **plist, struct poptOption **ppopts, poptContext *pc)
 153 {
 154   g_ptr_array_free(*plist, TRUE); *plist = NULL;
 155   g_free(*ppopts); *ppopts = NULL;
 156   poptFreeContext(*pc);
 157 }
 158
 159
 160 void lttv_option_parse(int argc, char **argv)
 161 {
 162   GPtrArray *list;
 163
 164   LttvOption *option;
 165
 166   int i, rc, first_arg;
 167
 168   struct poptOption *popts;
 169
 170   poptContext c;
 171
 172   i = 0;
 173
 174   first_arg = 0;
 175
 176   build_popts(&list, &popts, &c, argc, argv);
 177
 178   /* Parse options while not end of options event */
 179
 180   while((rc = poptGetNextOpt(c)) != -1) {
 181
 182     /* The option was recognized and the rc value returned is the argument
 183        position in the array. Call the associated hook if present. */
 184
 185     if(rc > 0) {
 186       option = (LttvOption *)(list->pdata[rc - 1]);
 187       if(option->hook != NULL) option->hook(option->hook_data);
 188       i++;
 189     }
 190
 191     else if(rc == POPT_ERROR_BADOPT && i != first_arg) {
 192
 193       /* Perhaps this option is newly added, restart parsing */
 194
 195       destroy_popts(&list, &popts, &c);
 196       build_popts(&list, &popts, &c, argc, argv);
 197
 198       /* Get back to the same argument */
 199
 200       first_arg = i;
 201       for(i = 0; i < first_arg; i++) poptGetNextOpt(c);
 202     }
 203
 204     else {
 205
 206       /* The option has some error and it is not because this is a newly
 207          added option not recognized. */
 208
 209       g_error("option %s: %s", poptBadOption(c,0), poptStrerror(rc));
 210       break;
 211     }
 212
 213   }
 214
 215   destroy_popts(&list, &popts, &c);
 216 }
 217