/*! @file slu_zdefs.h

 * \brief Header file for real operations

 * 

 * 
 

 * -- SuperLU routine (version 4.1) --

 * Univ. of California Berkeley, Xerox Palo Alto Research Center,

 * and Lawrence Berkeley National Lab.

 * November, 2010

 * 

 * Global data structures used in LU factorization -

 * 

 *   nsuper: #supernodes = nsuper + 1, numbered [0, nsuper].

 *   (xsup,supno): supno[i] is the supernode no to which i belongs;

 *	xsup(s) points to the beginning of the s-th supernode.

 *	e.g.   supno 0 1 2 2 3 3 3 4 4 4 4 4   (n=12)

 *	        xsup 0 1 2 4 7 12

 *	Note: dfs will be performed on supernode rep. relative to the new 

 *	      row pivoting ordering

 *

 *   (xlsub,lsub): lsub[*] contains the compressed subscript of

 *	rectangular supernodes; xlsub[j] points to the starting

 *	location of the j-th column in lsub[*]. Note that xlsub 

 *	is indexed by column.

 *	Storage: original row subscripts

 *

 *      During the course of sparse LU factorization, we also use

 *	(xlsub,lsub) for the purpose of symmetric pruning. For each

 *	supernode {s,s+1,...,t=s+r} with first column s and last

 *	column t, the subscript set

 *		lsub[j], j=xlsub[s], .., xlsub[s+1]-1

 *	is the structure of column s (i.e. structure of this supernode).

 *	It is used for the storage of numerical values.

 *	Furthermore,

 *		lsub[j], j=xlsub[t], .., xlsub[t+1]-1

 *	is the structure of the last column t of this supernode.

 *	It is for the purpose of symmetric pruning. Therefore, the

 *	structural subscripts can be rearranged without making physical

 *	interchanges among the numerical values.

 *

 *	However, if the supernode has only one column, then we

 *	only keep one set of subscripts. For any subscript interchange

 *	performed, similar interchange must be done on the numerical

 *	values.

 *

 *	The last column structures (for pruning) will be removed

 *	after the numercial LU factorization phase.

 *

 *   (xlusup,lusup): lusup[*] contains the numerical values of the

 *	rectangular supernodes; xlusup[j] points to the starting

 *	location of the j-th column in storage vector lusup[*]

 *	Note: xlusup is indexed by column.

 *	Each rectangular supernode is stored by column-major

 *	scheme, consistent with Fortran 2-dim array storage.

 *

 *   (xusub,ucol,usub): ucol[*] stores the numerical values of

 *	U-columns outside the rectangular supernodes. The row

 *	subscript of nonzero ucol[k] is stored in usub[k].

 *	xusub[i] points to the starting location of column i in ucol.

 *	Storage: new row subscripts; that is subscripts of PA.

 * 

 */

#ifndef __SUPERLU_zSP_DEFS /* allow multiple inclusions */

#define __SUPERLU_zSP_DEFS

/*

 * File name:		zsp_defs.h

 * Purpose:             Sparse matrix types and function prototypes

 * History:

 */

#ifdef _CRAY

#include <fortran.h></fortran.h>

#include <string.h></string.h>

#endif

/* Define my integer type int_t */

typedef int int_t; /* default */

#include <math.h></math.h>

#include <limits.h></limits.h>

#include "slu_Cnames.h"

#include "supermatrix.h"

#include "slu_util.h"

#include "slu_dcomplex.h"

typedef struct {

    int     *xsup;    /* supernode and column mapping */

    int     *supno;   

    int     *lsub;    /* compressed L subscripts */

    int	    *xlsub;

    doublecomplex  *lusup;   /* L supernodes */

    int     *xlusup;

    doublecomplex  *ucol;    /* U columns */

    int     *usub;

    int	    *xusub;

    int     nzlmax;   /* current max size of lsub */

    int     nzumax;   /*    "    "    "      ucol */

    int     nzlumax;  /*    "    "    "     lusup */

    int     n;        /* number of columns in the matrix */

    LU_space_t MemModel; /* 0 - system malloc'd; 1 - user provided */

    int     num_expansions;

    ExpHeader *expanders; /* Array of pointers to 4 types of memory */

    LU_stack_t stack;     /* use user supplied memory */

} GlobalLU_t;

/* -------- Prototypes -------- */

#ifdef __cplusplus

extern "C" {

#endif

/*! \brief Driver routines */

extern void

zgssv(superlu_options_t *, SuperMatrix *, int *, int *, SuperMatrix *,

      SuperMatrix *, SuperMatrix *, SuperLUStat_t *, int *);

extern void

zgssvx(superlu_options_t *, SuperMatrix *, int *, int *, int *,

       char *, double *, double *, SuperMatrix *, SuperMatrix *,

       void *, int, SuperMatrix *, SuperMatrix *,

       double *, double *, double *, double *,

       mem_usage_t *, SuperLUStat_t *, int *);

    /* ILU */

extern void

zgsisv(superlu_options_t *, SuperMatrix *, int *, int *, SuperMatrix *,

      SuperMatrix *, SuperMatrix *, SuperLUStat_t *, int *);

extern void

zgsisx(superlu_options_t *, SuperMatrix *, int *, int *, int *,

       char *, double *, double *, SuperMatrix *, SuperMatrix *,

       void *, int, SuperMatrix *, SuperMatrix *, double *, double *,

       mem_usage_t *, SuperLUStat_t *, int *);

/*! \brief Supernodal LU factor related */

extern void

zCreate_CompCol_Matrix(SuperMatrix *, int, int, int, doublecomplex *,

		       int *, int *, Stype_t, Dtype_t, Mtype_t);

extern void

zCreate_CompRow_Matrix(SuperMatrix *, int, int, int, doublecomplex *,

		       int *, int *, Stype_t, Dtype_t, Mtype_t);

extern void

zCopy_CompCol_Matrix(SuperMatrix *, SuperMatrix *);

extern void

zCreate_Dense_Matrix(SuperMatrix *, int, int, doublecomplex *, int,

		     Stype_t, Dtype_t, Mtype_t);

extern void

zCreate_SuperNode_Matrix(SuperMatrix *, int, int, int, doublecomplex *, 

		         int *, int *, int *, int *, int *,

			 Stype_t, Dtype_t, Mtype_t);

extern void

zCopy_Dense_Matrix(int, int, doublecomplex *, int, doublecomplex *, int);

extern void    countnz (const int, int *, int *, int *, GlobalLU_t *);

extern void    ilu_countnz (const int, int *, int *, GlobalLU_t *);

extern void    fixupL (const int, const int *, GlobalLU_t *);

extern void    zallocateA (int, int, doublecomplex **, int **, int **);

extern void    zgstrf (superlu_options_t*, SuperMatrix*,

                       int, int, int*, void *, int, int *, int *, 

                       SuperMatrix *, SuperMatrix *, SuperLUStat_t*, int *);

extern int     zsnode_dfs (const int, const int, const int *, const int *,

			     const int *, int *, int *, GlobalLU_t *);

extern int     zsnode_bmod (const int, const int, const int, doublecomplex *,

                              doublecomplex *, GlobalLU_t *, SuperLUStat_t*);

extern void    zpanel_dfs (const int, const int, const int, SuperMatrix *,

			   int *, int *, doublecomplex *, int *, int *, int *,

			   int *, int *, int *, int *, GlobalLU_t *);

extern void    zpanel_bmod (const int, const int, const int, const int,

                           doublecomplex *, doublecomplex *, int *, int *,

			   GlobalLU_t *, SuperLUStat_t*);

extern int     zcolumn_dfs (const int, const int, int *, int *, int *, int *,

			   int *, int *, int *, int *, int *, GlobalLU_t *);

extern int     zcolumn_bmod (const int, const int, doublecomplex *,

			   doublecomplex *, int *, int *, int,

                           GlobalLU_t *, SuperLUStat_t*);

extern int     zcopy_to_ucol (int, int, int *, int *, int *,

                              doublecomplex *, GlobalLU_t *);         

extern int     zpivotL (const int, const double, int *, int *, 

                         int *, int *, int *, GlobalLU_t *, SuperLUStat_t*);

extern void    zpruneL (const int, const int *, const int, const int,

			  const int *, const int *, int *, GlobalLU_t *);

extern void    zreadmt (int *, int *, int *, doublecomplex **, int **, int **);

extern void    zGenXtrue (int, int, doublecomplex *, int);

extern void    zFillRHS (trans_t, int, doublecomplex *, int, SuperMatrix *,

			  SuperMatrix *);

extern void    zgstrs (trans_t, SuperMatrix *, SuperMatrix *, int *, int *,

                        SuperMatrix *, SuperLUStat_t*, int *);

/* ILU */

extern void    zgsitrf (superlu_options_t*, SuperMatrix*, int, int, int*,

		        void *, int, int *, int *, SuperMatrix *, SuperMatrix *,

                        SuperLUStat_t*, int *);

extern int     zldperm(int, int, int, int [], int [], doublecomplex [],

                        int [],	double [], double []);

extern int     ilu_zsnode_dfs (const int, const int, const int *, const int *,

			       const int *, int *, GlobalLU_t *);

extern void    ilu_zpanel_dfs (const int, const int, const int, SuperMatrix *,

			       int *, int *, doublecomplex *, double *, int *, int *,

			       int *, int *, int *, int *, GlobalLU_t *);

extern int     ilu_zcolumn_dfs (const int, const int, int *, int *, int *,

				int *, int *, int *, int *, int *,

				GlobalLU_t *);

extern int     ilu_zcopy_to_ucol (int, int, int *, int *, int *,

                                  doublecomplex *, int, milu_t, double, int,

                                  doublecomplex *, int *, GlobalLU_t *, double *);

extern int     ilu_zpivotL (const int, const double, int *, int *, int, int *,

			    int *, int *, int *, double, milu_t,

                            doublecomplex, GlobalLU_t *, SuperLUStat_t*);

extern int     ilu_zdrop_row (superlu_options_t *, int, int, double,

                              int, int *, double *, GlobalLU_t *, 

                              double *, double *, int);

/*! \brief Driver related */

extern void    zgsequ (SuperMatrix *, double *, double *, double *,

			double *, double *, int *);

extern void    zlaqgs (SuperMatrix *, double *, double *, double,

                        double, double, char *);

extern void    zgscon (char *, SuperMatrix *, SuperMatrix *, 

		         double, double *, SuperLUStat_t*, int *);

extern double   zPivotGrowth(int, SuperMatrix *, int *, 

                            SuperMatrix *, SuperMatrix *);

extern void    zgsrfs (trans_t, SuperMatrix *, SuperMatrix *,

                       SuperMatrix *, int *, int *, char *, double *, 

                       double *, SuperMatrix *, SuperMatrix *,

                       double *, double *, SuperLUStat_t*, int *);

extern int     sp_ztrsv (char *, char *, char *, SuperMatrix *,

			SuperMatrix *, doublecomplex *, SuperLUStat_t*, int *);

extern int     sp_zgemv (char *, doublecomplex, SuperMatrix *, doublecomplex *,

			int, doublecomplex, doublecomplex *, int);

extern int     sp_zgemm (char *, char *, int, int, int, doublecomplex,

			SuperMatrix *, doublecomplex *, int, doublecomplex, 

			doublecomplex *, int);

extern         double dlamch_(char *);

/*! \brief Memory-related */

extern int     zLUMemInit (fact_t, void *, int, int, int, int, int,

                            double, SuperMatrix *, SuperMatrix *,

                            GlobalLU_t *, int **, doublecomplex **);

extern void    zSetRWork (int, int, doublecomplex *, doublecomplex **, doublecomplex **);

extern void    zLUWorkFree (int *, doublecomplex *, GlobalLU_t *);

extern int     zLUMemXpand (int, int, MemType, int *, GlobalLU_t *);

extern doublecomplex  *doublecomplexMalloc(int);

extern doublecomplex  *doublecomplexCalloc(int);

extern double  *doubleMalloc(int);

extern double  *doubleCalloc(int);

extern int     zmemory_usage(const int, const int, const int, const int);

extern int     zQuerySpace (SuperMatrix *, SuperMatrix *, mem_usage_t *);

extern int     ilu_zQuerySpace (SuperMatrix *, SuperMatrix *, mem_usage_t *);

/*! \brief Auxiliary routines */

extern void    zreadhb(int *, int *, int *, doublecomplex **, int **, int **);

extern void    zreadrb(int *, int *, int *, doublecomplex **, int **, int **);

extern void    zreadtriple(int *, int *, int *, doublecomplex **, int **, int **);

extern void    zCompRow_to_CompCol(int, int, int, doublecomplex*, int*, int*,

		                   doublecomplex **, int **, int **);

extern void    zfill (doublecomplex *, int, doublecomplex);

extern void    zinf_norm_error (int, SuperMatrix *, doublecomplex *);

extern void    PrintPerf (SuperMatrix *, SuperMatrix *, mem_usage_t *,

			 doublecomplex, doublecomplex, doublecomplex *, doublecomplex *, char *);

extern double  dqselect(int, double *, int);

/*! \brief Routines for debugging */

extern void    zPrint_CompCol_Matrix(char *, SuperMatrix *);

extern void    zPrint_SuperNode_Matrix(char *, SuperMatrix *);

extern void    zPrint_Dense_Matrix(char *, SuperMatrix *);

extern void    zprint_lu_col(char *, int, int, int *, GlobalLU_t *);

extern int     print_double_vec(char *, int, double *);

extern void    check_tempv(int, doublecomplex *);

#ifdef __cplusplus

  }

#endif

#endif /* __SUPERLU_zSP_DEFS */