alglib 0.1.1

Sign up to get free protection for your applications and to get access to all the features.
Files changed (255) hide show
  1. data/History.txt +7 -0
  2. data/Manifest.txt +253 -0
  3. data/README.txt +33 -0
  4. data/Rakefile +27 -0
  5. data/ext/Rakefile +24 -0
  6. data/ext/alglib.i +24 -0
  7. data/ext/alglib/Makefile +157 -0
  8. data/ext/alglib/airyf.cpp +372 -0
  9. data/ext/alglib/airyf.h +81 -0
  10. data/ext/alglib/alglib.cpp +8558 -0
  11. data/ext/alglib/alglib_util.cpp +19 -0
  12. data/ext/alglib/alglib_util.h +14 -0
  13. data/ext/alglib/ap.cpp +877 -0
  14. data/ext/alglib/ap.english.html +364 -0
  15. data/ext/alglib/ap.h +666 -0
  16. data/ext/alglib/ap.russian.html +442 -0
  17. data/ext/alglib/apvt.h +754 -0
  18. data/ext/alglib/bdss.cpp +1500 -0
  19. data/ext/alglib/bdss.h +251 -0
  20. data/ext/alglib/bdsvd.cpp +1339 -0
  21. data/ext/alglib/bdsvd.h +164 -0
  22. data/ext/alglib/bessel.cpp +1226 -0
  23. data/ext/alglib/bessel.h +331 -0
  24. data/ext/alglib/betaf.cpp +105 -0
  25. data/ext/alglib/betaf.h +74 -0
  26. data/ext/alglib/bidiagonal.cpp +1328 -0
  27. data/ext/alglib/bidiagonal.h +350 -0
  28. data/ext/alglib/binomialdistr.cpp +247 -0
  29. data/ext/alglib/binomialdistr.h +153 -0
  30. data/ext/alglib/blas.cpp +576 -0
  31. data/ext/alglib/blas.h +132 -0
  32. data/ext/alglib/cblas.cpp +226 -0
  33. data/ext/alglib/cblas.h +57 -0
  34. data/ext/alglib/cdet.cpp +138 -0
  35. data/ext/alglib/cdet.h +92 -0
  36. data/ext/alglib/chebyshev.cpp +216 -0
  37. data/ext/alglib/chebyshev.h +76 -0
  38. data/ext/alglib/chisquaredistr.cpp +157 -0
  39. data/ext/alglib/chisquaredistr.h +144 -0
  40. data/ext/alglib/cholesky.cpp +285 -0
  41. data/ext/alglib/cholesky.h +86 -0
  42. data/ext/alglib/cinverse.cpp +298 -0
  43. data/ext/alglib/cinverse.h +111 -0
  44. data/ext/alglib/clu.cpp +337 -0
  45. data/ext/alglib/clu.h +120 -0
  46. data/ext/alglib/correlation.cpp +280 -0
  47. data/ext/alglib/correlation.h +77 -0
  48. data/ext/alglib/correlationtests.cpp +726 -0
  49. data/ext/alglib/correlationtests.h +134 -0
  50. data/ext/alglib/crcond.cpp +826 -0
  51. data/ext/alglib/crcond.h +148 -0
  52. data/ext/alglib/creflections.cpp +310 -0
  53. data/ext/alglib/creflections.h +165 -0
  54. data/ext/alglib/csolve.cpp +312 -0
  55. data/ext/alglib/csolve.h +99 -0
  56. data/ext/alglib/ctrinverse.cpp +387 -0
  57. data/ext/alglib/ctrinverse.h +98 -0
  58. data/ext/alglib/ctrlinsolve.cpp +297 -0
  59. data/ext/alglib/ctrlinsolve.h +81 -0
  60. data/ext/alglib/dawson.cpp +234 -0
  61. data/ext/alglib/dawson.h +74 -0
  62. data/ext/alglib/descriptivestatistics.cpp +436 -0
  63. data/ext/alglib/descriptivestatistics.h +112 -0
  64. data/ext/alglib/det.cpp +140 -0
  65. data/ext/alglib/det.h +94 -0
  66. data/ext/alglib/dforest.cpp +1819 -0
  67. data/ext/alglib/dforest.h +316 -0
  68. data/ext/alglib/elliptic.cpp +497 -0
  69. data/ext/alglib/elliptic.h +217 -0
  70. data/ext/alglib/estnorm.cpp +429 -0
  71. data/ext/alglib/estnorm.h +107 -0
  72. data/ext/alglib/expintegrals.cpp +422 -0
  73. data/ext/alglib/expintegrals.h +108 -0
  74. data/ext/alglib/faq.english.html +258 -0
  75. data/ext/alglib/faq.russian.html +272 -0
  76. data/ext/alglib/fdistr.cpp +202 -0
  77. data/ext/alglib/fdistr.h +163 -0
  78. data/ext/alglib/fresnel.cpp +211 -0
  79. data/ext/alglib/fresnel.h +91 -0
  80. data/ext/alglib/gammaf.cpp +338 -0
  81. data/ext/alglib/gammaf.h +104 -0
  82. data/ext/alglib/gqgengauss.cpp +235 -0
  83. data/ext/alglib/gqgengauss.h +92 -0
  84. data/ext/alglib/gqgenhermite.cpp +268 -0
  85. data/ext/alglib/gqgenhermite.h +63 -0
  86. data/ext/alglib/gqgenjacobi.cpp +297 -0
  87. data/ext/alglib/gqgenjacobi.h +72 -0
  88. data/ext/alglib/gqgenlaguerre.cpp +265 -0
  89. data/ext/alglib/gqgenlaguerre.h +69 -0
  90. data/ext/alglib/gqgenlegendre.cpp +300 -0
  91. data/ext/alglib/gqgenlegendre.h +62 -0
  92. data/ext/alglib/gqgenlobatto.cpp +305 -0
  93. data/ext/alglib/gqgenlobatto.h +97 -0
  94. data/ext/alglib/gqgenradau.cpp +232 -0
  95. data/ext/alglib/gqgenradau.h +95 -0
  96. data/ext/alglib/hbisinv.cpp +480 -0
  97. data/ext/alglib/hbisinv.h +183 -0
  98. data/ext/alglib/hblas.cpp +228 -0
  99. data/ext/alglib/hblas.h +64 -0
  100. data/ext/alglib/hcholesky.cpp +339 -0
  101. data/ext/alglib/hcholesky.h +91 -0
  102. data/ext/alglib/hermite.cpp +114 -0
  103. data/ext/alglib/hermite.h +49 -0
  104. data/ext/alglib/hessenberg.cpp +370 -0
  105. data/ext/alglib/hessenberg.h +152 -0
  106. data/ext/alglib/hevd.cpp +247 -0
  107. data/ext/alglib/hevd.h +107 -0
  108. data/ext/alglib/hsschur.cpp +1316 -0
  109. data/ext/alglib/hsschur.h +108 -0
  110. data/ext/alglib/htridiagonal.cpp +734 -0
  111. data/ext/alglib/htridiagonal.h +180 -0
  112. data/ext/alglib/ialglib.cpp +6 -0
  113. data/ext/alglib/ialglib.h +9 -0
  114. data/ext/alglib/ibetaf.cpp +960 -0
  115. data/ext/alglib/ibetaf.h +125 -0
  116. data/ext/alglib/igammaf.cpp +430 -0
  117. data/ext/alglib/igammaf.h +157 -0
  118. data/ext/alglib/inv.cpp +274 -0
  119. data/ext/alglib/inv.h +115 -0
  120. data/ext/alglib/inverseupdate.cpp +480 -0
  121. data/ext/alglib/inverseupdate.h +185 -0
  122. data/ext/alglib/jacobianelliptic.cpp +164 -0
  123. data/ext/alglib/jacobianelliptic.h +94 -0
  124. data/ext/alglib/jarquebera.cpp +2271 -0
  125. data/ext/alglib/jarquebera.h +80 -0
  126. data/ext/alglib/kmeans.cpp +356 -0
  127. data/ext/alglib/kmeans.h +76 -0
  128. data/ext/alglib/laguerre.cpp +94 -0
  129. data/ext/alglib/laguerre.h +48 -0
  130. data/ext/alglib/lbfgs.cpp +1167 -0
  131. data/ext/alglib/lbfgs.h +218 -0
  132. data/ext/alglib/lda.cpp +434 -0
  133. data/ext/alglib/lda.h +133 -0
  134. data/ext/alglib/ldlt.cpp +1130 -0
  135. data/ext/alglib/ldlt.h +124 -0
  136. data/ext/alglib/leastsquares.cpp +1252 -0
  137. data/ext/alglib/leastsquares.h +290 -0
  138. data/ext/alglib/legendre.cpp +107 -0
  139. data/ext/alglib/legendre.h +49 -0
  140. data/ext/alglib/linreg.cpp +1185 -0
  141. data/ext/alglib/linreg.h +380 -0
  142. data/ext/alglib/logit.cpp +1523 -0
  143. data/ext/alglib/logit.h +333 -0
  144. data/ext/alglib/lq.cpp +399 -0
  145. data/ext/alglib/lq.h +160 -0
  146. data/ext/alglib/lu.cpp +462 -0
  147. data/ext/alglib/lu.h +119 -0
  148. data/ext/alglib/mannwhitneyu.cpp +4490 -0
  149. data/ext/alglib/mannwhitneyu.h +115 -0
  150. data/ext/alglib/minlm.cpp +918 -0
  151. data/ext/alglib/minlm.h +312 -0
  152. data/ext/alglib/mlpbase.cpp +3375 -0
  153. data/ext/alglib/mlpbase.h +589 -0
  154. data/ext/alglib/mlpe.cpp +1369 -0
  155. data/ext/alglib/mlpe.h +552 -0
  156. data/ext/alglib/mlptrain.cpp +1056 -0
  157. data/ext/alglib/mlptrain.h +283 -0
  158. data/ext/alglib/nearunityunit.cpp +91 -0
  159. data/ext/alglib/nearunityunit.h +17 -0
  160. data/ext/alglib/normaldistr.cpp +377 -0
  161. data/ext/alglib/normaldistr.h +175 -0
  162. data/ext/alglib/nsevd.cpp +1869 -0
  163. data/ext/alglib/nsevd.h +140 -0
  164. data/ext/alglib/pca.cpp +168 -0
  165. data/ext/alglib/pca.h +87 -0
  166. data/ext/alglib/poissondistr.cpp +143 -0
  167. data/ext/alglib/poissondistr.h +130 -0
  168. data/ext/alglib/polinterpolation.cpp +685 -0
  169. data/ext/alglib/polinterpolation.h +206 -0
  170. data/ext/alglib/psif.cpp +173 -0
  171. data/ext/alglib/psif.h +88 -0
  172. data/ext/alglib/qr.cpp +414 -0
  173. data/ext/alglib/qr.h +168 -0
  174. data/ext/alglib/ratinterpolation.cpp +134 -0
  175. data/ext/alglib/ratinterpolation.h +72 -0
  176. data/ext/alglib/rcond.cpp +705 -0
  177. data/ext/alglib/rcond.h +140 -0
  178. data/ext/alglib/reflections.cpp +504 -0
  179. data/ext/alglib/reflections.h +165 -0
  180. data/ext/alglib/rotations.cpp +473 -0
  181. data/ext/alglib/rotations.h +128 -0
  182. data/ext/alglib/rsolve.cpp +221 -0
  183. data/ext/alglib/rsolve.h +99 -0
  184. data/ext/alglib/sbisinv.cpp +217 -0
  185. data/ext/alglib/sbisinv.h +171 -0
  186. data/ext/alglib/sblas.cpp +185 -0
  187. data/ext/alglib/sblas.h +64 -0
  188. data/ext/alglib/schur.cpp +156 -0
  189. data/ext/alglib/schur.h +102 -0
  190. data/ext/alglib/sdet.cpp +193 -0
  191. data/ext/alglib/sdet.h +101 -0
  192. data/ext/alglib/sevd.cpp +116 -0
  193. data/ext/alglib/sevd.h +99 -0
  194. data/ext/alglib/sinverse.cpp +672 -0
  195. data/ext/alglib/sinverse.h +138 -0
  196. data/ext/alglib/spddet.cpp +138 -0
  197. data/ext/alglib/spddet.h +96 -0
  198. data/ext/alglib/spdgevd.cpp +842 -0
  199. data/ext/alglib/spdgevd.h +200 -0
  200. data/ext/alglib/spdinverse.cpp +509 -0
  201. data/ext/alglib/spdinverse.h +122 -0
  202. data/ext/alglib/spdrcond.cpp +421 -0
  203. data/ext/alglib/spdrcond.h +118 -0
  204. data/ext/alglib/spdsolve.cpp +275 -0
  205. data/ext/alglib/spdsolve.h +105 -0
  206. data/ext/alglib/spline2d.cpp +1192 -0
  207. data/ext/alglib/spline2d.h +301 -0
  208. data/ext/alglib/spline3.cpp +1264 -0
  209. data/ext/alglib/spline3.h +290 -0
  210. data/ext/alglib/srcond.cpp +595 -0
  211. data/ext/alglib/srcond.h +127 -0
  212. data/ext/alglib/ssolve.cpp +895 -0
  213. data/ext/alglib/ssolve.h +139 -0
  214. data/ext/alglib/stdafx.h +0 -0
  215. data/ext/alglib/stest.cpp +131 -0
  216. data/ext/alglib/stest.h +94 -0
  217. data/ext/alglib/studenttdistr.cpp +222 -0
  218. data/ext/alglib/studenttdistr.h +115 -0
  219. data/ext/alglib/studentttests.cpp +377 -0
  220. data/ext/alglib/studentttests.h +178 -0
  221. data/ext/alglib/svd.cpp +620 -0
  222. data/ext/alglib/svd.h +126 -0
  223. data/ext/alglib/tdbisinv.cpp +2608 -0
  224. data/ext/alglib/tdbisinv.h +228 -0
  225. data/ext/alglib/tdevd.cpp +1229 -0
  226. data/ext/alglib/tdevd.h +115 -0
  227. data/ext/alglib/tridiagonal.cpp +594 -0
  228. data/ext/alglib/tridiagonal.h +171 -0
  229. data/ext/alglib/trigintegrals.cpp +490 -0
  230. data/ext/alglib/trigintegrals.h +131 -0
  231. data/ext/alglib/trinverse.cpp +345 -0
  232. data/ext/alglib/trinverse.h +98 -0
  233. data/ext/alglib/trlinsolve.cpp +926 -0
  234. data/ext/alglib/trlinsolve.h +73 -0
  235. data/ext/alglib/tsort.cpp +405 -0
  236. data/ext/alglib/tsort.h +54 -0
  237. data/ext/alglib/variancetests.cpp +245 -0
  238. data/ext/alglib/variancetests.h +134 -0
  239. data/ext/alglib/wsr.cpp +6285 -0
  240. data/ext/alglib/wsr.h +96 -0
  241. data/ext/ap.i +97 -0
  242. data/ext/correlation.i +24 -0
  243. data/ext/extconf.rb +6 -0
  244. data/ext/logit.i +89 -0
  245. data/lib/alglib.rb +71 -0
  246. data/lib/alglib/correlation.rb +26 -0
  247. data/lib/alglib/linearregression.rb +63 -0
  248. data/lib/alglib/logit.rb +42 -0
  249. data/test/test_alglib.rb +52 -0
  250. data/test/test_correlation.rb +44 -0
  251. data/test/test_correlationtest.rb +45 -0
  252. data/test/test_linreg.rb +35 -0
  253. data/test/test_logit.rb +43 -0
  254. data/test/test_pca.rb +27 -0
  255. metadata +326 -0
@@ -0,0 +1,118 @@
1
+ /*************************************************************************
2
+ Copyright (c) 1992-2007 The University of Tennessee. All rights reserved.
3
+
4
+ Contributors:
5
+ * Sergey Bochkanov (ALGLIB project). Translation from FORTRAN to
6
+ pseudocode.
7
+
8
+ See subroutines comments for additional copyrights.
9
+
10
+ Redistribution and use in source and binary forms, with or without
11
+ modification, are permitted provided that the following conditions are
12
+ met:
13
+
14
+ - Redistributions of source code must retain the above copyright
15
+ notice, this list of conditions and the following disclaimer.
16
+
17
+ - Redistributions in binary form must reproduce the above copyright
18
+ notice, this list of conditions and the following disclaimer listed
19
+ in this license in the documentation and/or other materials
20
+ provided with the distribution.
21
+
22
+ - Neither the name of the copyright holders nor the names of its
23
+ contributors may be used to endorse or promote products derived from
24
+ this software without specific prior written permission.
25
+
26
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
27
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
28
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
29
+ A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
30
+ OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
31
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
32
+ LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
33
+ DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
34
+ THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
35
+ (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
36
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37
+ *************************************************************************/
38
+
39
+ #ifndef _spdrcond_h
40
+ #define _spdrcond_h
41
+
42
+ #include "ap.h"
43
+ #include "ialglib.h"
44
+
45
+ #include "trlinsolve.h"
46
+ #include "cholesky.h"
47
+ #include "estnorm.h"
48
+
49
+
50
+ /*************************************************************************
51
+ Condition number estimate of a symmetric positive definite matrix.
52
+
53
+ The algorithm calculates a lower bound of the condition number. In this case,
54
+ the algorithm does not return a lower bound of the condition number, but an
55
+ inverse number (to avoid an overflow in case of a singular matrix).
56
+
57
+ It should be noted that 1-norm and inf-norm of condition numbers of symmetric
58
+ matrices are equal, so the algorithm doesn't take into account the
59
+ differences between these types of norms.
60
+
61
+ Input parameters:
62
+ A - symmetric positive definite matrix which is given by its
63
+ upper or lower triangle depending on the value of
64
+ IsUpper. Array with elements [0..N-1, 0..N-1].
65
+ N - size of matrix A.
66
+ IsUpper - storage format.
67
+
68
+ Result:
69
+ 1/LowerBound(cond(A)), if matrix A is positive definite,
70
+ -1, if matrix A is not positive definite, and its condition number
71
+ could not be found by this algorithm.
72
+ *************************************************************************/
73
+ double spdmatrixrcond(const ap::real_2d_array& a, int n, bool isupper);
74
+
75
+
76
+ /*************************************************************************
77
+ Condition number estimate of a symmetric positive definite matrix given by
78
+ Cholesky decomposition.
79
+
80
+ The algorithm calculates a lower bound of the condition number. In this
81
+ case, the algorithm does not return a lower bound of the condition number,
82
+ but an inverse number (to avoid an overflow in case of a singular matrix).
83
+
84
+ It should be noted that 1-norm and inf-norm condition numbers of symmetric
85
+ matrices are equal, so the algorithm doesn't take into account the
86
+ differences between these types of norms.
87
+
88
+ Input parameters:
89
+ CD - Cholesky decomposition of matrix A,
90
+ output of SMatrixCholesky subroutine.
91
+ N - size of matrix A.
92
+
93
+ Result: 1/LowerBound(cond(A))
94
+ *************************************************************************/
95
+ double spdmatrixcholeskyrcond(const ap::real_2d_array& a, int n, bool isupper);
96
+
97
+
98
+ /*************************************************************************
99
+ Obsolete subroutine
100
+ *************************************************************************/
101
+ double rcondspd(ap::real_2d_array a, int n, bool isupper);
102
+
103
+
104
+ /*************************************************************************
105
+ Obsolete subroutine
106
+ *************************************************************************/
107
+ double rcondcholesky(const ap::real_2d_array& cd, int n, bool isupper);
108
+
109
+
110
+ void internalcholeskyrcond(const ap::real_2d_array& chfrm,
111
+ int n,
112
+ bool isupper,
113
+ bool isnormprovided,
114
+ double anorm,
115
+ double& rcond);
116
+
117
+
118
+ #endif
@@ -0,0 +1,275 @@
1
+
2
+ #include <stdafx.h>
3
+ #include "spdsolve.h"
4
+
5
+ /*************************************************************************
6
+ Solving a system of linear equations with a system matrix given by its
7
+ Cholesky decomposition.
8
+
9
+ The algorithm solves systems with a square matrix only.
10
+
11
+ Input parameters:
12
+ A - Cholesky decomposition of a system matrix (the result of
13
+ the SMatrixCholesky subroutine).
14
+ B - right side of a system.
15
+ Array whose index ranges within [0..N-1].
16
+ N - size of matrix A.
17
+ IsUpper - points to the triangle of matrix A in which the Cholesky
18
+ decomposition is stored. If IsUpper=True, the Cholesky
19
+ decomposition has the form of U'*U, and the upper triangle
20
+ of matrix A stores matrix U (in that case, the lower
21
+ triangle isn�t used and isn�t changed by the subroutine)
22
+ Similarly, if IsUpper = False, the Cholesky decomposition
23
+ has the form of L*L', and the lower triangle stores
24
+ matrix L.
25
+
26
+ Output parameters:
27
+ X - solution of a system.
28
+ Array whose index ranges within [0..N-1].
29
+
30
+ Result:
31
+ True, if the system is not singular. X contains the solution.
32
+ False, if the system is singular (there is a zero element on the main
33
+ diagonal). In this case, X doesn't contain a solution.
34
+
35
+ -- ALGLIB --
36
+ Copyright 2005-2008 by Bochkanov Sergey
37
+ *************************************************************************/
38
+ bool spdmatrixcholeskysolve(const ap::real_2d_array& a,
39
+ ap::real_1d_array b,
40
+ int n,
41
+ bool isupper,
42
+ ap::real_1d_array& x)
43
+ {
44
+ bool result;
45
+ int i;
46
+ double v;
47
+
48
+ ap::ap_error::make_assertion(n>0, "Error: N<=0 in SolveSystemCholesky");
49
+
50
+ //
51
+ // det(A)=0?
52
+ //
53
+ result = true;
54
+ for(i = 0; i <= n-1; i++)
55
+ {
56
+ if( a(i,i)==0 )
57
+ {
58
+ result = false;
59
+ return result;
60
+ }
61
+ }
62
+
63
+ //
64
+ // det(A)<>0
65
+ //
66
+ x.setbounds(0, n-1);
67
+ if( isupper )
68
+ {
69
+
70
+ //
71
+ // A = U'*U, solve U'*y = b first
72
+ //
73
+ b(0) = b(0)/a(0,0);
74
+ for(i = 1; i <= n-1; i++)
75
+ {
76
+ v = ap::vdotproduct(a.getcolumn(i, 0, i-1), b.getvector(0, i-1));
77
+ b(i) = (b(i)-v)/a(i,i);
78
+ }
79
+
80
+ //
81
+ // Solve U*x = y
82
+ //
83
+ b(n-1) = b(n-1)/a(n-1,n-1);
84
+ for(i = n-2; i >= 0; i--)
85
+ {
86
+ v = ap::vdotproduct(&a(i, i+1), &b(i+1), ap::vlen(i+1,n-1));
87
+ b(i) = (b(i)-v)/a(i,i);
88
+ }
89
+ ap::vmove(&x(0), &b(0), ap::vlen(0,n-1));
90
+ }
91
+ else
92
+ {
93
+
94
+ //
95
+ // A = L*L', solve L'*y = b first
96
+ //
97
+ b(0) = b(0)/a(0,0);
98
+ for(i = 1; i <= n-1; i++)
99
+ {
100
+ v = ap::vdotproduct(&a(i, 0), &b(0), ap::vlen(0,i-1));
101
+ b(i) = (b(i)-v)/a(i,i);
102
+ }
103
+
104
+ //
105
+ // Solve L'*x = y
106
+ //
107
+ b(n-1) = b(n-1)/a(n-1,n-1);
108
+ for(i = n-2; i >= 0; i--)
109
+ {
110
+ v = ap::vdotproduct(a.getcolumn(i, i+1, n-1), b.getvector(i+1, n-1));
111
+ b(i) = (b(i)-v)/a(i,i);
112
+ }
113
+ ap::vmove(&x(0), &b(0), ap::vlen(0,n-1));
114
+ }
115
+ return result;
116
+ }
117
+
118
+
119
+ /*************************************************************************
120
+ Solving a system of linear equations with a symmetric positive-definite
121
+ matrix by using the Cholesky decomposition.
122
+
123
+ The algorithm solves a system of linear equations whose matrix is symmetric
124
+ and positive-definite.
125
+
126
+ Input parameters:
127
+ A - upper or lower triangle part of a symmetric system matrix.
128
+ Array whose indexes range within [0..N-1, 0..N-1].
129
+ B - right side of a system.
130
+ Array whose index ranges within [0..N-1].
131
+ N - size of matrix A.
132
+ IsUpper - points to the triangle of matrix A in which the matrix is stored.
133
+
134
+ Output parameters:
135
+ X - solution of a system.
136
+ Array whose index ranges within [0..N-1].
137
+
138
+ Result:
139
+ True, if the system is not singular.
140
+ False, if the system is singular. In this case, X doesn't contain a
141
+ solution.
142
+
143
+ -- ALGLIB --
144
+ Copyright 2005-2008 by Bochkanov Sergey
145
+ *************************************************************************/
146
+ bool spdmatrixsolve(ap::real_2d_array a,
147
+ ap::real_1d_array b,
148
+ int n,
149
+ bool isupper,
150
+ ap::real_1d_array& x)
151
+ {
152
+ bool result;
153
+
154
+ result = spdmatrixcholesky(a, n, isupper);
155
+ if( !result )
156
+ {
157
+ return result;
158
+ }
159
+ result = spdmatrixcholeskysolve(a, b, n, isupper, x);
160
+ return result;
161
+ }
162
+
163
+
164
+ /*************************************************************************
165
+ Obsolete 1-bases subroutine
166
+ *************************************************************************/
167
+ bool solvesystemcholesky(const ap::real_2d_array& a,
168
+ ap::real_1d_array b,
169
+ int n,
170
+ bool isupper,
171
+ ap::real_1d_array& x)
172
+ {
173
+ bool result;
174
+ int i;
175
+ int im1;
176
+ int ip1;
177
+ double v;
178
+
179
+ ap::ap_error::make_assertion(n>0, "Error: N<=0 in SolveSystemCholesky");
180
+
181
+ //
182
+ // det(A)=0?
183
+ //
184
+ result = true;
185
+ for(i = 1; i <= n; i++)
186
+ {
187
+ if( a(i,i)==0 )
188
+ {
189
+ result = false;
190
+ return result;
191
+ }
192
+ }
193
+
194
+ //
195
+ // det(A)<>0
196
+ //
197
+ x.setbounds(1, n);
198
+ if( isupper )
199
+ {
200
+
201
+ //
202
+ // A = U'*U, solve U'*y = b first
203
+ //
204
+ b(1) = b(1)/a(1,1);
205
+ for(i = 2; i <= n; i++)
206
+ {
207
+ im1 = i-1;
208
+ v = ap::vdotproduct(a.getcolumn(i, 1, im1), b.getvector(1, im1));
209
+ b(i) = (b(i)-v)/a(i,i);
210
+ }
211
+
212
+ //
213
+ // Solve U*x = y
214
+ //
215
+ b(n) = b(n)/a(n,n);
216
+ for(i = n-1; i >= 1; i--)
217
+ {
218
+ ip1 = i+1;
219
+ v = ap::vdotproduct(&a(i, ip1), &b(ip1), ap::vlen(ip1,n));
220
+ b(i) = (b(i)-v)/a(i,i);
221
+ }
222
+ ap::vmove(&x(1), &b(1), ap::vlen(1,n));
223
+ }
224
+ else
225
+ {
226
+
227
+ //
228
+ // A = L*L', solve L'*y = b first
229
+ //
230
+ b(1) = b(1)/a(1,1);
231
+ for(i = 2; i <= n; i++)
232
+ {
233
+ im1 = i-1;
234
+ v = ap::vdotproduct(&a(i, 1), &b(1), ap::vlen(1,im1));
235
+ b(i) = (b(i)-v)/a(i,i);
236
+ }
237
+
238
+ //
239
+ // Solve L'*x = y
240
+ //
241
+ b(n) = b(n)/a(n,n);
242
+ for(i = n-1; i >= 1; i--)
243
+ {
244
+ ip1 = i+1;
245
+ v = ap::vdotproduct(a.getcolumn(i, ip1, n), b.getvector(ip1, n));
246
+ b(i) = (b(i)-v)/a(i,i);
247
+ }
248
+ ap::vmove(&x(1), &b(1), ap::vlen(1,n));
249
+ }
250
+ return result;
251
+ }
252
+
253
+
254
+ /*************************************************************************
255
+ Obsolete 1-bases subroutine
256
+ *************************************************************************/
257
+ bool solvespdsystem(ap::real_2d_array a,
258
+ ap::real_1d_array b,
259
+ int n,
260
+ bool isupper,
261
+ ap::real_1d_array& x)
262
+ {
263
+ bool result;
264
+
265
+ result = choleskydecomposition(a, n, isupper);
266
+ if( !result )
267
+ {
268
+ return result;
269
+ }
270
+ result = solvesystemcholesky(a, b, n, isupper, x);
271
+ return result;
272
+ }
273
+
274
+
275
+
@@ -0,0 +1,105 @@
1
+
2
+ #ifndef _spdsolve_h
3
+ #define _spdsolve_h
4
+
5
+ #include "ap.h"
6
+ #include "ialglib.h"
7
+
8
+ #include "cholesky.h"
9
+
10
+
11
+ /*************************************************************************
12
+ Solving a system of linear equations with a system matrix given by its
13
+ Cholesky decomposition.
14
+
15
+ The algorithm solves systems with a square matrix only.
16
+
17
+ Input parameters:
18
+ A - Cholesky decomposition of a system matrix (the result of
19
+ the SMatrixCholesky subroutine).
20
+ B - right side of a system.
21
+ Array whose index ranges within [0..N-1].
22
+ N - size of matrix A.
23
+ IsUpper - points to the triangle of matrix A in which the Cholesky
24
+ decomposition is stored. If IsUpper=True, the Cholesky
25
+ decomposition has the form of U'*U, and the upper triangle
26
+ of matrix A stores matrix U (in that case, the lower
27
+ triangle isn�t used and isn�t changed by the subroutine)
28
+ Similarly, if IsUpper = False, the Cholesky decomposition
29
+ has the form of L*L', and the lower triangle stores
30
+ matrix L.
31
+
32
+ Output parameters:
33
+ X - solution of a system.
34
+ Array whose index ranges within [0..N-1].
35
+
36
+ Result:
37
+ True, if the system is not singular. X contains the solution.
38
+ False, if the system is singular (there is a zero element on the main
39
+ diagonal). In this case, X doesn't contain a solution.
40
+
41
+ -- ALGLIB --
42
+ Copyright 2005-2008 by Bochkanov Sergey
43
+ *************************************************************************/
44
+ bool spdmatrixcholeskysolve(const ap::real_2d_array& a,
45
+ ap::real_1d_array b,
46
+ int n,
47
+ bool isupper,
48
+ ap::real_1d_array& x);
49
+
50
+
51
+ /*************************************************************************
52
+ Solving a system of linear equations with a symmetric positive-definite
53
+ matrix by using the Cholesky decomposition.
54
+
55
+ The algorithm solves a system of linear equations whose matrix is symmetric
56
+ and positive-definite.
57
+
58
+ Input parameters:
59
+ A - upper or lower triangle part of a symmetric system matrix.
60
+ Array whose indexes range within [0..N-1, 0..N-1].
61
+ B - right side of a system.
62
+ Array whose index ranges within [0..N-1].
63
+ N - size of matrix A.
64
+ IsUpper - points to the triangle of matrix A in which the matrix is stored.
65
+
66
+ Output parameters:
67
+ X - solution of a system.
68
+ Array whose index ranges within [0..N-1].
69
+
70
+ Result:
71
+ True, if the system is not singular.
72
+ False, if the system is singular. In this case, X doesn't contain a
73
+ solution.
74
+
75
+ -- ALGLIB --
76
+ Copyright 2005-2008 by Bochkanov Sergey
77
+ *************************************************************************/
78
+ bool spdmatrixsolve(ap::real_2d_array a,
79
+ ap::real_1d_array b,
80
+ int n,
81
+ bool isupper,
82
+ ap::real_1d_array& x);
83
+
84
+
85
+ /*************************************************************************
86
+ Obsolete 1-bases subroutine
87
+ *************************************************************************/
88
+ bool solvesystemcholesky(const ap::real_2d_array& a,
89
+ ap::real_1d_array b,
90
+ int n,
91
+ bool isupper,
92
+ ap::real_1d_array& x);
93
+
94
+
95
+ /*************************************************************************
96
+ Obsolete 1-bases subroutine
97
+ *************************************************************************/
98
+ bool solvespdsystem(ap::real_2d_array a,
99
+ ap::real_1d_array b,
100
+ int n,
101
+ bool isupper,
102
+ ap::real_1d_array& x);
103
+
104
+
105
+ #endif