nuts.h

Go to the documentation of this file.

#pragma once
#include "mesh.h"
#include <functional>
#include "block.h"
#include "point.h"
#include <assert.h>


namespace ts
{   
    class NUTS
    {
        private:

            ot::Mesh*  m_uiOctGrid;

            ot::Mesh* m_uiBlockGrid;

            unsigned int m_uiS;

            double m_uiTB;

            double m_uiTE;

            double m_uiDTMin;

            double m_uiDTMax;

            double m_uiCFL;

            MPI_Comm m_uiCommActive;

            MPI_Comm m_uiCommGlobal;

            bool isActive;

            Point m_uiPtMin;

            Point m_uiPtMax;
            
            double m_uiCT;

            std::vector<unsigned int> m_uiSendSMLevelTag;

            std::vector<unsigned int> m_uiRecvSMLevelTag;

            unsigned int ** m_uiSendCountByLev;
            
            unsigned int ** m_uiRecvCountByLev;

            unsigned int ** m_uiSendOffsetByLev;

            unsigned int ** m_uiRecvOffsetByLev;

            std::vector< std::pair<unsigned int,unsigned int> > m_uiSendNodeRSM;

            std::vector< std::pair<unsigned int,unsigned int> > m_uiRecvNodeRSM;

            

            

        private:
            void compBlockFlags();

            void computeBlockSM();




        public:

            NUTS(ot::Mesh* octMesh, unsigned int s, double tb, double te, Point ptMin, Point ptMax, double cfl=0.25);

            ~NUTS();

            template<typename T>
            void zip_to_unzip(T** in, T** out, bool isGSync=false, unsigned int dof=1);

            template<typename T>
            void unzip_to_zip(T** in, T** out, bool isGSync=false, unsigned int dof=1);


            template<typename T>
            void unzip_sync(T** in, unsigned int blk , unsigned int dof=1);
            
            template<typename T>
            void unzip_sync_all(T**in, unsigned int dof=1);

            void sync_ts();

            template<typename T>
            void timestep(const T** in, T** out, std::function<void(T,T**,T**)>f_rhs, const double*a, const double *b, const double* c, unsigned int dof=1);            



        
    };

    template<typename T>
    void NUTS::zip_to_unzip(T** in, T** out, bool isGSync, unsigned int dof)
    {

        if(!isGSync)
        { 
          for(unsigned int v=0;v<dof;v++)
            m_uiOctGrid->performGhostExchange(in[v]);

        } 
        
        for(unsigned int v=0;v<dof;v++)
          m_uiOctGrid->unzip(in[v],out[v]);

    }


    template<typename T>
    void NUTS::unzip_to_zip(T** in, T** out, bool isGSync, unsigned int dof)
    {
        for(unsigned int v=0;v<dof;v++)
          m_uiOctGrid->zip(in[v],out[v]);

        if(isGSync)
        {
            for(unsigned int v=0;v<dof;v++)
                m_uiOctGrid->performGhostExchange(in[v]);
        }
    }

    template<typename T>
    void NUTS::unzip_sync(T** in, unsigned int blk , unsigned int dof)
    {
        
        const unsigned int blockLocalBegin = m_uiBlockGrid ->getElementLocalBegin();
        const unsigned int blockLocalEnd   = m_uiBlockGrid->getElementLocalEnd();

        if(blk< blockLocalBegin  || blk>= blockLocalEnd)
            return;
        
        //const unsigned int ;


    }

    template<typename T>
    void NUTS::unzip_sync_all(T**in, unsigned int dof)
    {
        const unsigned int blockLocalBegin = m_uiBlockGrid ->getElementLocalBegin();
        const unsigned int blockLocalEnd   = m_uiBlockGrid->getElementLocalEnd();

        for(unsigned int blk = blockLocalBegin; blk<blockLocalEnd ; blk++)
            unzip_sync(in,blk,dof);

    }



}