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,98 @@
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 _ctrinverse_h
40
+ #define _ctrinverse_h
41
+
42
+ #include "ap.h"
43
+ #include "ialglib.h"
44
+
45
+ /*************************************************************************
46
+ Complex triangular matrix inversion
47
+
48
+ The subroutine inverts the following types of matrices:
49
+ * upper triangular
50
+ * upper triangular with unit diagonal
51
+ * lower triangular
52
+ * lower triangular with unit diagonal
53
+
54
+ In case of an upper (lower) triangular matrix, the inverse matrix will
55
+ also be upper (lower) triangular, and after the end of the algorithm,
56
+ the inverse matrix replaces the source matrix. The elements below (above)
57
+ the main diagonal are not changed by the algorithm.
58
+
59
+ If the matrix has a unit diagonal, the inverse matrix also has a unit
60
+ diagonal, the diagonal elements are not passed to the algorithm, they are
61
+ not changed by the algorithm.
62
+
63
+ Input parameters:
64
+ A - matrix.
65
+ Array whose indexes range within [0..N-1,0..N-1].
66
+ N - size of matrix A.
67
+ IsUpper - True, if the matrix is upper triangular.
68
+ IsUnitTriangular
69
+ - True, if the matrix has a unit diagonal.
70
+
71
+ Output parameters:
72
+ A - inverse matrix (if the problem is not degenerate).
73
+
74
+ Result:
75
+ True, if the matrix is not singular.
76
+ False, if the matrix is singular.
77
+
78
+ -- LAPACK routine (version 3.0) --
79
+ Univ. of Tennessee, Univ. of California Berkeley, NAG Ltd.,
80
+ Courant Institute, Argonne National Lab, and Rice University
81
+ February 29, 1992
82
+ *************************************************************************/
83
+ bool cmatrixtrinverse(ap::complex_2d_array& a,
84
+ int n,
85
+ bool isupper,
86
+ bool isunittriangular);
87
+
88
+
89
+ /*************************************************************************
90
+ Obsolete 1-based subroutine
91
+ *************************************************************************/
92
+ bool complexinvtriangular(ap::complex_2d_array& a,
93
+ int n,
94
+ bool isupper,
95
+ bool isunittriangular);
96
+
97
+
98
+ #endif
@@ -0,0 +1,297 @@
1
+
2
+ #include <stdafx.h>
3
+ #include "ctrlinsolve.h"
4
+
5
+ /*************************************************************************
6
+ Utility subroutine performing the "safe" solution of a system of linear
7
+ equations with triangular complex coefficient matrices.
8
+
9
+ The feature of an algorithm is that it could not cause an overflow or a
10
+ division by zero regardless of the matrix used as the input. If an overflow
11
+ is possible, an error code is returned.
12
+
13
+ The algorithm can solve systems of equations with upper/lower triangular
14
+ matrices, with/without unit diagonal, and systems of types A*x=b, A^T*x=b,
15
+ A^H*x=b.
16
+
17
+ Input parameters:
18
+ A - system matrix.
19
+ Array whose indexes range within [1..N, 1..N].
20
+ N - size of matrix A.
21
+ X - right-hand member of a system.
22
+ Array whose index ranges within [1..N].
23
+ IsUpper - matrix type. If it is True, the system matrix is the upper
24
+ triangular matrix and is located in the corresponding part
25
+ of matrix A.
26
+ Trans - problem type.
27
+ If Trans is:
28
+ * 0, A*x=b
29
+ * 1, A^T*x=b
30
+ * 2, A^H*x=b
31
+ IsUnit - matrix type. If it is True, the system matrix has a unit
32
+ diagonal (the elements on the main diagonal are not used
33
+ in the calculation process), otherwise the matrix is
34
+ considered to be a general triangular matrix.
35
+ CNORM - array which is stored in norms of rows and columns of the
36
+ matrix. If the array hasn't been filled up during previous
37
+ executions of an algorithm with the same matrix as the
38
+ input, it will be filled up by the subroutine. If the
39
+ array is filled up, the subroutine uses it without filling
40
+ it up again.
41
+ NORMIN - flag defining the state of column norms array. If True, the
42
+ array is filled up.
43
+ WORKA - working array whose index ranges within [1..N].
44
+ WORKX - working array whose index ranges within [1..N].
45
+
46
+ Output parameters (if the result is True):
47
+ X - solution. Array whose index ranges within [1..N].
48
+ CNORM - array of column norms whose index ranges within [1..N].
49
+
50
+ Result:
51
+ True, if the matrix is not singular and the algorithm finished its
52
+ work correctly without causing an overflow.
53
+ False, if the matrix is singular or the algorithm was cancelled
54
+ because of an overflow possibility.
55
+
56
+ Note:
57
+ The disadvantage of an algorithm is that sometimes it overestimates
58
+ an overflow probability. This is not a problem when solving ordinary
59
+ systems. If the elements of the matrix used as the input are close to
60
+ MaxRealNumber, a false overflow detection is possible, but in practice
61
+ such matrices can rarely be found.
62
+ You can find more reliable subroutines in the LAPACK library
63
+ (xLATRS subroutine ).
64
+
65
+ -- ALGLIB --
66
+ Copyright 31.03.2006 by Bochkanov Sergey
67
+ *************************************************************************/
68
+ bool complexsafesolvetriangular(const ap::complex_2d_array& a,
69
+ int n,
70
+ ap::complex_1d_array& x,
71
+ bool isupper,
72
+ int trans,
73
+ bool isunit,
74
+ ap::complex_1d_array& worka,
75
+ ap::complex_1d_array& workx)
76
+ {
77
+ bool result;
78
+ int i;
79
+ int l;
80
+ int j;
81
+ bool dolswp;
82
+ double ma;
83
+ double mx;
84
+ double v;
85
+ ap::complex t;
86
+ ap::complex r;
87
+ int i_;
88
+ int i1_;
89
+
90
+ ap::ap_error::make_assertion(trans>=0&&trans<=2, "ComplexSafeSolveTriangular: incorrect parameters!");
91
+ result = true;
92
+
93
+ //
94
+ // Quick return if possible
95
+ //
96
+ if( n<=0 )
97
+ {
98
+ return result;
99
+ }
100
+
101
+ //
102
+ // Main cycle
103
+ //
104
+ for(l = 1; l <= n; l++)
105
+ {
106
+
107
+ //
108
+ // Prepare subtask L
109
+ //
110
+ dolswp = false;
111
+ if( trans==0 )
112
+ {
113
+ if( isupper )
114
+ {
115
+ i = n+1-l;
116
+ i1_ = (i) - (1);
117
+ for(i_=1; i_<=l;i_++)
118
+ {
119
+ worka(i_) = a(i,i_+i1_);
120
+ }
121
+ i1_ = (i) - (1);
122
+ for(i_=1; i_<=l;i_++)
123
+ {
124
+ workx(i_) = x(i_+i1_);
125
+ }
126
+ dolswp = true;
127
+ }
128
+ if( !isupper )
129
+ {
130
+ i = l;
131
+ for(i_=1; i_<=l;i_++)
132
+ {
133
+ worka(i_) = a(i,i_);
134
+ }
135
+ for(i_=1; i_<=l;i_++)
136
+ {
137
+ workx(i_) = x(i_);
138
+ }
139
+ }
140
+ }
141
+ if( trans==1 )
142
+ {
143
+ if( isupper )
144
+ {
145
+ i = l;
146
+ for(i_=1; i_<=l;i_++)
147
+ {
148
+ worka(i_) = a(i_,i);
149
+ }
150
+ for(i_=1; i_<=l;i_++)
151
+ {
152
+ workx(i_) = x(i_);
153
+ }
154
+ }
155
+ if( !isupper )
156
+ {
157
+ i = n+1-l;
158
+ i1_ = (i) - (1);
159
+ for(i_=1; i_<=l;i_++)
160
+ {
161
+ worka(i_) = a(i_+i1_,i);
162
+ }
163
+ i1_ = (i) - (1);
164
+ for(i_=1; i_<=l;i_++)
165
+ {
166
+ workx(i_) = x(i_+i1_);
167
+ }
168
+ dolswp = true;
169
+ }
170
+ }
171
+ if( trans==2 )
172
+ {
173
+ if( isupper )
174
+ {
175
+ i = l;
176
+ for(i_=1; i_<=l;i_++)
177
+ {
178
+ worka(i_) = ap::conj(a(i_,i));
179
+ }
180
+ for(i_=1; i_<=l;i_++)
181
+ {
182
+ workx(i_) = x(i_);
183
+ }
184
+ }
185
+ if( !isupper )
186
+ {
187
+ i = n+1-l;
188
+ i1_ = (i) - (1);
189
+ for(i_=1; i_<=l;i_++)
190
+ {
191
+ worka(i_) = ap::conj(a(i_+i1_,i));
192
+ }
193
+ i1_ = (i) - (1);
194
+ for(i_=1; i_<=l;i_++)
195
+ {
196
+ workx(i_) = x(i_+i1_);
197
+ }
198
+ dolswp = true;
199
+ }
200
+ }
201
+ if( dolswp )
202
+ {
203
+ t = workx(l);
204
+ workx(l) = workx(1);
205
+ workx(1) = t;
206
+ t = worka(l);
207
+ worka(l) = worka(1);
208
+ worka(1) = t;
209
+ }
210
+ if( isunit )
211
+ {
212
+ worka(l) = 1;
213
+ }
214
+
215
+ //
216
+ // Test if workA[L]=0
217
+ //
218
+ if( worka(l)==0 )
219
+ {
220
+ result = false;
221
+ return result;
222
+ }
223
+
224
+ //
225
+ // Now we have:
226
+ //
227
+ // workA[1:L]*workX[1:L] = b[I]
228
+ //
229
+ // with known workA[1:L] and workX[1:L-1]
230
+ // and unknown workX[L]
231
+ //
232
+ t = 0;
233
+ if( l>=2 )
234
+ {
235
+ ma = 0;
236
+ for(j = 1; j <= l-1; j++)
237
+ {
238
+ ma = ap::maxreal(ma, ap::abscomplex(worka(j)));
239
+ }
240
+ mx = 0;
241
+ for(j = 1; j <= l-1; j++)
242
+ {
243
+ mx = ap::maxreal(mx, ap::abscomplex(workx(j)));
244
+ }
245
+ if( ap::maxreal(ma, mx)>1 )
246
+ {
247
+ v = ap::maxrealnumber/ap::maxreal(ma, mx);
248
+ v = v/(l-1);
249
+ if( v<ap::minreal(ma, mx) )
250
+ {
251
+ result = false;
252
+ return result;
253
+ }
254
+ }
255
+ t = 0.0;
256
+ for(i_=1; i_<=l-1;i_++)
257
+ {
258
+ t += worka(i_)*workx(i_);
259
+ }
260
+ }
261
+
262
+ //
263
+ // Now we have:
264
+ //
265
+ // workA[L]*workX[L] + T = b[I]
266
+ //
267
+ if( ap::maxreal(ap::abscomplex(t), ap::abscomplex(x(i)))>=0.5*ap::maxrealnumber )
268
+ {
269
+ result = false;
270
+ return result;
271
+ }
272
+ r = x(i)-t;
273
+
274
+ //
275
+ // Now we have:
276
+ //
277
+ // workA[L]*workX[L] = R
278
+ //
279
+ if( r!=0 )
280
+ {
281
+ if( log(ap::abscomplex(r))-log(ap::abscomplex(worka(l)))>=log(ap::maxrealnumber) )
282
+ {
283
+ result = false;
284
+ return result;
285
+ }
286
+ }
287
+
288
+ //
289
+ // X[I]
290
+ //
291
+ x(i) = r/worka(l);
292
+ }
293
+ return result;
294
+ }
295
+
296
+
297
+
@@ -0,0 +1,81 @@
1
+
2
+ #ifndef _ctrlinsolve_h
3
+ #define _ctrlinsolve_h
4
+
5
+ #include "ap.h"
6
+ #include "ialglib.h"
7
+
8
+ /*************************************************************************
9
+ Utility subroutine performing the "safe" solution of a system of linear
10
+ equations with triangular complex coefficient matrices.
11
+
12
+ The feature of an algorithm is that it could not cause an overflow or a
13
+ division by zero regardless of the matrix used as the input. If an overflow
14
+ is possible, an error code is returned.
15
+
16
+ The algorithm can solve systems of equations with upper/lower triangular
17
+ matrices, with/without unit diagonal, and systems of types A*x=b, A^T*x=b,
18
+ A^H*x=b.
19
+
20
+ Input parameters:
21
+ A - system matrix.
22
+ Array whose indexes range within [1..N, 1..N].
23
+ N - size of matrix A.
24
+ X - right-hand member of a system.
25
+ Array whose index ranges within [1..N].
26
+ IsUpper - matrix type. If it is True, the system matrix is the upper
27
+ triangular matrix and is located in the corresponding part
28
+ of matrix A.
29
+ Trans - problem type.
30
+ If Trans is:
31
+ * 0, A*x=b
32
+ * 1, A^T*x=b
33
+ * 2, A^H*x=b
34
+ IsUnit - matrix type. If it is True, the system matrix has a unit
35
+ diagonal (the elements on the main diagonal are not used
36
+ in the calculation process), otherwise the matrix is
37
+ considered to be a general triangular matrix.
38
+ CNORM - array which is stored in norms of rows and columns of the
39
+ matrix. If the array hasn't been filled up during previous
40
+ executions of an algorithm with the same matrix as the
41
+ input, it will be filled up by the subroutine. If the
42
+ array is filled up, the subroutine uses it without filling
43
+ it up again.
44
+ NORMIN - flag defining the state of column norms array. If True, the
45
+ array is filled up.
46
+ WORKA - working array whose index ranges within [1..N].
47
+ WORKX - working array whose index ranges within [1..N].
48
+
49
+ Output parameters (if the result is True):
50
+ X - solution. Array whose index ranges within [1..N].
51
+ CNORM - array of column norms whose index ranges within [1..N].
52
+
53
+ Result:
54
+ True, if the matrix is not singular and the algorithm finished its
55
+ work correctly without causing an overflow.
56
+ False, if the matrix is singular or the algorithm was cancelled
57
+ because of an overflow possibility.
58
+
59
+ Note:
60
+ The disadvantage of an algorithm is that sometimes it overestimates
61
+ an overflow probability. This is not a problem when solving ordinary
62
+ systems. If the elements of the matrix used as the input are close to
63
+ MaxRealNumber, a false overflow detection is possible, but in practice
64
+ such matrices can rarely be found.
65
+ You can find more reliable subroutines in the LAPACK library
66
+ (xLATRS subroutine ).
67
+
68
+ -- ALGLIB --
69
+ Copyright 31.03.2006 by Bochkanov Sergey
70
+ *************************************************************************/
71
+ bool complexsafesolvetriangular(const ap::complex_2d_array& a,
72
+ int n,
73
+ ap::complex_1d_array& x,
74
+ bool isupper,
75
+ int trans,
76
+ bool isunit,
77
+ ap::complex_1d_array& worka,
78
+ ap::complex_1d_array& workx);
79
+
80
+
81
+ #endif