rk4nlsm.h

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

#ifndef SFCSORTBENCH_RK4NLSM_H
#define SFCSORTBENCH_RK4NLSM_H


#include "rk.h"
#include "fdCoefficient.h"
#include "oct2vtk.h"
#include "checkPoint.h"
#include "mesh.h"
#include <string>
#include <iostream>
#include "nlsmUtils.h"
#include "parameters.h"
#include "nlsm.h"
#include "rhs.h"
#include "test/meshTestUtils.h"


static const double RK4_C[]={1.0/6.0,1.0/3.0,1.0/3.0,1.0/6.0};

static const double RK4_T[]={0,1.0/2.0,1.0/2.0,1.0};
static const double RK4_U[]={0.0,1.0/2.0,1.0/2.0,1.0};

namespace ode
{
    namespace solver
    {

        class RK4_NLSM : public RK
        {

        private:

            // variables for NLSM formulation.

            double ** m_uiVar;

            double ** m_uiPrevVar;

            double ** m_uiVarIm;

            double **m_uiUnzipVar;

            double **m_uiUnzipVarRHS;

            double *** m_uiStage;

            double ** m_uiConstraintVars;

            double ** m_uiUnzipConstraintVars;

            double ** m_uiSendNodeBuf;

            double ** m_uiRecvNodeBuf;

            MPI_Request ** m_uiSendReqs;

            MPI_Request ** m_uiRecvReqs;

            MPI_Status ** m_uiSendSts;

            MPI_Status ** m_uiRecvSts;




        public:
            RK4_NLSM(ot::Mesh *pMesh, double pTBegin, double pTEnd,double pTh);

            ~RK4_NLSM();

            void readConfigFile(const char * fName);

            void rkSolve();

            void restoreCheckPoint(const char * fNamePrefix,MPI_Comm comm);

        private:
            void applyInitialConditions(double ** zipIn);

            void initialGridConverge();

            void reallocateMPIResources();

            void performGhostExchangeVars(double** zipIn);

            void intergridTransferVars(double **& zipIn, const ot::Mesh* pnewMesh);

            void unzipVars(double ** zipIn , double **uzipOut);

            void unzipVars_async(double ** zipIn , double **uzipOut);

            void zipVars(double** uzipIn , double** zipOut);

            void writeToVTU(double **evolZipVarIn, double ** constrZipVarIn, unsigned int numEvolVars,unsigned int numConstVars,const unsigned int * evolVarIndices, const unsigned int * constVarIndices);

            void performSingleIteration();

            void applyBoundaryConditions();

            void storeCheckPoint(const char * fNamePrefix);


        };

    } // end of namespace solver
}// end of namespace ode





#endif //SFCSORTBENCH_RK4NLSM_H