Remove terms with kx=0 and ky<=0

1 files changed, 7 insertions, 9 deletions

diff --git a/src/init.c b/src/init.c
index cce97d2..cc482da 100644
--- a/src/init.c
+++ b/src/init.c
@@ -22,12 +22,10 @@ int init_random (
 
   // random init (set half, then the other half are the conjugates)
   for(kx=0;kx<=K1;kx++){
-    for(ky=-K2;ky<=K2;ky++){
-      if (kx!=0 || ky>0){
-	x=-0.5+((double) rand())/RAND_MAX;
-	y=-0.5+((double) rand())/RAND_MAX;
-	u0[klookup_sym(kx,ky,K2)]=x+y*I;
-      }
+    for(ky=(kx>0 ? -K2 : 1);ky<=K2;ky++){
+      x=-0.5+((double) rand())/RAND_MAX;
+      y=-0.5+((double) rand())/RAND_MAX;
+      u0[klookup_sym(kx,ky,K2)]=x+y*I;
     }
   }
 
@@ -39,7 +37,7 @@ int init_random (
     rescale=compute_enstrophy(u0, K1, K2, L);
   }
   for(kx=0;kx<=K1;kx++){
-    for(ky=-K2;ky<=K2;ky++){
+    for(ky=(kx>0 ? -K2 : 1);ky<=K2;ky++){
       u0[klookup_sym(kx,ky,K2)]=u0[klookup_sym(kx,ky,K2)]*sqrt(init_en/rescale);
     }
   }
@@ -60,7 +58,7 @@ int init_gaussian (
   double rescale;
 
   for(kx=0;kx<=K1;kx++){
-    for(ky=-K2;ky<=K2;ky++){
+    for(ky=(kx>0 ? -K2 : 1);ky<=K2;ky++){
       u0[klookup_sym(kx,ky,K2)]=(kx*kx+ky*ky)*exp(-(kx*kx+ky*ky));
     }
   }
@@ -73,7 +71,7 @@ int init_gaussian (
     rescale=compute_enstrophy(u0, K1, K2, L);
   }
   for(kx=0;kx<=K1;kx++){
-    for(ky=-K2;ky<=K2;ky++){
+    for(ky=(kx>0 ? -K2 : 1);ky<=K2;ky++){
       u0[klookup_sym(kx,ky,K2)]=u0[klookup_sym(kx,ky,K2)]*sqrt(init_en/rescale);
     }
   }