#include "neigh.h"

#include <iostream>

using namespace std;

int neigh::dirstep(int dir)
{
  int step=1;
  for(int idir=0; idir<dir; idir++) step*=len[idir];
  return step;
}

neigh::neigh(const int& dimension, ...)
{
  int totsize=1;
  len = new int[dimension];
  va_list length;

  va_start(length, dimension);
  for(int idim=0; idim < dimension; idim++) {
    len[idim] = va_arg(length, int);
    totsize *= len[idim];
  }
  va_end(length);

  nfield = new int[totsize*2*dimension];

  for( int ipos=0; ipos<totsize; ipos++ )
    for( int idir=0; idir<2*dimension; idir++) {
      if( idir < dimension ) { // positive direction
	if ( ( ipos/dirstep(idir%dimension) ) % len[idir%dimension] == len[idir%dimension]-1 )
	  nfield[ipos*2*dimension + idir] = ipos - dirstep(idir%dimension)*(len[idir%dimension]-1);
	else
	  nfield[ipos*2*dimension + idir] = ipos + dirstep(idir%dimension);
      }
      else { // negative direction
	if ( ( ipos/dirstep(idir%dimension) ) % len[idir%dimension] == 0 )
	  nfield[ipos*2*dimension + idir] = ipos + dirstep(idir%dimension)*(len[idir%dimension]-1);
	else
	  nfield[ipos*2*dimension + idir] = ipos - dirstep(idir%dimension);
      }
    }  
}