//
// This file is part of the Bones source-to-source compiler examples. The C-code
// is largely identical in terms of functionality and variable naming to the code
// found in PolyBench/C version 3.2. For more information on PolyBench/C or Bones
// please use the contact information below.
//
// == More information on PolyBench/C
// Contact............Louis-Noel Pouchet <pouchet@cse.ohio-state.edu>
// Web address........http://polybench.sourceforge.net/
// 
// == More information on Bones
// Contact............Cedric Nugteren <c.nugteren@tue.nl>
// Web address........http://parse.ele.tue.nl/bones/
//
// == File information
// Filename...........benchmark/doitgen.c
// Author.............Cedric Nugteren
// Last modified on...10-April-2012
//

#include "common.h"

// This is 'doitgen', a multiresolution analysis kernel
int main(void) {
	int i,j,k,p,q,r,s;
	
	// Declare arrays on the stack
	float A[NR][NQ][NP];
	float sum[NR][NQ][NP];
	float C4[NP][NP];
	
	// Set the input data
	for (i=0; i<NR; i++) { for (j=0; j<NQ; j++) { for (k=0; k<NP; k++) { A[i][j][k] = ((float) i*j + k) / NP; } } }
	for (i=0; i<NP; i++) { for (j=0; j<NP; j++) { C4[i][j] = ((float) i*j) / NP; } }
	
	// Perform the computation
	#pragma species kernel 0:NR-1,0:NQ-1,0:NP-1|chunk(0:0,0:0,0:NP-1) ^ 0:NP-1,0:NP-1|chunk(0:NP-1,0:0) -> 0:NR-1,0:NQ-1,0:NP-1|element
	for (r=0; r<NR; r++) {
		for (q=0; q<NQ; q++) {
			for (p=0; p<NP; p++) {
				sum[r][q][p] = 0;
				for (s=0; s<NP; s++) {
					sum[r][q][p] = sum[r][q][p] + A[r][q][s] * C4[s][p];
				}
			}
		}
	}
	#pragma species endkernel doitgen-part1
	#pragma species kernel 0:NR-1,0:NQ-1,0:NP-1|element -> 0:NR-1,0:NQ-1,0:NP-1|element
	for (r=0; r<NR; r++) {
		for (q=0; q<NQ; q++) {
			for (p=0; p<NP; p++) {
				A[r][q][p] = sum[r][q][p];
			}
		}
	}
	#pragma species endkernel doitgen-part2
	
	// Clean-up and exit the function
	fflush(stdout);
	return 0;
}