1c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This file is part of Eigen, a lightweight C++ template library 2c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// for linear algebra. 3c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 4c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Copyright (C) 2009-2010 Gael Guennebaud <gael.guennebaud@inria.fr> 5c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// 6c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// This Source Code Form is subject to the terms of the Mozilla 7c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// Public License v. 2.0. If a copy of the MPL was not distributed 8c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 10c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath#include "common.h" 11c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 12c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathstruct scalar_norm1_op { 13c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath typedef RealScalar result_type; 14c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath EIGEN_EMPTY_STRUCT_CTOR(scalar_norm1_op) 15c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath inline RealScalar operator() (const Scalar& a) const { return internal::norm1(a); } 16c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath}; 17c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathnamespace Eigen { 18c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath namespace internal { 19c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath template<> struct functor_traits<scalar_norm1_op > 20c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath { 21c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath enum { Cost = 3 * NumTraits<Scalar>::AddCost, PacketAccess = 0 }; 22c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath }; 23c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath } 24c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 25c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 26c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// computes the sum of magnitudes of all vector elements or, for a complex vector x, the sum 27c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// res = |Rex1| + |Imx1| + |Rex2| + |Imx2| + ... + |Rexn| + |Imxn|, where x is a vector of order n 28c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan KamathRealScalar EIGEN_CAT(EIGEN_CAT(REAL_SCALAR_SUFFIX,SCALAR_SUFFIX),asum_)(int *n, RealScalar *px, int *incx) 29c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 30c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// std::cerr << "__asum " << *n << " " << *incx << "\n"; 31c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Complex* x = reinterpret_cast<Complex*>(px); 32c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 33c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 34c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 35c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx==1) return vector(x,*n).unaryExpr<scalar_norm1_op>().sum(); 36c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else return vector(x,*n,std::abs(*incx)).unaryExpr<scalar_norm1_op>().sum(); 37c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 38c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 39c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// computes a dot product of a conjugated vector with another vector. 40c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathint EIGEN_BLAS_FUNC(dotcw)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy, RealScalar* pres) 41c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 42c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// std::cerr << "_dotc " << *n << " " << *incx << " " << *incy << "\n"; 43c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 44c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 45c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 46c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* x = reinterpret_cast<Scalar*>(px); 47c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* y = reinterpret_cast<Scalar*>(py); 48c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* res = reinterpret_cast<Scalar*>(pres); 49c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 50c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx==1 && *incy==1) *res = (vector(x,*n).dot(vector(y,*n))); 51c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx>0 && *incy>0) *res = (vector(x,*n,*incx).dot(vector(y,*n,*incy))); 52c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx<0 && *incy>0) *res = (vector(x,*n,-*incx).reverse().dot(vector(y,*n,*incy))); 53c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx>0 && *incy<0) *res = (vector(x,*n,*incx).dot(vector(y,*n,-*incy).reverse())); 54c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx<0 && *incy<0) *res = (vector(x,*n,-*incx).reverse().dot(vector(y,*n,-*incy).reverse())); 55c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return 0; 56c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 57c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 58c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// computes a vector-vector dot product without complex conjugation. 59c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathint EIGEN_BLAS_FUNC(dotuw)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy, RealScalar* pres) 60c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 61c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// std::cerr << "_dotu " << *n << " " << *incx << " " << *incy << "\n"; 62c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 63c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 64c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 65c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* x = reinterpret_cast<Scalar*>(px); 66c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* y = reinterpret_cast<Scalar*>(py); 67c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* res = reinterpret_cast<Scalar*>(pres); 68c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 69c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx==1 && *incy==1) *res = (vector(x,*n).cwiseProduct(vector(y,*n))).sum(); 70c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx>0 && *incy>0) *res = (vector(x,*n,*incx).cwiseProduct(vector(y,*n,*incy))).sum(); 71c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx<0 && *incy>0) *res = (vector(x,*n,-*incx).reverse().cwiseProduct(vector(y,*n,*incy))).sum(); 72c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx>0 && *incy<0) *res = (vector(x,*n,*incx).cwiseProduct(vector(y,*n,-*incy).reverse())).sum(); 73c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx<0 && *incy<0) *res = (vector(x,*n,-*incx).reverse().cwiseProduct(vector(y,*n,-*incy).reverse())).sum(); 74c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return 0; 75c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 76c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 77c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan KamathRealScalar EIGEN_CAT(EIGEN_CAT(REAL_SCALAR_SUFFIX,SCALAR_SUFFIX),nrm2_)(int *n, RealScalar *px, int *incx) 78c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 79c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// std::cerr << "__nrm2 " << *n << " " << *incx << "\n"; 80c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 81c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 82c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* x = reinterpret_cast<Scalar*>(px); 83c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 84c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx==1) 85c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return vector(x,*n).stableNorm(); 86c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 87c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return vector(x,*n,*incx).stableNorm(); 88c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 89c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 90c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathint EIGEN_CAT(EIGEN_CAT(SCALAR_SUFFIX,REAL_SCALAR_SUFFIX),rot_)(int *n, RealScalar *px, int *incx, RealScalar *py, int *incy, RealScalar *pc, RealScalar *ps) 91c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 92c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 93c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 94c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* x = reinterpret_cast<Scalar*>(px); 95c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* y = reinterpret_cast<Scalar*>(py); 96c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RealScalar c = *pc; 97c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RealScalar s = *ps; 98c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 99c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath StridedVectorType vx(vector(x,*n,std::abs(*incx))); 100c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath StridedVectorType vy(vector(y,*n,std::abs(*incy))); 101c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 102c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Reverse<StridedVectorType> rvx(vx); 103c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Reverse<StridedVectorType> rvy(vy); 104c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 105c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath // TODO implement mixed real-scalar rotations 106c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx<0 && *incy>0) internal::apply_rotation_in_the_plane(rvx, vy, JacobiRotation<Scalar>(c,s)); 107c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else if(*incx>0 && *incy<0) internal::apply_rotation_in_the_plane(vx, rvy, JacobiRotation<Scalar>(c,s)); 108c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else internal::apply_rotation_in_the_plane(vx, vy, JacobiRotation<Scalar>(c,s)); 109c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 110c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return 0; 111c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 112c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 113c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamathint EIGEN_CAT(EIGEN_CAT(SCALAR_SUFFIX,REAL_SCALAR_SUFFIX),scal_)(int *n, RealScalar *palpha, RealScalar *px, int *incx) 114c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath{ 115c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*n<=0) return 0; 116c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 117c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath Scalar* x = reinterpret_cast<Scalar*>(px); 118c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath RealScalar alpha = *palpha; 119c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 120c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath// std::cerr << "__scal " << *n << " " << alpha << " " << *incx << "\n"; 121c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 122c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath if(*incx==1) vector(x,*n) *= alpha; 123c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath else vector(x,*n,std::abs(*incx)) *= alpha; 124c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 125c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath return 0; 126c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath} 127c981c48f5bc9aefeffc0bcb0cc3934c2fae179ddNarayan Kamath 128