FEApp_Application.hpp

Go to the documentation of this file.

// $Id: FEApp_Application.hpp,v 1.8 2008/08/01 22:57:12 etphipp Exp $ 
// $Source: /space/CVS/Trilinos/packages/sacado/example/FEApp/FEApp_Application.hpp,v $ 
// @HEADER
// ***********************************************************************
// 
//                           Sacado Package
//                 Copyright (2006) Sandia Corporation
// 
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
// the U.S. Government retains certain rights in this software.
// 
// This library is free software; you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 2.1 of the
// License, or (at your option) any later version.
//  
// This library is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//  
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
// USA
// Questions? Contact David M. Gay (dmgay@sandia.gov) or Eric T. Phipps
// (etphipp@sandia.gov).
// 
// ***********************************************************************
// @HEADER

#ifndef FEAPP_APPLICATION_HPP
#define FEAPP_APPLICATION_HPP

#include <vector>

#include "Teuchos_RCP.hpp"
#include "Teuchos_ParameterList.hpp"
#include "Teuchos_SerialDenseMatrix.hpp"
#include "Epetra_Vector.h"
#include "Epetra_CrsMatrix.h"
#include "Epetra_Import.h"
#include "Epetra_Export.h"

#include "FEApp_NodeBC.hpp"
#include "FEApp_AbstractPDE.hpp"
#include "FEApp_AbstractQuadrature.hpp"
#include "FEApp_AbstractDiscretization.hpp"
#include "FEApp_AbstractProblem.hpp"
#include "FEApp_TemplateTypes.hpp"
#include "FEApp_InitPostOps.hpp"

#include "Sacado_ScalarParameterLibrary.hpp"
#include "Sacado_ScalarParameterVector.hpp"

#if SG_ACTIVE
#include "EpetraExt_BlockVector.h"
#include "EpetraExt_BlockCrsMatrix.h"
#include "Stokhos_OrthogPolyBasis.hpp"
#include "FEApp_BlockDiscretization.hpp"
#endif

#if SGFAD_ACTIVE
#include "Stokhos_TripleProduct.hpp"
#endif

namespace FEApp {

  class Application {
  public:

    Application(const std::vector<double>& coords,
    const Teuchos::RCP<const Epetra_Comm>& comm,
    const Teuchos::RCP<Teuchos::ParameterList>& params,
    bool is_transient);

    ~Application();

    Teuchos::RCP<const Epetra_Map> getMap() const;

    Teuchos::RCP<const Epetra_CrsGraph> getJacobianGraph() const;

    Teuchos::RCP<const Epetra_Vector> getInitialSolution() const;

    Teuchos::RCP<Sacado::ScalarParameterLibrary> getParamLib();

    bool isTransient() const;

    Teuchos::RCP<Epetra_Operator> createW() const;

    Teuchos::RCP<Epetra_Operator> createPrec() const;


    void computeGlobalResidual(const Epetra_Vector* xdot,
             const Epetra_Vector& x,
             const Sacado::ScalarParameterVector* p,
             Epetra_Vector& f);


    void computeGlobalJacobian(double alpha, double beta,
             const Epetra_Vector* xdot,
             const Epetra_Vector& x,
             const Sacado::ScalarParameterVector* p,
             Epetra_Vector* f,
             Epetra_Operator& jac);


    void computeGlobalPreconditioner(double alpha, double beta,
             const Epetra_Vector* xdot,
             const Epetra_Vector& x,
             const Sacado::ScalarParameterVector* p,
             Epetra_Vector* f,
             Epetra_Operator& jac);
    

    void computeGlobalTangent(double alpha, double beta,
            bool sum_derivs,
            const Epetra_Vector* xdot,
            const Epetra_Vector& x,
            Sacado::ScalarParameterVector* p,
            const Epetra_MultiVector* Vx,
            const Teuchos::SerialDenseMatrix<int,double>* Vp,
            Epetra_Vector* f,
            Epetra_MultiVector* JVx,
            Epetra_MultiVector* fVp);


    void computeGlobalSGResidual(const Epetra_Vector* sg_xdot,
         const Epetra_Vector& sg_x,
         const Sacado::ScalarParameterVector* p,
         Epetra_Vector& sg_f);


    void computeGlobalSGJacobian(double alpha, double beta,
         const Epetra_Vector* sg_xdot,
         const Epetra_Vector& sg_x,
         const Sacado::ScalarParameterVector* p,
         Epetra_Vector* sg_f,
         Epetra_Operator& sg_jac); 


    void computeGlobalSGPreconditioner(double alpha, double beta,
               const Epetra_Vector* sg_xdot,
               const Epetra_Vector& sg_x,
               const Sacado::ScalarParameterVector* p,
               Epetra_Vector* sg_f,
               Epetra_Operator& sg_jac); 

  private:
    
    Application(const Application&);

    Application& operator=(const Application&);

  protected:

    bool transient;
    
    Teuchos::RCP<FEApp::AbstractDiscretization> disc;

    std::vector< Teuchos::RCP<FEApp::NodeBC> > bc;

    Teuchos::RCP<const FEApp::AbstractQuadrature> quad;

    FEApp::AbstractPDE_TemplateManager<ValidTypes> pdeTM;

    Teuchos::RCP<Epetra_Vector> initial_x;

    Teuchos::RCP<Epetra_Import> importer;

    Teuchos::RCP<Epetra_Export> exporter;

    Teuchos::RCP<Epetra_Vector> overlapped_x;

    Teuchos::RCP<Epetra_Vector> overlapped_xdot;

    Teuchos::RCP<Epetra_Vector> overlapped_f;

    Teuchos::RCP<Epetra_CrsMatrix> overlapped_jac;

    Teuchos::RCP<Sacado::ScalarParameterLibrary> paramLib;

    bool enable_sg;

#if SG_ACTIVE

    std::string sg_solver_method;

    Teuchos::RCP<const Stokhos::OrthogPolyBasis<double> > sg_basis;

     typedef SGType::expansion_type::tp_type tp_type;
    Teuchos::RCP<const tp_type> Cijk;

    Teuchos::RCP<FEApp::BlockDiscretization> sg_disc;

    Teuchos::RCP<EpetraExt::BlockVector> sg_initial_x;

    Teuchos::RCP<Epetra_Import> sg_importer;

    Teuchos::RCP<Epetra_Export> sg_exporter;

    Teuchos::RCP<EpetraExt::BlockVector> sg_overlapped_x;

    Teuchos::RCP<EpetraExt::BlockVector> sg_overlapped_xdot;

    Teuchos::RCP<EpetraExt::BlockVector> sg_overlapped_f;

    Teuchos::RCP<EpetraExt::BlockCrsMatrix> sg_overlapped_jac;

    Teuchos::RCP<FEApp::SGResidualOp> sg_res_fill_op;

    Teuchos::RCP<FEApp::SGJacobianOp> sg_full_jac_fill_op;

    Teuchos::RCP<FEApp::SGMatrixFreeJacobianOp> sg_mf_jac_fill_op;

    Teuchos::RCP<Teuchos::ParameterList> precParams;

#endif

  };

}

#endif // FEAPP_APPLICATION_HPP

---