hydro.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 HYDRO_H
#define HYDRO_H
 
#include <blitz/array.h>
 
#include "boundary.h"
#include "parallel.h"
#include "poisson.h"
#include "plainvf.h"
#include "tseries.h"
#include "writer.h"
#include "reader.h"
#include "probes.h"
#include "sfield.h"
#include "vfield.h"
#include "parser.h"
#include "force.h"
#include "grid.h"
 
class hydro {
    public:
        vfield V;
 
        sfield P;
 
        force *vForcing;
 
        hydro(const grid &mesh, const parser &solParam, parallel &mpiParam);
 
        virtual void solvePDE();
        virtual real testPeriodic();
 
        virtual ~hydro() { };
 
    protected:
        int timeStepCount;
 
        int maxIterations;
 
        real time, dt;
 
        real hx, hy, hz;
        real hx2, hz2, hz2hx2;
        real hx2hy2, hy2hz2, hx2hy2hz2;
 
        const grid &mesh;
        const parser &inputParams;
 
        const real inverseRe;
 
        probes *dataProbe;
 
        boundary *uLft, *uRgt, *uFrn, *uBak, *uTop, *uBot;
        boundary *vLft, *vRgt, *vFrn, *vBak, *vTop, *vBot;
        boundary *wLft, *wRgt, *wFrn, *wBak, *wTop, *wBot;
 
        parallel &mpiData;
 
        plainsf Pp;
        plainsf mgRHS;
 
        plainvf nseRHS;
        plainvf velocityLaplacian;
        plainvf pressureGradient;
        plainvf guessedVelocity;
 
        void checkPeriodic();
        void setCoefficients();
 
        void initVBC();
        void imposeUBCs();
        void imposeVBCs();
        void imposeWBCs();
 
        void initVForcing();
 
        virtual void solveVx();
        virtual void solveVy();
        virtual void solveVz();
 
        virtual void timeAdvance();
};
 
class hydro_d2: public hydro {
    public:
        hydro_d2(const grid &mesh, const parser &solParam, parallel &mpiParam);
 
        void solvePDE();
        real testPeriodic();
 
        ~hydro_d2();
 
    private:
        multigrid_d2 mgSolver;
 
        void solveVx();
        void solveVz();
 
        void timeAdvance();
};
 
class hydro_d3: public hydro {
    public:
        hydro_d3(const grid &mesh, const parser &solParam, parallel &mpiParam);
 
        void solvePDE();
        real testPeriodic();
 
        ~hydro_d3();
 
    private:
        multigrid_d3 mgSolver;
 
#ifdef TIME_RUN
        real visc_time, nlin_time, intr_time, impl_time, prhs_time, pois_time;
#endif
 
        void solveVx();
        void solveVy();
        void solveVz();
 
        void timeAdvance();
};
 
#endif