src/smpi/include/smpi_f2c.hpp

   1 /* Handle Fortran - C conversion for MPI Types*/
   2
   3 /* Copyright (c) 2010-2021. 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
  15 namespace simgrid{
  16 namespace 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
  30 protected:
  31   static void allocate_lookup()
  32   {
  33     if (not f2c_lookup_)
  34       f2c_lookup_ = std::make_unique<f2c_lookup_type>();
  35   }
  36   static int f2c_id() { return f2c_id_; }
  37   static void f2c_id_increment() { f2c_id_++; }
  38   void mark_as_deleted() { deleted_ = true; };
  39
  40 public:
  41   bool deleted() const { return deleted_; }
  42   static f2c_lookup_type* lookup() { return f2c_lookup_.get(); }
  43   F2C();
  44   virtual ~F2C() = default;
  45
  46   int add_f();
  47   static void free_f(int id) { f2c_lookup_->erase(id); }
  48   int c2f();
  49
  50   // This method should be overridden in all subclasses to avoid casting the result when calling it.
  51   // For the default one, the MPI_*_NULL returned is assumed to be NULL.
  52   static F2C* f2c(int id);
  53   static void finish_initialization() { num_default_handles_ = f2c_lookup_->size(); }
  54   static f2c_lookup_type::size_type get_num_default_handles() { return num_default_handles_; }
  55 };
  56
  57 }
  58 }
  59
  60 #endif