checkPoint.h

//
// Created by milinda on 7/12/17.
//

#ifndef SFCSORTBENCH_CHECKPOINT_H
#define SFCSORTBENCH_CHECKPOINT_H

#include "TreeNode.h"
#include "mesh.h"
#include "json.hpp"
#include <iostream>
#include <fstream>
using json = nlohmann::json;
namespace io
{

  namespace checkpoint
  {

      int writeOctToFile(const char * fName,const ot::TreeNode* pNodes,const unsigned int num);

      int readOctFromFile(const char * fName,std::vector<ot::TreeNode> & pNodes);


      template <typename T>
      int writeVecToFile(const char * fName, const ot::Mesh* pMesh,const T* vec);

      template <typename T>
      int writeVecToFile(const char * fName, const ot::Mesh* pMesh,const T** vec,const unsigned int numVars);

      template <typename T>
      int readVecFromFile(const char * fName, const ot::Mesh* pMesh, T* vec);


      template <typename T>
      int readVecFromFile(const char * fName, const ot::Mesh* pMesh, T** vec, const unsigned int numVars);





  } // end of namespace checkpoint.

}// end of namespace io




// templated implementations
namespace io
{
  namespace checkpoint
  {
      template <typename T>
      int writeVecToFile(const char * fName, const ot::Mesh* pMesh,const T* vec)
      {
          unsigned int numNodes=pMesh->getNumLocalMeshNodes()+pMesh->getNumPreMeshNodes()+pMesh->getNumPostMeshNodes();
          unsigned int nLocalBegin=pMesh->getNodeLocalBegin();
          unsigned int nLocalEnd=pMesh->getNodeLocalEnd();

          FILE * outfile=fopen(fName,"w");
          if(outfile==NULL){std::cout<<fName<<" file open failed "<<std::endl; return  1;}

          fwrite(&numNodes,sizeof(unsigned int ),1,outfile);
          fwrite(&nLocalBegin,sizeof(unsigned int ),1,outfile);
          fwrite(&nLocalEnd,sizeof(unsigned int ),1,outfile);
          if(numNodes>0)
            fwrite((vec+nLocalBegin),sizeof(T),pMesh->getNumLocalMeshNodes(),outfile);

          fclose(outfile);
          return 0;

      }


      template <typename T>
      int writeVecToFile(const char * fName, const ot::Mesh* pMesh,const T** vec,const unsigned int numVars)
      {
          unsigned int numNodes=pMesh->getNumLocalMeshNodes()+pMesh->getNumPreMeshNodes()+pMesh->getNumPostMeshNodes();
          unsigned int nLocalBegin=pMesh->getNodeLocalBegin();
          unsigned int nLocalEnd=pMesh->getNodeLocalEnd();

          FILE * outfile=fopen(fName,"w");
          if(outfile==NULL){std::cout<<fName<<" file open failed "<<std::endl; return  1;}

          fwrite(&numNodes,sizeof(unsigned int ),1,outfile);
          fwrite(&nLocalBegin,sizeof(unsigned int ),1,outfile);
          fwrite(&nLocalEnd,sizeof(unsigned int ),1,outfile);
          if(numNodes>0)
              for(unsigned int i=0;i<numVars;i++)
              fwrite((vec[i]+nLocalBegin),sizeof(T),pMesh->getNumLocalMeshNodes(),outfile);

          fclose(outfile);
          return 0;

      }


      template <typename T>
      int readVecFromFile(const char * fName, const ot::Mesh* pMesh,T* vec)
      {

          unsigned int numNodes,nLocalBegin,nLocalEnd;

          FILE * infile = fopen(fName,"r");
          if(infile==NULL){std::cout<<fName<<" file open failed "<<std::endl; return  1;}
          fread(&numNodes,sizeof(unsigned int ),1,infile);
          fread(&nLocalBegin,sizeof(unsigned int ),1,infile);
          fread(&nLocalEnd,sizeof(unsigned int ),1,infile);

          if(numNodes!=(pMesh->getNumPreMeshNodes()+pMesh->getNumLocalMeshNodes()+pMesh->getNumPostMeshNodes())) {std::cout<<fName<<" file number of total node mismatched with the mesh. "<<std::endl; return 1;}
          if(nLocalBegin!=pMesh->getNodeLocalBegin()) {std::cout<<fName<<" file local node begin location mismatched with the mesh. "<<std::endl; return 1;}
          if(nLocalEnd!=pMesh->getNodeLocalEnd()) {std::cout<<fName<<" file local node end location mismatched with the mesh. "<<std::endl; return 1;}

          if(numNodes>0)
              fread((vec+nLocalBegin),sizeof(T),pMesh->getNumLocalMeshNodes(),infile);

          fclose(infile);
          return 0;
      }

      template <typename T>
      int readVecFromFile(const char * fName, const ot::Mesh* pMesh, T** vec, const unsigned int numVars)
      {
          unsigned int numNodes,nLocalBegin,nLocalEnd;

          FILE * infile = fopen(fName,"r");
          if(infile==NULL){std::cout<<fName<<" file open failed "<<std::endl; return  1;}
          fread(&numNodes,sizeof(unsigned int ),1,infile);
          fread(&nLocalBegin,sizeof(unsigned int ),1,infile);
          fread(&nLocalEnd,sizeof(unsigned int ),1,infile);

          if(numNodes!=(pMesh->getNumPreMeshNodes()+pMesh->getNumLocalMeshNodes()+pMesh->getNumPostMeshNodes())) {std::cout<<fName<<" file number of total node mismatched with the mesh. "<<std::endl; return 1;}
          if(nLocalBegin!=pMesh->getNodeLocalBegin()) {std::cout<<fName<<" file local node begin location mismatched with the mesh. "<<std::endl; return 1;}
          if(nLocalEnd!=pMesh->getNodeLocalEnd()) {std::cout<<fName<<" file local node end location mismatched with the mesh. "<<std::endl; return 1;}

          if(numNodes>0)
              for(unsigned int i=0;i<numVars;i++)
                fread((vec[i]+nLocalBegin),sizeof(T),pMesh->getNumLocalMeshNodes(),infile);

          fclose(infile);
          return 0;
      }


  } // end of namespace checkpoint

} // end of namespace io




#endif //SFCSORTBENCH_CHECKPOINT_H