2p/sequential/fluxdata.hh

Go to the documentation of this file.

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*****************************************************************************
 *   See the file COPYING for full copying permissions.                      *
 *                                                                           *
 *   This program is free software: you can redistribute it and/or modify    *
 *   it under the terms of the GNU General Public License as published by    *
 *   the Free Software Foundation, either version 3 of the License, or       *
 *   (at your option) any later version.                                     *
 *                                                                           *
 *   This program is distributed in the hope that it will be useful,         *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of          *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the            *
 *   GNU General Public License for more details.                            *
 *                                                                           *
 *   You should have received a copy of the GNU General Public License       *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.   *
 *****************************************************************************/
#ifndef DUMUX_FLUXDATA2P_HH
#define DUMUX_FLUXDATA2P_HH
 
#include "properties.hh"
 
namespace Dumux {
template<class TypeTag>
class FluxData2P
{
private:
    using Scalar = typename GET_PROP_TYPE(TypeTag, Scalar);
    using GridView = typename GET_PROP_TYPE(TypeTag, GridView);
 
    enum
    {
        dim = GridView::dimension
    };
 
    using Indices = typename GET_PROP_TYPE(TypeTag, ModelTraits)::Indices;
 
    enum
    {
        wPhaseIdx = Indices::wPhaseIdx, nPhaseIdx = Indices::nPhaseIdx
    };
 
    enum
    {
        numPhases = GET_PROP_VALUE(TypeTag, NumPhases)
    };
 
    using DimVector = Dune::FieldVector<Scalar, dim>;
    using VelocityVector = Dune::FieldVector<DimVector, 2*dim>;
 
    VelocityVector velocity_[numPhases];
    Scalar upwindPotential_[2 * dim][numPhases];
    bool velocityMarker_[2 * dim];
 
public:
 
    FluxData2P()
    {
        for (int fIdx = 0;  fIdx < 2*dim; fIdx++)
        {
            for (int phaseIdx = 0; phaseIdx < numPhases; phaseIdx++)
            {
                velocity_[phaseIdx][fIdx] = DimVector(0.0);
 
                upwindPotential_[fIdx][phaseIdx] = 0.0;
            }
            velocityMarker_[fIdx] = false;
        }
    }
 
    // functions returning the vectors of the primary variables
 
    const DimVector& velocity(int phaseIdx, int indexInInside)
    {
        return velocity_[phaseIdx][indexInInside];
    }
 
    const DimVector& velocity(int phaseIdx, int indexInInside) const
    {
        return velocity_[phaseIdx][indexInInside];
    }
 
    void setVelocity(int phaseIdx, int indexInInside, const DimVector& velocity)
    {
        velocity_[phaseIdx][indexInInside] = velocity;
    }
 
    void addVelocity(int phaseIdx, int indexInInside, const DimVector& velocity)
    {
        velocity_[phaseIdx][indexInInside] += velocity;
    }
 
    void resetVelocity()
    {
        for (int i = 0; i < 2 * dim; i++)
        {
            for (int j = 0; j < numPhases; j++)
            {
                velocity_[j][i] = 0.;
                upwindPotential_[i][j] = 0.;
            }
            velocityMarker_[i] = false;
        }
    }
 
    DimVector velocityTotal(int indexInInside)
    {
        return velocity_[wPhaseIdx][indexInInside]
                + velocity_[nPhaseIdx][indexInInside];
    }
 
    DimVector velocityTotal(int indexInInside) const
    {
        return velocity_[wPhaseIdx][indexInInside]
                + velocity_[nPhaseIdx][indexInInside];
    }
 
    void setVelocityMarker(int indexInInside)
    {
        velocityMarker_[indexInInside] = true;
    }
 
    bool haveVelocity(int indexInInside)
    {
        return velocityMarker_[indexInInside];
    }
 
    void resetVelocityMarker()
    {
        for (int i = 0; i < 2*dim; i++)
            velocityMarker_[i] = false;
    }
 
    bool isUpwindCell(int phaseIdx, int indexInInside)
    {
        return (upwindPotential_[indexInInside][phaseIdx] > 0.);
    }
 
    bool isUpwindCell(int phaseIdx, int indexInInside) const
    {
        return (upwindPotential_[indexInInside][phaseIdx] > 0.);
    }
 
    Scalar upwindPotential(int phaseIdx, int indexInInside)
    {
        return upwindPotential_[indexInInside][phaseIdx];
    }
 
    Scalar upwindPotential(int phaseIdx, int indexInInside) const
    {
        return upwindPotential_[indexInInside][phaseIdx];
    }
 
    void setUpwindPotential(int phaseIdx, int indexInInside, Scalar pot)
    {
        upwindPotential_[indexInInside][phaseIdx] = pot;
    }
 
    void addUpwindPotential(int phaseIdx, int indexInInside, Scalar pot)
    {
        upwindPotential_[indexInInside][phaseIdx] += pot;
    }
};
} // end namespace Dumux
#endif