BlaSparseCOO.c

Go to the documentation of this file.

 
#include <math.h>
#include <time.h>
 
#ifdef _OPENMP
#include <omp.h>
#endif
 
#include "fasp.h"
#include "fasp_functs.h"
 
/*---------------------------------*/
/*--      Public Functions       --*/
/*---------------------------------*/
 
dCOOmat fasp_dcoo_create (const INT  m,
                          const INT  n,
                          const INT  nnz)
{
    dCOOmat A;
    
    A.rowind = (INT *)fasp_mem_calloc(nnz, sizeof(INT));
    A.colind = (INT *)fasp_mem_calloc(nnz, sizeof(INT));
    A.val    = (REAL *)fasp_mem_calloc(nnz, sizeof(REAL));
    
    A.row = m; A.col = n; A.nnz = nnz;
    
    return A;
}
 
void fasp_dcoo_alloc (const INT  m,
                      const INT  n,
                      const INT  nnz,
                      dCOOmat   *A)
{
    
    if ( nnz > 0 ) {
        A->rowind = (INT *)fasp_mem_calloc(nnz, sizeof(INT));
        A->colind = (INT *)fasp_mem_calloc(nnz, sizeof(INT));
        A->val    = (REAL*)fasp_mem_calloc(nnz,sizeof(REAL));
    }
    else {
        A->rowind = NULL;
        A->colind = NULL;
        A->val    = NULL;
    }
    
    A->row = m; A->col = n; A->nnz = nnz;
    
    return;
}
 
void fasp_dcoo_free (dCOOmat *A)
{
    if (A==NULL) return;
    
    fasp_mem_free(A->rowind); A->rowind = NULL;
    fasp_mem_free(A->colind); A->colind = NULL;
    fasp_mem_free(A->val);    A->val    = NULL;
}
 
void fasp_dcoo_shift (dCOOmat   *A,
                      const INT  offset)
{
    const INT nnz = A->nnz;
    INT       i, *ai = A->rowind, *aj = A->colind;
    
    // Variables for OpenMP
    SHORT nthreads = 1, use_openmp = FALSE;
    INT myid, mybegin, myend;
    
#ifdef _OPENMP
    if (nnz > OPENMP_HOLDS) {
        use_openmp = TRUE;
        nthreads = fasp_get_num_threads();
    }
#endif
        
    if (use_openmp) {
#ifdef _OPENMP
#pragma omp parallel for private(myid, i, mybegin, myend)
#endif
        for (myid=0; myid<nthreads; myid++) {
            fasp_get_start_end(myid, nthreads, nnz, &mybegin, &myend);
            for (i=mybegin; i<myend; ++i) {
                ai[i]+=offset; aj[i]+=offset;
            }
        }
    }
    else {
        for (i=0;i<nnz;++i) {
            ai[i]+=offset; aj[i]+=offset;
        }
    }
}
 
/*---------------------------------*/
/*--        End of File          --*/
/*---------------------------------*/