1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2009-2010 Gael Guennebaud <[email protected]>
5 //
6 // This Source Code Form is subject to the terms of the Mozilla
7 // Public License v. 2.0. If a copy of the MPL was not distributed
8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9
10 #include "common.h"
11
12 // computes the sum of magnitudes of all vector elements or, for a complex vector x, the sum
13 // res = |Rex1| + |Imx1| + |Rex2| + |Imx2| + ... + |Rexn| + |Imxn|, where x is a vector of order n
EIGEN_BLAS_FUNC(asum)14 RealScalar EIGEN_BLAS_FUNC(asum)(int *n, RealScalar *px, int *incx)
15 {
16 // std::cerr << "_asum " << *n << " " << *incx << "\n";
17
18 Scalar* x = reinterpret_cast<Scalar*>(px);
19
20 if(*n<=0) return 0;
21
22 if(*incx==1) return make_vector(x,*n).cwiseAbs().sum();
23 else return make_vector(x,*n,std::abs(*incx)).cwiseAbs().sum();
24 }
25
EIGEN_CAT(i,EIGEN_BLAS_FUNC (amax))26 int EIGEN_CAT(i, EIGEN_BLAS_FUNC(amax))(int *n, RealScalar *px, int *incx)
27 {
28 if(*n<=0) return 0;
29 Scalar* x = reinterpret_cast<Scalar*>(px);
30
31 DenseIndex ret;
32 if(*incx==1) make_vector(x,*n).cwiseAbs().maxCoeff(&ret);
33 else make_vector(x,*n,std::abs(*incx)).cwiseAbs().maxCoeff(&ret);
34 return int(ret)+1;
35 }
36
EIGEN_CAT(i,EIGEN_BLAS_FUNC (amin))37 int EIGEN_CAT(i, EIGEN_BLAS_FUNC(amin))(int *n, RealScalar *px, int *incx)
38 {
39 if(*n<=0) return 0;
40 Scalar* x = reinterpret_cast<Scalar*>(px);
41
42 DenseIndex ret;
43 if(*incx==1) make_vector(x,*n).cwiseAbs().minCoeff(&ret);
44 else make_vector(x,*n,std::abs(*incx)).cwiseAbs().minCoeff(&ret);
45 return int(ret)+1;
46 }
47
48 // computes a vector-vector dot product.
EIGEN_BLAS_FUNC(dot)49 Scalar EIGEN_BLAS_FUNC(dot)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy)
50 {
51 // std::cerr << "_dot " << *n << " " << *incx << " " << *incy << "\n";
52
53 if(*n<=0) return 0;
54
55 Scalar* x = reinterpret_cast<Scalar*>(px);
56 Scalar* y = reinterpret_cast<Scalar*>(py);
57
58 if(*incx==1 && *incy==1) return (make_vector(x,*n).cwiseProduct(make_vector(y,*n))).sum();
59 else if(*incx>0 && *incy>0) return (make_vector(x,*n,*incx).cwiseProduct(make_vector(y,*n,*incy))).sum();
60 else if(*incx<0 && *incy>0) return (make_vector(x,*n,-*incx).reverse().cwiseProduct(make_vector(y,*n,*incy))).sum();
61 else if(*incx>0 && *incy<0) return (make_vector(x,*n,*incx).cwiseProduct(make_vector(y,*n,-*incy).reverse())).sum();
62 else if(*incx<0 && *incy<0) return (make_vector(x,*n,-*incx).reverse().cwiseProduct(make_vector(y,*n,-*incy).reverse())).sum();
63 else return 0;
64 }
65
66 // computes the Euclidean norm of a vector.
67 // FIXME
EIGEN_BLAS_FUNC(nrm2)68 Scalar EIGEN_BLAS_FUNC(nrm2)(int *n, RealScalar *px, int *incx)
69 {
70 // std::cerr << "_nrm2 " << *n << " " << *incx << "\n";
71 if(*n<=0) return 0;
72
73 Scalar* x = reinterpret_cast<Scalar*>(px);
74
75 if(*incx==1) return make_vector(x,*n).stableNorm();
76 else return make_vector(x,*n,std::abs(*incx)).stableNorm();
77 }
78
EIGEN_BLAS_FUNC(rot)79 int EIGEN_BLAS_FUNC(rot)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy, RealScalar *pc, RealScalar *ps)
80 {
81 // std::cerr << "_rot " << *n << " " << *incx << " " << *incy << "\n";
82 if(*n<=0) return 0;
83
84 Scalar* x = reinterpret_cast<Scalar*>(px);
85 Scalar* y = reinterpret_cast<Scalar*>(py);
86 Scalar c = *reinterpret_cast<Scalar*>(pc);
87 Scalar s = *reinterpret_cast<Scalar*>(ps);
88
89 StridedVectorType vx(make_vector(x,*n,std::abs(*incx)));
90 StridedVectorType vy(make_vector(y,*n,std::abs(*incy)));
91
92 Reverse<StridedVectorType> rvx(vx);
93 Reverse<StridedVectorType> rvy(vy);
94
95 if(*incx<0 && *incy>0) internal::apply_rotation_in_the_plane(rvx, vy, JacobiRotation<Scalar>(c,s));
96 else if(*incx>0 && *incy<0) internal::apply_rotation_in_the_plane(vx, rvy, JacobiRotation<Scalar>(c,s));
97 else internal::apply_rotation_in_the_plane(vx, vy, JacobiRotation<Scalar>(c,s));
98
99
100 return 0;
101 }
102
103 /*
104 // performs rotation of points in the modified plane.
105 int EIGEN_BLAS_FUNC(rotm)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy, RealScalar *param)
106 {
107 Scalar* x = reinterpret_cast<Scalar*>(px);
108 Scalar* y = reinterpret_cast<Scalar*>(py);
109
110 // TODO
111
112 return 0;
113 }
114
115 // computes the modified parameters for a Givens rotation.
116 int EIGEN_BLAS_FUNC(rotmg)(RealScalar *d1, RealScalar *d2, RealScalar *x1, RealScalar *x2, RealScalar *param)
117 {
118 // TODO
119
120 return 0;
121 }
122 */
123