/*
 * Copyright (C) 2016 Linaro Limited. All rights received.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package org.linaro.benchmarks.vector;

import org.openjdk.jmh.annotations.*;
import java.util.concurrent.TimeUnit;

@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MICROSECONDS)
@State(Scope.Benchmark)

public class TestSIMDMul {
  static final int LENGTH = 1024;
  static final int COUNT = 128;
  static int [] a = new int[LENGTH];
  static int [] b = new int[LENGTH];
  static int [] c = new int[LENGTH];
  static short [] sa = new short[LENGTH];
  static short [] sb = new short[LENGTH];
  static short [] sc = new short[LENGTH];
  static byte [] ba = new byte[LENGTH];
  static byte [] bb = new byte[LENGTH];
  static byte [] bc = new byte[LENGTH];
  static long [] la = new long[LENGTH];
  static long [] lb = new long[LENGTH];
  static long [] lc = new long[LENGTH];
  static float [] fa = new float[LENGTH];
  static float [] fb = new float[LENGTH];
  static float [] fc = new float[LENGTH];
  static double [] da = new double[LENGTH];
  static double [] db = new double[LENGTH];
  static double [] dc = new double[LENGTH];

  public static void TestSIMDMulInit() {
    for (int i = 0; i < LENGTH; i++) {
       a[i] = i + 3;
       b[i] = i + 2;
       c[i] = i + 1;
       sa[i] = (short)(i + 3);
       sb[i] = (short)(i + 2);
       sc[i] = (short)(i + 1);
       ba[i] = (byte)(i + 3);
       bb[i] = (byte)(i + 2);
       bc[i] = (byte)(i + 1);
       la[i] = (long)(i + 3);
       lb[i] = (long)(i + 2);
       lc[i] = (long)(i + 1);
       fa[i] = (float)(i + 3);
       fb[i] = (float)(i + 2);
       fc[i] = (float)(i + 1);
       da[i] = (double)(i + 3);
       db[i] = (double)(i + 2);
       dc[i] = (double)(i + 1);
    }
  }

  public static void vectMulInt() {
    for (int i = 0; i < LENGTH; i++) {
      c[i] = a[i] * b[i];
    }
  }

  public static void vectMulShort() {
    for (int i = 0; i < LENGTH; i++) {
      sc[i] = (short)(sa[i] * sb[i]);
    }
  }

  public static void vectMulByte() {
    for (int i = 0; i < LENGTH; i++) {
      bc[i] = (byte)(ba[i] * bb[i]);
    }
  }

  public static void vectMulLong() {
    for (int i = 0; i < LENGTH; i++) {
      lc[i] = la[i] * lb[i];
    }
  }

  public static void vectMulFloat() {
    for (int i = 0; i < LENGTH; i++) {
      fc[i] = fa[i] * fb[i];
    }
  }

  public static void vectMulDouble() {
    for (int i = 0; i < LENGTH; i++) {
      dc[i] = da[i] * db[i];
    }
  }

  @Setup
  public void setup()
  {
    TestSIMDMulInit();
  }

  @Benchmark
  public void testVectMulByte() {
    for (int i = 0; i < COUNT; i++) {
      vectMulByte();
    }
  }

  @Benchmark
  public void testVectMulShort() {
    for (int i = 0; i < COUNT; i++) {
      vectMulShort();
    }
  }

  @Benchmark
  public void tesVectMulInt() {
    for (int i = 0; i < COUNT; i++) {
      vectMulInt();
    }
  }

  @Benchmark
  public void testVectMulLong() {
    for (int i = 0; i < COUNT; i++) {
      vectMulLong();
    }
  }

  @Benchmark
  public void testVectMulFloat() {
    for (int i = 0; i < COUNT; i++) {
      vectMulFloat();
    }
  }

  @Benchmark
  public void testVectMulDouble() {
    for (int i = 0; i < COUNT; i++) {
      vectMulDouble();
    }
  }

}