inamuronewtonraphsondynamics.h

来自「open lattice boltzmann project www.open」· C头文件 代码 · 共 93 行

/*  This file is part of the OpenLB library
 *
 *  Copyright (C) 2006, 2007 Orestis Malaspinas, Jonas Latt
 *  Address: Rue General Dufour 24,  1211 Geneva 4, Switzerland 
 *  E-mail: [e-mail]
 *
 *  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 2
 *  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, write to the Free 
 *  Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 *  Boston, MA  02110-1301, USA.
*/

#ifndef INAMURO_NEWTON_RAPHSON_DYNAMICS_H
#define INAMURO_NEWTON_RAPHSON_DYNAMICS_H

#include "core/dynamics.h"

namespace olb {

/**
* This class computes the inamuro BC with general dynamics. It uses the formula from the
 * paper by Inamuro et al. but since there is no explict solution
 * for a lattice different from the D2Q9 and for a speed of sound
 * c_s=q/sqrt(3), we have to use a Newton-Raphson algorithm to
 * implement these boundary conditions.
*/
template<typename T, template<typename U> class Lattice, typename Dynamics, int direction, int orientation>
class InamuroNewtonRaphsonDynamics : public BasicDynamics<T,Lattice>
{
public:
    /// Constructor
    InamuroNewtonRaphsonDynamics(T omega_, Momenta<T,Lattice>& momenta_);
    /// Clone the object on its dynamic type.
    virtual InamuroNewtonRaphsonDynamics<T, Lattice, Dynamics, direction, orientation>* clone() const;
    /// Compute equilibrium distribution function
    virtual T computeEquilibrium(int iPop, T rho, const T u[Lattice<T>::d], T uSqr) const;
    /// Collision step
    virtual void collide(Cell<T,Lattice>& cell,
                         LatticeStatistics<T>& statistics);
    /// Collide with fixed velocity
    virtual void staticCollide(Cell<T,Lattice>& cell,
                               const T u[Lattice<T>::d],
                               LatticeStatistics<T>& statistics);
    /// Get local relaxation parameter of the dynamics
    virtual T getOmega() const;
    /// Set local relaxation parameter of the dynamics
    virtual void setOmega(T omega_);
    /// Get local value of any parameter
    virtual T getParameter(int whichParameter) const;
    /// Set local value of any parameter
    virtual void setParameter(int whichParameter, T value);

    void computeApproxMomentum(T approxMomentum[Lattice<T>::d],
        const Cell<T,Lattice> &cell,
        const T &rho, const T u[Lattice<T>::d], const T xi[Lattice<T>::d],
        const std::vector<int> knownIndexes,const std::vector<int> missingIndexes);

    /// compute the error (L^2 norm of (u-uApprox))
    T computeError(const T &rho,const T u[Lattice<T>::d], const T approxMomentum[Lattice<T>::d]);

    void computeGradGradError(T gradGradError[Lattice<T>::d][Lattice<T>::d],
        T gradError[Lattice<T>::d],
        const T &rho, const T u[Lattice<T>::d],const T xi[Lattice<T>::d],
        const T approxMomentum[Lattice<T>::d],
        const std::vector<int> missingIndexes);

    /// compute the new xi with the newton raphson algorithm
    bool newtonRaphson(T xi[Lattice<T>::d],
        const T gradError[Lattice<T>::d],
        const T gradGradError[Lattice<T>::d][Lattice<T>::d]);

    bool invert(const T a[2][2],T b[2][2]);

    bool invert(const T a[3][3],T b[3][3]);
private:
    Dynamics boundaryDynamics;
    T xi[Lattice<T>::d];
};

}

#endif