src/smpi/include/smpi_f2c.hpp

   1 /* Handle Fortran - C conversion for MPI Types*/
   2
   3 /* Copyright (c) 2010-2023. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 #ifndef SMPI_F2C_HPP_INCLUDED
  10 #define SMPI_F2C_HPP_INCLUDED
  11
  12 #include <memory>
  13 #include <unordered_map>
  14 #include <string>
  15
  16 namespace simgrid::smpi {
  17
  18 class F2C {
  19 private:
  20   using f2c_lookup_type = std::unordered_map<unsigned int, F2C*>;
  21
  22   // We use a single lookup table for every type.
  23   // Beware of collisions if id in mpif.h is not unique
  24   static std::unique_ptr<f2c_lookup_type> f2c_lookup_;
  25   static int f2c_id_;
  26   static f2c_lookup_type::size_type num_default_handles_;
  27   int my_f2c_id_ = -1;
  28   bool deleted_ = false;
  29   std::string call_location_;
  30
  31 protected:
  32   static void allocate_lookup()
  33   {
  34     if (not f2c_lookup_)
  35       f2c_lookup_ = std::make_unique<f2c_lookup_type>();
  36   }
  37   int f2c_id() const { return my_f2c_id_; }
  38   static int global_f2c_id() { return f2c_id_; }
  39   static void f2c_id_increment() { f2c_id_++; }
  40
  41 public:
  42   void mark_as_deleted() { deleted_ = true; };
  43   bool deleted() const { return deleted_; }
  44   static f2c_lookup_type* lookup() { return f2c_lookup_.get(); }
  45   F2C();
  46   virtual ~F2C() = default;
  47   virtual std::string name() const = 0;
  48
  49   int add_f();
  50   static void free_f(int id) { if(id!=-1) f2c_lookup_->erase(id); }
  51   int c2f();
  52
  53   // This method should be overridden in all subclasses to avoid casting the result when calling it.
  54   // For the default one, the MPI_*_NULL returned is assumed to be NULL.
  55   static F2C* f2c(int id);
  56   static void finish_initialization() { num_default_handles_ = f2c_lookup_->size(); }
  57   static f2c_lookup_type::size_type get_num_default_handles() { return num_default_handles_; }
  58   const std::string& call_location() const { return call_location_; }
  59 };
  60
  61 } // namespace simgrid::smpi
  62
  63 #endif