src/vtk.h

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    
    void output_vtk (scalar * list, int n, FILE * fp, bool linear)
    {
      fputs ("# vtk DataFile Version 2.0\n"
    	 "Basilisk\n"
    	 "ASCII\n"
    	 "DATASET STRUCTURED_GRID\n", fp);
      fprintf (fp, "DIMENSIONS %d %d 1\n", n, n);
      fprintf (fp, "POINTS %d double\n", n*n);
    
      double fn = n;
      double Delta = L0/fn;
      for (int i = 0; i < n; i++) {
        double x = Delta*i + X0 + Delta/2.;
        for (int j = 0; j < n; j++) {
          double y = Delta*j + Y0 + Delta/2.;
          fprintf (fp, "%g %g 0\n", x, y);
        }
      }
      fprintf (fp, "POINT_DATA %d\n", n*n);
      for (scalar s in list) {
        fprintf (fp, "SCALARS %s double\n", s.name);
        fputs ("LOOKUP_TABLE default\n", fp);
        double fn = n;
        double Delta = L0/fn;
        for (int i = 0; i < n; i++) {
          double x = Delta*i + X0 + Delta/2.;
          for (int j = 0; j < n; j++) {
    	double y = Delta*j + Y0 + Delta/2., v;
    	if (linear)
    	  v = interpolate (s, x, y);
    	else {
    	  Point point = locate (x, y);
    	  v = point.level >= 0 ? val(s) : nodata;
    	}
    	fprintf (fp, "%g\n", v);
          }
        }
      }
      fflush (fp);
    }