1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
|
/*
Copyright (C) 2008-2016 Michele Martone
This file is part of librsb.
librsb is free software; you can redistribute it and/or modify it
under the terms of the GNU Lesser General Public License as published
by the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
librsb is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
License for more details.
You should have received a copy of the GNU Lesser General Public
License along with librsb; see the file COPYING.
If not, see <http://www.gnu.org/licenses/>.
*/
/* @cond INNERDOC */
/*
* unsorted random matrix market generator
* */
/**
@file
@author Michele Martone
@brief
A toy program generating sparse matrices.
*/
//#include <stdlib.h> /* bsearch, calloc, malloc */
//#include <stdio.h> /* printf */
//#include "rsb.h"
#include "rsb_common.h"
#include "rsb_internals.h"
//#define RSB_CONDITIONAL_FREE(p) {if((p))free((p));(p)=NULL;}
int g_allow_duplicates;
static rsb_err_t gen_mm(rsb_coo_idx_t r, rsb_coo_idx_t c, rsb_nnz_idx_t nnz)
{
/*
* Since we could modify our implementation of bitmap data structures over time,
* there will always be a testing need, so we place here this routine.
* */
rsb_coo_idx_t * IA=NULL,*JA=NULL;
float *VA;
rsb_nnz_idx_t i,k;
rsb_nnz_idx_t duplicates = 0;
/* We generate a r x c bitmap and two integer index arrays */
IA = rsb__calloc(sizeof(rsb_coo_idx_t)*nnz);
JA = rsb__calloc(sizeof(rsb_coo_idx_t)*nnz);
VA = rsb__calloc(sizeof(float)*nnz);
if(!VA || !IA || !JA)goto err;
/* We populate the arrays with random coordinates (please note that there could be duplicates) */
for(k=0;k<nnz;++k) {IA[k]=rand()%r;}
for(k=0;k<nnz;++k) {JA[k]=rand()%c;}
for(k=0;k<nnz;++k) {VA[k]=(float)rand();}
/* with a very stupid algorithm for avoiding duplicates (COULD TAKE FOREVER!) */
duplicates = 0;
if(!g_allow_duplicates)
do
{
/* WARNING : THERE IS NO PROOF THIS COULD TERMINATE, AS IT IS DEPENDENT ON THE PSEUDORANDOM NUMER GENERATOR */
duplicates=0;
for(k=0;k<nnz;++k) for(i=0;i<k;++i)if(IA[i]==IA[k]) if(JA[i]==JA[k])
{
IA[k]=rand()%r;
JA[k]=rand()%c;
++duplicates;
}
}
while(duplicates);
printf("%s","%%MatrixMarket matrix coordinate real general\n");
printf("%d %d %d\n",r,c,nnz);
for(k=0;k<nnz;++k)
{
printf("%6d %6d %20g\n",IA[k]+1,JA[k]+1,VA[k]);
}
RSB_CONDITIONAL_FREE(IA);
RSB_CONDITIONAL_FREE(VA);
RSB_CONDITIONAL_FREE(JA);
return RSB_ERR_NO_ERROR;
err:
RSB_CONDITIONAL_FREE(IA);
RSB_CONDITIONAL_FREE(VA);
RSB_CONDITIONAL_FREE(JA);
return RSB_ERR_GENERIC_ERROR;
}
rsb_option options[] = {
{"nnz", required_argument, NULL, 'n'},
{"cols", required_argument, NULL, 'c'},
{"rows", required_argument, NULL, 'r'},
{"banded", required_argument, NULL, 'b'},
{"allow-duplicates", no_argument, NULL, 'd'},
{"generate-matrix", no_argument, NULL, 'g'},
{0,0,0,0}
};
//size_t strlen(char *s){char *ss=s;while(*s)++s;return s-ss;}
int rsb_genmm_main(int argc,char *argv[])
{
rsb_err_t errval = RSB_ERR_NO_ERROR;
int c;
int opt_index = 0;
rsb_coo_idx_t rows=0,cols=0;
rsb_nnz_idx_t nnz=0;
double want_percentage=0.0;
rsb_bool_t g_want_banded = 0;
rsb_bool_t g_diagonal = 0;
rsb_type_t typecode = RSB_NUMERICAL_TYPE_DEFAULT_INTEGER;
g_allow_duplicates = 0;
if(rsb_lib_init(RSB_NULL_INIT_OPTIONS) != RSB_ERR_NO_ERROR)
{
RSB_ERROR("initialization error!\n");
goto err;
}
for (;;)
{
c = rsb_getopt_long(argc, argv, "gDb:dr:c:n:", options, &opt_index);
if (c == -1)break;
switch (c)
{
case 'b':
g_want_banded = rsb__util_atoi(optarg);
g_want_banded++;/* just a trick */
break;
case 'D':
g_diagonal = 1;
break;
case 'd':
g_allow_duplicates = 1;
break;
case 'c':
cols = rsb__util_atoi(optarg);
break;
case 'n':
nnz = rsb__util_atoi(optarg);
if(*optarg && optarg[strlen(optarg)-1]=='%')want_percentage=nnz;
break;
case 'r':
rows = rsb__util_atoi(optarg);
break;
case 'g':
break;
}
}
if( g_want_banded != 0 )
{
/* we want a banded matrix */
struct rsb_mtx_t * mtxAp;
mtxAp = rsb__generate_banded(1, 1 , rows, cols, /*rows/4*/ g_want_banded-1 , NULL, typecode);
/* errval = rsb__generate_blocked_banded_mtx(rows,1,g_want_banded-1,g_want_banded-1,&mtxAp,typecode); */
if(!mtxAp) goto err;
rsb__do_file_mtx_save(mtxAp,NULL); /* FIXME: errval = rsb_pr..*/
return 0;
}
if( want_percentage )
{
want_percentage *= 0.01;
want_percentage *= rows;
want_percentage *= cols;
nnz = want_percentage ;
if(!nnz)++nnz;
}
if( g_diagonal )
{
/* we want a diagonal matrix */
struct rsb_mtx_t * mtxAp = NULL;
/* mtxAp = rsb_generate_diagonal( rows, NULL, typecode); */
errval = rsb__generate_blocked_banded_mtx(rows,1,0,0,&mtxAp,typecode);
if(!mtxAp) goto err;
rsb__do_file_mtx_save(mtxAp,NULL);
return RSB_PROGRAM_SUCCESS;
}
if( nnz < 1 )
{
fprintf(stderr,
"usage: %s -g -r rows -c cols \n"
"\t [ -n nonzeros [%%] ] "
"| [ -b bandwidth ] (-b for a banded matrix with 'bandwidth' wide bandwidth)\n"
"\t[-d ] (-d means that duplicates are allowed) !\n",
argv[0]
);
return RSB_PROGRAM_ERROR;
}
if( nnz > rows * cols )
{
fprintf(stderr,"can't generate more nonzeros than rows x columns!\n");
return RSB_PROGRAM_ERROR;
}
return RSB_ERR_TO_PROGRAM_ERROR(gen_mm(rows,cols,nnz));
err:
fprintf(stderr,"some error occurred during matrix generation\n");
/* no deallocation, though */
return RSB_PROGRAM_ERROR;
}
/*
int main(int argc,char *argv[])
{
return rsb_genmm_main(argc,argv);
}
*/
/* @endcond */
|