star-gs

sgs_args.c (5920B)

---

 /* Copyright (C) 2021-2023 Centre National de la Recherche Scientifique
  * Copyright (C) 2021-2023 INSA Lyon
  * Copyright (C) 2021-2023 Institut Mines Télécom Albi-Carmaux
  * Copyright (C) 2021-2023 |Méso|Star> (contact@meso-star.com)
  * Copyright (C) 2021-2023 Institut Pascal
  * Copyright (C) 2021-2023 PhotonLyX (info@photonlyx.com)
  * Copyright (C) 2021-2023 Université de Lorraine
  * Copyright (C) 2021-2023 Université Paul Sabatier
  * Copyright (C) 2021-2023 Université Toulouse - Jean Jaurès
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * the Free Software Foundation, either version 3 of the License, or
  * (at your option) any later version.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with this program. If not, see <http://www.gnu.org/licenses/>. */
 
 #define _POSIX_C_SOURCE 200112L /* strtok_r support */
 
 #include "sgs.h"
 #include "sgs_args.h"
 #include "sgs_log.h"
 
 #include <rsys/cstr.h>
 
 #include <getopt.h>
 #include <string.h>
 
 /*******************************************************************************
  * Helper functions
  ******************************************************************************/
 static void
 print_help(void)
 {
   printf("Usage: sgs [option] ...\n");
   printf("Calculate geometric sensitivity.\n");
   printf("\n");
 
   printf(
 "  -b <box-params:...>\n"
 "                 define an axis aligned bounding box\n"
 "                 Available box parameters are:\n");
   printf(
 "                 default\n"
 "                     use default settings.\n");
   printf(
 "                 low=x,y,z\n"
 "                     define the lower bound of the box\n"
 "                     (default: %g,%g,%g).\n",
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.lower[0],
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.lower[1],
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.lower[2]);
   printf(
 "                 upp=x,y,z\n"
 "                     define the upper bound of the box\n"
 "                     (default: %g,%g,%g).\n",
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.upper[0],
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.upper[1],
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.upper[2]);
 
   printf(
 "                 pi=t\n"
 "                   translation of the top face\n"
 "                   (default: %g).\n",
     SGS_GEOMETRY_BOX_ARGS_DEFAULT.pi);
   printf(
 "  -d             write geometry to stdout\n"
 "                 (VTK file format).\n");
   printf(
 "  -h             display this help and exit.\n");
   printf(
 "  -n nrealisations\n"
 "                 number of realisations (default: %li).\n",
     SGS_ARGS_DEFAULT.nrealisations);
   printf(
 "  -t nthreads    hint on the number of threads to use.\n"
 "                 Default assumes as many threads as CPU cores.\n");
   printf("\n");
 
   sgs_fprint_license(stdout);
 }
 
 static res_T
 parse_box_parameters(const char* str, void* ptr)
 {
   char buf[128];
   struct sgs_geometry_box_args* box = ptr;
   char* key;
   char* val;
   char* ctx;
   size_t len;
   res_T res = RES_OK;
   ASSERT(ptr && str);
 
   if(strlen(str) >= sizeof(buf) -1/*NULL char*/) {
     fprintf(stderr, SGS_LOG_ERROR_PREFIX
       "Could not duplicate the box option string `%s'.\n", str);
     res = RES_MEM_ERR;
     goto error;
   }
   strncpy(buf, str, sizeof(buf));
 
   key = strtok_r(buf, "=", &ctx);
   val = strtok_r(NULL, "", &ctx);
 
   if(!strcmp(key, "default")) {
     *box = SGS_GEOMETRY_BOX_ARGS_DEFAULT;
   } else if(!strcmp(key, "low")) {
     res = cstr_to_list_double(val, ',', box->lower, &len, 3);
     if(res == RES_OK && len != 3) res = RES_BAD_ARG;
   } else if(!strcmp(key, "upp")) {
     res = cstr_to_list_double(val, ',', box->upper, &len, 3);
     if(res == RES_OK && len != 3) res = RES_BAD_ARG;
   } else if(!strcmp(key, "pi")) {
     res = cstr_to_double(val, &box->pi);
   } else {
     fprintf(stderr, SGS_LOG_ERROR_PREFIX
       "Invalid box parameter `%s'.\n", key);
     res = RES_BAD_ARG;
     goto error;
   }
   if(res != RES_OK) {
     fprintf(stderr, SGS_LOG_ERROR_PREFIX
       "Error parsing the value of the `%s' box parameter: %s -- %s.\n",
       key, val, res_to_cstr(res));
     goto error;
   }
 
 exit:
   return res;
 error:
   goto exit;
 }
 
 /*******************************************************************************
  * sgs_args API
  ******************************************************************************/
 res_T
 sgs_args_init(struct sgs_args* args, int argc, char** argv)
 {
   int opt;
   res_T res = RES_OK;
   ASSERT(args && argc && argv);
 
   *args = SGS_ARGS_DEFAULT;
 
   while((opt = getopt(argc, argv, "b:dhn:t:")) != -1) {
     switch(opt) {
       case 'b':
         args->geom = SGS_GEOMETRY_BOX_ARGS_DEFAULT;
         res = cstr_parse_list
           (optarg, ':', parse_box_parameters, &args->geom);
         break;
       case 'd': args->dump_geometry = 1; break;
       case 'h':
         print_help();
         sgs_args_release(args);
         args->quit = 1;
         break;
       case 'n':
         res = cstr_to_long(optarg, &args->nrealisations);
         if(res == RES_OK && args->nrealisations <= 0) res = RES_BAD_ARG;
         break;
       case 't':
         res = cstr_to_uint(optarg, &args->nthreads);
         if(res == RES_OK && !args->nthreads) res = RES_BAD_ARG;
         break;
       default: res = RES_BAD_ARG; break;
     }
     if(res != RES_OK) {
       if(optarg) {
         fprintf(stderr, SGS_LOG_ERROR_PREFIX
           "%s: invalid option argument '%s' -- '%c'\n",
           argv[0], optarg, opt);
       }
       goto error;
     }
   }
 exit:
   return res;
 error:
   sgs_args_release(args);
   goto exit;
 }
 
 void
 sgs_args_release(struct sgs_args* args)
 {
   ASSERT(args);
   *args = SGS_ARGS_DEFAULT;
 }