fixing 1D case

Author: leios

src/operators.cu

@@ -198,25 +198,40 @@ void generate_p_space(Grid &par){
     int yDim = par.ival("yDim");
     int zDim = par.ival("zDim");
     double xMax = par.dval("xMax");
-    double yMax = par.dval("yMax");
+    double yMax = 0;
+    if (dimnum > 1){
+        yMax = par.dval("yMax");
+    }
     double zMax = 0;
     if (dimnum == 3){
         zMax = par.dval("zMax");
     }
+
     double pxMax = par.dval("pxMax");
-    double pyMax = par.dval("pyMax");
+    double pyMax = 0;
+    if (dimnum > 1){
+        pyMax = par.dval("pyMax");
+    }
     double pzMax = 0;
     if (dimnum == 3){
         pzMax = par.dval("pzMax");
     }
+
     double dx = par.dval("dx");
-    double dy = par.dval("dy");
+    double dy = 0;
+    if (dimnum > 1){
+        dy = par.dval("dy");
+    }
     double dz = 0;
     if (dimnum == 3){
         dz = par.dval("dz");
     }
+
     double dpx = par.dval("dpx");
-    double dpy = par.dval("dpy");
+    double dpy = 0;
+    if (dimnum > 1){
+        dpy = par.dval("dpy");
+    }
     double dpz = 0;
     if (dimnum == 3){
         dpz = par.dval("dpz");
@@ -284,6 +299,24 @@ void generate_p_space(Grid &par){
 
         }
 
+    }
+    else if (dimnum == 1){
+        for(int i=0; i<xDim/2; ++i){
+            x[i] = -xMax + i*dx;
+            x[i + (xDim/2)] = i*dx;
+
+            px[i] = i*dpx;
+            px[i + (xDim/2)] = -pxMax + i*dpx;
+
+        }
+
+        for(int i = 0; i < zDim; ++i){
+            z[i] = 0;
+            pz[i] = 0;
+            y[i] = 0;
+            py[i] = 0;
+        }
+
     }
     par.store("x",x);
     par.store("y",y);

src/unit_test.cu

@@ -1168,8 +1168,8 @@ void evolve_test(){
     par.store("esteps", esteps);
     par.store("gsteps", gsteps);
     par.store("printSteps", 30000);
-    par.store("write_file", false);
-    par.store("write_it", false);
+    par.store("write_file", true);
+    par.store("write_it", true);
     par.store("energy_calc", true);
     par.store("box_size", 0.00007);
     par.store("xDim", 64);
@@ -1178,7 +1178,7 @@ void evolve_test(){
 
 
     // Running through all the dimensions to check the energy
-    for (int i = 2; i <= 3; ++i){
+    for (int i = 1; i <= 3; ++i){
         if (i == 2){
             par.store("yDim", 64);
         }