FEApp_BlockDiscretization.hpp

Go to the documentation of this file.

// $Id: FEApp_BlockDiscretization.hpp,v 1.4 2008/08/01 22:57:12 etphipp Exp $ 
// $Source: /space/CVS/Trilinos/packages/sacado/example/FEApp/FEApp_BlockDiscretization.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_BLOCKDISCRETIZATION_HPP
#define FEAPP_BLOCKDISCRETIZATION_HPP

#include <vector>
#include "FEApp_TemplateTypes.hpp"

#if SG_ACTIVE

#include "Epetra_Comm.h"
#include "EpetraExt_BlockVector.h"
#include "EpetraExt_BlockCrsMatrix.h"
#include "EpetraExt_BlockUtility.h"

#include "FEApp_AbstractDiscretization.hpp"

#include "Stokhos_OrthogPolyBasis.hpp"

namespace FEApp {

  class BlockDiscretization : public FEApp::AbstractDiscretization {
  public:

    BlockDiscretization(
     const Teuchos::RCP<const Epetra_Comm>& comm,
     const Teuchos::RCP<const FEApp::AbstractDiscretization>& underlyingDisc_,
     const Teuchos::RCP<const Stokhos::OrthogPolyBasis<double> >& sg_basis_,
     bool makeJacobian);

    virtual ~BlockDiscretization();

    virtual void createMesh();

    virtual void createMaps();

    virtual void createJacobianGraphs();

    virtual Teuchos::RCP<const FEApp::Mesh> 
    getMesh() const; 

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

    virtual Teuchos::RCP<const Epetra_Map> 
    getOverlapMap() const;

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

    virtual Teuchos::RCP<const Epetra_CrsGraph> 
    getOverlapJacobianGraph() const;

    virtual int getNumNodesPerElement() const;

    Teuchos::RCP<EpetraExt::BlockCrsMatrix> 
    getJacobian();

    Teuchos::RCP<EpetraExt::BlockCrsMatrix> 
    getOverlapJacobian();


  private:

    BlockDiscretization(const BlockDiscretization&);

    BlockDiscretization& operator=(const BlockDiscretization&);

  protected:
    
    Teuchos::RCP<const FEApp::AbstractDiscretization> underlyingDisc;

    Teuchos::RCP<const Epetra_Comm> globalComm;

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

    Teuchos::RCP<const Epetra_Map> map;

    Teuchos::RCP<const Epetra_Map> overlap_map;

    Teuchos::RCP<const Epetra_CrsGraph> graph;

    Teuchos::RCP<const Epetra_CrsGraph> overlap_graph;

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

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

  };

}

#endif // SG_ACTIVE

#endif // FEAPP_CZERODISCRETIZATION_HPP

---