package my.my1;

import org.openjdk.jmh.annotations.Warmup;
import org.openjdk.jmh.infra.Blackhole;
import org.openjdk.jmh.annotations.Benchmark;
import org.openjdk.jmh.annotations.BenchmarkMode;
import org.openjdk.jmh.annotations.Fork;
import org.openjdk.jmh.annotations.Measurement;
import org.openjdk.jmh.annotations.Mode;
import org.openjdk.jmh.annotations.OutputTimeUnit;
import org.openjdk.jmh.annotations.Param;
import org.openjdk.jmh.annotations.Scope;
import org.openjdk.jmh.annotations.Setup;
import org.openjdk.jmh.annotations.State;
import org.openjdk.jmh.runner.Runner;
import org.openjdk.jmh.runner.RunnerException;
import org.openjdk.jmh.runner.options.Options;
import org.openjdk.jmh.runner.options.OptionsBuilder;
import java.util.concurrent.TimeUnit;

@Warmup(iterations = 5, time = 1, timeUnit = TimeUnit.SECONDS)
@Measurement(iterations = 5, time = 1, timeUnit = TimeUnit.SECONDS)
@Fork(value = 3, jvmArgsAppend = {"-XX:+UseParallelGC", "-Xms4g", "-Xmx4g"})
@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.NANOSECONDS)
@State(Scope.Benchmark)
public class MyBenchmark {
	
    @Param({"10000"})
    int size;
	
//	Param({"0", "1000", "10000"})
	@Param({"0"})
    int trashSize;

    Foo[] objArr;
	Foo[][] objArrTrash;
	int[] intArr;

    @Setup
    public void setup() {
		
		objArr = new Foo[size];
		
		objArrTrash = new Foo[trashSize][size];
		
		intArr = new int[size * 3];
		for(int i = 0; i < size; i++){
			intArr[i * 3] = 1;
			intArr[i * 3 + 1] = 2;
			intArr[i * 3 + 2] = 3;
			
			objArr[i] = new Foo();
			
			for(int j = 0; j < trashSize; j++){
				objArrTrash[j][i] = new Foo();
			}
		}
    }

    @Benchmark
    public void objArray(Blackhole blackhole) {
		long total = 0;
		for(int i = 0; i < size; i++){
			Foo obj = objArr[i];
			total += obj.getA() + obj.getB() + obj.getC();
		}
		blackhole.consume(total);
    }
	
    @Benchmark
    public void intArray(Blackhole blackhole) {
		long total = 0;
		for(int i = 0; i < size; i++){
			total += intArr[i*3] + intArr[i*3 + 1] + intArr[i*3 + 2];
		}
		blackhole.consume(total);
    }

    @Benchmark
    public void intArray2(Blackhole blackhole) {
		long total = 0;
		// IMHO Правильно так
		for(int i = 0; i < intArr.length; i++){
			total += intArr[i];
		}
		blackhole.consume(total);
    }
	
    @Benchmark
    public void intArray3(Blackhole blackhole) {
		long total = 0;
		// IMHO Или, в крайнем случае, так
		for(int i = 0; i < intArr.length-2; i+=3 ){
			total += intArr[i+0] + intArr[i+1] + intArr[i+2];
		}
		blackhole.consume(total);
    }
	
    public static class Foo {
		
		private int a;
		private int b;
		private int c;
		
		public int getA(){
			return this.a;
		}
		public int getB(){
			return this.b;
		}
		public int getC(){
			return this.c;
		}

		public Foo() {
			this.a = 1;
			this.b = 2;
			this.c = 3;
		}

		@Override
		public String toString() {
			return "Foo [a=" + a + ", b=" + b + ", c=" + c + "]";
		}

		@Override
		public int hashCode() {
			final int prime = 31;
			int result = 1;
			result = prime * result + a;
			result = prime * result + b;
			result = prime * result + c;
			return result;
		}

		@Override
		public boolean equals(Object obj) {
			if (this == obj)
				return true;
			if (obj == null)
				return false;
			if (getClass() != obj.getClass())
				return false;
			Foo other = (Foo) obj;
			if (a != other.a)
				return false;
			if (b != other.b)
				return false;
			if (c != other.c)
				return false;
			return true;
		}

	}



	public static void main(String[] args) throws RunnerException {

		Options options = new OptionsBuilder()
				.include( MyBenchmark.class.getSimpleName() ).build();
		new Runner(options).run();
	}
}