scano.java

来自「MPI for java for Distributed Programming」· Java 代码 · 共 83 行

package mpi.ccl_ObjSer.scanO; 

import mpi.ccl_ObjSer.test;
import mpi.*;
import mpi.*;
import java.io.*;

class complexNum implements Serializable{
  complexNum() {
  }
  
  float realPart;
  float imaginPart;
}

class complexAdd extends User_function{ 
  complexAdd() {
  }
  
  public void Call(Object invec, int inoffset, Object outvec, int outoffset,
                   int count, Datatype datatype){
    Object [] in_array = (Object[])invec;
    Object [] out_array = (Object[])outvec;

    for (int i = 0; i < count; i++){
      complexNum ocd = (complexNum)out_array[outoffset + i];
      complexNum icd = (complexNum)in_array[inoffset + i];

      ocd.realPart += icd.realPart;
      ocd.imaginPart += icd.imaginPart;
    }
  }
}

public class scanO {
  static public void main(String[] args) throws MPIException {
  }

  public scanO() {
  }

  public scanO(String[] args) throws Exception {
    
    final int MAXLEN = 10000;

    int i,j,k;
    complexNum out[] = new complexNum[MAXLEN];
    complexNum in[]  = new complexNum[MAXLEN];
    int myself,tasks;
    boolean bool = false;

    MPI.Init(args);
    myself = MPI.COMM_WORLD.Rank();
    tasks = MPI.COMM_WORLD.Size();

    for(i=0;i<MAXLEN;i++){  
      in[i] = new complexNum();
      out[i] = new complexNum();
      out[i].realPart = i;
      out[i].imaginPart = i;
    }

    complexAdd cadd = new complexAdd();
    Op op = new Op(cadd, bool);
    MPI.COMM_WORLD.Scan(out,0,in,0,MAXLEN,MPI.OBJECT,op);

    for(k=0;k<MAXLEN;k++) {
      if(in[k].realPart != k*(myself+1)) {  
        System.out.println("bad answer ("+(in[k].realPart)+") at index "+
      		     k+ 
      		     "(should be "+(k*(myself+1))+")");
        break; 
      }
    }


    MPI.COMM_WORLD.Barrier();
    if(myself == 0)  System.out.println("ScanO TEST COMPLETE");
    MPI.Finalize();

  }
}