svdu1v.c 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394
  1. /* svdu1v.c CCMATH mathematics library source code.
  2. *
  3. * Copyright (C) 2000 Daniel A. Atkinson All rights reserved.
  4. * This code may be redistributed under the terms of the GNU library
  5. * public license (LGPL). ( See the lgpl.license file for details.)
  6. * ------------------------------------------------------------------------
  7. */
  8. #include <stdlib.h>
  9. #include "ccmath.h"
  10. int svdu1v(double *d, double *a, int m, double *v, int n)
  11. {
  12. double *p, *p1, *q, *pp, *w, *e;
  13. double s, h, r, t, sv;
  14. int i, j, k, mm, nm, ms;
  15. if (m < n)
  16. return -1;
  17. w = (double *)calloc(m + n, sizeof(double));
  18. e = w + m;
  19. for (i = 0, mm = m, nm = n - 1, p = a; i < n; ++i, --mm, --nm, p += n + 1) {
  20. if (mm > 1) {
  21. sv = h = 0.;
  22. for (j = 0, q = p, s = 0.; j < mm; ++j, q += n) {
  23. w[j] = *q;
  24. s += *q * *q;
  25. }
  26. if (s > 0.) {
  27. h = sqrt(s);
  28. if (*p < 0.)
  29. h = -h;
  30. s += *p * h;
  31. s = 1. / s;
  32. t = 1. / (w[0] += h);
  33. sv = 1. + fabs(*p / h);
  34. for (k = 1, ms = n - i; k < ms; ++k) {
  35. for (j = 0, q = p + k, r = 0.; j < mm; q += n)
  36. r += w[j++] * *q;
  37. r *= s;
  38. for (j = 0, q = p + k; j < mm; q += n)
  39. *q -= r * w[j++];
  40. }
  41. for (j = 1, q = p; j < mm;)
  42. *(q += n) = t * w[j++];
  43. }
  44. *p = sv;
  45. d[i] = -h;
  46. }
  47. if (mm == 1)
  48. d[i] = *p;
  49. p1 = p + 1;
  50. sv = h = 0.;
  51. if (nm > 1) {
  52. for (j = 0, q = p1, s = 0.; j < nm; ++j, ++q)
  53. s += *q * *q;
  54. if (s > 0.) {
  55. h = sqrt(s);
  56. if (*p1 < 0.)
  57. h = -h;
  58. sv = 1. + fabs(*p1 / h);
  59. s += *p1 * h;
  60. s = 1. / s;
  61. t = 1. / (*p1 += h);
  62. for (k = n, ms = n * (m - i); k < ms; k += n) {
  63. for (j = 0, q = p1, pp = p1 + k, r = 0.; j < nm; ++j)
  64. r += *q++ * *pp++;
  65. r *= s;
  66. for (j = 0, q = p1, pp = p1 + k; j < nm; ++j)
  67. *pp++ -= r * *q++;
  68. }
  69. for (j = 1, q = p1 + 1; j < nm; ++j)
  70. *q++ *= t;
  71. }
  72. *p1 = sv;
  73. e[i] = -h;
  74. }
  75. if (nm == 1)
  76. e[i] = *p1;
  77. }
  78. ldvmat(a, v, n);
  79. atou1(a, m, n);
  80. qrbdu1(d, e, a, m, v, n);
  81. for (i = 0; i < n; ++i) {
  82. if (d[i] < 0.) {
  83. d[i] = -d[i];
  84. for (j = 0, p = v + i; j < n; ++j, p += n)
  85. *p = -*p;
  86. }
  87. }
  88. free(w);
  89. return 0;
  90. }