29 #ifndef ANASAZI_HELPER_TRAITS_HPP 30 #define ANASAZI_HELPER_TRAITS_HPP 49 template <
class ScalarType>
61 std::vector<
Value<ScalarType> >* RV, std::vector<int>* RO, std::vector<int>* RI );
82 template<
class ScalarType>
91 int curDim = (int)rRV.size();
99 if ( iRV[i] != MT_ZERO ) {
102 (*RV)[i].set(rRV[i], iRV[i]);
103 (*RV)[i+1].set(rRV[i+1], iRV[i+1]);
106 if ( (*RV)[i].imagpart < MT_ZERO ) {
109 (*RV)[i] = (*RV)[i+1];
110 (*RV)[i+1] = tmp_ritz;
112 int tmp_order = (*RO)[i];
113 (*RO)[i] = (*RO)[i+1];
114 (*RO)[i+1] = tmp_order;
117 RI->push_back(1); RI->push_back(-1);
122 (*RV)[i].set(rRV[i], MT_ZERO);
130 template<
class ScalarType>
143 int i = 0, curDim = S->
numRows();
145 ScalarType* s_ptr = S->
values();
146 while( i < curDim ) {
147 if ( iRV[i] != MT_ZERO ) {
148 temp = lapack_mag.
LAPY2( blas.
NRM2( curDim, s_ptr+i*curDim, 1 ),
149 blas.
NRM2( curDim, s_ptr+(i+1)*curDim, 1 ) );
150 blas.
SCAL( curDim, ST_ONE/temp, s_ptr+i*curDim, 1 );
151 blas.
SCAL( curDim, ST_ONE/temp, s_ptr+(i+1)*curDim, 1 );
154 temp = blas.
NRM2( curDim, s_ptr+i*curDim, 1 );
155 blas.
SCAL( curDim, ST_ONE/temp, s_ptr+i*curDim, 1 );
161 template<
class ScalarType>
174 int s_stride = S.
stride();
177 ScalarType* s_ptr = S.
values();
179 while( i < s_cols ) {
180 if ( iRV[i] != MT_ZERO ) {
181 (*RR)[i] = lapack_mag.
LAPY2( blas.
NRM2(s_rows, s_ptr + i*s_stride, 1),
182 blas.
NRM2(s_rows, s_ptr + (i+1)*s_stride, 1) );
183 (*RR)[i+1] = (*RR)[i];
186 (*RR)[i] = blas.
NRM2(s_rows, s_ptr + i*s_stride, 1);
192 #ifdef HAVE_TEUCHOS_COMPLEX 207 const std::vector<T>& rRV,
208 const std::vector<T>& iRV,
209 std::vector<
Value<ANSZI_CPLX_CLASS<T> > >* RV,
210 std::vector<int>* RO, std::vector<int>* RI );
213 const std::vector<T>& iRV,
217 const std::vector<T>& iRV,
219 std::vector<T>* RR );
223 void HelperTraits<ANSZI_CPLX_CLASS<T> >::sortRitzValues(
224 const std::vector<T>& rRV,
225 const std::vector<T>& iRV,
226 std::vector<Value<ANSZI_CPLX_CLASS<T> > >* RV,
227 std::vector<int>* RO, std::vector<int>* RI )
230 int curDim = (int)rRV.size();
237 while( i < curDim ) {
238 (*RV)[i].set(rRV[i], iRV[i]);
245 void HelperTraits<ANSZI_CPLX_CLASS<T> >::scaleRitzVectors(
246 const std::vector<T>& iRV,
250 typedef ANSZI_CPLX_CLASS<T> ST;
255 int i = 0, curDim = S->numRows();
257 ST* s_ptr = S->values();
258 while( i < curDim ) {
259 temp = blas.
NRM2( curDim, s_ptr+i*curDim, 1 );
260 blas.
SCAL( curDim, ST_ONE/temp, s_ptr+i*curDim, 1 );
266 void HelperTraits<ANSZI_CPLX_CLASS<T> >::computeRitzResiduals(
267 const std::vector<T>& iRV,
274 int s_stride = S.stride();
275 int s_rows = S.numRows();
276 int s_cols = S.numCols();
277 ANSZI_CPLX_CLASS<T>* s_ptr = S.values();
279 for (
int i=0; i<s_cols; ++i ) {
280 (*RR)[i] = blas.
NRM2(s_rows, s_ptr + i*s_stride, 1);
288 #endif // ANASAZI_HELPER_TRAITS_HPP ScalarType LAPY2(const ScalarType x, const ScalarType y) const
ScalarType * values() const
ScalarTraits< ScalarType >::magnitudeType NRM2(const OrdinalType n, const ScalarType *x, const OrdinalType incx) const
static void scaleRitzVectors(const std::vector< typename Teuchos::ScalarTraits< ScalarType >::magnitudeType > &iRV, Teuchos::SerialDenseMatrix< int, ScalarType > *S)
Helper function for correctly scaling the eigenvectors of the projected eigenproblem.
Namespace Anasazi contains the classes, structs, enums and utilities used by the Anasazi package...
static void sortRitzValues(const std::vector< typename Teuchos::ScalarTraits< ScalarType >::magnitudeType > &rRV, const std::vector< typename Teuchos::ScalarTraits< ScalarType >::magnitudeType > &iRV, std::vector< Value< ScalarType > > *RV, std::vector< int > *RO, std::vector< int > *RI)
Helper function for correctly storing the Ritz values when the eigenproblem is non-Hermitian.
OrdinalType numRows() const
This struct is used for storing eigenvalues and Ritz values, as a pair of real values.
Anasazi header file which uses auto-configuration information to include necessary C++ headers...
Types and exceptions used within Anasazi solvers and interfaces.
OrdinalType stride() const
static void computeRitzResiduals(const std::vector< typename Teuchos::ScalarTraits< ScalarType >::magnitudeType > &iRV, const Teuchos::SerialDenseMatrix< int, ScalarType > &S, std::vector< typename Teuchos::ScalarTraits< ScalarType >::magnitudeType > *RR)
Helper function for correctly computing the Ritz residuals of the projected eigenproblem.
OrdinalType numCols() const
void SCAL(const OrdinalType n, const ScalarType alpha, ScalarType *x, const OrdinalType incx) const
Class which defines basic traits for working with different scalar types.