/*****************************************************************************
*
* MODULE:       Grass PDE Numerical Library
* AUTHOR(S):    Soeren Gebbert, Berlin (GER) Dec 2006
* 		soerengebbert <at> gmx <dot> de
*               
* PURPOSE:      solute transport in porous media
* 		part of the gpde library
*
* COPYRIGHT:    (C) 2000 by the GRASS Development Team
*
*               This program is free software under the GNU General Public
*               License (>=v2). Read the file COPYING that comes with GRASS
*               for details.
*
*****************************************************************************/

#include "N_pde.h"

#ifndef _N_SOLUTE_TRANSPORT_H_
#define _N_SOLUTE_TRANSPORT_H_


typedef struct
{
    N_array_3d *c;		/*concentration */
    N_array_3d *c_start;	/*concentration at start */
    N_array_3d *diff_x;		/*x part of the diffusion tensor */
    N_array_3d *diff_y;		/*y part of the diffusion tensor */
    N_array_3d *diff_z;		/*z part of the diffusion tensor */
    N_array_3d *nf;		/*effective porosity */
    N_array_3d *cs;		/*concentration sources and sinks */
    N_array_3d *q;		/*well sources and sinks */
    N_array_3d *R;		/*retardation */
    N_array_3d *cin;		/*concentration input from wells */

    N_gradient_field_3d *grad;	/*velocity field */

    N_array_3d *status;		/*active/inactive/dirichlet cell status */

    N_array_3d *disp_xx;	/*x part of the dispersivity tensor */
    N_array_3d *disp_yy;	/*x part of the dispersivity tensor */
    N_array_3d *disp_zz;	/*x part of the dispersivity tensor */
    N_array_3d *disp_xy;	/*xy part of the dispersivity tensor */
    N_array_3d *disp_xz;	/*xz part of the dispersivity tensor */
    N_array_3d *disp_yz;	/*yz part of the dispersivity tensor */

    double dt;			/*calculation time */
    double al, at;		/*dispersivity length longditudinal and transversal */
    int stab;			/*stabilization criteria */

} N_solute_transport_data3d;


typedef struct
{
    N_array_2d *c;		/*concentration */
    N_array_2d *c_start;	/*concentration at start */
    N_array_2d *diff_x;		/*x part of the diffusion tensor */
    N_array_2d *diff_y;		/*y part of the diffusion tensor */
    N_array_2d *nf;		/*effective porosity */
    N_array_2d *cs;		/*concentration sources and sinks */
    N_array_2d *q;		/*well sources and sinks */
    N_array_2d *R;		/*retardation */
    N_array_2d *cin;		/*concentration  */

    N_gradient_field_2d *grad;	/*velocity field */

    N_array_2d *status;		/*active/inactive/dirichlet cell status */
    N_array_2d *top;		/* top surface of the aquifer */
    N_array_2d *bottom;		/* bottom surface of the aquifer */

    N_array_2d *disp_xx;	/*x part of the dispersivity tensor */
    N_array_2d *disp_yy;	/*x part of the dispersivity tensor */
    N_array_2d *disp_xy;	/*xy part of the dispersivity tensor */

    double dt;			/*calculation time */
    double al, at;		/*dispersivity length longditudinal and transversal */
    int stab;			/*stabilization criteria */

} N_solute_transport_data2d;


extern N_data_star *N_callback_solute_transport_3d(void *solutedata,
						   N_geom_data * geom,
						   int col, int row,
						   int depth);
extern N_data_star *N_callback_solute_transport_2d(void *solutedata,
						   N_geom_data * geom,
						   int col, int row);
extern N_solute_transport_data3d *N_alloc_solute_transport_data3d(int cols,
								  int rows,
								  int depths);
extern N_solute_transport_data2d *N_alloc_solute_transport_data2d(int cols,
								  int rows);
extern void N_free_solute_transport_data3d(N_solute_transport_data3d * data);
extern void N_free_solute_transport_data2d(N_solute_transport_data2d * data);

/*compute the dispersivity tensor */
extern void N_calc_solute_transport_disptensor_2d(N_solute_transport_data2d *
						  data);
extern void N_calc_solute_transport_disptensor_3d(N_solute_transport_data3d *
						  data);
extern void N_calc_solute_transport_transmission_2d(N_solute_transport_data2d
						    * data);
extern void N_calc_solute_transport_transmission_3d(N_solute_transport_data3d
						    * data);
#endif