star-gs

sgs.c (5077B)

---

 /* 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/>. */
 
 #include "sgs.h"
 #include "sgs_args.h"
 #include "sgs_c.h"
 #include "sgs_log.h"
 
 #include <star/s3d.h>
 
 #include <rsys/cstr.h>
 #include <rsys/double3.h>
 #include <rsys/mem_allocator.h>
 
 /*******************************************************************************
  * Helper functions
  ******************************************************************************/
 static res_T
 setup_scene(struct sgs* sgs, const struct sgs_args* args)
 {
   res_T res = RES_OK;
 
   res = sgs_geometry_box_create(sgs, &args->geom, &sgs->scene.geom);
   if(res != RES_OK) goto error;
 
   d3_sub(sgs->scene.D, args->geom.upper, args->geom.lower);
 
   /* Setup the 2D lower/upper bounds of the receiver */
   sgs->scene.recv_min[0] = sgs->scene.D[0]*1.0/8.0 + args->geom.lower[0];
   sgs->scene.recv_min[1] = sgs->scene.D[1]*3.0/8.0 + args->geom.lower[1];
   sgs->scene.recv_max[0] = sgs->scene.D[0]*3.0/8.0 + args->geom.lower[0];
   sgs->scene.recv_max[1] = sgs->scene.D[1]*5.0/8.0 + args->geom.lower[1];
 
 exit:
   return res;
 error:
   goto exit;
 }
 
 /*******************************************************************************
  * sgs API
  ******************************************************************************/
 res_T
 sgs_create
   (struct mem_allocator* mem_allocator,
    const struct sgs_args* args,
    struct sgs** out_sgs)
 {
   struct sgs* sgs = NULL;
   struct mem_allocator* allocator = NULL;
   res_T res = RES_OK;
   ASSERT(args && out_sgs);
 
   allocator = mem_allocator ? mem_allocator : &mem_default_allocator;
   sgs = MEM_CALLOC(allocator, 1, sizeof(struct sgs));
   if(!sgs) {
     fprintf(stderr, SGS_LOG_ERROR_PREFIX
       "Could not allocate the sgs data structure.\n");
     res = RES_MEM_ERR;
     goto error;
   }
   sgs->allocator = allocator;
   sgs->nrealisations = (size_t)args->nrealisations;
   sgs->nthreads = args->nthreads;
   sgs->dump_geometry = args->dump_geometry;
 
   setup_logger(sgs);
 
   res = s3d_device_create(&sgs->logger, sgs->allocator, 0, &sgs->s3d);
   if(res != RES_OK) {
     sgs_log_err(sgs, "Error creating the Star-3D device -- %s.\n",
       res_to_cstr(res));
     goto error;
   }
 
   res = setup_scene(sgs, args);
   if(res != RES_OK) goto error;
 
 exit:
   *out_sgs = sgs;
   return res;
 error:
   if(sgs) {
     sgs_release(sgs);
     sgs = NULL;
   }
   goto exit;
 }
 
 void
 sgs_release(struct sgs* sgs)
 {
   ASSERT(sgs);
 
   if(sgs->s3d) S3D(device_ref_put(sgs->s3d));
   if(sgs->scene.geom) sgs_geometry_ref_put(sgs->scene.geom);
   logger_release(&sgs->logger);
   MEM_RM(sgs->allocator, sgs);
 }
 
 struct mem_allocator*
 sgs_get_allocator(struct sgs* sgs)
 {
   ASSERT(sgs);
   return sgs->allocator;
 }
 
 struct s3d_device*
 sgs_get_s3d_device(struct sgs* sgs)
 {
   ASSERT(sgs);
   return sgs->s3d;
 }
 
 void
 sgs_fprint_copyright(FILE* stream)
 {
   ASSERT(stream);
   fprintf(stream,
 "Copyright (C) 2021-2023 Centre National de la Recherche Scientifique\n"
 "Copyright (C) 2021-2023 INSA Lyon\n"
 "Copyright (C) 2021-2023 Institut Mines Télécom Albi-Carmaux\n"
 "Copyright (C) 2021-2023 |Méso|Star> (contact@meso-star.com)\n"
 "Copyright (C) 2021-2023 Institut Pascal\n"
 "Copyright (C) 2021-2023 PhotonLyX (info@photonlyx.com)\n"
 "Copyright (C) 2021-2023 Université de Lorraine\n"
 "Copyright (C) 2021-2023 Université Paul Sabatier\n"
 "Copyright (C) 2021-2023 Université Toulouse - Jean Jaurès\n");
 }
 
 void
 sgs_fprint_license(FILE* stream)
 {
   ASSERT(stream);
   fprintf(stream,
 "Star Geometric Sensitivity is free software released under the GNU GPL\n"
 "license, version 3 or later. You are free to change or redistribute it\n"
 "under certain conditions <http://gnu.org/licenses/gpl.html>.\n");
 }
 
 res_T
 sgs_run(struct sgs* sgs)
 {
   res_T res = RES_OK;
   ASSERT(sgs);
 
   if(sgs->dump_geometry) {
     res = sgs_geometry_dump_vtk(sgs->scene.geom, stdout);
     if(res != RES_OK) goto error;
   } else {
     res = compute_sensitivity_translation(sgs);
     if(res != RES_OK) goto error;
   }
 
 exit:
   return res;
 error:
   goto exit;
 }