Examples of cern.colt.matrix.DoubleMatrix2D.zAssign8Neighbors()

cern.colt.matrix.DoubleMatrix2D.zAssign8Neighbors()

ighbors cern.colt.function.Double9Function f = new cern.colt.function.Double9Function() { public final double apply( double a00, double a01, double a02, double a10, double a11, double a12, double a20, double a21, double a22) { return beta*a11 + alpha*(a00+a01+a02 + a10+a12 + a20+a21+a22); } }; A.zAssign8Neighbors(B,f); // 4 neighbors cern.colt.function.Double9Function g = new cern.colt.function.Double9Function() { public final double apply( double a00, double a01, double a02, double a10, double a11, double a12, double a20, double a21, double a22) { return beta*a11 + alpha*(a01+a10+a12+a21); } C.zAssign8Neighbors(B,g); // fast, even though it doesn't look like it }; @param B the matrix to hold the results. @param function the function to be applied to the 9 cells. @throws NullPointerException if function==null. @throws IllegalArgumentException if rows() != B.rows() || columns() != B.columns().

};
cern.colt.Timer timer = new cern.colt.Timer().start();


System.out.println("benchmarking stencil...");
for (int i=0; i<runs; i++) {
  A.zAssign8Neighbors(A,function);
}
//A.zSum4Neighbors(A,alpha,beta,runs);
timer.stop().display();
//System.out.println("A="+A);
A=null;

  if (convergenceIterations%2 != 0) convergenceIterations++; // odd -> make it even


  int i=0;
  while (i<maxIterations) { // do two steps at a time for efficiency
    A.zAssign8Neighbors(B,function);
    B.zAssign8Neighbors(A,function);
    i=i+2;
    if (i%convergenceIterations == 0 && hasConverged!=null) {
      if (hasConverged.apply(A)) return i;
    }
  }

  if (convergenceIterations%2 != 0) convergenceIterations++; // odd -> make it even


  int i=0;
  while (i<maxIterations) { // do two steps at a time for efficiency
    A.zAssign8Neighbors(B,function);
    B.zAssign8Neighbors(A,function);
    i=i+2;
    if (i%convergenceIterations == 0 && hasConverged!=null) {
      if (hasConverged.apply(A)) return i;
    }
  }

};
cern.colt.Timer timer = new cern.colt.Timer().start();


System.out.println("benchmarking stencil...");
for (int i=0; i<runs; i++) {
  A.zAssign8Neighbors(A,function);
}
//A.zSum4Neighbors(A,alpha,beta,runs);
timer.stop().display();
//System.out.println("A="+A);
A=null;

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.