| 1 | 
 | 
|---|
| 2 | /* 
 | 
|---|
| 3 |  * These routines are based on the work of Edward T. Seidl at the
 | 
|---|
| 4 |  * National Institutes of Health.
 | 
|---|
| 5 |  */
 | 
|---|
| 6 | 
 | 
|---|
| 7 | #include <stdio.h>
 | 
|---|
| 8 | #include <stdlib.h>
 | 
|---|
| 9 | #include <math.h>
 | 
|---|
| 10 | #include <string.h>
 | 
|---|
| 11 | #include <math/scmat/cmatrix.h>
 | 
|---|
| 12 | 
 | 
|---|
| 13 | static void ludcmp(double**, int, int*, double*);
 | 
|---|
| 14 | static void lubksb(double**, int, int*, double*);
 | 
|---|
| 15 | static void symm_lu_decomp(double**, int, double*);
 | 
|---|
| 16 | static void symm_lu_back_sub(double**, int, double*);
 | 
|---|
| 17 | 
 | 
|---|
| 18 | static void tred2(int dim,double**,double*,double*,int);
 | 
|---|
| 19 | static void tqli(int dim,double*,double**,double*,int,double);
 | 
|---|
| 20 | static void eigsort(int dim,double*,double**);
 | 
|---|
| 21 | 
 | 
|---|
| 22 | double**
 | 
|---|
| 23 | cmat_new_square_matrix(int n)
 | 
|---|
| 24 | {
 | 
|---|
| 25 |   double *mat;
 | 
|---|
| 26 |   double **r;
 | 
|---|
| 27 |   if (n == 0) return 0;
 | 
|---|
| 28 |   mat = (double*) malloc(sizeof(double)*n*n);
 | 
|---|
| 29 |   if (!mat) return 0;
 | 
|---|
| 30 |   r = (double**) malloc(sizeof(double*)*n);
 | 
|---|
| 31 |   if (!r) {
 | 
|---|
| 32 |     free(mat);
 | 
|---|
| 33 |     return 0;
 | 
|---|
| 34 |   }
 | 
|---|
| 35 |   cmat_matrix_pointers(r,mat,n,n);
 | 
|---|
| 36 |   return r;
 | 
|---|
| 37 | }
 | 
|---|
| 38 | 
 | 
|---|
| 39 | double**
 | 
|---|
| 40 | cmat_new_rect_matrix(int n,int m)
 | 
|---|
| 41 | {
 | 
|---|
| 42 |   double *mat;
 | 
|---|
| 43 |   double **r;
 | 
|---|
| 44 |   if (n == 0 || m == 0) return 0;
 | 
|---|
| 45 |   mat = (double*) malloc(sizeof(double)*n*m);
 | 
|---|
| 46 |   if (!mat) return 0;
 | 
|---|
| 47 |   r = (double**) malloc(sizeof(double*)*n);
 | 
|---|
| 48 |   if (!r) {
 | 
|---|
| 49 |     free(mat);
 | 
|---|
| 50 |     return 0;
 | 
|---|
| 51 |   }
 | 
|---|
| 52 |   cmat_matrix_pointers(r,mat,n,m);
 | 
|---|
| 53 |   return r;
 | 
|---|
| 54 | }
 | 
|---|
| 55 | 
 | 
|---|
| 56 | /* this deletes both square and triangular matrices */
 | 
|---|
| 57 | void
 | 
|---|
| 58 | cmat_delete_matrix(double**m)
 | 
|---|
| 59 | {
 | 
|---|
| 60 |   if (m) {
 | 
|---|
| 61 |       free(m[0]);
 | 
|---|
| 62 |       free(m);
 | 
|---|
| 63 |     }
 | 
|---|
| 64 | }
 | 
|---|
| 65 | 
 | 
|---|
| 66 | void
 | 
|---|
| 67 | cmat_transpose_square_matrix(double**matrix, int n)
 | 
|---|
| 68 | {
 | 
|---|
| 69 |   int i,j;
 | 
|---|
| 70 |   for (i=0; i<n; i++) {
 | 
|---|
| 71 |       for (j=0; j<i; j++) {
 | 
|---|
| 72 |           double tmp = matrix[i][j];
 | 
|---|
| 73 |           matrix[i][j] = matrix[j][i];
 | 
|---|
| 74 |           matrix[j][i] = tmp;
 | 
|---|
| 75 |         }
 | 
|---|
| 76 |     }
 | 
|---|
| 77 | }
 | 
|---|
| 78 | 
 | 
|---|
| 79 | void
 | 
|---|
| 80 | cmat_matrix_pointers(double**ptrs,double*matrix,int nrow, int ncol)
 | 
|---|
| 81 | {
 | 
|---|
| 82 |   int i;
 | 
|---|
| 83 |   for (i=0; i<nrow; i++) ptrs[i] = &matrix[i*ncol];
 | 
|---|
| 84 | }
 | 
|---|
| 85 | 
 | 
|---|
| 86 | /*
 | 
|---|
| 87 |  * a contains pointers to the an area of contiguous storage.
 | 
|---|
| 88 |  * Its dimensions are nr by nc.  On exit it will be transposed,
 | 
|---|
| 89 |  * however the a vector of double* is itself unchanged.  Another
 | 
|---|
| 90 |  * vector is needed to access the storage or a must be updated
 | 
|---|
| 91 |  * after this routine is called.
 | 
|---|
| 92 |  */
 | 
|---|
| 93 | void
 | 
|---|
| 94 | cmat_transpose_matrix(double**a, int nr, int nc)
 | 
|---|
| 95 | {
 | 
|---|
| 96 |   int i,j;
 | 
|---|
| 97 |   double* tmpp;
 | 
|---|
| 98 |   double* tmp;
 | 
|---|
| 99 | 
 | 
|---|
| 100 |   if (nr == 0 || nc == 0) return;
 | 
|---|
| 101 | 
 | 
|---|
| 102 |   if (nr == nc) {
 | 
|---|
| 103 |       cmat_transpose_square_matrix(a,nr);
 | 
|---|
| 104 |       return;
 | 
|---|
| 105 |     };
 | 
|---|
| 106 | 
 | 
|---|
| 107 |   tmp = (double*) malloc(sizeof(double)*nr*nc);
 | 
|---|
| 108 |   if (!tmp && nr && nc) {
 | 
|---|
| 109 |     fprintf(stderr,"cmat_transpose_matrix: malloc failed\n");
 | 
|---|
| 110 |     abort();
 | 
|---|
| 111 |   }
 | 
|---|
| 112 | 
 | 
|---|
| 113 |   tmpp = tmp;
 | 
|---|
| 114 |   for (i=0; i<nc; i++) {
 | 
|---|
| 115 |       for (j=0; j<nr; j++) {
 | 
|---|
| 116 |           *tmpp = a[j][i];
 | 
|---|
| 117 |           tmpp++;
 | 
|---|
| 118 |         }
 | 
|---|
| 119 |     }
 | 
|---|
| 120 | 
 | 
|---|
| 121 |   memcpy(a[0],tmp,sizeof(double)*nr*nc);
 | 
|---|
| 122 | 
 | 
|---|
| 123 |   if (tmp) free(tmp);
 | 
|---|
| 124 | }
 | 
|---|
| 125 | 
 | 
|---|
| 126 | /* a is symmetric if sym is true */
 | 
|---|
| 127 | double
 | 
|---|
| 128 | cmat_determ(double** a, int sym, int dim)
 | 
|---|
| 129 | {
 | 
|---|
| 130 |   int i;
 | 
|---|
| 131 |   double det=0;
 | 
|---|
| 132 | 
 | 
|---|
| 133 |   if (sym) {
 | 
|---|
| 134 |     symm_lu_decomp(a,dim,&det);
 | 
|---|
| 135 |   } else {
 | 
|---|
| 136 |     int *indx= (int*) malloc(sizeof(int)*dim);
 | 
|---|
| 137 |     ludcmp(a,dim,indx,&det);
 | 
|---|
| 138 |     free(indx);
 | 
|---|
| 139 |   }
 | 
|---|
| 140 | 
 | 
|---|
| 141 |   if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 142 | 
 | 
|---|
| 143 |   for (i=0; i < dim; i++) det *= a[i][i];
 | 
|---|
| 144 | 
 | 
|---|
| 145 |   return det;
 | 
|---|
| 146 | }
 | 
|---|
| 147 | 
 | 
|---|
| 148 | /* a is symmetric if sym is true */
 | 
|---|
| 149 | double
 | 
|---|
| 150 | cmat_solve_lin(double** a, int sym, double* b, int dim)
 | 
|---|
| 151 | {
 | 
|---|
| 152 |   int i;
 | 
|---|
| 153 |   double det=0;
 | 
|---|
| 154 | 
 | 
|---|
| 155 |   if (sym) {
 | 
|---|
| 156 |     symm_lu_decomp(a,dim,&det);
 | 
|---|
| 157 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 158 |     symm_lu_back_sub(a,dim,b);
 | 
|---|
| 159 |   } else {
 | 
|---|
| 160 |     int *indx= (int*) malloc(sizeof(int)*dim);
 | 
|---|
| 161 |     ludcmp(a,dim,indx,&det);
 | 
|---|
| 162 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 163 |     lubksb(a,dim,indx,b);
 | 
|---|
| 164 |     free(indx);
 | 
|---|
| 165 |   }
 | 
|---|
| 166 | 
 | 
|---|
| 167 |   for(i=0; i < dim; i++) det *= a[i][i];
 | 
|---|
| 168 |   if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 169 | 
 | 
|---|
| 170 |   return det;
 | 
|---|
| 171 | }
 | 
|---|
| 172 | 
 | 
|---|
| 173 | double 
 | 
|---|
| 174 | cmat_invert(double**a, int sym, int dim)
 | 
|---|
| 175 | {
 | 
|---|
| 176 |   int i,j;
 | 
|---|
| 177 |   double det=0;
 | 
|---|
| 178 |   double **y;
 | 
|---|
| 179 |   double *b;
 | 
|---|
| 180 | 
 | 
|---|
| 181 |   b = (double*) malloc(sizeof(double)*dim);
 | 
|---|
| 182 |   y = cmat_new_square_matrix(dim);
 | 
|---|
| 183 | 
 | 
|---|
| 184 |   if (sym) {
 | 
|---|
| 185 |     symm_lu_decomp(a,dim,&det);
 | 
|---|
| 186 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 187 | 
 | 
|---|
| 188 |     for (i=0; i < dim; i++) det *= a[i][i];
 | 
|---|
| 189 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 190 | 
 | 
|---|
| 191 |     for (i=0; i < dim; i++) {
 | 
|---|
| 192 |       for (j=0; j < dim; j++) b[j]=0;
 | 
|---|
| 193 |       b[i]=1;
 | 
|---|
| 194 |       symm_lu_back_sub(a,dim,b);
 | 
|---|
| 195 |       for (j=0; j < dim; j++) y[j][i]=b[j];
 | 
|---|
| 196 |     }
 | 
|---|
| 197 | 
 | 
|---|
| 198 |     for (i=0; i < dim; i++)
 | 
|---|
| 199 |       for (j=0; j <= i; j++)
 | 
|---|
| 200 |         a[i][j] = y[i][j];
 | 
|---|
| 201 | 
 | 
|---|
| 202 |   } else {
 | 
|---|
| 203 |     int *indx= (int*) malloc(sizeof(int)*dim);
 | 
|---|
| 204 | 
 | 
|---|
| 205 |     ludcmp(a,dim,indx,&det);
 | 
|---|
| 206 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 207 | 
 | 
|---|
| 208 |     for (i=0; i < dim; i++) det *= a[i][i];
 | 
|---|
| 209 |     if (fabs(det) < 1.0e-16) return 0;
 | 
|---|
| 210 | 
 | 
|---|
| 211 |     for (i=0; i < dim; i++) {
 | 
|---|
| 212 |       memset(b,0,sizeof(double)*dim);
 | 
|---|
| 213 |       b[i]=1;
 | 
|---|
| 214 |       lubksb(a,dim,indx,b);
 | 
|---|
| 215 |       for (j=0; j < dim; j++) y[j][i]=b[j];
 | 
|---|
| 216 |     }
 | 
|---|
| 217 | 
 | 
|---|
| 218 |     for (i=0; i < dim; i++)
 | 
|---|
| 219 |       for (j=0; j < dim; j++)
 | 
|---|
| 220 |         a[i][j] = y[i][j];
 | 
|---|
| 221 |     free(indx);
 | 
|---|
| 222 |   }
 | 
|---|
| 223 | 
 | 
|---|
| 224 |   free(b);
 | 
|---|
| 225 |   cmat_delete_matrix(y);
 | 
|---|
| 226 | 
 | 
|---|
| 227 |   return det;
 | 
|---|
| 228 | }
 | 
|---|
| 229 | 
 | 
|---|
| 230 | static void
 | 
|---|
| 231 | ludcmp(double** a, int n, int *indx, double *d)
 | 
|---|
| 232 | {
 | 
|---|
| 233 |   int i,imax=0,j,k;
 | 
|---|
| 234 |   double big,dum,sum,temp;
 | 
|---|
| 235 | 
 | 
|---|
| 236 |   double* vv = (double*) malloc(sizeof(double)*n);
 | 
|---|
| 237 | 
 | 
|---|
| 238 |   *d = 1.0;
 | 
|---|
| 239 | 
 | 
|---|
| 240 |   for (i=0; i < n ; i++) {
 | 
|---|
| 241 |     big=0.0;
 | 
|---|
| 242 |     for (j=0; j < n; j++) if ((temp=fabs(a[i][j])) > big) big=temp;
 | 
|---|
| 243 | #if 1
 | 
|---|
| 244 |     if (big == 0.0) {
 | 
|---|
| 245 |       *d = 0.0;
 | 
|---|
| 246 |       free(vv);
 | 
|---|
| 247 |       return;
 | 
|---|
| 248 |       }
 | 
|---|
| 249 | #else
 | 
|---|
| 250 |     if(big==0.0) big=1.0e-16;
 | 
|---|
| 251 | #endif
 | 
|---|
| 252 |     vv[i] = 1.0/big;
 | 
|---|
| 253 |     }
 | 
|---|
| 254 | 
 | 
|---|
| 255 |   for (j=0; j < n ; j++) {
 | 
|---|
| 256 |     for (i=0; i < j ; i++) {
 | 
|---|
| 257 |       sum = a[i][j];
 | 
|---|
| 258 |       for (k=0; k < i ; k++) sum -= a[i][k]*a[k][j];
 | 
|---|
| 259 |       a[i][j] = sum;
 | 
|---|
| 260 |       }
 | 
|---|
| 261 | 
 | 
|---|
| 262 |     big = 0.0;
 | 
|---|
| 263 |     for (i=j ; i < n ; i++) {
 | 
|---|
| 264 |       sum=a[i][j];
 | 
|---|
| 265 |       for (k=0; k < j ; k++) sum -= a[i][k]*a[k][j];
 | 
|---|
| 266 |       a[i][j] = sum;
 | 
|---|
| 267 |       if ((dum=vv[i]*fabs(sum)) >= big) {
 | 
|---|
| 268 |         big = dum;
 | 
|---|
| 269 |         imax = i;
 | 
|---|
| 270 |         }
 | 
|---|
| 271 |       }
 | 
|---|
| 272 | 
 | 
|---|
| 273 |     if (j != imax) {
 | 
|---|
| 274 |       for (k=0; k < n; k++) {
 | 
|---|
| 275 |         dum=a[imax][k];
 | 
|---|
| 276 |         a[imax][k]=a[j][k];
 | 
|---|
| 277 |         a[j][k]=dum;
 | 
|---|
| 278 |         }
 | 
|---|
| 279 |       *d = -(*d);
 | 
|---|
| 280 |       vv[imax]=vv[j];
 | 
|---|
| 281 |       }
 | 
|---|
| 282 | 
 | 
|---|
| 283 |     indx[j]=imax;
 | 
|---|
| 284 |     if (a[j][j] == 0.0) a[j][j] = 1.0e-20;
 | 
|---|
| 285 |     if (j != n-1) {
 | 
|---|
| 286 |       dum = 1.0/a[j][j];
 | 
|---|
| 287 |       for (i=j+1; i < n ; i++) a[i][j] *= dum;
 | 
|---|
| 288 |       }
 | 
|---|
| 289 |     }
 | 
|---|
| 290 |   free(vv);
 | 
|---|
| 291 |   }
 | 
|---|
| 292 | 
 | 
|---|
| 293 | static void
 | 
|---|
| 294 | lubksb(double** a, int n, int *indx, double* b)
 | 
|---|
| 295 | {
 | 
|---|
| 296 |   int i,ii=0,ip,j;
 | 
|---|
| 297 |   int t=0;
 | 
|---|
| 298 |   double sum;
 | 
|---|
| 299 | 
 | 
|---|
| 300 |   for (i=0; i < n ; i++) {
 | 
|---|
| 301 |     ip = indx[i];
 | 
|---|
| 302 |     sum = b[ip];
 | 
|---|
| 303 |     b[ip]=b[i];
 | 
|---|
| 304 | 
 | 
|---|
| 305 |     if(t) {
 | 
|---|
| 306 |       for (j=ii; j <= i-1 ; j++)
 | 
|---|
| 307 |         sum -= a[i][j]*b[j];
 | 
|---|
| 308 |       }
 | 
|---|
| 309 |     else if(sum) {
 | 
|---|
| 310 |       ii=i;
 | 
|---|
| 311 |       t++;
 | 
|---|
| 312 |       }
 | 
|---|
| 313 | 
 | 
|---|
| 314 |     b[i]=sum;
 | 
|---|
| 315 |     }
 | 
|---|
| 316 | 
 | 
|---|
| 317 |   for (i=n-1; i >= 0 ; i--) {
 | 
|---|
| 318 |     sum = b[i];
 | 
|---|
| 319 |     for (j=i+1; j < n ; j++) sum -= a[i][j]*b[j];
 | 
|---|
| 320 |     b[i] = sum/a[i][i];
 | 
|---|
| 321 |     }
 | 
|---|
| 322 |   }
 | 
|---|
| 323 | 
 | 
|---|
| 324 | /*
 | 
|---|
| 325 |  * this is LU decomposition where A is a symmetric matrix
 | 
|---|
| 326 |  * when A is symmetric, then 
 | 
|---|
| 327 |  *   beta(i,j) = A(i,j) - sum_k(i-1) beta(k,i)*beta(k,j)/beta(k,k)
 | 
|---|
| 328 |  *   alpha(i,j) = beta(j,i)/beta(j,j)
 | 
|---|
| 329 |  *
 | 
|---|
| 330 |  * since we're storing beta in a, the indices of beta will be switched
 | 
|---|
| 331 |  * since alpha is expressed in terms of beta, we don't store it
 | 
|---|
| 332 |  *
 | 
|---|
| 333 |  * so we have
 | 
|---|
| 334 |  *   beta(i,j) = A(i,j) - sum_k(i-1) beta(i,k)*beta(j,k)/beta(k,k)
 | 
|---|
| 335 |  *   alpha(i,j) = beta(i,j)/beta(j,j)
 | 
|---|
| 336 |  */
 | 
|---|
| 337 | 
 | 
|---|
| 338 | static void
 | 
|---|
| 339 | symm_lu_decomp(double** a, int n, double *d)
 | 
|---|
| 340 | {
 | 
|---|
| 341 |   int i,j,k;
 | 
|---|
| 342 |   double tmp;
 | 
|---|
| 343 | 
 | 
|---|
| 344 |   double* v = (double*) malloc(sizeof(double)*n);
 | 
|---|
| 345 |   memset(v,0,sizeof(double)*n);
 | 
|---|
| 346 | 
 | 
|---|
| 347 |   /* check for singular matrix */
 | 
|---|
| 348 |   for (i=0; i < n; i++) {
 | 
|---|
| 349 |     for (j=0; j < i; j++) {
 | 
|---|
| 350 |       v[i] = ((tmp=fabs(a[i][j])) > v[i]) ? tmp : v[i];
 | 
|---|
| 351 |       v[j] = (tmp > v[j]) ? tmp : v[j];
 | 
|---|
| 352 |     }
 | 
|---|
| 353 |     v[i] = ((tmp=fabs(a[i][i])) > v[i]) ? tmp : v[i];
 | 
|---|
| 354 |   }
 | 
|---|
| 355 | 
 | 
|---|
| 356 |   for (i=0; i < n; i++) {
 | 
|---|
| 357 |     if (fabs(v[i]) < 1.0e-16) {
 | 
|---|
| 358 |       fprintf(stderr,"\n  warning: singular matrix in symm_lu_decomp\n");
 | 
|---|
| 359 |       *d = 0.0;
 | 
|---|
| 360 |       return;
 | 
|---|
| 361 |     }
 | 
|---|
| 362 |   }
 | 
|---|
| 363 | 
 | 
|---|
| 364 |   free(v);
 | 
|---|
| 365 | 
 | 
|---|
| 366 |   *d = 1.0;
 | 
|---|
| 367 | 
 | 
|---|
| 368 |   for (i=0; i < n ; i++) {
 | 
|---|
| 369 |     /* check to make sure we're not going to blow up */
 | 
|---|
| 370 |     if (i < n-1) {
 | 
|---|
| 371 |       tmp = 0; 
 | 
|---|
| 372 |       for (k=0; k < i-1; k++)
 | 
|---|
| 373 |         tmp += a[i][k]*a[i][k]/a[k][k];
 | 
|---|
| 374 |       if (fabs(a[i][i]-tmp) < 1.0e-16) {
 | 
|---|
| 375 |         fprintf(stderr,"\n  warning: singular matrix in symm_lu_decomp 2\n");
 | 
|---|
| 376 |         *d = 0;
 | 
|---|
| 377 |         return;
 | 
|---|
| 378 |       }
 | 
|---|
| 379 |     }
 | 
|---|
| 380 |     for (j=i; j < n; j++) {
 | 
|---|
| 381 |       tmp = 0;
 | 
|---|
| 382 |       for (k=0; k <= i-1; k++)
 | 
|---|
| 383 |         tmp -= a[i][k]*a[j][k]/a[k][k];
 | 
|---|
| 384 |       a[j][i] += tmp;
 | 
|---|
| 385 |     }
 | 
|---|
| 386 |   }
 | 
|---|
| 387 | }
 | 
|---|
| 388 | 
 | 
|---|
| 389 | static void
 | 
|---|
| 390 | symm_lu_back_sub(double** a, int n, double* b)
 | 
|---|
| 391 | {
 | 
|---|
| 392 |   int i,j;
 | 
|---|
| 393 |   double sum;
 | 
|---|
| 394 | 
 | 
|---|
| 395 |  /* form y(i) = bi - sum_j(i-1) alpha(i,j)*y(j)
 | 
|---|
| 396 |   * alpha(i,j) = beta(j,i)/beta(j,j), but beta is stored lower instead of 
 | 
|---|
| 397 |   * upper triangle, so alpha(i,j) = beta(i,j)/beta(j,j)
 | 
|---|
| 398 |   */
 | 
|---|
| 399 |   for (i=0; i < n ; i++) {
 | 
|---|
| 400 |     sum = 0;
 | 
|---|
| 401 |     for (j=0; j < i; j++)
 | 
|---|
| 402 |       sum += (a[i][j]/a[j][j]) * b[j];
 | 
|---|
| 403 |     b[i] -= sum;
 | 
|---|
| 404 |   }
 | 
|---|
| 405 | 
 | 
|---|
| 406 |  /* now form x(i) = 1/beta(i,i)*[y(i) - sum_j=i+1(N) beta(i,j)*x(j)]
 | 
|---|
| 407 |   * is really ...[...beta(j,i)*x(j)]
 | 
|---|
| 408 |   */
 | 
|---|
| 409 |   for (i=n-1; i >= 0 ; i--) {
 | 
|---|
| 410 |     sum = b[i];
 | 
|---|
| 411 |     for (j=i+1; j < n ; j++) sum -= a[j][i]*b[j];
 | 
|---|
| 412 |     b[i] = sum/a[i][i];
 | 
|---|
| 413 |   }
 | 
|---|
| 414 | }
 | 
|---|
| 415 | 
 | 
|---|
| 416 | /*
 | 
|---|
| 417 |  * This does c(t) (+)= a(t) * b(t), where the (t) means the transpose
 | 
|---|
| 418 |  * of the matrix can be optionally used and the (+) means that accumulation
 | 
|---|
| 419 |  * is optional.  The dimensions of the matrices is as follows:
 | 
|---|
| 420 |  * a(nr,nl) (if ta then a(nl,nr))
 | 
|---|
| 421 |  * b(nl,nc) (if tb then b(nc,nl))
 | 
|---|
| 422 |  * c(nr,nc) (if tc then c(nc,nr))
 | 
|---|
| 423 |  */
 | 
|---|
| 424 | void
 | 
|---|
| 425 | cmat_mxm(double** a, int ta, double** b, int tb, double** c, int tc,
 | 
|---|
| 426 |          int nr, int nl, int nc, int add)
 | 
|---|
| 427 | {
 | 
|---|
| 428 |   int odd_nr,odd_nc;
 | 
|---|
| 429 |   int i,j,k;
 | 
|---|
| 430 |   double t00,t01,t10,t11;
 | 
|---|
| 431 |   double *att,*bt;
 | 
|---|
| 432 |   double *at1,*bt1;
 | 
|---|
| 433 |   double** old_a = 0;
 | 
|---|
| 434 |   double** old_b = 0;
 | 
|---|
| 435 | 
 | 
|---|
| 436 |   odd_nr = (nr)%2;
 | 
|---|
| 437 |   odd_nc = (nc)%2;
 | 
|---|
| 438 | 
 | 
|---|
| 439 |   if(ta) {
 | 
|---|
| 440 |       cmat_transpose_matrix(a,nl,nr);
 | 
|---|
| 441 |       if (nr > nl) {
 | 
|---|
| 442 |           old_a = a;
 | 
|---|
| 443 |           a = (double**) malloc(nr*sizeof(double*));
 | 
|---|
| 444 |           if (!a) {
 | 
|---|
| 445 |             fprintf(stderr,"cmat_mxm: malloc a failed\n");
 | 
|---|
| 446 |             abort();
 | 
|---|
| 447 |           }
 | 
|---|
| 448 |           a[0] = old_a[0];
 | 
|---|
| 449 |         }
 | 
|---|
| 450 |       cmat_matrix_pointers(a,a[0],nr,nl);
 | 
|---|
| 451 |     }
 | 
|---|
| 452 |   if(!tb) {
 | 
|---|
| 453 |       cmat_transpose_matrix(b,nl,nc);
 | 
|---|
| 454 |       if (nc > nl) {
 | 
|---|
| 455 |           old_b = b;
 | 
|---|
| 456 |           b = (double**) malloc(nc*sizeof(double*));
 | 
|---|
| 457 |           if (!b) {
 | 
|---|
| 458 |             fprintf(stderr,"cmat_mxm: malloc b failed\n");
 | 
|---|
| 459 |             abort();
 | 
|---|
| 460 |           }
 | 
|---|
| 461 |           b[0] = old_b[0];
 | 
|---|
| 462 |         }
 | 
|---|
| 463 |       cmat_matrix_pointers(b,b[0],nc,nl);
 | 
|---|
| 464 |     }
 | 
|---|
| 465 | 
 | 
|---|
| 466 |   for(j=0; j < nc-1 ; j+=2) {
 | 
|---|
| 467 |     for(i=0; i < nr-1 ; i+=2) {
 | 
|---|
| 468 |       att=a[i]; bt=b[j];
 | 
|---|
| 469 |       at1=a[i+1]; bt1=b[j+1];
 | 
|---|
| 470 |       if(add) {
 | 
|---|
| 471 |         if(tc) {
 | 
|---|
| 472 |           t00 = c[j][i];
 | 
|---|
| 473 |           t01 = c[j+1][i];
 | 
|---|
| 474 |           t10 = c[j][i+1];
 | 
|---|
| 475 |           t11 = c[j+1][i+1];
 | 
|---|
| 476 |           }
 | 
|---|
| 477 |         else {
 | 
|---|
| 478 |           t00 = c[i][j];
 | 
|---|
| 479 |           t01 = c[i][j+1];
 | 
|---|
| 480 |           t10 = c[i+1][j];
 | 
|---|
| 481 |           t11 = c[i+1][j+1];
 | 
|---|
| 482 |           }
 | 
|---|
| 483 |         }
 | 
|---|
| 484 |       else
 | 
|---|
| 485 |         t00=t01=t10=t11=0.0;
 | 
|---|
| 486 |       for(k=nl; k ; k--,att++,bt++,at1++,bt1++) {
 | 
|---|
| 487 |         t00 += *att * *bt;
 | 
|---|
| 488 |         t01 += *att * *bt1;
 | 
|---|
| 489 |         t10 += *at1 * *bt;
 | 
|---|
| 490 |         t11 += *at1 * *bt1;
 | 
|---|
| 491 |         }
 | 
|---|
| 492 |       if(tc) {
 | 
|---|
| 493 |         c[j][i]=t00;
 | 
|---|
| 494 |         c[j+1][i]=t01;
 | 
|---|
| 495 |         c[j][i+1]=t10;
 | 
|---|
| 496 |         c[j+1][i+1]=t11;
 | 
|---|
| 497 |         }
 | 
|---|
| 498 |       else {
 | 
|---|
| 499 |         c[i][j]=t00;
 | 
|---|
| 500 |         c[i][j+1]=t01;
 | 
|---|
| 501 |         c[i+1][j]=t10;
 | 
|---|
| 502 |         c[i+1][j+1]=t11;
 | 
|---|
| 503 |         }
 | 
|---|
| 504 |       }
 | 
|---|
| 505 |     if(odd_nr) {
 | 
|---|
| 506 |       att=a[i]; bt=b[j];
 | 
|---|
| 507 |       bt1=b[j+1];
 | 
|---|
| 508 |       if(add) {
 | 
|---|
| 509 |         if(tc) {
 | 
|---|
| 510 |           t00 = c[j][i];
 | 
|---|
| 511 |           t01 = c[j+1][i];
 | 
|---|
| 512 |           }
 | 
|---|
| 513 |         else {
 | 
|---|
| 514 |           t00 = c[i][j];
 | 
|---|
| 515 |           t01 = c[i][j+1];
 | 
|---|
| 516 |           }
 | 
|---|
| 517 |         }
 | 
|---|
| 518 |       else t00=t01=0.0;
 | 
|---|
| 519 |       for(k= nl; k ; k--,att++,bt++,bt1++) {
 | 
|---|
| 520 |         t00 += *att * *bt;
 | 
|---|
| 521 |         t01 += *att * *bt1;
 | 
|---|
| 522 |         }
 | 
|---|
| 523 |       if(tc) {
 | 
|---|
| 524 |         c[j][i]=t00;
 | 
|---|
| 525 |         c[j+1][i]=t01;
 | 
|---|
| 526 |         }
 | 
|---|
| 527 |       else {
 | 
|---|
| 528 |         c[i][j]=t00;
 | 
|---|
| 529 |         c[i][j+1]=t01;
 | 
|---|
| 530 |         }
 | 
|---|
| 531 |       }
 | 
|---|
| 532 |     }
 | 
|---|
| 533 |   if(odd_nc) {
 | 
|---|
| 534 |     for(i=0; i < nr-1 ; i+=2) {
 | 
|---|
| 535 |       att=a[i]; bt=b[j];
 | 
|---|
| 536 |       at1=a[i+1];
 | 
|---|
| 537 |       if(add) {
 | 
|---|
| 538 |         if(tc) {
 | 
|---|
| 539 |           t00 = c[j][i];
 | 
|---|
| 540 |           t10 = c[j][i+1];
 | 
|---|
| 541 |           }
 | 
|---|
| 542 |         else {
 | 
|---|
| 543 |           t00 = c[i][j];
 | 
|---|
| 544 |           t10 = c[i+1][j];
 | 
|---|
| 545 |           }
 | 
|---|
| 546 |         }
 | 
|---|
| 547 |       else t00=t10=0.0;
 | 
|---|
| 548 |       for(k= nl; k ; k--,att++,bt++,at1++) {
 | 
|---|
| 549 |         t00 += *att * *bt;
 | 
|---|
| 550 |         t10 += *at1 * *bt;
 | 
|---|
| 551 |         }
 | 
|---|
| 552 |       if(tc) {
 | 
|---|
| 553 |         c[j][i]=t00;
 | 
|---|
| 554 |         c[j][i+1]=t10;
 | 
|---|
| 555 |         }
 | 
|---|
| 556 |       else {
 | 
|---|
| 557 |         c[i][j]=t00;
 | 
|---|
| 558 |         c[i+1][j]=t10;
 | 
|---|
| 559 |         }
 | 
|---|
| 560 |       }
 | 
|---|
| 561 |     if(odd_nr) {
 | 
|---|
| 562 |       att=a[i]; bt=b[j];
 | 
|---|
| 563 |       if(add) t00 = (tc) ? c[j][i] : c[i][j];
 | 
|---|
| 564 |       else t00=0.0;
 | 
|---|
| 565 |       for(k=nl; k ; k--,att++,bt++) t00 += *att * *bt;
 | 
|---|
| 566 |       if(tc) c[j][i]=t00;
 | 
|---|
| 567 |       else c[i][j]=t00;
 | 
|---|
| 568 |       }
 | 
|---|
| 569 |     }
 | 
|---|
| 570 | 
 | 
|---|
| 571 |   if(ta) {
 | 
|---|
| 572 |       cmat_transpose_matrix(a,nr,nl);
 | 
|---|
| 573 |       if (old_a) {
 | 
|---|
| 574 |           free(a);
 | 
|---|
| 575 |           a = old_a;
 | 
|---|
| 576 |         }
 | 
|---|
| 577 |       cmat_matrix_pointers(a,a[0],nr,nl);
 | 
|---|
| 578 |     }
 | 
|---|
| 579 |   if(!tb) {
 | 
|---|
| 580 |       cmat_transpose_matrix(b,nc,nl);
 | 
|---|
| 581 |       if (old_b) {
 | 
|---|
| 582 |           free(b);
 | 
|---|
| 583 |           b = old_b;
 | 
|---|
| 584 |         }
 | 
|---|
| 585 |       cmat_matrix_pointers(b,b[0],nl,nc);
 | 
|---|
| 586 |     }
 | 
|---|
| 587 |   }
 | 
|---|
| 588 | 
 | 
|---|
| 589 | /*
 | 
|---|
| 590 |  * a is symmetric (na,na) in a triangular storage format
 | 
|---|
| 591 |  * b is rectangular (na,nb)
 | 
|---|
| 592 |  * a (+)= b * transpose(b) (+= if add)
 | 
|---|
| 593 |  */
 | 
|---|
| 594 | void
 | 
|---|
| 595 | cmat_symmetric_mxm(double**a,int na, /* a is (na,na) */
 | 
|---|
| 596 |                    double**b,int nb, /* b is (na,nb) */
 | 
|---|
| 597 |                    int add)
 | 
|---|
| 598 | {
 | 
|---|
| 599 |   int i,j,k;
 | 
|---|
| 600 |   for (i=0; i<na; i++) {
 | 
|---|
| 601 |       double*ai=a[i];
 | 
|---|
| 602 |       for (j=0; j<=i; j++) {
 | 
|---|
| 603 |           double*bi=b[i];
 | 
|---|
| 604 |           double*bj=b[j];
 | 
|---|
| 605 |           double tmp;
 | 
|---|
| 606 |           if (add) tmp = 0.0;
 | 
|---|
| 607 |           else tmp = ai[j];
 | 
|---|
| 608 |           for (k=nb; k; k--,bi++,bj++) {
 | 
|---|
| 609 |               tmp += *bi * *bj;
 | 
|---|
| 610 |             }
 | 
|---|
| 611 |           ai[j] = tmp;
 | 
|---|
| 612 |         }
 | 
|---|
| 613 |     }
 | 
|---|
| 614 | }
 | 
|---|
| 615 | 
 | 
|---|
| 616 | /*
 | 
|---|
| 617 |  * a is symmetric (na,na) in a triangular storage format
 | 
|---|
| 618 |  * b is symmetric (nb,nb) in a triangular storage format
 | 
|---|
| 619 |  * a (+)= c * b * transpose(c) (+= if add)
 | 
|---|
| 620 |  */
 | 
|---|
| 621 | void
 | 
|---|
| 622 | cmat_transform_symmetric_matrix(double**a,int na, /* a is (na,na) */
 | 
|---|
| 623 |                                 double**b,int nb, /* b is (nb,nb) */
 | 
|---|
| 624 |                                 double**c,        /* c is (na,nb) */
 | 
|---|
| 625 |                                 int add)
 | 
|---|
| 626 | {
 | 
|---|
| 627 |   int i,j,k;
 | 
|---|
| 628 |   double**t;
 | 
|---|
| 629 |   double* brow;
 | 
|---|
| 630 | 
 | 
|---|
| 631 |   /* create a temporary matrix, t */
 | 
|---|
| 632 |   t = cmat_new_rect_matrix(na,nb);
 | 
|---|
| 633 | 
 | 
|---|
| 634 |   /* t = transpose(b * transpose(c)) */
 | 
|---|
| 635 |   brow = (double*) malloc(sizeof(double)*nb);
 | 
|---|
| 636 |   if (!brow) {
 | 
|---|
| 637 |     fprintf(stderr,"cmat_transform_symmetric_matrix: malloc brow failed\n");
 | 
|---|
| 638 |     abort();
 | 
|---|
| 639 |   }
 | 
|---|
| 640 |   for (i=0; i<nb; i++) {
 | 
|---|
| 641 |       for (k=0; k<=i; k++) brow[k] = b[i][k];
 | 
|---|
| 642 |       for (   ; k<nb; k++) brow[k] = b[k][i];
 | 
|---|
| 643 |       for (j=0; j<na; j++) {
 | 
|---|
| 644 |           double*bi = brow;
 | 
|---|
| 645 |           double*cj = c[j];
 | 
|---|
| 646 |           double tmp = 0.0;
 | 
|---|
| 647 |           for (k=nb; k; k--,bi++,cj++) tmp += *bi * *cj;
 | 
|---|
| 648 |           t[j][i] = tmp;
 | 
|---|
| 649 |         }
 | 
|---|
| 650 |     }
 | 
|---|
| 651 |   free(brow);
 | 
|---|
| 652 | 
 | 
|---|
| 653 |   /* a = c * transpose(t) */
 | 
|---|
| 654 |   for (i=0; i<na; i++) {
 | 
|---|
| 655 |       for (j=0; j<=i; j++) {
 | 
|---|
| 656 |           double*ci = c[i];
 | 
|---|
| 657 |           double*tj = t[j];
 | 
|---|
| 658 |           double tmp;
 | 
|---|
| 659 |           if (add) tmp = a[i][j];
 | 
|---|
| 660 |           else tmp = 0.0;
 | 
|---|
| 661 |           for (k=nb; k; k--,ci++,tj++) tmp += *ci * *tj;
 | 
|---|
| 662 |           a[i][j] = tmp;
 | 
|---|
| 663 |         }
 | 
|---|
| 664 |     }
 | 
|---|
| 665 | 
 | 
|---|
| 666 |   /* delete the temporary */
 | 
|---|
| 667 |   cmat_delete_matrix(t);
 | 
|---|
| 668 | }
 | 
|---|
| 669 | 
 | 
|---|
| 670 | /*
 | 
|---|
| 671 |  * a is symmetric (na,na) in a triangular storage format
 | 
|---|
| 672 |  * b is diagonal (nb,nb) in a vector storage format
 | 
|---|
| 673 |  * a (+)= c * b * transpose(c) (+= if add)
 | 
|---|
| 674 |  */
 | 
|---|
| 675 | void
 | 
|---|
| 676 | cmat_transform_diagonal_matrix(double**a,int na, /* a is (na,na) */
 | 
|---|
| 677 |                                double*b,int nb,  /* b is (nb,nb) */
 | 
|---|
| 678 |                                double**c,        /* c is (na,nb) */
 | 
|---|
| 679 |                                int add)
 | 
|---|
| 680 | {
 | 
|---|
| 681 |   int i,j,k;
 | 
|---|
| 682 |   double t;
 | 
|---|
| 683 | 
 | 
|---|
| 684 |   for (i=0; i < na; i++) {
 | 
|---|
| 685 |     for (j=0; j <= i; j++) {
 | 
|---|
| 686 |       t=0;
 | 
|---|
| 687 |       for (k=0; k < nb; k++)
 | 
|---|
| 688 |         t += c[i][k] * c[j][k] * b[k];
 | 
|---|
| 689 |       if (add)
 | 
|---|
| 690 |         a[i][j] += t;
 | 
|---|
| 691 |       else
 | 
|---|
| 692 |         a[i][j] = t;
 | 
|---|
| 693 |     }
 | 
|---|
| 694 |   }
 | 
|---|
| 695 | }
 | 
|---|
| 696 | 
 | 
|---|
| 697 | /*
 | 
|---|
| 698 |  * Argument a contains pointers to the rows of a symmetrix matrix.  The
 | 
|---|
| 699 |  * in each row is the row number + 1.  These rows are stored in
 | 
|---|
| 700 |  * contiguous memory starting with 0.  Evecs also contains pointers to
 | 
|---|
| 701 |  * contiguous memory.  N is the dimension.
 | 
|---|
| 702 |  */
 | 
|---|
| 703 | void
 | 
|---|
| 704 | cmat_diag(double**a, double*evals, double**evecs, int n,
 | 
|---|
| 705 |               int matz, double tol)
 | 
|---|
| 706 | {
 | 
|---|
| 707 |   int i,j;
 | 
|---|
| 708 |   int diagonal=1;
 | 
|---|
| 709 |   double*fv1;
 | 
|---|
| 710 | 
 | 
|---|
| 711 |  /* I'm having problems with diagonalizing matrices which are already
 | 
|---|
| 712 |   * diagonal.  So let's first check to see if _a_ is diagonal, and if it
 | 
|---|
| 713 |   * is, then just return the diagonal elements in evals and a unit matrix
 | 
|---|
| 714 |   * in evecs
 | 
|---|
| 715 |   */
 | 
|---|
| 716 | 
 | 
|---|
| 717 |   for (i=1; i < n; i++) {
 | 
|---|
| 718 |     for (j=0; j < i; j++) {
 | 
|---|
| 719 |       if (fabs(a[i][j]) > tol) diagonal=0;
 | 
|---|
| 720 |       }
 | 
|---|
| 721 |     }
 | 
|---|
| 722 | 
 | 
|---|
| 723 |   if (diagonal) {
 | 
|---|
| 724 |     for(i=0; i < n; i++) {
 | 
|---|
| 725 |       evals[i] = a[i][i];
 | 
|---|
| 726 |       evecs[i][i] = 1.0;
 | 
|---|
| 727 | 
 | 
|---|
| 728 |       for(j=0; j < i; j++) {
 | 
|---|
| 729 |         evecs[i][j] = evecs[j][i] = 0.0;
 | 
|---|
| 730 |         }
 | 
|---|
| 731 |       }
 | 
|---|
| 732 |     eigsort(n,evals,evecs);
 | 
|---|
| 733 |     return;
 | 
|---|
| 734 |     }
 | 
|---|
| 735 | 
 | 
|---|
| 736 |   fv1 = (double*) malloc(sizeof(double)*n);
 | 
|---|
| 737 |   if (!fv1) {
 | 
|---|
| 738 |     fprintf(stderr,"cmat_diag: malloc fv1 failed\n");
 | 
|---|
| 739 |     abort();
 | 
|---|
| 740 |   }
 | 
|---|
| 741 | 
 | 
|---|
| 742 |   for(i=0; i < n; i++) {
 | 
|---|
| 743 |       for(j=0; j <= i; j++) {
 | 
|---|
| 744 |           evecs[i][j] = evecs[j][i] = a[i][j];
 | 
|---|
| 745 |         }
 | 
|---|
| 746 |     }
 | 
|---|
| 747 | 
 | 
|---|
| 748 |   tred2(n,evecs,evals,fv1,1);
 | 
|---|
| 749 | 
 | 
|---|
| 750 |   cmat_transpose_square_matrix(evecs,n);
 | 
|---|
| 751 |   tqli(n,evals,evecs,fv1,1,tol);
 | 
|---|
| 752 |   cmat_transpose_square_matrix(evecs,n);
 | 
|---|
| 753 | 
 | 
|---|
| 754 |   eigsort(n,evals,evecs);
 | 
|---|
| 755 | 
 | 
|---|
| 756 |   free(fv1);
 | 
|---|
| 757 |   }
 | 
|---|
| 758 | 
 | 
|---|
| 759 | #define dsign(a,b) (((b) >= 0.0) ? fabs(a) : -fabs(a))
 | 
|---|
| 760 | 
 | 
|---|
| 761 | static void
 | 
|---|
| 762 | tred2(int n,double** a,double* d,double* e,int matz)
 | 
|---|
| 763 | {
 | 
|---|
| 764 |   int i,j,k,l;
 | 
|---|
| 765 |   double f,g,h,hh,scale,scale_inv,h_inv;
 | 
|---|
| 766 |   if (n == 1) return;
 | 
|---|
| 767 | 
 | 
|---|
| 768 |   for(i=n-1; i > 0; i--) {
 | 
|---|
| 769 |     l = i-1;
 | 
|---|
| 770 |     h = 0.0;
 | 
|---|
| 771 |     scale = 0.0;
 | 
|---|
| 772 |     if(l) {
 | 
|---|
| 773 |       for(k=0; k <= l; k++) scale += fabs(a[i][k]);
 | 
|---|
| 774 |       if (scale == 0.0) e[i] = a[i][l];
 | 
|---|
| 775 |       else {
 | 
|---|
| 776 |         scale_inv=1.0/scale;
 | 
|---|
| 777 |         for (k=0; k <= l; k++) {
 | 
|---|
| 778 |           a[i][k] *= scale_inv;
 | 
|---|
| 779 |           h += a[i][k]*a[i][k];
 | 
|---|
| 780 |           }
 | 
|---|
| 781 |         f=a[i][l];
 | 
|---|
| 782 |         g= -(dsign(sqrt(h),f));
 | 
|---|
| 783 |         e[i] = scale*g;
 | 
|---|
| 784 |         h -= f*g;
 | 
|---|
| 785 |         a[i][l] = f-g;
 | 
|---|
| 786 |         f = 0.0;
 | 
|---|
| 787 |         h_inv=1.0/h;
 | 
|---|
| 788 |         for (j=0; j <= l; j++) {
 | 
|---|
| 789 |           if (matz) a[j][i] = a[i][j]*h_inv;
 | 
|---|
| 790 |           g = 0.0;
 | 
|---|
| 791 |           for (k=0; k <= j; k++) g += a[j][k]*a[i][k];
 | 
|---|
| 792 |           if (l > j) for (k=j+1; k <= l; k++) g += a[k][j]*a[i][k];
 | 
|---|
| 793 |           e[j] = g*h_inv;
 | 
|---|
| 794 |           f += e[j]*a[i][j];
 | 
|---|
| 795 |           }
 | 
|---|
| 796 |         hh = f/(h+h);
 | 
|---|
| 797 |         for (j=0; j <= l; j++) {
 | 
|---|
| 798 |           f = a[i][j];
 | 
|---|
| 799 |           g = e[j] - hh*f;
 | 
|---|
| 800 |           e[j] = g;
 | 
|---|
| 801 |           for (k=0; k <= j; k++) a[j][k] -= (f*e[k] + g*a[i][k]);
 | 
|---|
| 802 |           }
 | 
|---|
| 803 |         }
 | 
|---|
| 804 |       }
 | 
|---|
| 805 |     else {
 | 
|---|
| 806 |       e[i] = a[i][l];
 | 
|---|
| 807 |       }
 | 
|---|
| 808 |     d[i] = h;
 | 
|---|
| 809 |     }
 | 
|---|
| 810 |   if(matz) d[0] = 0.0;
 | 
|---|
| 811 |   e[0] = 0.0;
 | 
|---|
| 812 | 
 | 
|---|
| 813 |   for(i=0; i < n; i++) {
 | 
|---|
| 814 |     l = i-1;
 | 
|---|
| 815 |     if (matz) {
 | 
|---|
| 816 |       if(d[i]) {
 | 
|---|
| 817 |         for(j=0; j <= l; j++) {
 | 
|---|
| 818 |           g = 0.0;
 | 
|---|
| 819 |           for(k=0; k <= l; k++) g += a[i][k]*a[k][j];
 | 
|---|
| 820 |           for(k=0; k <= l; k++) a[k][j] -= g*a[k][i];
 | 
|---|
| 821 |           }
 | 
|---|
| 822 |         }
 | 
|---|
| 823 |       }
 | 
|---|
| 824 |     d[i] = a[i][i];
 | 
|---|
| 825 |     if(matz) {
 | 
|---|
| 826 |       a[i][i] = 1.0;
 | 
|---|
| 827 |       if(l >= 0) for (j=0; j<= l; j++) a[i][j] = a[j][i] = 0.0;
 | 
|---|
| 828 |       }
 | 
|---|
| 829 |     }
 | 
|---|
| 830 |   }
 | 
|---|
| 831 | 
 | 
|---|
| 832 | static void
 | 
|---|
| 833 | tqli(int n, double* d, double** z, double* e, int matz, double toler)
 | 
|---|
| 834 | {
 | 
|---|
| 835 |   register int k;
 | 
|---|
| 836 |   int i,l,m,iter;
 | 
|---|
| 837 |   double g,r,s,c,p,f,b;
 | 
|---|
| 838 |   double azi;
 | 
|---|
| 839 | 
 | 
|---|
| 840 |   f=0.0;
 | 
|---|
| 841 |   if (n == 1) {
 | 
|---|
| 842 |     d[0]=z[0][0];
 | 
|---|
| 843 |     z[0][0] = 1.0;
 | 
|---|
| 844 |     return;
 | 
|---|
| 845 |     }
 | 
|---|
| 846 | 
 | 
|---|
| 847 |   for (i=1; i < n ; i++) e[i-1] = e[i];
 | 
|---|
| 848 |   e[n-1] = 0.0;
 | 
|---|
| 849 |   for (l=0; l < n; l++) {
 | 
|---|
| 850 |     iter = 0;
 | 
|---|
| 851 | L1:
 | 
|---|
| 852 |     for (m=l; m < n-1;m++) if (fabs(e[m]) < toler) goto L2;
 | 
|---|
| 853 |     m=n-1;
 | 
|---|
| 854 | L2:
 | 
|---|
| 855 |     if (m != l) {
 | 
|---|
| 856 |       if (iter++ == 30) {
 | 
|---|
| 857 |         fprintf (stderr,"tqli not converging %d %g\n",l,e[l]);
 | 
|---|
| 858 |         continue;
 | 
|---|
| 859 |         }
 | 
|---|
| 860 | 
 | 
|---|
| 861 |       g = (d[l+1]-d[l])/(2.0*e[l]);
 | 
|---|
| 862 |       r = sqrt(g*g + 1.0);
 | 
|---|
| 863 |       g = d[m] - d[l] + e[l]/((g + dsign(r,g)));
 | 
|---|
| 864 |       s=1.0;
 | 
|---|
| 865 |       c=1.0;
 | 
|---|
| 866 |       p=0.0;
 | 
|---|
| 867 |       for (i=m-1; i >= l; i--) {
 | 
|---|
| 868 |         f = s*e[i];
 | 
|---|
| 869 |         b = c*e[i];
 | 
|---|
| 870 |         if (fabs(f) >= fabs(g)) {
 | 
|---|
| 871 |           c = g/f;
 | 
|---|
| 872 |           r = sqrt(c*c + 1.0);
 | 
|---|
| 873 |           e[i+1] = f*r;
 | 
|---|
| 874 |           s=1.0/r;
 | 
|---|
| 875 |           c *= s;
 | 
|---|
| 876 |           }
 | 
|---|
| 877 |         else {
 | 
|---|
| 878 |           s = f/g;
 | 
|---|
| 879 |           r = sqrt(s*s + 1.0);
 | 
|---|
| 880 |           e[i+1] = g*r;
 | 
|---|
| 881 |           c = 1.0/r;
 | 
|---|
| 882 |           s *= c;
 | 
|---|
| 883 |           }
 | 
|---|
| 884 |         g = d[i+1] - p;
 | 
|---|
| 885 |         r = (d[i]-g)*s + 2.0*c*b;
 | 
|---|
| 886 |         p = s*r;
 | 
|---|
| 887 |         d[i+1] = g+p;
 | 
|---|
| 888 |         g = c*r-b;
 | 
|---|
| 889 | 
 | 
|---|
| 890 |         if (matz) {
 | 
|---|
| 891 |           double *zi = z[i];
 | 
|---|
| 892 |           double *zi1 = z[i+1];
 | 
|---|
| 893 |           for (k=n; k ; k--,zi++,zi1++) {
 | 
|---|
| 894 |             azi = *zi;
 | 
|---|
| 895 |             f = *zi1;
 | 
|---|
| 896 |             *zi1 = azi*s + c*f;
 | 
|---|
| 897 |             *zi = azi*c - s*f;
 | 
|---|
| 898 |             }
 | 
|---|
| 899 |           }
 | 
|---|
| 900 |         }
 | 
|---|
| 901 | 
 | 
|---|
| 902 |       d[l] -= p;
 | 
|---|
| 903 |       e[l] = g;
 | 
|---|
| 904 |       e[m] = 0.0;
 | 
|---|
| 905 |       goto L1;
 | 
|---|
| 906 |       }
 | 
|---|
| 907 |     }
 | 
|---|
| 908 |   }
 | 
|---|
| 909 | 
 | 
|---|
| 910 | static void
 | 
|---|
| 911 | eigsort(int n, double* d, double** v)
 | 
|---|
| 912 | {
 | 
|---|
| 913 |   int i,j,k;
 | 
|---|
| 914 |   double p;
 | 
|---|
| 915 | 
 | 
|---|
| 916 |   for(i=0; i < n-1 ; i++) {
 | 
|---|
| 917 |     k=i;
 | 
|---|
| 918 |     p=d[i];
 | 
|---|
| 919 |     for(j=i+1; j < n; j++) {
 | 
|---|
| 920 |       if(d[j] < p) {
 | 
|---|
| 921 |         k=j;
 | 
|---|
| 922 |         p=d[j];
 | 
|---|
| 923 |         }
 | 
|---|
| 924 |       }
 | 
|---|
| 925 |     if(k != i) {
 | 
|---|
| 926 |       d[k]=d[i];
 | 
|---|
| 927 |       d[i]=p;
 | 
|---|
| 928 |       for(j=0; j < n; j++) {
 | 
|---|
| 929 |         p=v[j][i];
 | 
|---|
| 930 |         v[j][i]=v[j][k];
 | 
|---|
| 931 |         v[j][k]=p;
 | 
|---|
| 932 |         }
 | 
|---|
| 933 |       }
 | 
|---|
| 934 |     }
 | 
|---|
| 935 |   }
 | 
|---|
| 936 | 
 | 
|---|
| 937 | void
 | 
|---|
| 938 | cmat_schmidt(double **C, double *S, int nrow, int nc)
 | 
|---|
| 939 | {
 | 
|---|
| 940 |   int i,j,ij;
 | 
|---|
| 941 |   int m;
 | 
|---|
| 942 |   double vtmp;
 | 
|---|
| 943 |   double *v = (double*) malloc(sizeof(double)*nrow);
 | 
|---|
| 944 | 
 | 
|---|
| 945 |   if (!v) {
 | 
|---|
| 946 |     fprintf(stderr,"cmat_schmidt: could not malloc v(%d)\n",nrow);
 | 
|---|
| 947 |     abort();
 | 
|---|
| 948 |   }
 | 
|---|
| 949 |   
 | 
|---|
| 950 |   for (m=0; m < nc; m++) {
 | 
|---|
| 951 |     v[0] = C[0][m] * S[0];
 | 
|---|
| 952 | 
 | 
|---|
| 953 |     for (i=ij=1; i < nrow; i++) {
 | 
|---|
| 954 |       for (j=0,vtmp=0.0; j < i; j++,ij++) {
 | 
|---|
| 955 |         vtmp += C[j][m]*S[ij];
 | 
|---|
| 956 |         v[j] += C[i][m]*S[ij];
 | 
|---|
| 957 |       }
 | 
|---|
| 958 |       v[i] = vtmp + C[i][m]*S[ij];
 | 
|---|
| 959 |       ij++;
 | 
|---|
| 960 |     }
 | 
|---|
| 961 | 
 | 
|---|
| 962 |     for (i=0,vtmp=0.0; i < nrow; i++)
 | 
|---|
| 963 |       vtmp += v[i]*C[i][m];
 | 
|---|
| 964 | 
 | 
|---|
| 965 |     if (!vtmp) {
 | 
|---|
| 966 |       fprintf(stderr,"cmat_schmidt: bogus\n");
 | 
|---|
| 967 |       abort();
 | 
|---|
| 968 |     }
 | 
|---|
| 969 | 
 | 
|---|
| 970 |     if (vtmp < 1.0e-15)
 | 
|---|
| 971 |       vtmp = 1.0e-15;
 | 
|---|
| 972 | 
 | 
|---|
| 973 |     vtmp = 1.0/sqrt(vtmp);
 | 
|---|
| 974 |     
 | 
|---|
| 975 |     for (i=0; i < nrow; i++) {
 | 
|---|
| 976 |       v[i] *= vtmp;
 | 
|---|
| 977 |       C[i][m] *= vtmp;
 | 
|---|
| 978 |     }
 | 
|---|
| 979 | 
 | 
|---|
| 980 |     if (m < nc-1) {
 | 
|---|
| 981 |       for (i=m+1,vtmp=0.0; i < nc; i++) {
 | 
|---|
| 982 |         for (j=0,vtmp=0.0; j < nrow; j++)
 | 
|---|
| 983 |           vtmp += v[j] * C[j][i];
 | 
|---|
| 984 |         for (j=0; j < nrow; j++)
 | 
|---|
| 985 |           C[j][i] -= vtmp * C[j][m];
 | 
|---|
| 986 |       }
 | 
|---|
| 987 |     }
 | 
|---|
| 988 |   }
 | 
|---|
| 989 | }
 | 
|---|
| 990 | 
 | 
|---|
| 991 | /* Returns the number of linearly independent vectors
 | 
|---|
| 992 |    orthogonal wrt S. */
 | 
|---|
| 993 | int
 | 
|---|
| 994 | cmat_schmidt_tol(double **C, double *S, int nrow, int ncol,
 | 
|---|
| 995 |                  double tolerance, double *res)
 | 
|---|
| 996 | {
 | 
|---|
| 997 |   int i,j,ij;
 | 
|---|
| 998 |   int m;
 | 
|---|
| 999 |   double vtmp;
 | 
|---|
| 1000 |   int northog = 0;
 | 
|---|
| 1001 |   double *v = (double*) malloc(sizeof(double)*nrow);
 | 
|---|
| 1002 | 
 | 
|---|
| 1003 |   if (res) *res = 1.0;
 | 
|---|
| 1004 | 
 | 
|---|
| 1005 |   if (!v) {
 | 
|---|
| 1006 |     fprintf(stderr,"cmat_schmidt_tol: could not malloc v(%d)\n",nrow);
 | 
|---|
| 1007 |     abort();
 | 
|---|
| 1008 |   }
 | 
|---|
| 1009 | 
 | 
|---|
| 1010 |   /* Orthonormalize the columns of C wrt S. */
 | 
|---|
| 1011 |   for (m=0; m < ncol; m++) {
 | 
|---|
| 1012 |     v[0] = C[0][m] * S[0];
 | 
|---|
| 1013 | 
 | 
|---|
| 1014 |     for (i=ij=1; i < nrow; i++) {
 | 
|---|
| 1015 |       for (j=0,vtmp=0.0; j < i; j++,ij++) {
 | 
|---|
| 1016 |         vtmp += C[j][m]*S[ij];
 | 
|---|
| 1017 |         v[j] += C[i][m]*S[ij];
 | 
|---|
| 1018 |       }
 | 
|---|
| 1019 |       v[i] = vtmp + C[i][m]*S[ij];
 | 
|---|
| 1020 |       ij++;
 | 
|---|
| 1021 |     }
 | 
|---|
| 1022 | 
 | 
|---|
| 1023 |     for (i=0,vtmp=0.0; i < nrow; i++)
 | 
|---|
| 1024 |       vtmp += v[i]*C[i][m];
 | 
|---|
| 1025 | 
 | 
|---|
| 1026 |     if (vtmp < tolerance) continue;
 | 
|---|
| 1027 | 
 | 
|---|
| 1028 |     if (res && (m == 0 || vtmp < *res)) *res = vtmp;
 | 
|---|
| 1029 | 
 | 
|---|
| 1030 |     vtmp = 1.0/sqrt(vtmp);
 | 
|---|
| 1031 |     
 | 
|---|
| 1032 |     for (i=0; i < nrow; i++) {
 | 
|---|
| 1033 |       v[i] *= vtmp;
 | 
|---|
| 1034 |       C[i][northog] = C[i][m] * vtmp;
 | 
|---|
| 1035 |     }
 | 
|---|
| 1036 | 
 | 
|---|
| 1037 |     for (i=m+1,vtmp=0.0; i < ncol; i++) {
 | 
|---|
| 1038 |         for (j=0,vtmp=0.0; j < nrow; j++)
 | 
|---|
| 1039 |             vtmp += v[j] * C[j][i];
 | 
|---|
| 1040 |         for (j=0; j < nrow; j++)
 | 
|---|
| 1041 |             C[j][i] -= vtmp * C[j][northog];
 | 
|---|
| 1042 |       }
 | 
|---|
| 1043 |     northog++;
 | 
|---|
| 1044 |   }
 | 
|---|
| 1045 |   return northog;
 | 
|---|
| 1046 | }
 | 
|---|