xref: /phasta/phSolver/common/test/phIOwriteFields.cc (revision 3964a29a7f38426c31f34f31a5251ddd14e65cbd)
1 #include <mpi.h>
2 #include <stdio.h>
3 #include <stdlib.h>
4 #include <unistd.h>
5 #include <cassert>
6 #include "phiostats.h"
7 #include "phIO.h"
8 #include "phstream.h" //for makeRStream and makeGRStream
9 #include "syncio.h"
10 #include "posixio.h"
11 #include "streamio.h"
12 
13 int main(int argc, char* argv[]) {
14   MPI_Init(&argc,&argv);
15   int rank;
16   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
17   int size;
18   MPI_Comm_size(MPI_COMM_WORLD, &size);
19   if( argc != 2 ) {
20     fprintf(stderr, "Usage: %s <numSyncFiles>\n",argv[0]);
21     MPI_Finalize();
22     return 1;
23   }
24   const char* phrase = "number of fishes";
25   const char* type = "double";
26   const char* iotype = "binary";
27   int zero = 0;
28   int one = 1;
29   int numFish = 0;
30   double fishWeight = 1.23;
31   int nfiles = 1;
32   int ppf = size/nfiles;
33   const char* filename[2] = {"water-dat.", "water.dat."};
34   rstream rs = makeRStream();
35      phio_fp file[3];
36   const char* modes[3]={"syncio", "posixio", "streamio"};
37   syncio_setup_write(nfiles, one, ppf, &(file[0]));
38   posixio_setup(&(file[1]), 'w');
39   streamio_setup_r(&(file[2]), rs, 'w');
40   fprintf(stderr, "%s\n" ,"Outside loop 1.0");
41   for(int i=0; i<3; i++) {
42     fprintf(stderr, "%s\n" ,"Within the i loop");
43     if(!rank) fprintf(stderr, "%s\n", modes[i]);
44     fprintf(stderr, "%s\n" ,"Before phastaio");
45     phastaio_initStats();
46     fprintf(stderr, "%s\n" ,"Opening files with ", filename[i], file[i] );
47     phio_openfile(filename[i], file[i]);
48     fprintf(stderr, "%s\n" ,"Entering for loop for ", atoi(argv[1]) );
49    // const char* str = "Number of times "+ nfiles ;
50     for (int j = 0; j < nfiles ; j++) {
51       fprintf(stderr,"%s\n", "Inside loop");
52       fprintf(stderr,"%d\n",atoi(argv[1]));
53       const char* str = "Number of times " +j;
54       fprintf(stderr,"%s\n", "Writing the header time - " );
55       fprintf(stderr,"%d\n",zero);
56       fprintf(stderr,"%d\n",one);
57       fprintf(stderr,"%s\n",file[i] );
58       fprintf(stderr,"%s\n","Should have printed the file" );
59       fprintf(stderr,"%s\n",type);
60       fprintf(stderr,"%s\n",iotype);
61       fprintf(stderr,"%s\n","Opening the file" );
62       phio_writeheader(file[i], str, &zero, &one, &zero, type, iotype);
63       fprintf(stderr,"%s\n",str );
64       fprintf(stderr,"%d\n",j );
65       fprintf(stderr,"%s\n", "Writing the data block time - ");
66       fprintf(stderr,"%d\n",j );
67       phio_writedatablock(file[i], str, &fishWeight, &zero, type, iotype);
68     }
69     phio_closefile(file[i]);
70     phastaio_printStats();
71   }
72   syncio_setup_read(nfiles, &(file[0]));
73   posixio_setup(&(file[1]), 'r');
74   streamio_setup_r(&(file[2]), rs, 'r');
75   for(int i=0; i<3; i++) {
76     if(!rank) fprintf(stderr, "%s\n", modes[i]);
77     phastaio_initStats();
78     phio_openfile(filename[i], file[i]);
79     //Str was added
80    // const char* str = "Number of times "+ nfiles ;
81    // for (int j = 0; j < nfiles ; j++) {
82     phio_readheader(file[i], phrase, &numFish, &one, type, iotype);
83     //  phio_readheader(file[i], str, &numFish, &one, type, iotype);
84      //Changing argument from file[i] to file[j]
85     assert(!numFish);
86     phio_readdatablock(file[i], phrase, &fishWeight, &numFish, type, iotype);
87     //  phio_readdatablock(file[i], str, &fishWeight, &numFish, type, iotype);
88     assert(fishWeight == 1.23);
89     phio_closefile(file[i]);
90     phastaio_printStats();
91   }
92   MPI_Finalize();
93   return 0;
94 }
95