92 строки
3.0 KiB
C
92 строки
3.0 KiB
C
#include "ompi_config.h"
|
|
|
|
#ifdef HAVE_TARGETCONDITIONALS_H
|
|
#include <TargetConditionals.h>
|
|
#endif
|
|
|
|
#include "mpi.h"
|
|
#include "mpi_Message.h"
|
|
#include "mpiJava.h"
|
|
|
|
JNIEXPORT void JNICALL Java_mpi_Message_init(JNIEnv *e, jclass c)
|
|
{
|
|
ompi_java_setStaticLongField(e, c, "NULL", (jlong)MPI_MESSAGE_NULL);
|
|
ompi_java_setStaticLongField(e, c, "NO_PROC", (jlong)MPI_MESSAGE_NO_PROC);
|
|
ompi_java.MessageHandle = (*e)->GetFieldID(e, c, "handle", "J");
|
|
}
|
|
|
|
JNIEXPORT jobject JNICALL Java_mpi_Message_mProbe(
|
|
JNIEnv *env, jobject jthis, jint source, jint tag, jlong comm)
|
|
{
|
|
MPI_Message message = (MPI_Message)((*env)->GetLongField(
|
|
env, jthis, ompi_java.MessageHandle));
|
|
int rc;
|
|
MPI_Status status;
|
|
rc = MPI_Mprobe(source, tag, (MPI_Comm)comm, &message, &status);
|
|
|
|
if(ompi_java_exceptionCheck(env, rc))
|
|
return NULL;
|
|
|
|
(*env)->SetLongField(env, jthis, ompi_java.MessageHandle, (jlong)message);
|
|
jobject stat = ompi_java_status_new(&status, env);
|
|
return stat;
|
|
}
|
|
|
|
JNIEXPORT jobject JNICALL Java_mpi_Message_imProbe(
|
|
JNIEnv *env, jobject jthis, jint source, jint tag, jlong comm)
|
|
{
|
|
MPI_Message message = (MPI_Message)((*env)->GetLongField(
|
|
env, jthis, ompi_java.MessageHandle));
|
|
int rc, flag;
|
|
MPI_Status status;
|
|
rc = MPI_Improbe(source, tag, (MPI_Comm)comm, &flag, &message, &status);
|
|
|
|
if(ompi_java_exceptionCheck(env, rc) || !flag)
|
|
return NULL;
|
|
|
|
(*env)->SetLongField(env, jthis, ompi_java.MessageHandle, (jlong)message);
|
|
jobject stat = ompi_java_status_new(&status, env);
|
|
return stat;
|
|
}
|
|
|
|
JNIEXPORT void JNICALL Java_mpi_Message_mRecv(
|
|
JNIEnv *env, jobject jthis, jobject buf, jint offset, jint count,
|
|
jobject jType, jobject stat)
|
|
{
|
|
MPI_Message msg = (MPI_Message)((*env)->GetLongField(
|
|
env, jthis, ompi_java.MessageHandle));
|
|
|
|
MPI_Datatype type = (MPI_Datatype)((*env)->GetLongField(
|
|
env, jType, ompi_java.DatatypeHandle));
|
|
|
|
int bType = (*env)->GetIntField(env, jType, ompi_java.DatatypeBaseType);
|
|
void *bufPtr, *bufBase;
|
|
bufPtr = ompi_java_getBufPtr(&bufBase, env, buf, bType, offset);
|
|
|
|
MPI_Status status;
|
|
int rc = MPI_Mrecv(bufPtr, count, type, &msg, &status);
|
|
|
|
if(!ompi_java_exceptionCheck(env, rc))
|
|
{
|
|
(*env)->SetLongField(env, jthis, ompi_java.MessageHandle, (jlong)msg);
|
|
ompi_java_status_set(&status, env, stat);
|
|
}
|
|
|
|
ompi_java_releaseBufPtr(env, buf, bufBase, bType);
|
|
}
|
|
|
|
JNIEXPORT jlong JNICALL Java_mpi_Message_imRecv(
|
|
JNIEnv *env, jobject jthis, jobject buf, jint count, jlong type)
|
|
{
|
|
MPI_Message msg = (MPI_Message)((*env)->GetLongField(
|
|
env, jthis, ompi_java.MessageHandle));
|
|
|
|
void *ptr = ompi_java_getDirectBufferAddress(env, buf);
|
|
MPI_Request request;
|
|
|
|
int rc = MPI_Imrecv(ptr, count, (MPI_Datatype)type, &msg, &request);
|
|
ompi_java_exceptionCheck(env, rc);
|
|
(*env)->SetLongField(env, jthis, ompi_java.MessageHandle, (jlong)msg);
|
|
return (jlong)request;
|
|
}
|