#include "phIO.h"
#include "phComm.h"
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <phastaIO.h>
#include <sstream>
#include <string>

namespace {
  void appendRank(std::stringstream& ss, const char* phrase) {
    ss << phrase << "@" << phcomm_rank()+1;
  }
  std::string appendSync(const char* phrase) {
    std::stringstream ss;
    appendRank(ss,phrase);
    ss << "?";
    return ss.str();
  }
  std::string appendSyncWrite(const char* phrase) {
    std::stringstream ss;
    appendRank(ss,phrase);
    return ss.str();
  }
  std::string appendColor(const char* phrase, int numFiles) {
    const int color = computeColor(phcomm_rank(), phcomm_size(), numFiles);
    std::stringstream ss;
    ss << phrase << color+1;
    return ss.str();
  }
}

void phio_readheader(
    int* fileDescriptor,
    const  char keyphrase[],
    void* valueArray,
    int*  nItems,
    const char  datatype[],
    const char  iotype[] ) {
  std::string syncPhrase = appendSync(keyphrase);
  readheader(fileDescriptor, syncPhrase.c_str(),
      valueArray, nItems, datatype, iotype);
}

void phio_writeheader(
      const int* fileDescriptor,
      const char keyphrase[],
      const void* valueArray,
      const int* nItems,
      const int* ndataItems,
      const char datatype[],
      const char iotype[] ) {
  std::string syncPhrase = appendSyncWrite(keyphrase);
  writeheader(fileDescriptor, syncPhrase.c_str(),
      valueArray, nItems, ndataItems, datatype, iotype);
}


void phio_readdatablock(
    int*  fileDescriptor,
    const char keyphrase[],
    void* valueArray,
    int*  nItems,
    const char  datatype[],
    const char  iotype[] ) {
  std::string syncPhrase = appendSync(keyphrase);
  readdatablock(fileDescriptor, syncPhrase.c_str(),
      valueArray, nItems, datatype, iotype);
}

void phio_writedatablock(
    const int* fileDescriptor,
    const char keyphrase[],
    const void* valueArray,
    const int* nItems,
    const char datatype[],
    const char iotype[]) {
  std::string syncPhrase = appendSyncWrite(keyphrase);
  writedatablock(fileDescriptor, syncPhrase.c_str(),
      valueArray, nItems, datatype, iotype);
}

void phio_openfile_read(
    const char filename[],
    int* numFiles,
    int* fileDescriptor) {
  std::string syncName = appendColor(filename, *numFiles);
  int nfields=0;
  int nppf=0;
  queryphmpiio(syncName.c_str(), &nfields, &nppf);
  const char* mode = "read";
  initphmpiio(&nfields, &nppf, numFiles, fileDescriptor, mode); 
  openfile(syncName.c_str(), mode, fileDescriptor);
}

void phio_openfile_write(
    const char filename[],
    int* numFiles,
    int* numFields,
    int* numPPF,
    int* fileDescriptor) {
  std::string syncName = appendColor(filename, *numFiles);
  //TODO - define a good upper bound
  assert(*numFields > 0 && *numFields < 1024);
  assert(*numPPF > 0 && *numPPF < 1024);
  const char* mode = "write";
  initphmpiio(numFields, numPPF, numFiles, fileDescriptor, mode); 
  openfile(syncName.c_str(), mode, fileDescriptor);
}

void phio_restartname(int* step, char* filename) {
  std::stringstream ss;
  ss << "restart-dat." << *step << '.';
  std::string s = ss.str();
  strcpy(filename, s.c_str()); 
}

void phio_closefile_read(int* fileDescriptor) {
  const char* mode = "read";
  closefile(fileDescriptor, mode);
  finalizephmpiio(fileDescriptor);
}

void phio_closefile_write(int* fileDescriptor) {
  const char* mode = "write";
  closefile(fileDescriptor, mode);
  finalizephmpiio(fileDescriptor);
}