/*==============================================================================

   Copyright (C) 1998 - 2002 Rachid Touzani

   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; Version 2 of the License.

   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., 59 Temple Place - Suite 330
   Boston, MA  02111-1307, USA

  ==============================================================================*/

#ifndef __PNS2_H
#define __PNS2_H

#include "OFELI.h"

using namespace OFELI;
void ReadParam(char *file, char *proj, double &max_time, double &deltat,
               int &verbose, int &output_flag, int &plot_file, int &save_flag,
               int &init_flag, int &bc_flag, int &body_flag, int &bound_flag, 
               double &visc, double &dens, double &tolerance);
void GetArgs(int argc, char **argv, char *proj, double &max_time, double &deltat, 
             int &verbose, int &output_flag, int &plot_file, int &save_flag, 
             int &init_flag, int &bc_flag, int &body_flag, int &bound_flag);
void InitialV(double deltat, Mesh &ms, SpMatrix<> &A, const Vect<> &M,
              const ILUPrec< double,SpMatrix<> > &P, Vect<> &b, Vect<> &u, 
              Vect<> &ub, Vect<> &p, Vect<> &q);
void VMatrix(double deltat, double dens, double visc, const Mesh &ms, SpMatrix<> &Av);
void Momentum(int step, double time, double deltat, Mesh &ms, VDF &vdf,
              Vect<> &u, Vect<> &v, Vect<> &ub, Vect<> &c, const SpMatrix<> &Av, 
              const ILUPrec< double,SpMatrix<> > &Pv, const Vect<> &p,
              double dens, double visc, double &cfl, int verbose);
void PMatrix(double dt, const Mesh &ms, SpMatrix<> &A, Vect<> &M);
int CalPres(double deltat, const Mesh &ms, const SpMatrix<> &A,
            const ILUPrec< double,SpMatrix<> > &P, const Vect<> &u, const Vect<> &ub, 
            Vect<> &p, Vect<> &q);
void UpdateV(const Mesh &ms, double dt, const Vect<> &M, const Vect<> &q, Vect<> &u, Vect<> &ub);
void StreamF(const Mesh &ms, const Vect<> &u, NodeVect<> &psi);
void Vorticity(double q, const Mesh &ms, const Vect<> &u, const Vect<> &M, NodeVect<> &w);

#endif