1 files changed, 83 insertions, 0 deletions
diff --git a/src/rawSpeed.cc b/src/rawSpeed.cc
new file mode 100644
index 0000000..9f7e4b0
--- /dev/null
+++ b/src/rawSpeed.cc
@@ -0,0 +1,83 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include "IEEEIO.hh"
+#include "vatoi.hh"
+#include <sys/time.h>
+double MPI_Wtime(){
+  timeval tv;
+  gettimeofday(&tv,0);
+  return tv.tv_sec + tv.tv_usec/1000000.0l;
+}
+
+class CommandLine {
+public:
+  int dims[3];
+  char *programname;
+  int bufsize;
+  int ntrials;
+  void setDefaults(){
+    dims[0]=dims[1]=dims[2]=128;
+    bufsize=0;
+    ntrials=40;
+  }
+  CommandLine(int argc,char *argv[]){
+    setDefaults();
+    parse(argc,argv);
+  }
+  void parse(int argc,char *argv[]){
+    programname=argv[0];
+    for(int i=1;i<argc;i++){
+      char *flag = argv[i],*val=argv[i+1];
+      if(*flag=='-') flag++;
+      else continue; // ignore non-flags
+      if(!strcmp(flag,"dims")){
+	if(++i<argc) sscanf(val,"%u,%u,%u",dims,dims+1,dims+2);
+      }
+      else if(!strcmp(flag,"buffer")) {
+	if(++i<argc)
+	  bufsize = vatoi(val);
+      }
+      else if(*flag=='n') {
+	ntrials=atoi(val);
+	i++;
+      }
+      else if(*flag=='h') usage();
+    }
+  }
+  void usage(){
+    printf("Usage: %s <-dims int,int,int> -n <ntrials>\n",
+	   programname);
+    printf("\tdims: Set the dimensions of the dataset.\n");
+    printf("\tn: Set number of trials of writing the dataset.\n");
+    printf("\tbuffer: Set buffer cache size in bytes.\n");
+  }
+  void printStatus(FILE *f=stdout){
+    fprintf(f,"%s: dims=%u,%u,%u\n",
+	   programname,
+	   dims[0],dims[1],dims[2]);
+  }
+};
+
+int main(int argc,char *argv[]){
+  CommandLine cmdln(argc,argv);
+
+  int size=IObase::nElements(3,cmdln.dims);
+  int i;
+  float *data = new float[size];
+  for(i=0;i<size;i++) data[i] = (float)i;
+  IEEEIO *file = new IEEEIO("rawdata.raw",IObase::Write);
+  if(cmdln.bufsize>0) file->bufferOn(cmdln.bufsize);
+  double stime = MPI_Wtime();
+  for(i=0;i<cmdln.ntrials;i++){ // do this 40 times
+    file->reserveStream(IObase::Float32,3,cmdln.dims);
+    file->writeStream(data,size);
+  }
+  if(file) delete file;  // make sure the file is deleted
+  double etime = MPI_Wtime();
+  printf("Elapsed time to write=%lf\n",etime-stime);
+  printf("IO performance = %f Megabytes/sec\n",
+	 (float)(IObase::nBytes(IObase::Float32,3,cmdln.dims))*(float)(cmdln.ntrials)/(1024*1024*(etime-stime)));
+  delete data;
+}
+
+