Stokhos_PCEAnasaziKL.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Stokhos Package
//
// Copyright 2009 NTESS and the Stokhos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef STOKHOS_PCE_ANASAZI_KL_HPP
#define STOKHOS_PCE_ANASAZI_KL_HPP

#include "Stokhos_ConfigDefs.h"
#ifdef HAVE_STOKHOS_ANASAZI

#include "Teuchos_ParameterList.hpp"
#include "Stokhos_PCECovarianceOp.hpp"
#include "AnasaziBasicEigenproblem.hpp"
#include "AnasaziEpetraAdapter.hpp"

namespace Stokhos {


  class PCEAnasaziKL {
  public:

    PCEAnasaziKL(const Stokhos::VectorOrthogPoly<Epetra_Vector>& X_poly,
     int num_KL_);

    PCEAnasaziKL(const Teuchos::RCP<const EpetraExt::BlockVector>& X,
     const Stokhos::OrthogPolyBasis<int,double>& basis,
     int num_KL_);

    PCEAnasaziKL(const Teuchos::RCP<const Epetra_MultiVector>& X,
     const Stokhos::OrthogPolyBasis<int,double>& basis,
     int num_KL_);
    
    virtual ~PCEAnasaziKL() {}

    Teuchos::ParameterList getDefaultParams() const;

    bool computeKL(Teuchos::ParameterList& anasazi_params);

    Teuchos::Array<double> getEigenvalues() const;

    Teuchos::RCP<Epetra_MultiVector> getEigenvectors() const;

  private:

    PCEAnasaziKL(const PCEAnasaziKL&);

    PCEAnasaziKL& operator=(const PCEAnasaziKL&);

  protected:

    typedef double ScalarType;
    typedef Teuchos::ScalarTraits<ScalarType>          SCT;
    typedef SCT::magnitudeType               MagnitudeType;
    typedef Epetra_MultiVector                          MV;
    typedef Epetra_Operator                             OP;
    typedef Anasazi::MultiVecTraits<ScalarType,MV>     MVT;
    typedef Anasazi::OperatorTraits<ScalarType,MV,OP>  OPT;

    Teuchos::RCP<Stokhos::PCECovarianceOp> covOp;

    int num_KL;

    Teuchos::RCP<Anasazi::BasicEigenproblem<ScalarType,MV,OP> > anasazi_problem;

    // Eigen solution
    Anasazi::Eigensolution<ScalarType,MV> sol;

  }; // PCEAnasaziKL

} // namespace Stokhos

#endif // HAVE_STOKHOS_ANASAZI

#endif // STOKHOS_PCE_ANASAZI_KL_HPP