mpi2/io_utils.cpp at master · jonqdoe/mpi2 · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#include "globals.h"


void write_kspace_data( const char *lbl , complex<double> *kdt ) {
  int i, j , nn[Dim] ;
  FILE *otp ;
  double kv[Dim], k2 ;

  char nm[30] ;
  sprintf( nm, "%s.p%d.dat" , lbl, myrank ) ;
  otp = fopen( nm , "w" ) ;

  for ( i=1 ; i<ML ; i++ ) {
    unstack( i , nn ) ;

    k2 = get_k( i , kv ) ;

    for ( j=0 ; j<Dim ; j++ )
      fprintf( otp , "%lf " , kv[j] ) ;

    fprintf( otp , "%1.5e %1.5e %1.5e %1.5e\n" , abs(kdt[i]), sqrt(k2),
        real(kdt[i]) , imag(kdt[i]) ) ;

    if ( Dim == 2 && nn[0] == Nx[0]-1 )
      fprintf( otp , "\n" ) ;
  }

  fclose( otp ) ;


}

void write_grid_data( const char *lbl , double *dat ) {

  int i, j, nn[Dim] ;
  FILE *otp ;
  double r[Dim] ;
  char nm[30] ;
  sprintf( nm, "%s.p%d.dat" , lbl, myrank ) ;

  otp = fopen( nm , "w" ) ;

  for ( i=0 ; i<ML ; i++ ) {
    get_r( i, r ) ;
    unstack_local( i, nn ) ;

    for ( j=0 ; j<Dim ; j++ )
      fprintf( otp , "%lf " , r[j] ) ;

    fprintf( otp , "%1.8e \n" , dat[i] ) ;

    if ( Dim == 2 && nn[0] == Nx[0]-1 )
      fprintf( otp , "\n" ) ;
  }

  fclose( otp ) ;

}

void write_Sfield_data( const char *lbl , double ***dat ) {

  int i, j, nn[Dim] ;
  FILE *otp ;
  double r[Dim] ;
  char nm[30] ;
  sprintf( nm, "%s.p%d.dat" , lbl, myrank ) ;

  otp = fopen( nm , "w" ) ;

  for ( i=0 ; i<ML ; i++ ) {
    get_r( i, r ) ;
    unstack_local( i, nn ) ;

    for ( j=0 ; j<Dim ; j++ )
      fprintf( otp , "%lf " , r[j] ) ;

    fprintf( otp , "%1.16e \n" , dat[i][0][0] ) ;

    if ( Dim == 2 && nn[0] == Nx[0]-1 )
      fprintf( otp , "\n" ) ;
  }

  fclose( otp ) ;

}