// -*- C++ -*-
#define N 10

generic main()
{
  stream long[N] A1;
  stream long[N] A2;
  stream long[N] B1;
  stream long[N] B2;
  
  genera1  (output_stream A1);
  genera2  (output_stream A2);  
  cross    (input_stream A1, A2  output_stream B1, B2);
  fine1    (input_stream B1);
  fine2    (input_stream B2);
}


genera1(output_stream long A1[N]) {
  fgen1(output_stream A1);
}


proc fgen1(output_stream long A1[N])
inc<"iostream">
$c++{
  long a[N];
  
  // start regression test
  std::cerr << "STARTING fgen1" << std::endl;

  for (int i=0;i<N;i++) 
    a[i] = i;
  
  assist_out(A1, a);
}c++$


genera2(output_stream long A2[N]) {
  fgen2(output_stream A2);
}


proc fgen2(output_stream long A2[N])
inc<"iostream">
$c++{
  long a[N];

  // start regression test
  std::cerr << "STARTING fgen2" << std::endl;

  for (int i=0;i<N;i++)
    a[i] = 10 + i;
  
  assist_out(A2,a);
}c++$


parmod cross(input_stream long A1[N], long A2[N]  output_stream long B1[N], long B2[N]) {

  topology array [i:N] Pv;

  attribute long S[N] scatter S[*i0] onto Pv[i0];

  stream long s1;
  stream long s2;

  do input_section {
    guard1: on , , A1 {
      distribution A1[*j] scatter to S[j];
    }
    guard2: on , , A2 {
      distribution A2[*k] scatter to S[k];
    }  
  } while (true)
  
  virtual_processors {
    elab1(in guard1 out s1) {
      VP i {
	f_p1(in S[i] out s1);
      }
    }
    elab2(in guard2 out s2) {
      VP i {
	f_p2(in S[i] out s2);
      }
    }
  }
  
  output_section {
    collects s1 from ALL Pv[i] {  
      int el1;
      int B1_[N];
      AST_FOR_EACH(el1) {
	B1_[i]=el1;
      }
      assist_out (B1, B1_);
    }<>;

    collects s2 from ALL Pv[i] {
      int el2;
      int B2_[N];
      AST_FOR_EACH(el2) {
	B2_[i]=el2;
      }
      assist_out (B2, B2_);
    }<>;
  } 
}


fine1(input_stream long B1[N])
 
inc<"iostream">
$c++{
  int ok = 0;
  
  /*
    std::cerr << "Fine1: " << std::endl;
    for (int i=0; i<N; i++)
    std::cerr << B1[i] << " ";
    std::cerr << std::endl;
  */
  
  for (int i=0;i<N;i++) if (B1[i] != i) ok = -1;
  if (ok==0) 
    std::cerr << "ENDING with SUCCESS fine1 array_scatter.ast" << std::endl;
  else  
    std::cerr << "ENDING with FAILURE fine1 array_scatter.ast" << std::endl;
}c++$


fine2(input_stream long B2[N])
 
inc<"iostream">
$c++{
  int ok = 0;

  /*
    std::cerr << "Fine2: " << std::endl;
    for (int i=0; i<N; i++)
    std::cerr << B2[i] << " ";
    std::cerr << std::endl;
  */

  for (int i=0;i<N;i++) if (B2[i] != (10 + i)) ok = -1;
  if (ok==0) 
    std::cerr << "ENDING with SUCCESS fine2 array_scatter.ast" << std::endl;
  else  
    std::cerr << "ENDING with FAILURE fine2 array_scatter.ast" << std::endl;
}c++$


proc f_p1(in long A1 out long B1)
 
inc<"iostream">
$c++{
  long b;
  b = A1;
  B1 = b;
}c++$


proc f_p2(in long A2 out long B2)
 
inc<"iostream">
$c++{
  B2 = A2;
}c++$