sandbox/Antoonvh/slicer.h

    Obtain a slice of 3D data

    Three functions to obtain a square XZ,YZ,ZY slice, respectively, of scalar s. The data is outputted in a file with fname in a regular grid fashion (maxlevel\ ^2).

    It should work for all grids and parallization styles.

    void sliceXZ(char * fname,scalar s,double yp, int maxlevel){
      FILE *fpver =fopen (fname,"w"); 
      int nn = (1<<maxlevel);
      double ** field = matrix_new (nn, nn, sizeof(double));
      double stp = L0/(double)nn;
      for (int i = 0; i < nn; i++)
        {
          double xp = stp*i + X0 + stp/2.;
          for (int j = 0; j < nn; j++) 
    	{
    	  double zp = stp*j + Z0 + stp/2.;
    	  Point point = locate (xp, yp,zp);
    	  field[i][j] = point.level >= 0 ? s[] : nodata;
    	}
        }
      if (pid() == 0){ // master
    #if _MPI
        MPI_Reduce (MPI_IN_PLACE, field[0], sq(nn), MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
        for (int i = 0; i < nn; i++) {
          for (int j = 0; j < nn; j++) {
    	fprintf (fpver, "%g\t", field[i][j]);
          }
          fputc ('\n', fpver);
        }
        fflush (fpver);
      }
    #if _MPI
      else // slave
        MPI_Reduce (field[0], NULL, nn*nn, MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
      matrix_free (field);
    }
    
    void sliceYZ(char * fname,scalar s,double xp, int maxlevel){
      FILE *fpver =fopen (fname,"w"); 
      int nn = (1<<maxlevel);
      double ** field = matrix_new (nn, nn, sizeof(double));
      double stp = L0/(double)nn;
      for (int i = 0; i < nn; i++)
        {
          double yp = stp*i + Y0 + stp/2.;
          for (int j = 0; j < nn; j++) 
    	{
    	  double zp = stp*j + Z0 + stp/2.;
    	  Point point = locate (xp, yp,zp);
    	  field[i][j] = point.level >= 0 ? s[] : nodata;
    	}
        }
      if (pid() == 0){ // master
    #if _MPI
        MPI_Reduce (MPI_IN_PLACE, field[0], sq(nn), MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
        for (int i = 0; i < nn; i++) {
          for (int j = 0; j < nn; j++) {
    	fprintf (fpver, "%g\t", field[i][j]);
          }
          fputc ('\n', fpver);
        }
        fflush (fpver);
      }
    #if _MPI
      else // slave
        MPI_Reduce (field[0], NULL, nn*nn, MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
      matrix_free (field);
    }
    
    void sliceXY(char * fname,scalar s,double zp, int maxlevel){
      FILE *fpver =fopen (fname,"w"); 
      int nn = (1<<maxlevel);
      double ** field = matrix_new (nn, nn, sizeof(double));
      double stp = L0/(double)nn;
      for (int i = 0; i < nn; i++)
        {
          double xp = stp*i + X0 + stp/2.;
          for (int j = 0; j < nn; j++) 
    	{
    	  double yp = stp*j + Y0 + stp/2.;
    	  Point point = locate (xp, yp,zp);
    	  field[i][j] = point.level >= 0 ? s[] : nodata;
    	}
        }
      if (pid() == 0){ // master
    #if _MPI
        MPI_Reduce (MPI_IN_PLACE, field[0], sq(nn), MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
        for (int i = 0; i < nn; i++) {
          for (int j = 0; j < nn; j++) {
    	fprintf (fpver, "%g\t", field[i][j]);
          }
          fputc ('\n', fpver);
        }
        fflush (fpver);
      }
    #if _MPI
      else // slave
        MPI_Reduce (field[0], NULL, nn*nn, MPI_DOUBLE, MPI_MIN, 0,
    		MPI_COMM_WORLD);
    #endif
      matrix_free (field);
    }