Stokhos_ParallelData.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_PARALLEL_DATA_HPP
#define STOKHOS_PARALLEL_DATA_HPP

#include "Teuchos_RCP.hpp"
#include "Stokhos_OrthogPolyBasis.hpp"
#include "Stokhos_Sparse3Tensor.hpp"
#include "EpetraExt_MultiComm.h"
#include "Teuchos_ParameterList.hpp"
#include "Stokhos_EpetraSparse3Tensor.hpp"

namespace Stokhos {

  class ParallelData {
  public:

    ParallelData(
      const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int,double> >& basis,
      const Teuchos::RCP<const Stokhos::Sparse3Tensor<int,double> >& Cijk,
      const Teuchos::RCP<const Epetra_Comm>& globalComm,
      Teuchos::ParameterList& params);

    ParallelData(
      const Teuchos::RCP<const Stokhos::OrthogPolyBasis<int,double> >& basis,
      const Teuchos::RCP<const Stokhos::Sparse3Tensor<int,double> >& Cijk,
      const Teuchos::RCP<const EpetraExt::MultiComm>& globalMultiComm,
      Teuchos::ParameterList& params);

    ~ParallelData() {}

    Teuchos::RCP<const EpetraExt::MultiComm> 
    getMultiComm() const { return globalMultiComm; }

    Teuchos::RCP<const Epetra_Comm> 
    getStochasticComm() const { return stoch_comm; }

    Teuchos::RCP<const Epetra_Comm> 
    getSpatialComm() const { return spatial_comm; }

    Teuchos::RCP<const Stokhos::EpetraSparse3Tensor>
    getEpetraCijk() const { return epetraCijk; }

  protected:

    Teuchos::RCP<const EpetraExt::MultiComm> globalMultiComm;

    Teuchos::RCP<const Epetra_Comm> stoch_comm;

    Teuchos::RCP<const Epetra_Comm> spatial_comm;

    Teuchos::RCP<Stokhos::EpetraSparse3Tensor> epetraCijk;

  }; // class ParallelData

  Teuchos::RCP<const EpetraExt::MultiComm> 
  buildMultiComm(const Epetra_Comm& globalComm,
     int num_global_stochastic_blocks,
     int num_spatial_procs = -1);

  Teuchos::RCP<const Epetra_Comm> 
  getSpatialComm(
    const Teuchos::RCP<const EpetraExt::MultiComm>& globalMultiComm);

  Teuchos::RCP<const Epetra_Comm> 
  getStochasticComm(
    const Teuchos::RCP<const EpetraExt::MultiComm>& globalMultiComm);
    

} // namespace Stokhos

#endif // STOKHOS_PARALLEL_DATA_HPP