root/ompi/mpi/c/pack_size.c

DEFINITIONS

1. MPI_Pack_size

   1 /*
   2  * Copyright (c) 2004-2007 The Trustees of Indiana University and Indiana
   3  *                         University Research and Technology
   4  *                         Corporation.  All rights reserved.
   5  * Copyright (c) 2004-2005 The University of Tennessee and The University
   6  *                         of Tennessee Research Foundation.  All rights
   7  *                         reserved.
   8  * Copyright (c) 2004-2008 High Performance Computing Center Stuttgart,
   9  *                         University of Stuttgart.  All rights reserved.
  10  * Copyright (c) 2004-2005 The Regents of the University of California.
  11  *                         All rights reserved.
  12  * Copyright (c) 2006      Cisco Systems, Inc.  All rights reserved.
  13  * Copyright (c) 2015      Research Organization for Information Science
  14  *                         and Technology (RIST). All rights reserved.
  15  * $COPYRIGHT$
  16  *
  17  * Additional copyrights may follow
  18  *
  19  * $HEADER$
  20  */
  21 #include "ompi_config.h"
  22 #include <stdio.h>
  23 
  24 #include "ompi/mpi/c/bindings.h"
  25 #include "ompi/runtime/params.h"
  26 #include "ompi/communicator/communicator.h"
  27 #include "ompi/errhandler/errhandler.h"
  28 #include "ompi/datatype/ompi_datatype.h"
  29 #include "opal/datatype/opal_convertor.h"
  30 #include "ompi/memchecker.h"
  31 
  32 #if OMPI_BUILD_MPI_PROFILING
  33 #if OPAL_HAVE_WEAK_SYMBOLS
  34 #pragma weak MPI_Pack_size = PMPI_Pack_size
  35 #endif
  36 #define MPI_Pack_size PMPI_Pack_size
  37 #endif
  38 
  39 static const char FUNC_NAME[] = "MPI_Pack_size";
  40 
  41 int MPI_Pack_size(int incount, MPI_Datatype datatype, MPI_Comm comm,
  42                   int *size)
  43 {
  44     opal_convertor_t local_convertor;
  45     size_t length;
  46 
  47     MEMCHECKER(
  48         memchecker_datatype(datatype);
  49         memchecker_comm(comm);
  50     );
  51 
  52     if (MPI_PARAM_CHECK) {
  53         OMPI_ERR_INIT_FINALIZE(FUNC_NAME);
  54         if (ompi_comm_invalid(comm)) {
  55             return OMPI_ERRHANDLER_INVOKE(MPI_COMM_WORLD, MPI_ERR_COMM,
  56                                           FUNC_NAME);
  57         } else if (NULL == size) {
  58             return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_ARG, FUNC_NAME);
  59         } else if (MPI_DATATYPE_NULL == datatype || NULL == datatype) {
  60             return OMPI_ERRHANDLER_INVOKE(comm, MPI_ERR_TYPE, FUNC_NAME);
  61         }
  62     }
  63 
  64     OPAL_CR_ENTER_LIBRARY();
  65 
  66     OBJ_CONSTRUCT( &local_convertor, opal_convertor_t );
  67     /* the resulting convertor will be set to the position ZERO */
  68     opal_convertor_copy_and_prepare_for_send( ompi_mpi_local_convertor, &(datatype->super),
  69                                               incount, NULL, 0, &local_convertor );
  70 
  71     opal_convertor_get_packed_size( &local_convertor, &length );
  72     *size = (int)length;
  73     OBJ_DESTRUCT( &local_convertor );
  74 
  75     OPAL_CR_EXIT_LIBRARY();
  76 
  77     return MPI_SUCCESS;
  78 }