PRACE NPT

#include "mpiargs.h"

void cntdpls( int scnts[], int sdpls[], int rcnts[], int rdpls[], int dir )

// ---------------------------------------------------------------------

// --- 'cntdpls' computes the element counts and displacements to be

//      used in 'MPI_Alltoallv' based on the data in external array

//      'sizes'.

// ---------------------------------------------------------------------

{

   int dim1, dim2, k;

// ---------------------------------------------------------------------

   if ( dir == 0 ) {

      dim1 = 0;

      dim2 = 1;

   }

   else {

      dim1 = 1;

      dim2 = 0;

   }

   for( k = 0; k < nodes; k++ ) {

      scnts[k] = sizes[me][dim1]*sizes[k][dim2];

   }

   sdpls[0] = 0;

   for( k = 1; k < nodes; k++ ) {

      sdpls[k] = sdpls[k-1] + scnts[k-1];

   }

   for( k = 0; k < nodes; k++ ) {

      rcnts[k] = sizes[me][dim2]*sizes[k][dim1];

   }

   rdpls[0] = 0;

   for( k = 1; k < nodes; k++ ) {

      rdpls[k] = rdpls[k-1] + rcnts[k-1];

   }

}