async_io.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) NumeriCor GmbH
// SPDX-License-Identifier: LicenseRef-APL-1.1
 
#ifndef _ASYNC_IO_H
#define _ASYNC_IO_H
 
namespace opencarp {
namespace async {
 
#define ASYNC_TAG 1001
 
#define ASYNC_CMD_EXIT 1
#define ASYNC_CMD_REGISTER_OUTPUT 2
#define ASYNC_CMD_OUTPUT 3
 
struct intercomm_layout {
  int loc_rank = 0;                    
  int loc_size = 0;                    
  int rem_size = 0;                    
 
  inline void setup(MPI_Comm ic)
  {
    loc_rank = get_rank();
    loc_size = get_size();
    rem_size = get_remote_size(ic);
  }
};
 
struct async_IO_queue {
  SF::vector<IGBheader*>             IGBs;     
  SF::vector<SF::vector<long int>>   layouts;  
  SF::vector<SF::commgraph<size_t>>  cg;       
  SF::vector<SF::vector<mesh_int_t>> perm_b;   
  SF::vector<SF::vector<mesh_int_t>> perm_a;   
 
  inline int add(IGBheader* igb,
                 const SF::vector<long int> & lt,
                 const SF::commgraph<size_t> & c,
                 const SF::vector<mesh_int_t> & pb,
                 const SF::vector<mesh_int_t> & pa)
  {
    int id = cg.size();
 
    IGBs   .push_back(igb);
    layouts.push_back(lt);
    cg     .push_back(c);
    perm_b .push_back(pb);
    perm_a .push_back(pa);
 
    return id;
  }
};
 
// functions used by IO nodes
void IO_poll_for_output(async_IO_queue & io_queue);
void IO_register_output(async_IO_queue & io_queue);
void IO_get_sender_ranks(const intercomm_layout & il, SF::vector<int> & sender);
void IO_prepare_sort(const SF::vector<mesh_int_t> & inp_idx,
                     SF::commgraph<size_t> & grph,
                     SF::vector<mesh_int_t> & perm_before_comm,
                     SF::vector<mesh_int_t> & perm_after_comm);
void IO_do_output(async_IO_queue & io_queue);
 
 
// functions used by COMPUTE nodes
int  COMPUTE_register_output(const SF::vector<mesh_int_t> & idx,
                             const int dpn,
                             const char* name,
                             const char* units);
void COMPUTE_send_exit_flag();
int  COMPUTE_get_receive_rank(const intercomm_layout & il);
void COMPUTE_do_output(SF_real* data, const int lsize, const int IO_id);
void COMPUTE_do_output(SF_real* dat, const SF::vector<mesh_int_t> & idx, const int IO_id);
 
}}
 
#endif