X-Git-Url: http://git.treefish.org/~alex/phys/latlib.git/blobdiff_plain/385a5a6cbcab9001633efd700b2ffdeaf1aaaf6d..d3e332b13eccd2fa7e467770f6d973b4f34904cf:/neigh.cpp

diff --git a/neigh.cpp b/neigh.cpp
index a25d1e7..8e4084c 100644
--- a/neigh.cpp
+++ b/neigh.cpp
@@ -1,28 +1,44 @@
 #include "neigh.h"
 
-neigh::neigh(const int& dimension, const int& length)
+#include <iostream>
+
+using namespace std;
+
+int neigh::dirstep(int dir)
 {
-  nfield = new int[ipow(dimension,length)*2*dimension];
+  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 < ipow(length,dimension); ipos++)
-    {
-      for (int idir = 0; idir < 2*dimension; idir++)
-	{
-	  if ( idir < dimension ) //pos direction
-	    {
-	      if ( ( ipos/ipow(length,idir%dimension) ) % length == length-1 )
-		nfield[ipos*2*dimension + idir] = ipos - ipow(length,idir%dimension)*(length-1);
-	      else
-		nfield[ipos*2*dimension + idir] = ipos + ipow(length,idir%dimension);
-	    }
-	  else //negative dir
-	    {
-	      if ( ( ipos/ipow(length,idir%dimension) ) % length == 0 )
-		nfield[ipos*2*dimension + idir] = ipos + ipow(length,idir%dimension)*(length-1);
-	      else
-		nfield[ipos*2*dimension + idir] = ipos - ipow(length,idir%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);
+      }
+    }  
 }