mpid_type_indexed.c

刚才是说明 现在是安装程序在 LINUX环境下进行编程的MPICH安装文件

/* -*- Mode: C; c-basic-offset:4 ; -*- */

/*
 *  (C) 2001 by Argonne National Laboratory.
 *      See COPYRIGHT in top-level directory.
 */

#include <mpiimpl.h>
#include <mpid_dataloop.h>
#include <stdlib.h>
#include <assert.h>

/* #define MPID_TYPE_ALLOC_DEBUG */

/*@
  MPID_Type_indexed - create an indexed datatype
 
  Input Parameters:
+ count - number of blocks in vector
. blocklength_array - number of elements in each block
. displacement_array - offsets of blocks from start of type (see next
  parameter for units)
. dispinbytes - if nonzero, then displacements are in bytes, otherwise
  they in terms of extent of oldtype
- oldtype - type (using handle) of datatype on which vector is based

  Output Parameters:
. newtype - handle of new indexed datatype

  Return Value:
  0 on success, -1 on failure.

  This routine calls MPID_Dataloop_copy() to create the loops for this
  new datatype.  It calls MPIU_Handle_obj_alloc() to allocate space for the new
  datatype.
@*/

int MPID_Type_indexed(int count,
		      int *blocklength_array,
		      void *displacement_array,
		      int dispinbytes,
		      MPI_Datatype oldtype,
		      MPI_Datatype *newtype)
{
    int mpi_errno = MPI_SUCCESS;
    int i, new_loopsize, el_count, *iptr;
    char *curpos;

    MPID_Datatype *new_dtp;
    struct MPID_Dataloop *dlp;
    MPI_Aint *aptr;

    MPI_Aint min_lb = 0, max_ub = 0, eff_disp, el_extent, el_size;

    /* allocate new datatype object and handle */
    new_dtp = (MPID_Datatype *) MPIU_Handle_obj_alloc(&MPID_Datatype_mem);
    if (!new_dtp) {
	mpi_errno = MPIR_Err_create_code(MPI_ERR_OTHER, "**nomem", 0);
	return mpi_errno;
    }

    /* Note: handle is filled in by MPIU_Handle_obj_alloc() */
    MPIU_Object_set_ref(new_dtp, 1);
    new_dtp->is_permanent = 0;
    new_dtp->is_committed = 0;
    new_dtp->attributes   = 0;
    new_dtp->cache_id     = 0;
    new_dtp->name[0]      = 0;

    /* builtins are handled differently than user-defined types because they
     * have no associated dataloop or datatype structure.
     */
    if (HANDLE_GET_KIND(oldtype) == HANDLE_KIND_BUILTIN) {
	/* get old values directly from the handle using bit ops */
	el_size = MPID_Datatype_get_basic_size(oldtype);
	el_extent = el_size; /* needed at bottom of function at the very least */

	el_count = blocklength_array[0]; /* also going to count # of oldtypes */
	if (dispinbytes)
	    eff_disp = ((MPI_Aint *) displacement_array)[0];
	else
	    eff_disp = ((MPI_Aint *) displacement_array)[0] * el_size;

	min_lb = eff_disp;
	max_ub = min_lb + (MPI_Aint) blocklength_array[0] * el_size;
	
	for (i=1; i < count; i++) {
	    MPI_Aint tmp_lb, tmp_ub; /* mostly used for clarity; could remove */

	    el_count += blocklength_array[i]; /* add more oldtypes */

	    if (dispinbytes)
		eff_disp = ((MPI_Aint *) displacement_array)[i];
	    else
		eff_disp = ((MPI_Aint *) displacement_array)[i] * el_size;
	    
	    /* calculate ub and lb for this block */
	    tmp_lb = eff_disp;
	    tmp_ub = eff_disp + (MPI_Aint) blocklength_array[i] * el_size;
	    if (tmp_lb < min_lb) min_lb = tmp_lb;
	    if (tmp_ub > max_ub) max_ub = tmp_ub;
	}

	/* fill in remainder of new datatype */
	new_dtp->size           = el_count * el_size;
	new_dtp->has_sticky_ub  = 0;
	new_dtp->has_sticky_lb  = 0;
	new_dtp->loopinfo_depth = 1;

	new_dtp->lb             = min_lb;
	new_dtp->ub             = max_ub;
	new_dtp->true_lb        = min_lb;
	new_dtp->true_ub        = max_ub;
	new_dtp->extent         = max_ub - min_lb;
	new_dtp->alignsize      = el_size;
	new_dtp->n_elements     = el_count;
	new_dtp->is_contig      = 0; /* ??? */

	/* allocate dataloop
	 * 
	 * Note: we allocate space for displacements as MPI_Aints, because
	 * that is how they are stored in the loop (despite being passed in
	 * as ints).
	 *
	 * We'll need to do a conversion in here too.
	 *
	 * TODO: PADDING???
	 */
	new_loopsize = sizeof(struct MPID_Dataloop) + count * (sizeof(MPI_Aint) + sizeof(int));

	dlp = (struct MPID_Dataloop *)MPIU_Malloc(new_loopsize);
	if (dlp == NULL) assert(0);

	new_dtp->loopinfo       = dlp;
	new_dtp->loopsize       = new_loopsize;

	/* fill in dataloop, noting that this is a leaf.  no need to copy. */
	/* NOTE: element size in kind is off. */
	dlp->kind                       = DLOOP_KIND_INDEXED | DLOOP_FINAL_MASK | (el_size << DLOOP_ELMSIZE_SHIFT);
	dlp->handle                     = new_dtp->handle;
	dlp->loop_params.i_t.count      = count;
	dlp->el_extent                  = el_size; /* extent = size for basic types */
	dlp->el_size                    = el_size;
    }
    else /* user-defined base type */ {
	MPI_Aint el_lb, el_ub;
	MPID_Datatype *old_dtp;
	    
	MPID_Datatype_get_ptr(oldtype, old_dtp);
	el_size   = old_dtp->size;
	el_extent = old_dtp->extent;
	el_lb     = old_dtp->lb;
	el_ub     = old_dtp->ub;

	/* get some starting values for lb, ub */
	el_count = blocklength_array[0]; /* also going to count # of oldtypes */
	if (dispinbytes)
	    eff_disp = ((MPI_Aint *) displacement_array)[0];
	else
	    eff_disp = ((MPI_Aint *) displacement_array)[0] * el_extent;
	
	/* MPID_DATATYPE_BLOCK_LB_UB() is defined in mpid_datatype.h */
	MPID_DATATYPE_BLOCK_LB_UB((MPI_Aint) blocklength_array[0],
				  eff_disp,
				  el_lb,
				  el_ub,
				  el_extent,
				  min_lb,
				  max_ub);
	
	/* find smallest lb, largest ub */
	for (i=1; i < count; i++) {
	    MPI_Aint tmp_lb, tmp_ub; /* mostly used for clarity; could remove */

	    el_count += blocklength_array[i]; /* add more oldtypes */

	    if (dispinbytes)
		eff_disp = ((MPI_Aint *) displacement_array)[i];
	    else
		eff_disp = ((MPI_Aint *) displacement_array)[i] * el_extent;
	    
	    /* calculate ub and lb for this block */
	    MPID_DATATYPE_BLOCK_LB_UB((MPI_Aint) blocklength_array[i],
				      eff_disp,
				      el_lb,
				      el_ub,
				      el_extent,
				      tmp_lb,
				      tmp_ub);

	    if (tmp_lb < min_lb) min_lb = tmp_lb;
	    if (tmp_ub > max_ub) max_ub = tmp_ub;
	}

	new_dtp->size           = el_count * el_size; /* in bytes */
	new_dtp->lb             = min_lb;
	new_dtp->ub             = max_ub;
	new_dtp->true_lb        = min_lb + (old_dtp->true_lb - el_lb);
	new_dtp->true_ub        = max_ub + (old_dtp->true_ub - el_ub);
	new_dtp->extent         = max_ub - min_lb;

	new_dtp->has_sticky_ub  = old_dtp->has_sticky_ub;
	new_dtp->has_sticky_lb  = old_dtp->has_sticky_lb;
	new_dtp->loopinfo_depth = old_dtp->loopinfo_depth + 1;
	new_dtp->alignsize      = old_dtp->alignsize;
	new_dtp->n_elements     = el_count * old_dtp->n_elements;

	new_dtp->is_contig = 0; /* TODO: FIX THIS */

	/* allocate space for dataloop */
	new_loopsize = old_dtp->loopsize + sizeof(struct MPID_Dataloop) + 
	    count * (sizeof(MPI_Aint) + sizeof(int));

	dlp = (struct MPID_Dataloop *) MPIU_Malloc(new_loopsize);
	if (dlp == NULL) assert(0);

	new_dtp->loopinfo = dlp;
	new_dtp->loopsize = new_loopsize;

	/* fill in top part of dataloop */
	dlp->kind                       = DLOOP_KIND_INDEXED | (old_dtp->size << DLOOP_ELMSIZE_SHIFT); /* WRONG I THINK */
	dlp->handle                     = new_dtp->handle; /* filled in by MPIU_Handle_obj_alloc */
	dlp->loop_params.i_t.count      = count;
	dlp->el_extent                  = el_extent;
	dlp->el_size                    = el_size;

	/* copy in old dataloop */
	curpos = (char *) dlp; /* NEED TO PAD? */
	curpos += new_loopsize - old_dtp->loopsize;

	MPID_Dataloop_copy(curpos, old_dtp->loopinfo, old_dtp->loopsize);
	dlp->loop_params.i_t.dataloop = (struct MPID_Dataloop *) curpos;
    }

    /* copy in blocklength and displacement parameters (in that order) */
    curpos = (char *) dlp;
    curpos += sizeof(struct MPID_Dataloop);

    dlp->loop_params.i_t.blocksize_array = (int *) curpos;
    iptr = (int *) curpos;
    for (i=0; i < count; i++) {
	iptr[i] = blocklength_array[i];
    }
    curpos += count * sizeof(int);

    dlp->loop_params.i_t.offset_array = (MPI_Aint *) curpos;
    aptr = (MPI_Aint *) curpos;
    for (i=0; i < count; i++) {
	if (dispinbytes) /* hindexed */
	    aptr[i] = (MPI_Aint) ((MPI_Aint *) displacement_array)[i];
	else /* indexed */
	    aptr[i] = ((MPI_Aint) ((int *) displacement_array)[i]) * el_extent;
    }

    /* return handle to new datatype in last parameter */
    *newtype = new_dtp->handle;

#ifdef MPID_TYPE_ALLOC_DEBUG
    MPIU_dbg_printf("(h)indexed type %x created.\n", new_dtp->handle);
#endif
    return MPI_SUCCESS;
}