/*--------------------------------------------*/
void COO2CSR(coo_t *coo, csr_t *csr) {
int n,nnz;
/*------- Allocate CSR */
n = coo->n;
nnz = coo->nnz;
malloc_csr(n, nnz, csr);
/*-------- COO -> CSR */
coocsr_(&coo->n, &coo->nnz, coo->val, coo->ir,
coo->jc, csr->a, csr->ja, csr->ia);
}
void read_coo(char *data_file, int *proc_config,
int *N_global, int *n_nonzeros,
double **val, int **bindx,
double **x, double **b, double **xexact)
#undef DEBUG
/* read ASCII data file:
line 1: N_global, number of entries (%d,%d)
line 2-...: i,j,real (%d, %d, %f)
*/
{
FILE *data ;
int i, n_entries, N_columns;
int ii, jj ;
int kk = 0;
int max_ii = 0, max_jj = 0;
int ione = 1;
double value;
double *cnt;
int *pntr, *indx1, *pntr1;
double *val1;
int MAXBLOCKSIZE = 25;
if(proc_config[AZ_node] == 0)
{
data = fopen(data_file,"r") ;
fscanf(data, "%d %d %d", N_global, &N_columns, &n_entries) ;
if (N_columns != *N_global)
perror("Matrix dimensions must be the same");
printf("Reading from file: %s\n",data_file);
printf("Number of equations = %d\n",*N_global);
printf("Number of entries = %d\n",n_entries);
*bindx = (int *) calloc(n_entries+1,sizeof(int)) ;
*val = (double *) calloc(n_entries+1,sizeof(double)) ;
pntr1 = (int *) calloc(n_entries+1,sizeof(int)) ;
indx1 = (int *) calloc(n_entries+1,sizeof(int)) ;
val1 = (double *) calloc(n_entries+1,sizeof(double)) ;
pntr = (int *) calloc(n_entries+1,sizeof(int)) ;
if ((pntr) == NULL)
perror("Error: Not enough space to create matrix");
while(!feof(data))
{
fscanf(data, "%d %d %lf", &ii, &jj, &value) ;
max_ii = AZ_MAX(max_ii,ii);
max_jj = AZ_MAX(max_jj,jj);
#ifdef DEBUG
printf("Entry %d, %d = %lf.\n",ii,jj,value);
#endif
(*bindx)[kk] = ii;
pntr[kk] = jj;
(*val)[kk] = value;
kk++;
}
*n_nonzeros = kk-1;
*N_global = max_ii;
if (max_ii != max_jj) perror("Error: Number of rows and columns not equal");
printf("Number of nonzeros = %d\n",*n_nonzeros);
/* Convert real part in the following way:
- Convert COO to CSR
- CSR to CSC
- CSC to CSR (columns are now in ascending order)
- CSR to MSR
*/
coocsr_(N_global,n_nonzeros, *val, *bindx, pntr, val1, indx1, pntr1);
csrcsc_(N_global,&ione,&ione,
val1,indx1,pntr1,
*val,*bindx,pntr);
csrcsc_(N_global,&ione,&ione,
*val,*bindx,pntr,
val1,indx1,pntr1);
csrmsr_(N_global,val1,indx1,pntr1,
*val,*bindx,
*val,*bindx);
/* Finally, convert bindx vectors to zero base */
for (i=0;i<*n_nonzeros+1;i++)
(*bindx)[i] -= 1;
*b = (double *) calloc((*N_global)*MAXBLOCKSIZE,sizeof(double)) ;
*x = (double *) calloc((*N_global)*MAXBLOCKSIZE,sizeof(double)) ;
if ((*x) == NULL)
perror("Error: Not enough space to create matrix");
/* Set RHS to a random vector, initial guess to zero */
for (i=0;i<*N_global;i++)
{
(*b)[i] = drand48();
(*x)[i] = 0.0;
}
}
/* Release unneeded space */
free((void *) pntr);
free((void *) val1);
free((void *) indx1);
free((void *) pntr1);
/* end read_coo */
}
