/*! \file
Copyright (c) 2003, The Regents of the University of California, through
Lawrence Berkeley National Laboratory (subject to receipt of any required
approvals from U.S. Dept. of Energy)

All rights reserved.

The source code is distributed under BSD license, see the file License.txt
at the top-level directory.
*/

/*! @file zutil.c
 * \brief Matrix utility functions
 *
 * <pre>
 * -- SuperLU routine (version 3.1) --
 * Univ. of California Berkeley, Xerox Palo Alto Research Center,
 * and Lawrence Berkeley National Lab.
 * August 1, 2008
 *
 * Copyright (c) 1994 by Xerox Corporation.  All rights reserved.
 *
 * THIS MATERIAL IS PROVIDED AS IS, WITH ABSOLUTELY NO WARRANTY
 * EXPRESSED OR IMPLIED.  ANY USE IS AT YOUR OWN RISK.
 *
 * Permission is hereby granted to use or copy this program for any
 * purpose, provided the above notices are retained on all copies.
 * Permission to modify the code and to distribute modified code is
 * granted, provided the above notices are retained, and a notice that
 * the code was modified is included with the above copyright notice.
 * </pre>
 */

#include <math.h>
#include "slu_zdefs.h"

void zCreate_CompCol_Matrix(SuperMatrix* A, int m, int n, int_t nnz, doublecomplex* nzval, int_t* rowind, int_t* colptr, Stype_t stype, Dtype_t dtype, Mtype_t mtype) {
    NCformat* Astore;

    A->Stype = stype;
    A->Dtype = dtype;
    A->Mtype = mtype;
    A->nrow = m;
    A->ncol = n;
    A->Store = (void*)SUPERLU_MALLOC(sizeof(NCformat));
    if(!(A->Store)) ABORT("SUPERLU_MALLOC fails for A->Store");
    Astore = A->Store;
    Astore->nnz = nnz;
    Astore->nzval = nzval;
    Astore->rowind = rowind;
    Astore->colptr = colptr;
}

void zCreate_CompRow_Matrix(SuperMatrix* A, int m, int n, int_t nnz, doublecomplex* nzval, int_t* colind, int_t* rowptr, Stype_t stype, Dtype_t dtype, Mtype_t mtype) {
    NRformat* Astore;

    A->Stype = stype;
    A->Dtype = dtype;
    A->Mtype = mtype;
    A->nrow = m;
    A->ncol = n;
    A->Store = (void*)SUPERLU_MALLOC(sizeof(NRformat));
    if(!(A->Store)) ABORT("SUPERLU_MALLOC fails for A->Store");
    Astore = A->Store;
    Astore->nnz = nnz;
    Astore->nzval = nzval;
    Astore->colind = colind;
    Astore->rowptr = rowptr;
}

/*! \brief Copy matrix A into matrix B. */
void zCopy_CompCol_Matrix(SuperMatrix* A, SuperMatrix* B) {
    NCformat *Astore, *Bstore;
    int ncol, nnz, i;

    B->Stype = A->Stype;
    B->Dtype = A->Dtype;
    B->Mtype = A->Mtype;
    B->nrow = A->nrow;
    ;
    B->ncol = ncol = A->ncol;
    Astore = (NCformat*)A->Store;
    Bstore = (NCformat*)B->Store;
    Bstore->nnz = nnz = Astore->nnz;
    for(i = 0; i < nnz; ++i)
        ((doublecomplex*)Bstore->nzval)[i] = ((doublecomplex*)Astore->nzval)[i];
    for(i = 0; i < nnz; ++i) Bstore->rowind[i] = Astore->rowind[i];
    for(i = 0; i <= ncol; ++i) Bstore->colptr[i] = Astore->colptr[i];
}

void zCreate_Dense_Matrix(SuperMatrix* X, int m, int n, doublecomplex* x, int ldx, Stype_t stype, Dtype_t dtype, Mtype_t mtype) {
    DNformat* Xstore;

    X->Stype = stype;
    X->Dtype = dtype;
    X->Mtype = mtype;
    X->nrow = m;
    X->ncol = n;
    X->Store = (void*)SUPERLU_MALLOC(sizeof(DNformat));
    if(!(X->Store)) ABORT("SUPERLU_MALLOC fails for X->Store");
    Xstore = (DNformat*)X->Store;
    Xstore->lda = ldx;
    Xstore->nzval = (doublecomplex*)x;
}

void zCopy_Dense_Matrix(int M, int N, doublecomplex* X, int ldx, doublecomplex* Y, int ldy) {
    /*! \brief Copies a two-dimensional matrix X to another matrix Y.
     */
    int i, j;

    for(j = 0; j < N; ++j)
        for(i = 0; i < M; ++i)
            Y[i + j * ldy] = X[i + j * ldx];
}

void zCreate_SuperNode_Matrix(SuperMatrix* L, int m, int n, int_t nnz, doublecomplex* nzval, int_t* nzval_colptr, int_t* rowind, int_t* rowind_colptr, int* col_to_sup, int* sup_to_col, Stype_t stype, Dtype_t dtype, Mtype_t mtype) {
    SCformat* Lstore;

    L->Stype = stype;
    L->Dtype = dtype;
    L->Mtype = mtype;
    L->nrow = m;
    L->ncol = n;
    L->Store = (void*)SUPERLU_MALLOC(sizeof(SCformat));
    if(!(L->Store)) ABORT("SUPERLU_MALLOC fails for L->Store");
    Lstore = L->Store;
    Lstore->nnz = nnz;
    Lstore->nsuper = col_to_sup[n];
    Lstore->nzval = nzval;
    Lstore->nzval_colptr = nzval_colptr;
    Lstore->rowind = rowind;
    Lstore->rowind_colptr = rowind_colptr;
    Lstore->col_to_sup = col_to_sup;
    Lstore->sup_to_col = sup_to_col;
}

/*! \brief Convert a row compressed storage into a column compressed storage.
 */
void zCompRow_to_CompCol(int m, int n, int_t nnz, doublecomplex* a, int_t* colind, int_t* rowptr, doublecomplex** at, int_t** rowind, int_t** colptr) {
    register int i, j, col, relpos;
    int_t* marker;

    /* Allocate storage for another copy of the matrix. */
    *at = (doublecomplex*)doublecomplexMalloc(nnz);
    *rowind = (int_t*)intMalloc(nnz);
    *colptr = (int_t*)intMalloc(n + 1);
    marker = (int_t*)intCalloc(n);

    /* Get counts of each column of A, and set up column pointers */
    for(i = 0; i < m; ++i)
        for(j = rowptr[i]; j < rowptr[i + 1]; ++j) ++marker[colind[j]];
    (*colptr)[0] = 0;
    for(j = 0; j < n; ++j) {
        (*colptr)[j + 1] = (*colptr)[j] + marker[j];
        marker[j] = (*colptr)[j];
    }

    /* Transfer the matrix into the compressed column storage. */
    for(i = 0; i < m; ++i) {
        for(j = rowptr[i]; j < rowptr[i + 1]; ++j) {
            col = colind[j];
            relpos = marker[col];
            (*rowind)[relpos] = i;
            (*at)[relpos] = a[j];
            ++marker[col];
        }
    }

    SUPERLU_FREE(marker);
}

void zPrint_CompCol_Matrix(char* what, SuperMatrix* A) {
    NCformat* Astore;
    register int_t i;
    register int n;
    double* dp;

    printf("\nCompCol matrix %s:\n", what);
    printf("Stype %d, Dtype %d, Mtype %d\n", A->Stype, A->Dtype, A->Mtype);
    n = A->ncol;
    Astore = (NCformat*)A->Store;
    dp = (double*)Astore->nzval;
    printf("nrow %d, ncol %d, nnz %ld\n", (int)A->nrow, (int)A->ncol, (long)Astore->nnz);
    printf("nzval: ");
    for(i = 0; i < 2 * Astore->colptr[n]; ++i) printf("%f  ", dp[i]);
    printf("\nrowind: ");
    for(i = 0; i < Astore->colptr[n]; ++i) printf("%ld  ", (long)Astore->rowind[i]);
    printf("\ncolptr: ");
    for(i = 0; i <= n; ++i) printf("%ld  ", (long)Astore->colptr[i]);
    printf("\n");
    fflush(stdout);
}

void zPrint_SuperNode_Matrix(char* what, SuperMatrix* A) {
    SCformat* Astore;
    register int_t i, j, k, c, d, n, nsup;
    double* dp;
    int *col_to_sup, *sup_to_col;
    int_t *rowind, *rowind_colptr;

    printf("\nSuperNode matrix %s:\n", what);
    printf("Stype %d, Dtype %d, Mtype %d\n", A->Stype, A->Dtype, A->Mtype);
    n = A->ncol;
    Astore = (SCformat*)A->Store;
    dp = (double*)Astore->nzval;
    col_to_sup = Astore->col_to_sup;
    sup_to_col = Astore->sup_to_col;
    rowind_colptr = Astore->rowind_colptr;
    rowind = Astore->rowind;
    printf("nrow %d, ncol %d, nnz %lld, nsuper %d\n", (int)A->nrow, (int)A->ncol, (long long)Astore->nnz, (int)Astore->nsuper);
    printf("nzval:\n");
    for(k = 0; k <= Astore->nsuper; ++k) {
        c = sup_to_col[k];
        nsup = sup_to_col[k + 1] - c;
        for(j = c; j < c + nsup; ++j) {
            d = Astore->nzval_colptr[j];
            for(i = rowind_colptr[c]; i < rowind_colptr[c + 1]; ++i) {
                printf("%d\t%d\t%e\t%e\n", (int)rowind[i], (int)j, dp[d], dp[d + 1]);
                d += 2;
            }
        }
    }
#if 0
    for (i = 0; i < 2*Astore->nzval_colptr[n]; ++i) printf("%f  ", dp[i]);
#endif
    printf("\nnzval_colptr: ");
    for(i = 0; i <= n; ++i) printf("%lld  ", (long long)Astore->nzval_colptr[i]);
    printf("\nrowind: ");
    for(i = 0; i < Astore->rowind_colptr[n]; ++i)
        printf("%lld  ", (long long)Astore->rowind[i]);
    printf("\nrowind_colptr: ");
    for(i = 0; i <= n; ++i) printf("%lld  ", (long long)Astore->rowind_colptr[i]);
    printf("\ncol_to_sup: ");
    for(i = 0; i < n; ++i) printf("%d  ", col_to_sup[i]);
    printf("\nsup_to_col: ");
    for(i = 0; i <= Astore->nsuper + 1; ++i)
        printf("%d  ", sup_to_col[i]);
    printf("\n");
    fflush(stdout);
}

void zPrint_Dense_Matrix(char* what, SuperMatrix* A) {
    DNformat* Astore = (DNformat*)A->Store;
    register int i, j, lda = Astore->lda;
    double* dp;

    printf("\nDense matrix %s:\n", what);
    printf("Stype %d, Dtype %d, Mtype %d\n", A->Stype, A->Dtype, A->Mtype);
    dp = (double*)Astore->nzval;
    printf("nrow %d, ncol %d, lda %d\n", (int)A->nrow, (int)A->ncol, lda);
    printf("\nnzval: ");
    for(j = 0; j < A->ncol; ++j) {
        for(i = 0; i < 2 * A->nrow; ++i) printf("%f  ", dp[i + j * 2 * lda]);
        printf("\n");
    }
    printf("\n");
    fflush(stdout);
}

/*! \brief Diagnostic print of column "jcol" in the U/L factor.
 */
void zprint_lu_col(char* msg, int jcol, int pivrow, int_t* xprune, GlobalLU_t* Glu) {
    int_t i, k;
    int *xsup, *supno, fsupc;
    int_t *xlsub, *lsub;
    doublecomplex* lusup;
    int_t* xlusup;
    doublecomplex* ucol;
    int_t *usub, *xusub;

    xsup = Glu->xsup;
    supno = Glu->supno;
    lsub = Glu->lsub;
    xlsub = Glu->xlsub;
    lusup = (doublecomplex*)Glu->lusup;
    xlusup = Glu->xlusup;
    ucol = (doublecomplex*)Glu->ucol;
    usub = Glu->usub;
    xusub = Glu->xusub;

    printf("%s", msg);
    printf("col %d: pivrow %d, supno %d, xprune %lld\n", jcol, pivrow, supno[jcol], (long long)xprune[jcol]);

    printf("\tU-col:\n");
    for(i = xusub[jcol]; i < xusub[jcol + 1]; i++)
        printf("\t%d%10.4f, %10.4f\n", (int)usub[i], ucol[i].r, ucol[i].i);
    printf("\tL-col in rectangular snode:\n");
    fsupc = xsup[supno[jcol]]; /* first col of the snode */
    i = xlsub[fsupc];
    k = xlusup[jcol];
    while(i < xlsub[fsupc + 1] && k < xlusup[jcol + 1]) {
        printf("\t%d\t%10.4f, %10.4f\n", (int)lsub[i], lusup[k].r, lusup[k].i);
        i++;
        k++;
    }
    fflush(stdout);
}

/*! \brief Check whether tempv[] == 0. This should be true before and after calling any numeric routines, i.e., "panel_bmod" and "column_bmod".
 */
void zcheck_tempv(int n, doublecomplex* tempv) {
    int i;

    for(i = 0; i < n; i++) {
        if((tempv[i].r != 0.0) || (tempv[i].i != 0.0)) {
            fprintf(stderr, "tempv[%d] = {%f, %f}\n", i, tempv[i].r, tempv[i].i);
            ABORT("zcheck_tempv");
        }
    }
}

void zGenXtrue(int n, int nrhs, doublecomplex* x, int ldx) {
    int i, j;
    for(j = 0; j < nrhs; ++j)
        for(i = 0; i < n; ++i) {
            x[i + j * ldx].r = 1.0;
            x[i + j * ldx].i = 0.0;
        }
}

/*! \brief Let rhs[i] = sum of i-th row of A, so the solution vector is all 1's
 */
void zFillRHS(trans_t trans, int nrhs, doublecomplex* x, int ldx, SuperMatrix* A, SuperMatrix* B) {
    DNformat* Bstore;
    doublecomplex* rhs;
    doublecomplex one = {1.0, 0.0};
    doublecomplex zero = {0.0, 0.0};
    int ldc;
    char transc[1];

    // Astore = A->Store;
    // Aval   = (doublecomplex *) Astore->nzval;
    Bstore = B->Store;
    rhs = Bstore->nzval;
    ldc = Bstore->lda;

    if(trans == NOTRANS) *(unsigned char*)transc = 'N';
    else *(unsigned char*)transc = 'T';

    sp_zgemm(transc, "N", A->nrow, nrhs, A->ncol, one, A, x, ldx, zero, rhs, ldc);
}

/*! \brief Fills a doublecomplex precision array with a given value.
 */
void zfill(doublecomplex* a, int alen, doublecomplex dval) {
    register int i;
    for(i = 0; i < alen; i++) a[i] = dval;
}

/*! \brief Check the inf-norm of the error vector
 */
void zinf_norm_error(int nrhs, SuperMatrix* X, doublecomplex* xtrue) {
    DNformat* Xstore;
    double err, xnorm;
    doublecomplex *Xmat, *soln_work;
    doublecomplex temp;
    int i, j;

    Xstore = X->Store;
    Xmat = Xstore->nzval;

    for(j = 0; j < nrhs; j++) {
        soln_work = &Xmat[j * Xstore->lda];
        err = xnorm = 0.0;
        for(i = 0; i < X->nrow; i++) {
            z_sub(&temp, &soln_work[i], &xtrue[i]);
            err = SUPERLU_MAX(err, z_abs(&temp));
            xnorm = SUPERLU_MAX(xnorm, z_abs(&soln_work[i]));
        }
        err = err / xnorm;
        printf("||X - Xtrue||/||X|| = %e\n", err);
    }
}

/*! \brief Print performance of the code. */
void zPrintPerf(SuperMatrix* L, SuperMatrix* U, mem_usage_t* mem_usage, double rpg, double rcond, double* ferr, const double* berr, const char* equed, SuperLUStat_t* stat) {
    SCformat* Lstore;
    NCformat* Ustore;
    double* utime;
    flops_t* ops;

    utime = stat->utime;
    ops = stat->ops;

    if(utime[FACT] != 0.)
        printf("Factor flops = %e\tMflops = %8.2f\n", ops[FACT], ops[FACT] * 1e-6 / utime[FACT]);
    printf("Identify relaxed snodes	= %8.2f\n", utime[RELAX]);
    if(utime[SOLVE] != 0.)
        printf("Solve flops = %.0f, Mflops = %8.2f\n", ops[SOLVE], ops[SOLVE] * 1e-6 / utime[SOLVE]);

    Lstore = (SCformat*)L->Store;
    Ustore = (NCformat*)U->Store;
    printf("\tNo of nonzeros in factor L = %lld\n", (long long)Lstore->nnz);
    printf("\tNo of nonzeros in factor U = %lld\n", (long long)Ustore->nnz);
    printf("\tNo of nonzeros in L+U = %lld\n", (long long)Lstore->nnz + Ustore->nnz);

    printf("L\\U MB %.3f\ttotal MB needed %.3f\n", mem_usage->for_lu / 1e6, mem_usage->total_needed / 1e6);
    printf("Number of memory expansions: %d\n", stat->expansions);

    printf("\tFactor\tMflops\tSolve\tMflops\tEtree\tEquil\tRcond\tRefine\n");
    printf("PERF:%8.2f%8.2f%8.2f%8.2f%8.2f%8.2f%8.2f%8.2f\n", utime[FACT], ops[FACT] * 1e-6 / utime[FACT], utime[SOLVE], ops[SOLVE] * 1e-6 / utime[SOLVE], utime[ETREE], utime[EQUIL], utime[RCOND], utime[REFINE]);

    printf("\tRpg\t\tRcond\t\tFerr\t\tBerr\t\tEquil?\n");
    printf("NUM:\t%e\t%e\t%e\t%e\t%s\n", rpg, rcond, ferr[0], berr[0], equed);
}

int print_doublecomplex_vec(const char* what, int n, const doublecomplex* vec) {
    int i;
    printf("%s: n %d\n", what, n);
    for(i = 0; i < n; ++i) printf("%d\t%f%f\n", i, vec[i].r, vec[i].i);
    return 0;
}
