eigensolver/anasaziHelper.h

Go to the documentation of this file.

#pragma once

#define HAVE_CONFIG_H 

#include "AnasaziOperator.hpp"
#include <AnasaziMultiVec.hpp> 
#include "operator.hh"
#include <Teuchos_SerialDenseMatrix.hpp>

namespace concepts {

template <class ScalarT>
class AnasaziOp : public Anasazi::Operator<ScalarT> {
  public:


    AnasaziOp(Teuchos::RCP<Operator<ScalarT> > op)
      : op(op)
    { }
    
    virtual ~AnasaziOp() {};
    

    
    virtual void Apply ( const Anasazi::MultiVec<ScalarT>& x, Anasazi::MultiVec<ScalarT>& y ) const;

    
    const Operator<ScalarT>& getOp() { 
      return *op;
    }

  private:
    Teuchos::RCP<Operator<ScalarT> > op;
    
};

template <class ScalarT>
class AnasaziMV : public Anasazi::MultiVec<ScalarT> {
public:
  static const bool NO_ALLOCATE = false;
  typedef Vector<ScalarT> VectorT; 

  AnasaziMV(int dim = 0, int nvecs = 0, bool allocate = true);
  virtual ~AnasaziMV();

private:
  AnasaziMV(const AnasaziMV<ScalarT>& other);
  AnasaziMV<ScalarT>& operator=(const AnasaziMV<ScalarT> other);


public:

  virtual AnasaziMV<ScalarT>* Clone( const int numvecs ) const;

  virtual AnasaziMV<ScalarT>* CloneCopy() const;
  
  virtual AnasaziMV<ScalarT>* CloneCopy ( const std::vector<int>& index ) const;
  
  virtual AnasaziMV<ScalarT>* CloneView ( const std::vector<int>& index );


  
  int GetVecLength () const { return dim; }

  virtual int GetNumberVecs () const { return nvecs; }




  virtual void MvTimesMatAddMv ( ScalarT alpha, const Anasazi::MultiVec<ScalarT>& A, 
    const Teuchos::SerialDenseMatrix<int,ScalarT>& B, ScalarT beta );

  virtual void MvAddMv ( ScalarT alpha,const Anasazi::MultiVec<ScalarT>& A, 
      ScalarT beta, const Anasazi::MultiVec<ScalarT>& B );

  virtual void MvTransMv ( ScalarT alpha, const Anasazi::MultiVec<ScalarT>& A, 
      Teuchos::SerialDenseMatrix<int,ScalarT>& B
#ifdef HAVE_ANASAZI_EXPERIMENTAL
         , ConjType conj = Anasazi::CONJ
#endif         
         ) const;
  
  virtual void MvDot ( const Anasazi::MultiVec<ScalarT>& A, std::vector<ScalarT> & b 
#ifdef HAVE_ANASAZI_EXPERIMENTAL
      , ConjType conj = Anasazi::CONJ
#endif
      ) const;
  



  virtual void MvNorm ( 
      std::vector<typename Teuchos::ScalarTraits<ScalarT>::magnitudeType> & normvec 
      ) const;



  
  virtual void SetBlock ( const Anasazi::MultiVec<ScalarT>& A, const std::vector<int>& index );
  
  virtual void MvScale ( ScalarT alpha );

  virtual void MvScale ( const std::vector<ScalarT>& alpha );

  virtual void MvRandom ();

  virtual void MvInit ( ScalarT alpha );




  virtual void MvPrint ( std::ostream& os ) const;
  
  VectorT* getVec(int i)
  {
    return vecs[i];
  }

  const VectorT* getVec(int i) const
  {
    return vecs[i];
  }

private:
  const int dim;
  int nvecs;
  VectorT** vecs;
  bool allocate;
};

}