static char rcsid[] = "$Id: prlfor005.c,v 1.6 2002/04/03 11:51:48 a-hasega Exp $";
/* 
 * $RWC_Release: Omni-1.6 $
 * $RWC_Copyright:
 *  Omni Compiler Software Version 1.5-1.6
 *  Copyright (C) 2002 PC Cluster Consortium
 *  
 *  This software is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License version
 *  2.1 published by the Free Software Foundation.
 *  
 *  Omni Compiler Software Version 1.0-1.4
 *  Copyright (C) 1999, 2000, 2001.
 *   Tsukuba Research Center, Real World Computing Partnership, Japan.
 *  
 *  Please check the Copyright and License information in the files named
 *  COPYRIGHT and LICENSE under the top  directory of the Omni Compiler
 *  Software release kit.
 *  
 *  
 *  $
 */
/* parallel for 005:
 * guided スケジューリングを指定した場合の動作を確認
 */

#include "omniConfig.h"
#include <omp.h>
#include "omni.h"


#define	MAX_CHUNK	10
#define LOOPNUM		(MAX_CHUNK*thds)

int ret_same (int);

int	thds;
int	*buf;


void
clear ()
{
  int lp;
  
  for (lp=0; lp<=LOOPNUM; lp++) {
    buf[lp] = -1;
  }
}


int
check_result (int s)
{
  int	lp, lp2, id;

  int	err = 0;


  for (lp=0; lp<LOOPNUM; ) {
    id = buf[lp];
    if (id<0 && thds<=id) {
      err += 1;
    }
    for (lp2=0; lp2<(LOOPNUM-lp); lp2++) {
      if (buf[lp+lp2] != id) {
	if (lp2+1<s) {
	  err += 1;
	}
      }
    }
    lp += lp2;
  }
  if (buf[LOOPNUM] != -1) {
    err += 1;
  }

  return err;
}


main ()
{
  int	lp, ln;

  int	chunk = 3;
  int	errors = 0;


  thds = omp_get_max_threads ();
  if (thds == 1) {
    printf ("should be run this program on multi threads.\n");
    exit (0);
  }
  buf = (int *) malloc (sizeof (int) * (LOOPNUM + 1));
  if (buf == NULL) {
    printf ("can not allocate memory.\n");
    exit (1);
  }
  omp_set_dynamic (0);
  ln = LOOPNUM;

  clear ();
  #pragma omp parallel for schedule (guided)
  for (lp=0; lp<LOOPNUM; lp++) {
    buf[lp] = omp_get_thread_num ();
  }
  errors += check_result (1);

  clear ();
  #pragma omp parallel for schedule (guided,1)
  for (lp=0; lp<LOOPNUM; lp++) {
    buf[lp] = omp_get_thread_num ();
  }
  errors += check_result (1);

  clear ();
  #pragma omp parallel for schedule (guided,2)
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (2);

  clear ();
  #pragma omp parallel for schedule (guided,chunk)
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (chunk);

  clear ();
  #pragma omp parallel for schedule (guided,2+1)
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (2+1);

  clear ();
  #pragma omp parallel for schedule (guided,ret_same(3))
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (ret_same(3));

  clear ();
  #pragma omp parallel for schedule (guided,ret_same(3)+1)
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (ret_same(3)+1);

  clear ();
  #pragma omp parallel for schedule (guided,ln+1)
  for (lp=0; lp<LOOPNUM; lp++) {
    if (lp != 0) {
      while (buf[lp-1] == -1) {
	#pragma omp flush
      }
    }
    buf[lp] = omp_get_thread_num ();
    #pragma omp flush
  }
  errors += check_result (LOOPNUM+1);

  if (errors == 0) {
    printf ("parallel for 005 : SUCCESS\n");
    return 0;
  } else {
    printf ("parallel for 005 : FAILED\n");
    return 1;
  }
}


int
ret_same (int i)
{
  return i;
};