grid.h

Go to the documentation of this file.

/********************************************************************************************************************************************
 * Saras
 * 
 * Copyright (C) 2019, Mahendra K. Verma
 *
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     1. Redistributions of source code must retain the above copyright
 *        notice, this list of conditions and the following disclaimer.
 *     2. Redistributions in binary form must reproduce the above copyright
 *        notice, this list of conditions and the following disclaimer in the
 *        documentation and/or other materials provided with the distribution.
 *     3. Neither the name of the copyright holder nor the
 *        names of its contributors may be used to endorse or promote products
 *        derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 ********************************************************************************************************************************************
 */
#ifndef GRID_H
#define GRID_H
 
#include <math.h>
#include <string>
#include <vector>
#include <blitz/array.h>
 
#include "parallel.h"
 
class grid {
    private:
        blitz::TinyVector<real, 3> thBeta;
 
        blitz::Array<real, 1> xi;
        blitz::Array<real, 1> et;
        blitz::Array<real, 1> zt;
 
        blitz::Array<real, 1> xiGlo;
        blitz::Array<real, 1> etGlo;
        blitz::Array<real, 1> ztGlo;
 
        void resizeGrid();
        void makeSizeArray();
        void setDomainSizes();
        void globalXiEtaZeta();
 
        void createUniformGrid();
        void createTanHypGrid(int dim);
 
        void checkAnisotropy();
        void gatherGlobal();
 
        void computeGlobalLimits();
 
    public:
        const parser &inputParams;
 
        const parallel &rankData;
 
        blitz::TinyVector<int, 3> collocCoreSize;
 
        blitz::TinyVector<int, 3> collocFullSize;
 
        blitz::TinyVector<int, 3> staggrCoreSize;
 
        blitz::TinyVector<int, 3> staggrFullSize;
 
        blitz::TinyVector<int, 3> padWidths;
 
        blitz::TinyVector<int, 3> globalSize;
 
        blitz::TinyVector<int, 3> subarrayEnds;
 
        blitz::TinyVector<int, 3> subarrayStarts;
 
        real dXi;
 
        real dEt;
 
        real dZt;
 
        real xLen;
 
        real yLen;
 
        real zLen;
 
        blitz::Array<int, 1> sizeArray;
 
        blitz::TinyVector<int, 3> sizeIndex;
 
        blitz::RectDomain<3> collocCoreDomain;
 
        blitz::RectDomain<3> collocFullDomain;
 
        blitz::RectDomain<3> staggrCoreDomain;
 
        blitz::RectDomain<3> staggrFullDomain;
 
        /*****************************************************************************************************************************************************/
 
        blitz::Array<real, 1> xColloc;
 
        blitz::Array<real, 1> yColloc;
 
        blitz::Array<real, 1> zColloc;
 
        blitz::Array<real, 1> xStaggr;
 
        blitz::Array<real, 1> yStaggr;
 
        blitz::Array<real, 1> zStaggr;
 
        /*****************************************************************************************************************************************************/
 
        blitz::Array<real, 1> xCollocGlobal;
 
        blitz::Array<real, 1> yCollocGlobal;
 
        blitz::Array<real, 1> zCollocGlobal;
 
        blitz::Array<real, 1> xStaggrGlobal;
 
        blitz::Array<real, 1> yStaggrGlobal;
 
        blitz::Array<real, 1> zStaggrGlobal;
 
        /*****************************************************************************************************************************************************/
 
        blitz::Array<real, 1> xi_xColloc;
 
        blitz::Array<real, 1> xi_xStaggr;
 
        blitz::Array<real, 1> xixxColloc;
 
        blitz::Array<real, 1> xixxStaggr;
 
        blitz::Array<real, 1> xix2Colloc;
 
        blitz::Array<real, 1> xix2Staggr;
 
        /*****************************************************************************************************************************************************/
 
        blitz::Array<real, 1> et_yColloc;
 
        blitz::Array<real, 1> et_yStaggr;
 
        blitz::Array<real, 1> etyyColloc;
 
        blitz::Array<real, 1> etyyStaggr;
 
        blitz::Array<real, 1> ety2Colloc;
 
        blitz::Array<real, 1> ety2Staggr;
 
        /*****************************************************************************************************************************************************/
 
        blitz::Array<real, 1> zt_zColloc;
 
        blitz::Array<real, 1> zt_zStaggr;
 
        blitz::Array<real, 1> ztzzColloc;
 
        blitz::Array<real, 1> ztzzStaggr;
 
        blitz::Array<real, 1> ztz2Colloc;
 
        blitz::Array<real, 1> ztz2Staggr;
 
        grid(const parser &solParam, parallel &parallelData);
 
        inline bool pointInDomain(blitz::TinyVector<int, 3> gloIndex) const {
            if ((gloIndex(0) < subarrayEnds(0)) and (gloIndex(0) >= subarrayStarts(0)) and (gloIndex(1) < subarrayEnds(1)) and (gloIndex(1) >= subarrayStarts(1))) return true;
 
            return false;
        };
 
        inline blitz::TinyVector<int, 3> glo2loc(blitz::TinyVector<int, 3> gloIndex) const {
            blitz::TinyVector<int, 3> locIndex;
 
            if (pointInDomain(gloIndex)) {
                locIndex(0) = gloIndex(0) % collocCoreSize(0);
                locIndex(1) = gloIndex(1) % collocCoreSize(1);
                locIndex(2) = gloIndex(2);
            } else {
                locIndex = 0, 0, 0;
            }
 
            return locIndex;
        };
 
        inline blitz::TinyVector<int, 3> loc2glo(blitz::TinyVector<int, 3> locIndex) const {
            blitz::TinyVector<int, 3> gloIndex;
 
            gloIndex(0) = rankData.xRank*collocCoreSize(0) + locIndex(0);
            gloIndex(1) = rankData.yRank*collocCoreSize(1) + locIndex(1);
            gloIndex(2) = locIndex(2);
 
            return gloIndex;
        };
};
 
#endif