xref: /phasta/phSolver/common/test/phIOwriteFields.cc (revision 0a242d7cf942d5a60cebedb353bc827503389fbe)
1f1d1f61eSPranav Subramanian #include <mpi.h>
2f1d1f61eSPranav Subramanian #include <stdio.h>
3f1d1f61eSPranav Subramanian #include <stdlib.h>
4f1d1f61eSPranav Subramanian #include <unistd.h>
5f1d1f61eSPranav Subramanian #include <cassert>
6f1d1f61eSPranav Subramanian #include "phiostats.h"
7f1d1f61eSPranav Subramanian #include "phIO.h"
8f1d1f61eSPranav Subramanian #include "phstream.h" //for makeRStream and makeGRStream
9f1d1f61eSPranav Subramanian #include "syncio.h"
10f1d1f61eSPranav Subramanian #include "posixio.h"
11f1d1f61eSPranav Subramanian #include "streamio.h"
12f1d1f61eSPranav Subramanian 
13f1d1f61eSPranav Subramanian int main(int argc, char* argv[]) {
14f1d1f61eSPranav Subramanian   MPI_Init(&argc,&argv);
15f1d1f61eSPranav Subramanian   int rank;
16f1d1f61eSPranav Subramanian   MPI_Comm_rank(MPI_COMM_WORLD, &rank);
17f1d1f61eSPranav Subramanian   int size;
18f1d1f61eSPranav Subramanian   MPI_Comm_size(MPI_COMM_WORLD, &size);
19f1d1f61eSPranav Subramanian   if( argc != 2 ) {
20f1d1f61eSPranav Subramanian     fprintf(stderr, "Usage: %s <numSyncFiles>\n",argv[0]);
21f1d1f61eSPranav Subramanian     MPI_Finalize();
22f1d1f61eSPranav Subramanian     return 1;
23f1d1f61eSPranav Subramanian   }
24f1d1f61eSPranav Subramanian   const char* phrase = "number of fishes";
25f1d1f61eSPranav Subramanian   const char* type = "double";
26f1d1f61eSPranav Subramanian   const char* iotype = "binary";
27f1d1f61eSPranav Subramanian   int zero = 0;
28f1d1f61eSPranav Subramanian   int one = 1;
29f1d1f61eSPranav Subramanian   int numFish = 0;
30f1d1f61eSPranav Subramanian   double fishWeight = 1.23;
31*0a242d7cSPranav Subramanian   int nfiles = 1;
32f1d1f61eSPranav Subramanian   int ppf = size/nfiles;
33f1d1f61eSPranav Subramanian   const char* filename[2] = {"water-dat.", "water.dat."};
34f1d1f61eSPranav Subramanian   rstream rs = makeRStream();
35f1d1f61eSPranav Subramanian   phio_fp file[3];
36f1d1f61eSPranav Subramanian   const char* modes[3]={"syncio", "posixio", "streamio"};
37f1d1f61eSPranav Subramanian   syncio_setup_write(nfiles, one, ppf, &(file[0]));
38f1d1f61eSPranav Subramanian   posixio_setup(&(file[1]), 'w');
39f1d1f61eSPranav Subramanian   streamio_setup_r(&(file[2]), rs, 'w');
40f1d1f61eSPranav Subramanian   for(int i=0; i<3; i++) {
41f1d1f61eSPranav Subramanian     if(!rank) fprintf(stderr, "%s\n", modes[i]);
42f1d1f61eSPranav Subramanian     phastaio_initStats();
43f1d1f61eSPranav Subramanian     phio_openfile(filename[i], file[i]);
44*0a242d7cSPranav Subramanian     for (int j = 0; j < atoi(argv[1]) ; j++) {
45f1d1f61eSPranav Subramanian       phio_writeheader(file[i], phrase, &zero, &one, &zero, type, iotype);
46f1d1f61eSPranav Subramanian       phio_writedatablock(file[i], phrase, &fishWeight, &zero, type, iotype);
47*0a242d7cSPranav Subramanian     }
48f1d1f61eSPranav Subramanian     phio_closefile(file[i]);
49f1d1f61eSPranav Subramanian     phastaio_printStats();
50f1d1f61eSPranav Subramanian   }
51f1d1f61eSPranav Subramanian   syncio_setup_read(nfiles, &(file[0]));
52f1d1f61eSPranav Subramanian   posixio_setup(&(file[1]), 'r');
53f1d1f61eSPranav Subramanian   streamio_setup_r(&(file[2]), rs, 'r');
54f1d1f61eSPranav Subramanian   for(int i=0; i<3; i++) {
55f1d1f61eSPranav Subramanian     if(!rank) fprintf(stderr, "%s\n", modes[i]);
56f1d1f61eSPranav Subramanian     phastaio_initStats();
57f1d1f61eSPranav Subramanian     phio_openfile(filename[i], file[i]);
58f1d1f61eSPranav Subramanian     phio_readheader(file[i], phrase, &numFish, &one, type, iotype);
59f1d1f61eSPranav Subramanian     assert(!numFish);
60f1d1f61eSPranav Subramanian     phio_readdatablock(file[i], phrase, &fishWeight, &numFish, type, iotype);
61f1d1f61eSPranav Subramanian     assert(fishWeight == 1.23);
62f1d1f61eSPranav Subramanian     phio_closefile(file[i]);
63f1d1f61eSPranav Subramanian     phastaio_printStats();
64f1d1f61eSPranav Subramanian   }
65f1d1f61eSPranav Subramanian   MPI_Finalize();
66f1d1f61eSPranav Subramanian   return 0;
67f1d1f61eSPranav Subramanian }
68