StepFilterTest.java

/* 
 * Copyright 2012 Phil Pratt-Szeliga and other contributors
 * http://chirrup.org/
 * 
 * See the file LICENSE for copying permission.
 */

package org.trifort.rootbeer.testcases.rootbeertest.gpurequired;

import java.util.ArrayList;
import java.util.List;

import org.trifort.rootbeer.runtime.Kernel;
import org.trifort.rootbeer.test.TestSerialization;

public class StepFilterTest implements TestSerialization {

  @Override
  public List<Kernel> create() {
    int size = 1600*1200/4;
    short[] a0 = new short[size];
    short[] b0 = new short[size];
    short[] a1 = new short[size];
    short[] b1 = new short[size];
    short[] a2 = new short[size];
    short[] b2 = new short[size];
    short[] a3 = new short[size];
    short[] b3 = new short[size];
    short[][] kx = {{-1,-2,-1},{0,0,0},{1,2,1}};
    short[][] ky = {{-1,0,1}, {-2,0,2},{-1,0,1}};
    for(int i = 0; i < size; ++i){
      a0[i] = (short) i;
      a1[i] = (short) i;
      a2[i] = (short) i;
      a3[i] = (short) i;
    }
    List<Kernel> jobs = new ArrayList<Kernel>();
    int num = 1675;
    for(int i = 0; i < size * 4; i += num){
      short[] a;
      short[] b;
      if(i < size){
        a = a0;
        b = b0;
      } else if (i < size * 2){
        a = a1;
        b = b1;
      } else if(i < size * 3){
        a = a2;
        b = b2;
      } else {
        a = a3;
        b = b3;
      }
      StepFilterRunOnGpu curr = new StepFilterRunOnGpu(a, b, kx, ky, i, num);
      jobs.add(curr);
    }
    return jobs;
  }

  @Override
  public boolean compare(Kernel lhs, Kernel rhs) {
    StepFilterRunOnGpu blhs = (StepFilterRunOnGpu) lhs;
    StepFilterRunOnGpu brhs = (StepFilterRunOnGpu) rhs;
    return blhs.compare(brhs);
  }
}